♻️ 重构用户祈愿数据库相关处理

src/plugins/Sqlite/index.ts

@@ -18,7 +18,7 @@ import {
   insertRecordData,
   insertRoleData,
 } from "./sql/insertData";
-import { importUIAFData, importUIGFData } from "./sql/updateData";
+import { importUIAFData } from "./sql/updateData";
 
 class Sqlite {
   /**
@@ -115,15 +115,13 @@ class Sqlite {
 
   /**
    * @description 插入 Account 数据
-   * @since Beta v0.4.1
+   * @since Beta v0.4.7
    * @param {TGApp.User.Account.Game[]} accounts
    * @returns {Promise<void>}
    */
   public async saveAccount(accounts: TGApp.User.Account.Game[]): Promise<void> {
     const db = await this.getDB();
-    // 为了防止多账号的情况，先清空数据表
-    const clear = "DELETE FROM GameAccount WHERE 1=1;";
-    await db.execute(clear);
+    await db.execute("DELETE FROM GameAccount WHERE true;");
     for (const a of accounts) {
       const sql = insertGameAccountData(a);
       await db.execute(sql);
@@ -399,98 +397,6 @@ class Sqlite {
     return res;
   }
 
-  /**
-   * @description 获取已有 uid 列表
-   * @since Beta v0.3.3
-   * @returns {Promise<string[]>}
-   */
-  public async getUidList(): Promise<string[]> {
-    const db = await this.getDB();
-    const sql = "SELECT DISTINCT uid FROM GachaRecords";
-    const res: Array<{ uid: string }> = await db.select(sql);
-    return res.map((item) => item.uid);
-  }
-
-  /**
-   * @description 获取指定 uid 的用户角色数据
-   * @since Beta v0.3.3
-   * @param {string} uid 用户 uid
-   * @returns {Promise<TGApp.Sqlite.GachaRecords.SingleTable[]>}
-   */
-  public async getGachaRecords(uid: string): Promise<TGApp.Sqlite.GachaRecords.SingleTable[]> {
-    const db = await this.getDB();
-    const sql = `SELECT *
-                 FROM GachaRecords
-                 WHERE uid = '${uid}'`;
-    return await db.select(sql);
-  }
-
-  /**
-   * @description 删除指定 uid 的祈愿数据
-   * @since Beta v0.3.3
-   * @param {string} uid 用户 uid
-   * @returns {Promise<void>}
-   */
-  public async deleteGachaRecords(uid: string): Promise<void> {
-    const db = await this.getDB();
-    const sql = `DELETE
-                 FROM GachaRecords
-                 WHERE uid = '${uid}'`;
-    await db.execute(sql);
-  }
-
-  /**
-   * @description 合并祈愿数据
-   * @since Beta v0.3.3
-   * @param {string} uid UID
-   * @param {TGApp.Plugins.UIGF.GachaItem[]} data UIGF 数据
-   * @returns {Promise<void>}
-   */
-  public async mergeUIGF(uid: string, data: TGApp.Plugins.UIGF.GachaItem[]): Promise<void> {
-    const db = await this.getDB();
-    const sql = importUIGFData(uid, data);
-    for (const item of sql) {
-      await db.execute(item);
-    }
-  }
-
-  /**
-   * @description 判断今天是否是某个角色的生日
-   * @since Beta v0.3.6
-   * @returns {Promise<false|string>}
-   */
-  public async isBirthday(): Promise<false | string> {
-    const db = await this.getDB();
-    const dateNow = new Date();
-    const date = `${dateNow.getMonth() + 1},${dateNow.getDate()}`;
-    const sql = `SELECT name
-                 FROM AppCharacters
-                 WHERE birthday = '${date}';`;
-    const res: Array<{ name: string }> = await db.select(sql);
-    if (res.length === 0) return false;
-    return res.map((item) => item.name).join("、");
-  }
-
-  /**
-   * @description 用于检测祈愿增量更新的 gacha id
-   * @since Beta v0.4.4
-   * @param {string} uid 用户 uid
-   * @param {string} type 卡池类型
-   * @returns {Promise<string|undefined>}
-   */
-  async getGachaCheck(uid: string, type: string): Promise<string | undefined> {
-    const db = await this.getDB();
-    const sql = `SELECT id
-                 FROM GachaRecords
-                 WHERE uid = '${uid}'
-                   AND gachaType = '${type}'
-                 ORDER BY id DESC
-                 LIMIT 1;`;
-    const res: Array<{ id: string }> = await db.select(sql);
-    if (res.length === 0) return undefined;
-    return res[0].id;
-  }
-
   /**
    * @description 检测特定表是否存在
    * @since Beta v0.4.5

src/plugins/Sqlite/modules/userGacha.ts

@@ -0,0 +1,147 @@
+/**
+ * @file plugins/Sqlite/modules/userGacha.ts
+ * @description 用户祈愿模块
+ * @since Beta v0.4.7
+ */
+
+import { AppCharacterData, AppWeaponData } from "../../../data/index";
+import TGSqlite from "../index";
+import { importUIGFData } from "../sql/updateData";
+
+type gachaItemTypeRes =
+  | ["角色", TGApp.App.Character.WikiBriefInfo]
+  | ["武器", TGApp.App.Weapon.WikiBriefInfo]
+  | ["未知", "未知"];
+
+/**
+ * @description 根据 item_id 获取角色/武器类型
+ * @since Beta v0.4.7
+ * @param {string} item_id - item_id
+ * @return {gachaItemTypeRes}
+ */
+function getGachaItemType(item_id: string): gachaItemTypeRes {
+  const findAvatar = AppCharacterData.find((i) => i.id === item_id);
+  if (findAvatar !== undefined) return ["角色", findAvatar];
+  const findWeapon = AppWeaponData.find((i) => i.id === item_id);
+  if (findWeapon !== undefined) return ["武器", findWeapon];
+  return ["未知", "未知"];
+}
+
+/**
+ * @description 转换祈愿数据，防止多语言
+ * @since Beta v0.4.7
+ * @param {TGApp.Plugins.UIGF.GachaItem} gacha - UIGF数据
+ * @return {TGApp.Plugins.UIGF.GachaItem} 转换后的数据
+ */
+function transGacha(gacha: TGApp.Plugins.UIGF.GachaItem): TGApp.Plugins.UIGF.GachaItem {
+  const type = getGachaItemType(gacha.item_id);
+  let res = gacha;
+  res.item_type = type[0];
+  if (type[0] === "角色") {
+    const data: TGApp.App.Character.WikiBriefInfo = type[1];
+    res = {
+      gacha_type: gacha.gacha_type,
+      item_id: gacha.item_id,
+      count: gacha.count ?? "1",
+      time: gacha.time,
+      name: data.name,
+      item_type: "角色",
+      rank_type: data.star.toString(),
+      id: gacha.id,
+      uigf_gacha_type: gacha.uigf_gacha_type,
+    };
+  } else if (type[0] === "武器") {
+    const data: TGApp.App.Weapon.WikiBriefInfo = type[1];
+    res = {
+      gacha_type: gacha.gacha_type,
+      item_id: gacha.item_id,
+      count: gacha.count ?? "1",
+      time: gacha.time,
+      name: data.name,
+      item_type: "武器",
+      rank_type: data.star.toString(),
+      id: gacha.id,
+      uigf_gacha_type: gacha.uigf_gacha_type,
+    };
+  }
+  return res;
+}
+
+/**
+ * @description 获取数据库的uid列表
+ * @since Beta v0.4.7
+ * @return {Promise<string[]>}
+ */
+async function getUidList(): Promise<string[]> {
+  const db = await TGSqlite.getDB();
+  type resType = Array<{ uid: string }>;
+  const res = await db.select<resType>("SELECT DISTINCT uid FROM GachaRecords;");
+  return res.map((i) => i.uid);
+}
+
+/**
+ * @description 获取检测增量更新的记录 ID
+ * @since Beta v0.4.7
+ * @param {string} uid - UID
+ * @param {string} type - 类型
+ * @returns {Promise<string|undefined>}
+ */
+async function getGachaCheck(uid: string, type: string): Promise<string | undefined> {
+  const db = await TGSqlite.getDB();
+  type resType = Array<{ id: string }>;
+  const res = await db.select<resType>(
+    "SELECT id FROM GachaRecords WHERE uid = ? AND gachaType = ? ORDER BY id DESC LIMIT 1;",
+    [uid, type],
+  );
+  if (res.length === 0) return undefined;
+  return res[0].id;
+}
+
+/**
+ * @description 获取用户祈愿记录
+ * @since Beta v0.4.7
+ * @param {string} uid - UID
+ * @return {Promise<TGApp.Sqlite.GachaRecords.SingleTable[]>}
+ */
+async function getGachaRecords(uid: string): Promise<TGApp.Sqlite.GachaRecords.SingleTable[]> {
+  const db = await TGSqlite.getDB();
+  return await db.select("SELECT * FROM GachaRecords WHERE uid = ?;", [uid]);
+}
+
+/**
+ * @description 删除指定UID的祈愿记录
+ * @since Beta v0.4.7
+ * @param {string} uid - UID
+ * @return {Promise<void>}
+ */
+async function deleteGachaRecords(uid: string): Promise<void> {
+  const db = await TGSqlite.getDB();
+  await db.execute("DELETE FROM GachaRecords WHERE uid = ?;", [uid]);
+}
+
+/**
+ * @description 合并祈愿数据
+ * @since Beta v0.4.7
+ * @param {string} uid - UID
+ * @param {TGApp.Plugins.UIGF.GachaItem[]} data - UIGF数据
+ * @return {Promise<void>}
+ */
+async function mergeUIGF(uid: string, data: TGApp.Plugins.UIGF.GachaItem[]): Promise<void> {
+  const db = await TGSqlite.getDB();
+  for (const gacha of data) {
+    const trans = transGacha(gacha);
+    console.log(trans);
+    const sql = importUIGFData(uid, trans);
+    await db.execute(sql);
+  }
+}
+
+const TSUserGacha = {
+  getUidList,
+  getGachaCheck,
+  getGachaRecords,
+  deleteGachaRecords,
+  mergeUIGF,
+};
+
+export default TSUserGacha;

src/plugins/Sqlite/sql/updateData.ts

@@ -1,11 +1,9 @@
 /**
- * @file plugins Sqlite sql updateData.ts
+ * @file plugins/Sqlite/sql/updateData.ts
  * @description 更新数据
- * @author BTMuli <bt-muli@outlook.com>
- * @since Alpha v0.2.0
+ * @since Beta v0.4.7
  */
 
-// utils
 import minifySql from "../../../utils/minifySql";
 
 /**
@@ -26,15 +24,21 @@ export function importUIAFData(data: TGApp.Plugins.UIAF.Achievement[]): string[]
         .replace("T", " ")
         .slice(0, 19);
       sql = `
-        UPDATE Achievements
-        SET isCompleted = 1, completedTime = '${completedTime}', progress = ${achievement.current}, updated = datetime('now', 'localtime')
-        WHERE id = ${achievement.id} AND (isCompleted = 0 OR completedTime != '${completedTime}' OR progress != ${achievement.current});
+          UPDATE Achievements
+          SET isCompleted   = 1,
+              completedTime = '${completedTime}',
+              progress      = ${achievement.current},
+              updated       = datetime('now', 'localtime')
+          WHERE id = ${achievement.id}
+            AND (isCompleted = 0 OR completedTime != '${completedTime}' OR progress != ${achievement.current});
       `;
     } else {
       sql = `
-        UPDATE Achievements
-        SET progress = ${achievement.current}, updated = datetime('now', 'localtime')
-        WHERE id = ${achievement.id} AND progress != ${achievement.current};
+          UPDATE Achievements
+          SET progress = ${achievement.current},
+              updated  = datetime('now', 'localtime')
+          WHERE id = ${achievement.id}
+            AND progress != ${achievement.current};
       `;
     }
     return sqlRes.push(minifySql(sql));
@@ -43,37 +47,30 @@ export function importUIAFData(data: TGApp.Plugins.UIAF.Achievement[]): string[]
 }
 
 /**
- * @description 导入UIGF数据
- * @since Alpha v0.2.3
+ * @description 导入UIGF数据-单项
+ * @since Beta v0.4.7
  * @param {string} uid - UID
- * @param {TGApp.Plugins.UIGF.GachaItem[]} data - UIGF数据
- * @returns {string[]} sql
+ * @param {TGApp.Plugins.UIGF.GachaItem} gacha - UIGF数据
+ * @returns {string} sql
  */
-export function importUIGFData(uid: string, data: TGApp.Plugins.UIGF.GachaItem[]): string[] {
-  const sqlRes: string[] = [];
-  data.forEach((gacha) => {
-    const sql = `
+export function importUIGFData(uid: string, gacha: TGApp.Plugins.UIGF.GachaItem): string {
+  const sql = `
       INSERT INTO GachaRecords (uid, gachaType, itemId, count, time, name, type, rank, id, uigfType, updated)
-      VALUES ('${uid}', '${gacha.gacha_type}', '${gacha.item_id ?? null}', '${
-        gacha.count ?? null
-      }', '${gacha.time}',
-              '${gacha.name}', '${gacha.item_type ?? null}', '${gacha.rank_type ?? null}', '${
-                gacha.id
-              }',
+      VALUES ('${uid}', '${gacha.gacha_type}', '${gacha.item_id ?? null}', '${gacha.count ?? null}', '${gacha.time}',
+              '${gacha.name}', '${gacha.item_type ?? null}', '${gacha.rank_type ?? null}', '${gacha.id}',
               '${gacha.uigf_gacha_type}', datetime('now', 'localtime'))
-      ON CONFLICT (id) DO UPDATE SET
-        uid       = '${uid}',
-        gachaType = '${gacha.gacha_type}',
-        uigfType  = '${gacha.uigf_gacha_type}',
-        time      = '${gacha.time}',
-        itemId    = '${gacha.item_id ?? null}',
-        count     = '${gacha.count ?? null}',
-        name      = '${gacha.name}',
-        type  = '${gacha.item_type ?? null}',
-        rank  = '${gacha.rank_type ?? null}',
-        updated   = datetime('now', 'localtime'); 
-    `;
-    sqlRes.push(minifySql(sql));
-  });
-  return sqlRes;
+      ON CONFLICT (id)
+          DO UPDATE
+          SET uid       = '${uid}',
+              gachaType = '${gacha.gacha_type}',
+              uigfType  = '${gacha.uigf_gacha_type}',
+              time      = '${gacha.time}',
+              itemId    = '${gacha.item_id ?? null}',
+              count     = '${gacha.count ?? null}',
+              name      = '${gacha.name}',
+              type      = '${gacha.item_type ?? null}',
+              rank      = '${gacha.rank_type ?? null}',
+              updated   = datetime('now', 'localtime');
+  `;
+  return minifySql(sql);
 }