oak-general-business/scripts/getAmapArea.ts

/**
 * 用于更新高德数据库中的adCode信息到本地src/data/area.json中
 */
import 'isomorphic-fetch';
import assert from 'assert';
import { writeFileSync } from 'fs-extra';
import { FormCreateData } from 'oak-domain/lib/types/Entity';
import { OpSchema as Area } from '../src/general-app-domain/Area/Schema';


const KEY = '4f3d4499850ba51429b9dece8cedd8d2';

async function acquireAmap(keyWords: string, subdistrict: number) {
    const url = `https://restapi.amap.com/v3/config/district?keywords=${encodeURIComponent(keyWords)}&subdistrict=${subdistrict}&key=${KEY}`;

    const res = await fetch(url);
    return res.json();
}


function processRepeatedAdCode(districts: any[], parentAdCode: string) {
    // 高德的street的adCode和父亲相同，这里统一处理一下
    let jdCount = 0;
    let zCount = 100;
    let oCount = 400;
    let xCount = 200;
    const districts2 = districts.map(
        (ele) => {
            const { name, adcode } = ele;
            let ele2 = ele;
            if (adcode === parentAdCode) {
                if (name.endsWith('街道')) {
                    ele2 = Object.assign(ele, { adcode: parseInt(adcode) * 1000 + jdCount });
                    jdCount++;
                }
                else if (name.endsWith('镇')) {
                    ele2 = Object.assign(ele, { adcode: parseInt(adcode) * 1000 + zCount });
                    zCount++;
                }
                else if (name.endsWith('乡')) {
                    ele2 = Object.assign(ele, { adcode: parseInt(adcode) * 1000 + xCount });
                    xCount++;
                }
                else {
                    ele2 = Object.assign(ele, { adcode: parseInt(adcode) * 1000 + oCount });
                    oCount++;
                }
            }
            return ele2;
        }
    );
    return districts2;
}

/**
 * 省份直接取到街道，然后存储
 * @param provinceName
 * @returns {*}
 */
/* function getProvince(provinceName) {
    return acquireAmap(provinceName, 3)
        .then(
            (result) => {
                const { districts: cities, adcode: provinceCode } = result.districts[0];
                cities.forEach(
                    (city) => {
                        const { districts, adcode: cityCode } = city;
                        const districts2 = processRepeatedAdCode(districts, cityCode);
                        saveDistricts(districts2, cityCode);
                        districts2.forEach(
                            (district) => {
                                const { districts: streets, adcode: districtCode } = district;
                                const streets2 = processRepeatedAdCode(streets, districtCode);
                                saveDistricts(streets2, districtCode);
                            }
                        );
                    }
                );
            }
        );
} */


async function main(provinceName?: string) {
    const areaDebug: FormCreateData<Area>[] = [];
    const areaTotal: FormCreateData<Area>[] = [];
    function saveAreas(areas: any[], parentId: string | null, depth: 0 | 1 | 2 | 3 | 4, dest: FormCreateData<Area>[]) {
        areas.forEach(
            (ele) => {
                const { adcode, center, citycode, level, name } = ele;
                const coords = center.split(',');
                dest.push({
                    code: adcode,
                    level,
                    parentId,
                    name,
                    depth,
                    id: adcode,
                    center: {
                        type: 'point',
                        coordinate: [parseFloat(coords[0]), parseFloat(coords[1])],
                    },
                });
            }
        );
    }
    
    const result = await acquireAmap('中国', 1);
    const { districts } = result;
    const country = districts[0];
    saveAreas([country], null, 0, areaDebug);
    saveAreas([country], null, 0, areaTotal);
    const { districts: provinces, adcode: countryCode } = country;

    // debug 省只存一个
    const provinces2 = provinceName ? provinces.filter((ele: any) => ele.name === provinceName) : provinces.slice(0, 1);
    saveAreas(provinces, countryCode, 1, areaTotal);
    saveAreas(provinces2, countryCode, 1, areaDebug);
    
    // debug的数据
    for (const dist of provinces2) {
        const result2 = await acquireAmap(dist.name, 3);
        const { districts: cities, adcode: provinceCode } = result2.districts[0];
        const cities2 =  cities.sort((ele1: any, ele2: any) => parseInt(ele1.adcode) - parseInt(ele2.adcode)).slice(0, 2);        // debug 城市取一个
        saveAreas(cities2, provinceCode, 2, areaDebug);
        for (const city of cities2) {
            const { districts, adcode: cityCode } = city;
            const districts2 = processRepeatedAdCode(districts, cityCode);
            // const districts3 = districts2.slice(0, 3);      // debug district取三个
            saveAreas(districts2, cityCode, 3, areaDebug);
            districts2.forEach(
                (district) => {
                    const { districts: streets, adcode: districtCode } = district;
                    const streets2 = processRepeatedAdCode(streets, districtCode);
                    const streets3 = streets2.slice(0, 3);
                    saveAreas(streets2, districtCode, 4, areaDebug);           // street取三个
                }
            );
        }
    }

    // 完整的数据
    for (const dist of provinces) {
        const result2 = await acquireAmap(dist.name, 3);
        const { districts: cities, adcode: provinceCode } = result2.districts[0];
        saveAreas(cities, provinceCode, 2, areaTotal);
        for (const city of cities) {
            const { districts, adcode: cityCode } = city;
            const districts2 = processRepeatedAdCode(districts, cityCode);
            saveAreas(districts2, cityCode, 3, areaTotal);
            districts2.forEach(
                (district) => {
                    const { districts: streets, adcode: districtCode } = district;
                    const streets2 = processRepeatedAdCode(streets, districtCode);
                    saveAreas(streets2, districtCode, 4, areaTotal);
                }
            );
        }
    }

    writeFileSync(`${__dirname}/../src/data/area-debug.json`, JSON.stringify(areaDebug));
    writeFileSync(`${__dirname}/../src/data/area.json`, JSON.stringify(areaTotal));
}


main(process.argv[2]).then(
    () => console.log('success')
);
/* 
acquireAmap('中国', 1)
    .then(
        (result) => {
            saveDistricts(result.districts, null);
            const country = result.districts[0];
            const { districts: provinces, adcode: countryCode } = country;
            saveDistricts(provinces, countryCode);
            function provIter(idx) {
                if (idx === provinces.length) {
                    return Promise.resolve();
                }
                const prov = provinces[idx];
                return getProvince(prov.name)
                    .then(
                        () => {
                            console.log(`${prov.name}处理完毕`);
                            return provIter(idx + 1);
                        }
                    );
            }

            return provIter(0);
        }
    )
    .catch(
        (err) => {
            console.error(err);
            process.exit(-1);
        }
    ); */