#!/bin/bash

GADM_VERSION="4.1"
GADM_BASEPATH="https://geodata.ucdavis.edu/gadm"

mapshaper="./node_modules/mapshaper/bin/mapshaper"
ATA_URL="https://media.githubusercontent.com/media/wmgeolab/geoBoundaries/905b0baf5f4fb3b9ccf45293647dcacdb2b799d4/releaseData/gbOpen/ATA/ADM0/geoBoundaries-ATA-ADM0_simplified.geojson"

# Caspian Sea: "XCA"
countries=(
  "AFG" "XAD" "ALA" "ALB" "DZA" "ASM" "AND" "AGO" "AIA" "ATG" "ARG" "ARM" "ABW" "AUS" "AUT" "AZE"
  "BHS" "BHR" "BGD" "BRB" "BLR" "BEL" "BLZ" "BEN" "BMU" "BTN" "BOL" "BES" "BIH" "BWA" "BVT" "BRA" "IOT" "VGB" "BRN" "BGR" "BFA" "BDI" "KHM"
  "CMR" "CAN" "CPV"  "CYM" "CAF" "TCD" "CHL" "CHN" "CXR" "XCL" "CCK" "COL" "COM" "COK" "CRI" "CIV" "HRV" "CUB" "CUW" "CYP" "CZE" "COD"
  "DNK" "DJI" "DMA" "DOM" "ECU" "EGY" "SLV" "GNQ" "ERI" "EST" "ETH" "FLK" "FRO" "FJI" "FIN" "FRA" "GUF" "PYF" "ATF"
  "GAB" "GMB" "GEO" "DEU" "GHA" "GIB" "GRC" "GRL" "GRD" "GLP" "GUM" "GTM" "GGY" "GIN" "GNB" "GUY" "HTI" "HMD" "HND" "HUN"
  "ISL" "IND" "IDN" "IRN" "IRQ" "IRL" "IMN" "ISR" "ITA" "JAM" "JPN" "JEY" "JOR" "KAZ" "KEN" "KIR" "XKO" "KWT" "KGZ"
"LAO" "LVA" "LBN" "LSO" "LBR" "LBY" "LIE" "LTU" "LUX" "SXM"
"MKD" "MDG" "MWI" "MYS" "MDV" "MLI" "MLT" "MHL" "MTQ" "MRT" "MUS" "MYT" "MEX" "FSM" "MDA" "MCO" "MNG" "MNE" "MSR" "MAR" "MOZ" "MMR"
"NAM" "NRU" "NPL" "NLD" "NCL" "NZL" "NIC" "NER" "NGA" "NIU" "NFK" "PRK" "ZNC" "MNP" "NOR" "OMN"
"PAK" "PLW" "PSE" "PAN" "PNG" "PRY" "PER" "PHL" "PCN" "POL" "PRT" "PRI" "QAT" "COG" "REU" "ROU" "RUS" "RWA" "BLM" "MAF"
"SHN" "KNA" "LCA" "SPM" "VCT" "WSM" "SMR" "STP" "SAU" "SEN" "SRB" "SYC" "SLE" "SGP" "SVK" "SVN" "SLB" "SOM" "ZAF" "SGS" "KOR" "SSD" "ESP"
"LKA" "SDN" "SUR" "SJM" "SWZ" "SWE" "CHE" "SYR" "TWN" "TJK" "TZA" "THA" "TLS" "TGO" "TKL" "TON" "TTO" "TUN" "TUR" "TKM" "TCA" "TUV" "UGA"
"UKR" "ARE" "GBR" "USA" "UMI" "URY" "UZB" "VUT" "VAT" "VEN" "VNM" "VIR" "WLF" "ESH" "YEM" "ZMB" "ZWE"
)


url_0() {
    local country="$1"
    echo "${GADM_BASEPATH}/gadm${GADM_VERSION}/json/gadm${GADM_VERSION//./}_${country}_0.json"
}

download_0() {
    local url=$(url_0 "$1")
    local output_dir="./temp/0"
    mkdir -p "$output_dir"
    if [ -f "$output_dir/$1.json" ]; then
        echo "File Exists: $1"
    else
        echo "File Download: $1"
        wget -q -O "$output_dir/$1.json" "$url"
    fi
}


url_1() {
    local country="$1"
    echo "${GADM_BASEPATH}/gadm${GADM_VERSION}/json/gadm${GADM_VERSION//./}_${country}_1.json.zip"
}

download_1() {
    local url=$(url_1 "$1")
    local output_dir="./temp/1"
    mkdir -p "$output_dir"
    if [ -f "$output_dir/$1.json.zip" ]; then
        echo "File Exists: $1"
    else
        echo "File Download: $1"
        wget -q -O "$output_dir/$1.json.zip" "$url"
    fi
    if [ -f "$output_dir/gadm41_${1}_1.json" ]; then
        echo "File Exists: $1"
    else
        echo "File Extract: $1"
        unzip -q -o "$output_dir/$1.json.zip" -d "$output_dir"
    fi
    if [ -f "$output_dir/gadm41_${1}_1.json" ]; then
        jq '.features[] |= . + {properties: (.properties | .GID_1 = (.GID_0 + "_" + (
            if .HASC_1 != "NA" then (.HASC_1 | split(".") | .[-1])
            elif .ISO_1 != "NA" then (.ISO_1 | split("-") | .[-1])
            elif .CC_1 != "NA" then (.CC_1)
            elif .NAME_1 != "NA" then (.NAME_1)
            elif .GID_1 != "NA" then (.GID_1 | split(".") | .[-1])
            else .GID_1
            end
            )))}' "$output_dir/gadm41_${1}_1.json"  > "$output_dir/$1.json.1"
        sed -E 's/"[gadm41_]*([A-Z]*)_1"/"\1"/g' "$output_dir/$1.json.1" > "$output_dir/$1.json"
    fi
}


generate_map() {
    local OUT_FILE="$1"      # First argument: Output destination path
    local PROJ_ARGS="$2"     # Second argument: Projection parameters
    shift 2                  # Remove the first two arguments, leaving only the files
    local FILES_TO_RUN=("$@") # Capture all remaining arguments as the file array

    echo "Generating: $OUT_FILE using projection [$PROJ_ARGS]"
    echo "Processing ${#FILES_TO_RUN[@]} files..."

    # Each feature is tagged with the place it belongs to and whether it is a
    # country outline or one of its regions. packmap.js groups on that; nothing
    # downstream needs the features split into layers.
    local JS_PIPELINE_LOGIC="
        const conflicts = {
                'Z01': 'IND', 'Z04': 'IND', 'Z05': 'IND', 'Z07': 'IND', 'Z09': 'IND',
                'Z02': 'CHN', 'Z03': 'CHN','Z08': 'CHN',
                'Z06': 'PAK'
        };
        let rawCode = GID_0 || 'UNK';
        let cCode = conflicts[rawCode] ? conflicts[rawCode] : rawCode;
        // Antarctica's boundary file carries no country code at either level.
        if (cCode === 'UNK') cCode = 'ATA';
        let isGidMissing = (!GID || GID === 'undefined' || GID === 'null' || GID === '');
        CODE = isGidMissing ? cCode : GID;
        LEVEL = isGidMissing ? 0 : 1;
    "

    "$mapshaper" -i "${FILES_TO_RUN[@]}" combine-files \
        -snap \
        -merge-layers force\
        -proj $PROJ_ARGS densify \
        -simplify 10% weighted keep-shapes \
        -filter-islands min-area=0 \
        -sort 'this.area' ascending \
        -each "$JS_PIPELINE_LOGIC" \
        -o "$OUT_FILE" \
        format=geojson

}

build_maps() {
    input_files=()
    
    input_files+=("./temp/1/ATA.json")
    for country in "${countries[@]}"
    do
        input_file1="./temp/1/${country}.json"
        if [ -f "$input_file1" ]; then
            input_files+=("$input_file1")
        fi
    done
    input_files+=("./temp/0/ATA.json")
    for country in "${countries[@]}"
    do
        input_file0="./temp/0/${country}.json"
        if [ -f "$input_file0" ]; then
            input_files+=("$input_file0")
        fi
    done

    generate_map "./mapdata/loxim01.geojson" "loxim" "${input_files[@]}"
    generate_map "./mapdata/webmercator01.geojson" "webmercator" "${input_files[@]}"
    generate_map "./mapdata/aeqd01.geojson" "aeqd +lat_0=90" "${input_files[@]}"

    # The app reads the packed form; the GeoJSON above is only an intermediate.
    node packmap.js ./mapdata/loxim01.geojson ./mapdata/webmercator01.geojson ./mapdata/aeqd01.geojson
}

do_1() {
    for country in "${countries[@]}"
    do
        download_1 "$country"
    done
    wget -q -O "./temp/1/ATA.json" "$ATA_URL"
}
do_0() {
    for country in "${countries[@]}"
    do
        download_0 "$country"
    done
    wget -q -O "./temp/0/ATA.json" "$ATA_URL"
}
# do_0
# do_1
build_maps

#CUW, CCK, XCL, CXR, IOT, BVT, ABW, FLK, GIB, HMD, KIR, SXM, MDV, MCO, NIU, NFK, PCN, MAF, SGS, VAT