miao-plugin/tools/artis-data.js

import fs from 'fs'
import cheerio from 'cheerio'
import fetch from 'node-fetch'
import { Data } from '#miao'
import lodash from 'lodash'
import request from 'request'
import HttpsProxyAgent from 'https-proxy-agent'

let agent = new HttpsProxyAgent('http://localhost:4780')

const artiIdx = {
  Flower: 1,
  Plume: 2,
  Sands: 3,
  Goblet: 4,
  Circlet: 5
}

async function getSets (id) {
  const url = `https://genshin.honeyhunterworld.com/i_${id}/?lang=CHS`
  let req = await fetch(url, { agent })
  let txt = await req.text()
  let sTxt = /sortable_data.push\((.*)\)/.exec(txt)
  let ret = {}

  if (sTxt && sTxt[1]) {
    // eslint-disable-next-line no-eval
    let arrs = eval(sTxt[1])
    lodash.forEach(arrs, (ds) => {
      let $ = cheerio.load(ds.join(''))
      let a = $('a:eq(1)')
      let idRet = /i_(.+)\//.exec(a.attr('href'))
      let nRet = /^(\w+)\s/.exec($('a:last').text())
      if (nRet && idRet) {
        ret[artiIdx[nRet[1]]] = {
          id: idRet[1],
          name: a.text()
        }
      }
    })
  }
  return ret
}

async function down (sets = '') {
  const url = 'https://genshin.honeyhunterworld.com/fam_art_set/?lang=CHS'
  let req = await fetch(url, { agent })
  let txt = await req.text()
  if (sets) {
    sets = sets.split(',')
  }

  let ret = {}

  let sTxt = /sortable_data.push\((.*)\)/.exec(txt)
  if (sTxt && sTxt[1]) {
    // eslint-disable-next-line no-eval
    let txt = sTxt[1]
    txt = txt.replace(/<script>.+<\/script>/g, '')
    let tmp
    try {
      tmp = eval(txt)
    } catch (e) {
    }

    lodash.forEach(tmp, (ds) => {
      let na = cheerio.load(ds[0])('a:last')
      let idRet = /i_(\w+)\//.exec(na.attr('href'))
      if (idRet && idRet[1]) {
        let effect = {}
        let $ = cheerio.load(`<div>${ds[2]}</div>`)

        $('span').each(function () {
          let setRet = /(\d)-Piece:\s*(.*)。?$/.exec($(this).text())
          if (setRet && setRet[1]) {
            effect[setRet[1]] = setRet[2]
          }
        })
        tmp = {
          id: idRet[1],
          name: na.find('img').attr('alt'),
          sets: {},
          effect
        }
        ret[tmp.id] = tmp
      }
    })
  }
  let imgs = []
  for (let idx in ret) {
    let ds = ret[idx]
    if (sets) {
      if (!sets.includes(ds.name)) {
        continue
      }
    }
    ds.sets = await getSets(ds.id)
    console.log(`arti ${ds.id}:${ds.name} Done`)
    Data.createDir(`resources/meta/artifact/imgs/${ds.name}`, 'miao')

    lodash.forEach(ds.sets, (s, idx) => {
      imgs.push({
        url: `img/i_${s.id}.webp`,
        file: `${ds.name}/${idx}.webp`
      })
    })
  }
  Data.createDir('resources/meta/artifact', 'miao')
  Data.writeJSON('resources/meta/artifact/data.json', ret, 'miao')

  const _path = process.cwd()
  const _root = _path + '/plugins/miao-plugin/'
  const _aRoot = _root + 'resources/meta/artifact/imgs/'
  await Data.asyncPool(5, imgs, async function (ds) {
    if (fs.existsSync(`${_aRoot}/${ds.file}`)) {
      // console.log(`已存在，跳过 ${ds.file}`)
      return true
    }

    try {
      let stream = fs.createWriteStream(`${_aRoot}/${ds.file}.tmp`)
      await request('https://genshin.honeyhunterworld.com/' + ds.url).pipe(stream)
      return new Promise((resolve) => {
        stream.on('finish', () => {
          fs.rename(`${_aRoot}/${ds.file}.tmp`, `${_aRoot}/${ds.file}`, () => {
            console.log(`图像下载成功: ${ds.file}`)
            resolve()
          })
        })
      })
    } catch (e) {
      console.log(`图像下载失败: ${ds.file}`)
      console.log(e)
      return false
    }
  })
}

await down()