進一步完善對外 API

readme.md

@@ -0,0 +1,11 @@
+## TheWhiteSilk
+
+Under developing...
+
+### TODO
+
+* [x] 接口完備
+* [ ] 對外接口
+* [ ] 單元測試
+* [ ] 基本可用
+* [ ] 負載均衡

src/api/ApiManager.ts

@@ -2,6 +2,7 @@ import BaseApi from './BaseApi.ts'
 import HttpServerLike from '../types/HttpServerLike.ts'
 import UserApi from "./UserApi.ts"
 import SocketIo from "socket.io"
+import UnknownFunction from "../types/UnknownFunction.ts";
 
 export default class ApiManager {
     static httpServer: HttpServerLike
@@ -14,9 +15,22 @@ export default class ApiManager {
     static getHttpServer() {
         return this.httpServer
     }
+    static getSocketIoServer() {
+        return this.socketIoServer
+    }
     static initAllApis() {
         this.apis_instance = {
             user: new UserApi()
         }
     }
+    static initEvents() {
+        const io = this.socketIoServer
+        io.on('connection', (socket) => {
+            socket.on("The_White_Silk", (name: string, args: {}, callback: UnknownFunction) => {
+                if (name == null || args == null) return callback({
+                    
+                })
+            })
+        })
+    }
 }

src/api/BaseApi.ts

@@ -1,4 +1,5 @@
 import UnknownFunction from '../types/UnknownFunction.ts'
+import ApiManager from "./ApiManager.ts";
 
 export default abstract class BaseApi {
     abstract getName(): string
@@ -7,6 +8,7 @@ export default abstract class BaseApi {
     }
     abstract onInit(): void
     registerEvent(name: string, func: UnknownFunction) {
+        const io = ApiManager.getSocketIoServer()
         
     }
 }

src/data/FileManager.ts

@@ -1,38 +1,39 @@
-import { DatabaseSync } from "node:sqlite"
+import { DatabaseSync, SQLInputValue } from "node:sqlite"
 import { Buffer } from "node:buffer"
 import path from 'node:path'
 import crypto from 'node:crypto'
 import fs_sync from 'node:fs'
-
+import chalk from "chalk"
 import { fileTypeFromBuffer } from 'file-type'
 
-export default class FileManager {
+import config from "../config.ts"
-    static FileBean = class {
+
+class FileBean {
     declare count: number
     declare name: string
     declare hash: string
     declare mime: string
-        declare chatid: string | null
+    declare chatid?: string
     declare upload_time: number
     declare last_used_time: number
-    }
+}
 
-    static File = class {
+class File {
-        declare bean: FileManager.FileBean
+    declare bean: FileBean
-        constructor(bean: FileManager.FileBean) {
+    constructor(bean: FileBean) {
         this.bean = bean
     }
-        private setAttr(key: string, value: unknown): void {
+    private setAttr(key: string, value: SQLInputValue) {
         FileManager.database.prepare(`UPDATE ${FileManager.table_name} SET ${key} = ? WHERE count = ?`).run(value, this.bean.count)
         this.bean[key] = value
     }
-        getMime(): string {
+    getMime() {
         return this.bean.mime
     }
-        getName(): string {
+    getName() {
         return this.bean.name
     }
-        getFilePath(): string {
+    getFilePath() {
         const hash = this.bean.hash
         return path.join(
             config.data_path,
@@ -43,23 +44,27 @@ export default class FileManager {
             this.bean.hash
         )
     }
-        getChatId(): string | null {
+    getChatId() {
         return this.bean.chatid
     }
-        getUploadTime(): number {
+    getUploadTime() {
         return this.bean.upload_time
     }
-        getLastUsedTime(): number {
+    getLastUsedTime() {
         return this.bean.last_used_time
     }
-        readSync(): Buffer {
+    readSync() {
         this.setAttr("last_used_time", Date.now())
         return fs_sync.readFileSync(this.getFilePath())
     }
-    }
+}
+
+export default class FileManager {
+    static FileBean = FileBean
+    static File = File
 
     static table_name: string = "FileReferences"
-    private static database: DatabaseSync = FileManager.init()
+    static database: DatabaseSync = FileManager.init()
     private static init(): DatabaseSync {
         const db: DatabaseSync = new DatabaseSync(path.join(config.data_path, FileManager.table_name + '.db'))
         db.exec(`
@@ -76,15 +81,15 @@ export default class FileManager {
         return db
     }
 
-    static uploadFile(fileName: string, data: Buffer, chatId: string | null) {
+    static async uploadFile(fileName: string, data: Buffer, chatId?: string) {
         const hash = crypto.createHash('sha256').update(data).digest('hex')
         try {
             return FileManager.findByHash(hash)
-        } catch(_e) {
+        } catch (_e) {
             // Do nothing...
         }
 
-        const mime = fileTypeFromBuffer(data)
+        const mime = (await fileTypeFromBuffer(data))?.mime || 'application/octet-stream'
         fs_sync.writeFileSync(
             path.join(
                 config.data_path,
@@ -110,7 +115,7 @@ export default class FileManager {
                     fileName,
                     hash,
                     mime,
-                    chatId,
+                    chatId || null,
                     Date.now(),
                     -1
                 ).lastInsertRowid
@@ -118,15 +123,15 @@ export default class FileManager {
         )
     }
 
-    private static findAllBeansByCondition(condition: string, ...args: unknown[]): FileManager.FileBean[] {
+    private static findAllBeansByCondition(condition: string, ...args: SQLInputValue[]): FileBean[] {
-        return FileManager.database.prepare(`SELECT * FROM ${FileManager.table_name} WHERE ${condition};`).all(...args)
+        return FileManager.database.prepare(`SELECT * FROM ${FileManager.table_name} WHERE ${condition};`).all(...args) as unknown as FileBean[]
     }
-    static findByHash(hash: string): FileManager.File {
+    static findByHash(hash: string): File {
         const beans = FileManager.findAllBeansByCondition('hash = ?', hash)
         if (beans.length == 0)
             throw new Error(`找不到 hash 为 ${hash} 的文件`)
         else if (beans.length > 1)
-            console.error(chalk.red(`警告: 查询 hash = ${id} 时, 查询到多个相同 Hash 的文件`))
+            console.error(chalk.red(`警告: 查询 hash = ${hash} 时, 查询到多个相同 Hash 的文件`))
         return new FileManager.File(beans[0])
     }
 }

src/types/ApiCallbackMessage.ts

@@ -0,0 +1,4 @@
+type ApiCallbackMessage = {
+    msg: string,
+    code: 200 | 400 | 401 | 403 | 404 | 500 | 501
+}