build-origin.js

const axios = require('axios')
const pinyin = require('pinyin')
const fs = require('fs')

const loadOverall = async data => {
  const overall = data
    .match(/window.getStatisticsService = (.*?)}catch/)[1]
  // const result = JSON.parse(overall)
  // const numbers = result.countRemark.match(/\d+/g)
  // console.log(result)
  // result.confirmed = numbers[0]
  // result.suspect = numbers[1]
  // result.death = numbers[2]
  // result.cured = numbers[3]
  fs.writeFileSync('./src/data/overall.json', overall)
}

const cityMap = {
  '恩施州': '恩施土家族苗族自治州',
  '西双版纳': '西双版纳傣族自治州',
  '大理': '大理白族自治州',
  '红河': '红河哈尼族彝族自治州',
  '德宏': '德宏傣族景颇族自治州',
}

const loadCityList = async data => {
  const cityList = data
    .match(/window.getAreaStat = (.*?)}catch/)[1]
  const provinces = JSON.parse(cityList)
  const result = JSON.stringify(provinces.map(p => {
    if (p.provinceShortName === '陕西') {
      p.pinyin = 'shanxi1'
    } else if (p.provinceShortName === '重庆') {
      p.pinyin = 'chongqing'
    }
    return {
      pinyin: pinyin(p.provinceShortName, {
        style: pinyin.STYLE_NORMAL
      }).map(x => x[0]).join(''),
      name: p.provinceShortName,
      ...p,
      cities: p.cities.map(city => {
        let fullCityName = city.cityName
        if (p.provinceShortName === '北京' || p.provinceShortName === '上海') {
          fullCityName = city.cityName + '区'
        } else {
          if (city.cityName.length > 2 && /(市|州|区|旗)/.test(city.cityName)) {
            fullCityName = city.cityName
          } else {
            fullCityName = city.cityName + '市'
          }
        }
        fullCityName = cityMap[city.cityName] || fullCityName
        return {
          ...city,
          fullCityName
        }
      })
    }
  }), null, 2)
  fs.writeFileSync('./src/data/area.json', result)
}

let times = 0
async function request () {
  return axios.request('https://3g.dxy.cn/newh5/view/pneumonia').then(({ data: html }) => {
    return Promise.all([
      loadCityList(html),
      loadOverall(html)
    ])
  }).catch(e => {
    console.log('Retry')
    if (times++ > 10) {
      throw new Error(e)
    }
    return request()
  })
}

request().catch(e => {
  console.log(e)
  process.exit(1)
})