feat: support github oauth

2024-10-14 00:02:58 +08:00
parent 2ed57d598b
commit 272830f6ae
18 changed files with 395 additions and 74 deletions
@@ -0,0 +1,45 @@
+import type { NewsItem } from "@shared/types"
+import type { Database } from "db0"
+import type { CacheInfo } from "../types"
+
+export class Cache {
+  private db
+  constructor(db: Database) {
+    this.db = db
+  }
+
+  async init() {
+    await this.db.prepare(`
+      CREATE TABLE IF NOT EXISTS cache (
+        id TEXT PRIMARY KEY,
+        updated INTEGER,
+        data TEXT
+      );
+    `).run()
+    logger.success(`init cache table`)
+  }
+
+  async set(key: string, value: NewsItem[]) {
+    const now = Date.now()
+    await this.db.prepare(
+      `INSERT OR REPLACE INTO cache (id, data, updated) VALUES (?, ?, ?)`,
+    ).run(key, JSON.stringify(value), now)
+    logger.success(`set ${key} cache`)
+  }
+
+  async get(key: string): Promise<CacheInfo> {
+    const row: any = await this.db.prepare(`SELECT id, data, updated FROM cache WHERE id = ?`).get(key)
+    const r = row
+      ? {
+          ...row,
+          data: JSON.parse(row.data),
+        }
+      : undefined
+    logger.success(`get ${key} cache`)
+    return r
+  }
+
+  async delete(key: string) {
+    return await this.db.prepare(`DELETE FROM cache WHERE id = ?`).run(key)
+  }
+}
@@ -0,0 +1,65 @@
+import type { Database } from "@ourongxing/db0"
+import type { UserInfo } from "#/types"
+
+export class UserTable {
+  private db
+  constructor(db: Database) {
+    this.db = db
+  }
+
+  async init() {
+    await this.db.prepare(`
+      CREATE TABLE IF NOT EXISTS user (
+        id TEXT PRIMARY KEY,
+        email TEXT,
+        data TEXT,
+        type TEXT,
+        created INTEGER,
+        updated INTEGER
+      );
+    `).run()
+    logger.success(`init user table`)
+  }
+
+  async addUser(id: string, email: string, type: "github") {
+    const u = await this.getUser(id)
+    const now = Date.now()
+    if (!u) {
+      await this.db.prepare(`INSERT INTO user (id, email, data, type, created, updated) VALUES (?, ?, ?, ?, ?, ?)`)
+        .run(id, email, "", type, now, now)
+      logger.success(`add user ${id}`)
+    } else if (u.email !== email && u.type !== type) {
+      await this.db.prepare(`REPLACE INTO user (id, email, updated) VALUES (?, ?, ?)`).run(id, email, now)
+      logger.success(`update user ${id} email`)
+    }
+  }
+
+  async getUser(id: string) {
+    return (await this.db.prepare(`SELECT id, email, data, created, updated FROM user WHERE id = ?`).get(id)) as UserInfo
+  }
+
+  async setData(key: string, value: string) {
+    const now = Date.now()
+    const state = await this.db.prepare(
+      `REPLACE INTO user (id, data, updated) VALUES (?, ?, ?)`,
+    ).run(key, JSON.stringify(value), now)
+    if (!state.success) throw new Error(`set user ${key} data failed`)
+    logger.success(`set ${key} cache`)
+  }
+
+  async getData(id: string) {
+    const row: any = await this.db.prepare(`SELECT data, update FROM user WHERE id = ?`).get(id)
+    if (!row || !row.data) throw new Error(`user ${id} not found`)
+    logger.success(`get ${id} data`)
+    return row.data as {
+      data: string
+      updated: number
+    }
+  }
+
+  async deleteUser(key: string) {
+    const state = await this.db.prepare(`DELETE FROM user WHERE id = ?`).run(key)
+    if (!state.success) throw new Error(`delete user ${key} failed`)
+    logger.success(`delete user ${key}`)
+  }
+}