Merge branch 'pr/Gemini2035/252' into Gemini2035-guan/temp1

Author: guansss

package.json

@@ -14,9 +14,10 @@
     "lint:fix": "npm-run-all --parallel lint:fix:eslint lint:fix:prettier",
     "lint:fix:eslint": "eslint --fix src/**/*",
     "lint:fix:prettier": "prettier --write src/**/*",
-    "scripts:update": "yarn install && npm-run-all --parallel scripts:update-operators scripts:update-operator-avatars",
+    "scripts:update": "yarn install && npm-run-all --parallel scripts:update-operators scripts:update-operator-avatars scripts:update-prof-icons",
     "scripts:update-operators": "esno scripts/update-operators.ts",
-    "scripts:update-operator-avatars": "esno scripts/update-operator-avatars.ts"
+    "scripts:update-operator-avatars": "esno scripts/update-operator-avatars.ts",
+    "scripts:update-prof-icons": "yarn install && esno scripts/update-prof-icons.ts"
   },
   "dependencies": {
     "@blueprintjs/core": "^4.15.1",

public/assets/prof-icons/CASTER.png

@@ -1,8 +1,21 @@
+import { access } from 'fs/promises'
 import { uniq, uniqBy } from 'lodash-es'
 import fetch from 'node-fetch'
-import pinyin from 'pinyin'
+import { pinyin } from 'pinyin'
 import simplebig from 'simplebig'
 
+type Profession = { id: string; name: string }
+type Professions = (Profession & { sub: Profession[] })[]
+
+export async function fileExists(file: string) {
+  try {
+    await access(file)
+    return true
+  } catch (e) {
+    return false
+  }
+}
+
 function pinyinify(name: string) {
   return [
     pinyin(name, {
@@ -19,14 +32,15 @@ function pinyinify(name: string) {
 }
 
 function transformOperatorName(name: string) {
-  const cleanedSimplifiedName = name.replace(/[”“"]/g, '')
+  const cleanedName = name.replace(/[”“"]/g, '')
 
   const traditional = simplebig.s2t(name) as string
   const cleanedTraditional = traditional.replace(/[”“"]/g, '')
+
   return {
     name,
     alias: uniq([
-      ...pinyinify(cleanedSimplifiedName),
+      ...pinyinify(cleanedName),
       traditional,
       cleanedTraditional,
       ...pinyinify(cleanedTraditional),
@@ -36,23 +50,69 @@ function transformOperatorName(name: string) {
 
 const CHARACTER_TABLE_JSON_URL =
   'https://raw.githubusercontent.com/Kengxxiao/ArknightsGameData/master/zh_CN/gamedata/excel/character_table.json'
+const UNIEQUIP_TABLE_JSON_URL =
+  'https://raw.githubusercontent.com/Kengxxiao/ArknightsGameData/master/zh_CN/gamedata/excel/uniequip_table.json'
 
 const CHARACTER_BLOCKLIST = [
+  'char_512_aprot', // 暮落(集成战略)：It's just not gonna be there.
   'token_10012_rosmon_shield', // 迷迭香的战术装备：It's just not gonna be there.
 ]
 
-export async function getOperatorNames() {
-  const resp = (await fetch(CHARACTER_TABLE_JSON_URL).then((res) =>
-    res.json(),
-  )) as any
-  const ids = Object.keys(resp)
+const PROFESSION_NAMES = {
+  MEDIC: '医疗',
+  WARRIOR: '近卫',
+  SPECIAL: '特种',
+  SNIPER: '狙击',
+  PIONEER: '先锋',
+  TANK: '重装',
+  CASTER: '术师',
+  SUPPORT: '辅助',
+}
+
+async function json(url: string) {
+  return (await (await fetch(url)).json()) as any
+}
+
+export async function getOperators() {
+  const [charTable, uniequipTable] = await Promise.all([
+    json(CHARACTER_TABLE_JSON_URL),
+    json(UNIEQUIP_TABLE_JSON_URL),
+  ])
+
+  const { subProfDict } = uniequipTable
+
+  const opIds = Object.keys(charTable)
+  const professions: Professions = []
   const result = uniqBy(
-    ids.flatMap((el) => {
-      const op = resp[el]
+    opIds.flatMap((id) => {
+      const op = charTable[id]
       if (['TRAP'].includes(op.profession)) return []
+
+      if (!['TOKEN'].includes(op.profession)) {
+        const prof = professions.find((p) => p.id === op.profession)
+        if (!prof) {
+          professions.push({
+            id: op.profession,
+            name: PROFESSION_NAMES[op.profession],
+            sub: [
+              {
+                id: op.subProfessionId,
+                name: subProfDict[op.subProfessionId].subProfessionName,
+              },
+            ],
+          })
+        } else if (!prof.sub.find((p) => p.id === op.subProfessionId)) {
+          prof.sub.push({
+            id: op.subProfessionId,
+            name: subProfDict[op.subProfessionId].subProfessionName,
+          })
+        }
+      }
+
       return [
         {
-          id: el,
+          id: id,
+          subProf: op.subProfessionId,
           ...transformOperatorName(op.name),
           alt_name: op.appellation,
         },
@@ -62,5 +122,8 @@ export async function getOperatorNames() {
   ).sort((a, b) => {
     return pinyin.compare(a.name, b.name)
   })
-  return result.filter((el) => !CHARACTER_BLOCKLIST.includes(el.id))
+  return {
+    professions,
+    operators: result.filter((el) => !CHARACTER_BLOCKLIST.includes(el.id)),
+  }
 }