From 271543a0028bb3b2a80b538ec03000b50f2dba13 Mon Sep 17 00:00:00 2001 From: BTMuli Date: Fri, 19 May 2023 09:35:25 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20=E6=A0=BC=E5=BC=8F=E5=8C=96?= =?UTF-8?q?=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/utils/TGSqlite.ts | 278 +++++++++++++++++++++++------------------- 1 file changed, 154 insertions(+), 124 deletions(-) diff --git a/src/utils/TGSqlite.ts b/src/utils/TGSqlite.ts index 2e8aad92..0bc7bd09 100644 --- a/src/utils/TGSqlite.ts +++ b/src/utils/TGSqlite.ts @@ -12,20 +12,20 @@ import { importUIAFData, initSQLiteData, initSQLiteTable } from "./TGSql"; import { getUiafStatus } from "./UIAF"; class TGSqlite { /** - * @description 数据库地址 - * @private - * @type {string} - * @memberof TGSqlite - * @since Alpha v0.1.4 - */ + * @description 数据库地址 + * @private + * @type {string} + * @memberof TGSqlite + * @since Alpha v0.1.4 + */ private readonly dbPath: string = "sqlite:tauri-genshin.db"; /** - * @description 数据库包含的表 - * @private - * @type {string[]} - * @memberof TGSqlite - * @since Alpha v0.1.4 - */ + * @description 数据库包含的表 + * @private + * @type {string[]} + * @memberof TGSqlite + * @since Alpha v0.1.4 + */ private readonly tables: string[] = [ "AppData", "Achievements", @@ -35,12 +35,12 @@ class TGSqlite { ]; /** - * @description 初始化数据库 - * @memberof TGSqlite - * @since Alpha v0.1.4 - * @returns {Promise} - * @memberof TGSqlite - */ + * @description 初始化数据库 + * @memberof TGSqlite + * @since Alpha v0.1.4 + * @returns {Promise} + * @memberof TGSqlite + */ public async init (): Promise { const db = await Database.load(this.dbPath); const sqlT = initSQLiteTable(); @@ -55,11 +55,11 @@ class TGSqlite { } /** - * @description 获取数据库信息 - * @memberof TGSqlite - * @since Alpha v0.1.4 - * @returns {Promise<{ key: string, value: string, updated: string }[]>} - */ + * @description 获取数据库信息 + * @memberof TGSqlite + * @since Alpha v0.1.4 + * @returns {Promise<{ key: string, value: string, updated: string }[]>} + */ public async getAppData (): Promise> { const db = await Database.load(this.dbPath); const sql = "SELECT * FROM AppData;"; @@ -69,14 +69,14 @@ class TGSqlite { } /** - * @description 封装-根据 table keys 获取数据 - * @memberof TGSqlite - * @since Alpha v0.2.0 - * @param {string} table 表名 - * @param {string} keyName 键名 - * @param {string} keyValue 键值 - * @returns {Promise} 数据 - */ + * @description 封装-根据 table keys 获取数据 + * @memberof TGSqlite + * @since Alpha v0.2.0 + * @param {string} table 表名 + * @param {string} keyName 键名 + * @param {string} keyValue 键值 + * @returns {Promise} 数据 + */ public async getDataByKey (table: string, keyName: string, keyValue: string): Promise { const db = await Database.load(this.dbPath); const sql = `SELECT * FROM ${table} WHERE ${keyName}='${keyValue}';`; @@ -86,12 +86,12 @@ class TGSqlite { } /** - * @description 封装-保存数据 - * @memberof TGSqlite - * @since Alpha v0.2.0 - * @param {string} sql sql语句 - * @returns {Promise} - */ + * @description 封装-保存数据 + * @memberof TGSqlite + * @since Alpha v0.2.0 + * @param {string} sql sql语句 + * @returns {Promise} + */ public async saveData (sql: string): Promise { const db = await Database.load(this.dbPath); await db.execute(sql); @@ -99,46 +99,68 @@ class TGSqlite { } /** - * @description 输入 cookie - * @memberof TGSqlite - * @since Alpha v0.2.0 - * @param {string} cookie - * @returns {Promise} - */ + * @description 输入 cookie + * @memberof TGSqlite + * @since Alpha v0.2.0 + * @param {string} cookie + * @returns {Promise} + */ public async inputCookie (cookie: string): Promise { const db = await Database.load(this.dbPath); const sql = ` - INSERT INTO AppData (key, value, updated) - VALUES ('cookie', '${cookie}', datetime('now', 'localtime')) - ON CONFLICT(key) DO UPDATE SET value = '${cookie}', updated = datetime('now', 'localtime');`; + INSERT INTO AppData (key, value, updated) + VALUES ('cookie', '${cookie}', datetime('now', 'localtime')) + ON CONFLICT(key) DO UPDATE SET value = '${cookie}',updated = datetime('now', 'localtime'); + `; await db.execute(sql); await db.close(); } /** - * @description 保存 appData - * @memberof TGSqlite - * @since Alpha v0.2.0 - * @param {string} key - * @param {string} value - * @returns {Promise} - */ + * @description 获取 cookie + * @memberof TGSqlite + * @since Alpha v0.2.0 + * @returns {Promise} + */ + public async getCookie (): Promise { + const db = await Database.load(this.dbPath); + const sql = "SELECT value FROM AppData WHERE key='cookie';"; + const cookieSelect: Array<{ value: string }> = await db.select(sql); + await db.close(); + const cookieGet = JSON.parse(cookieSelect[0].value); + let res = ""; + const cookieKeys = Object.keys(cookieGet); + for (const key of cookieKeys) { + if (cookieGet[key] !== "") res += `${key}=${cookieGet[key]};`; + } + return res; + } + + /** + * @description 保存 appData + * @memberof TGSqlite + * @since Alpha v0.2.0 + * @param {string} key + * @param {string} value + * @returns {Promise} + */ public async saveAppData (key: string, value: string): Promise { const db = await Database.load(this.dbPath); const sql = ` - INSERT INTO AppData (key, value, updated) - VALUES ('${key}', '${value}', datetime('now', 'localtime')) - ON CONFLICT(key) DO UPDATE SET value = '${value}', updated = datetime('now', 'localtime');`; + INSERT INTO AppData (key, value, updated) + VALUES ('${key}', '${value}', datetime('now', 'localtime')) + ON CONFLICT(key) DO UPDATE SET value = '${value}',updated = datetime('now', 'localtime'); + `; await db.execute(sql); await db.close(); } /** - * @description 已有数据表跟触发器不变的情况下，更新数据库数据 - * @memberof TGSqlite - * @since Alpha v0.1.4 - * @returns {Promise} - */ + * @description 已有数据表跟触发器不变的情况下，更新数据库数据 + * @memberof TGSqlite + * @since Alpha v0.1.4 + * @returns {Promise} + */ public async update (): Promise { const db = await Database.load(this.dbPath); const sqlD = await initSQLiteData(); @@ -149,33 +171,33 @@ class TGSqlite { } /** - * @description 检测数据库完整性 - * @memberof TGSqlite - * @since Alpha v0.1.4 - * @returns {Promise} - */ + * @description 检测数据库完整性 + * @memberof TGSqlite + * @since Alpha v0.2.0 + * @returns {Promise} + */ public async check (): Promise { const db = await Database.load(this.dbPath); - let isVertified = false; + let isVerified = false; // 检测数据表是否都存在 const sqlT = "SELECT name FROM sqlite_master WHERE type='table' ORDER BY name;"; const res: Array<{ name: string }> = await db.select(sqlT); // 考虑到 sqlite_sequence 表，所以需要 +1 if (res.length === this.tables.length + 1) { if (this.tables.every((item) => res.map((i) => i.name).includes(item))) { - isVertified = true; + isVerified = true; } } await db.close(); - return isVertified; + return isVerified; } /** - * @description 重置数据库 - * @memberof TGSqlite - * @since Alpha v0.1.4 - * @returns {Promise} - */ + * @description 重置数据库 + * @memberof TGSqlite + * @since Alpha v0.1.4 + * @returns {Promise} + */ public async reset (): Promise { const db = await Database.load(this.dbPath); this.tables.map(async (item) => { @@ -187,11 +209,11 @@ class TGSqlite { } /** - * @description 获取数据库版本及构建时间 - * @memberof TGSqlite - * @since Alpha v0.1.4 - * @returns {Promise<{ version: string, buildTime: string }>} - */ + * @description 获取数据库版本及构建时间 + * @memberof TGSqlite + * @since Alpha v0.1.4 + * @returns {Promise<{ version: string, buildTime: string }>} + */ public async getMetadata (): Promise<{ version: string, buildTime: string }> { const db = await Database.load(this.dbPath); const sql = "SELECT * FROM AppData WHERE key='appVersion' OR key='dataUpdated';"; @@ -203,48 +225,51 @@ class TGSqlite { } /** - * @description 获取成就系列列表 - * @memberof TGSqlite - * @since Alpha v0.1.4 - * @returns {Promise} - */ + * @description 获取成就系列列表 + * @memberof TGSqlite + * @since Alpha v0.1.4 + * @returns {Promise} + */ public async getAchievementSeries (): Promise { const db = await Database.load(this.dbPath); - const sql = "SELECT * FROM AchievementSeries ORDER BY `order` ASC;"; + const sql = "SELECT * FROM AchievementSeries ORDER BY `order`;"; const res: BTMuli.SQLite.AchievementSeries[] = await db.select(sql); await db.close(); return res; } /** - * @description 获取成就系列对应的名片 - * @memberof TGSqlite - * @since Alpha v0.1.4 - * @param {number} seriesId 系列 ID - * @returns {Promise} - */ + * @description 获取成就系列对应的名片 + * @memberof TGSqlite + * @since Alpha v0.1.4 + * @param {number} seriesId 系列 ID + * @returns {Promise} + */ public async getNameCard (seriesId: number): Promise { const db = await Database.load(this.dbPath); - const sql = `SELECT * FROM NameCard WHERE name=(SELECT nameCard FROM AchievementSeries WHERE id=${seriesId});`; + const sql = `SELECT * + FROM NameCard + WHERE name = (SELECT nameCard FROM AchievementSeries WHERE id = ${seriesId}); + `; const res: BTMuli.SQLite.NameCard[] = await db.select(sql); await db.close(); return res[0]; } /** - * @description 获取成就列表 - * @memberof TGSqlite - * @param {number} [seriesId] 系列 ID - * @since Alpha v0.1.4 - * @returns {Promise} - */ + * @description 获取成就列表 + * @memberof TGSqlite + * @param {number} [seriesId] 系列 ID + * @since Alpha v0.1.4 + * @returns {Promise} + */ public async getAchievements (seriesId?: number): Promise { const db = await Database.load(this.dbPath); let sql; if (seriesId) { - sql = `SELECT * FROM Achievements WHERE series=${seriesId} ORDER BY isCompleted ASC, \`order\` ASC;`; + sql = `SELECT * FROM Achievements WHERE series=${seriesId} ORDER BY isCompleted, \`order\`;`; } else { - sql = "SELECT * FROM Achievements ORDER BY isCompleted ASC, `order` ASC;"; + sql = "SELECT * FROM Achievements ORDER BY isCompleted, `order`;"; } const res: BTMuli.SQLite.Achievements[] = await db.select(sql); await db.close(); @@ -252,11 +277,11 @@ class TGSqlite { } /** - * @description 获取成就概况 - * @since Alpha v0.1.4 - * @memberof TGSqlite - * @returns {Promise<{total:number,fin:number}>} - */ + * @description 获取成就概况 + * @since Alpha v0.1.4 + * @memberof TGSqlite + * @returns {Promise<{total:number,fin:number}>} + */ public async getAchievementsOverview (): Promise<{ total: number, fin: number }> { const db = await Database.load(this.dbPath); const sql = "SELECT SUM(totalCount) AS total, SUM(finCount) AS fin FROM AchievementSeries;"; @@ -266,20 +291,25 @@ class TGSqlite { } /** - * @description 查询成就 - * @memberof TGSqlite - * @param {string} keyword 关键词 - * @since Alpha v0.1.4 - * @returns {Promise} - */ + * @description 查询成就 + * @memberof TGSqlite + * @param {string} keyword 关键词 + * @since Alpha v0.1.4 + * @returns {Promise} + */ public async searchAchievements (keyword: string): Promise { const db = await Database.load(this.dbPath); let sql; if (keyword.startsWith("v")) { const version = keyword.replace("v", ""); - sql = `SELECT * FROM Achievements WHERE version LIKE '%${version}%' ORDER BY isCompleted ASC, \`order\` ASC;`; + sql = `SELECT * FROM Achievements WHERE version LIKE '%${version}%' ORDER BY isCompleted, \`order\`;`; } else { - sql = `SELECT * FROM Achievements WHERE name LIKE '%${keyword}%' OR description LIKE '%${keyword}%' ORDER BY isCompleted ASC, \`order\` ASC;`; + sql = `SELECT * + FROM Achievements + WHERE name LIKE '%${keyword}%' + OR description LIKE '%${keyword}%' + ORDER BY isCompleted, \`order\`; + `; } const res: BTMuli.SQLite.Achievements[] = await db.select(sql); await db.close(); @@ -287,12 +317,12 @@ class TGSqlite { } /** - * @description 合并 UIAF 数据 - * @memberof TGSqlite - * @param {BTMuli.UIAF.Achievement[]} achievements UIAF 数据 - * @since Alpha v0.1.4 - * @returns {Promise} - */ + * @description 合并 UIAF 数据 + * @memberof TGSqlite + * @param {BTMuli.UIAF.Achievement[]} achievements UIAF 数据 + * @since Alpha v0.1.4 + * @returns {Promise} + */ public async mergeUIAF (achievements: TGPlugin.UIAF.Achievement[]): Promise { const db = await Database.load(this.dbPath); const sql = importUIAFData(achievements); @@ -303,11 +333,11 @@ class TGSqlite { } /** - * @description 获取 UIAF 数据 - * @memberof TGSqlite - * @since Alpha v0.1.4 - * @returns {Promise} - */ + * @description 获取 UIAF 数据 + * @memberof TGSqlite + * @since Alpha v0.1.4 + * @returns {Promise} + */ public async getUIAF (): Promise { const db = await Database.load(this.dbPath); const sql = "SELECT * FROM Achievements WHERE isCompleted = 1 OR progress > 0";