chore: MAKE LINIT HAPPT

src/data/FileManager.ts

@@ -1,38 +1,38 @@
-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 { 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 +43,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,7 +80,7 @@ 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)
@@ -84,7 +88,7 @@ export default class FileManager {
             // Do nothing...
         }
 
-        const mime = fileTypeFromBuffer(data)
+        const mime = (await fileTypeFromBuffer(data))?.mime || 'application/octet-stream'
         fs_sync.writeFileSync(
             path.join(
                 config.data_path,
@@ -118,15 +122,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])
     }
 }