1. 修复 ignore
2. 推翻旧后端代码全部重写, 使用 SQLite
This commit is contained in:
CrescentLeaf
2025-08-01 22:53:05 +08:00
parent 0ecd6c6585
commit d0b5890b99
9 changed files with 283 additions and 1 deletions

2
.gitignore vendored
View File

@@ -1,4 +1,4 @@
# 配置文件 # 配置文件
thewhitesilk_config.json thewhitesilk_config.json
# **默认**数据目录 # **默认**数据目录
data/ thewhitesilk_data/

17
src/config.ts Normal file
View File

@@ -0,0 +1,17 @@
import fs from 'node:fs/promises'
import chalk from 'chalk'
let config = {
data_path: "./thewhitesilk_data"
}
try {
config = JSON.parse(await fs.readFile('thewhitesilk_config.json'))
} catch (_e) {
console.log(chalk.yellow("配置文件貌似不存在, 正在创建..."))
await fs.writeFile('thewhitesilk_config.json', JSON.stringify(config))
}
await fs.mkdir(config.data_path, { recursive: true })
export default config

64
src/data/Chat.ts Normal file
View File

@@ -0,0 +1,64 @@
import { DatabaseSync } from "node:sqlite"
import { Buffer } from "node:buffer"
import path from 'node:path'
import config from '../config.ts'
import ChatBean from './ChatBean.ts'
/**
* Chat.ts - Wrapper and manager
* Wrap with ChatBean to directly update database
* Manage the database by itself (static)
*/
export default class Chat {
static table_name: string = "Chat"
private static database: DatabaseSync = Chat.init()
private static init(): DatabaseSync {
const db: DatabaseSync = new DatabaseSync(path.join(config.data_path, 'Chats.db'))
db.exec(`
CREATE TABLE IF NOT EXISTS ${Chat.table_name} (
/* 序号 */ count INTEGER PRIMARY KEY AUTOINCREMENT,
/* Chat ID, 哈希 */ id TEXT,
/* 设置 */ settings TEXT NOT NULL
);
`)
return db
}
private static findAllByCondition(condition: string, ...args: unknown[]): UserBean[] {
return database.prepare(`SELECT count, id FROM ${User.table_name} WHERE ${condition}`).all(...args)
}
declare bean: ChatBean
constructor(bean: ChatBean) {
this.bean = bean
}
private setAttr(key: string, value: unknown): void {
User.database.prepare(`UPDATE ${User.table_name} SET ${key} = ? WHERE id = ?`).run(value, this.bean.id)
this.bean[key] = value
}
getUserName(): string {
return this.bean.username
}
setUserName(userName: string): void {
this.setAttr("username", userName)
}
getNickName(): string {
return this.bean.nickname
}
setNickName(nickName: string): void {
this.setAttr("nickname", nickName)
}
getAvatar(): Uint8Array {
return this.bean.avatar
}
setAvatar(avatar: Buffer): void {
this.setAttr("avatar", avatar)
}
getContacts() {
}
}

3
src/data/ChatBean.ts Normal file
View File

@@ -0,0 +1,3 @@
export default interface ChatBean {
id: string
}

101
src/data/User.ts Normal file
View File

@@ -0,0 +1,101 @@
import { DatabaseSync } from "node:sqlite"
import { Buffer } from "node:buffer"
import path from 'node:path'
import crypto from 'node:crypto'
import config from '../config.ts'
import UserBean from './UserBean.ts'
/**
* User.ts - Wrapper and manager
* Wrap with UserBean to directly update database
* Manage the database by itself (static)
*/
export default class User {
static table_name: string = "Users"
private static database: DatabaseSync = User.init()
private static init(): DatabaseSync {
const db: DatabaseSync = new DatabaseSync(path.join(config.data_path, 'Users.db'))
db.exec(`
CREATE TABLE IF NOT EXISTS ${TABEL_NAME} (
/* 序号 */ count INTEGER PRIMARY KEY AUTOINCREMENT,
/* 用户 ID, 哈希 */ id TEXT
/* 注册时间, 时间戳 */ registered_time INT8 NOT NULL,
/* 用戶名, 可選 */ username TEXT,
/* 昵称 */ nickname TEXT NOT NULL,
/* 头像, 可选 */ avatar BLOB,
/* 设置 */ settings TEXT NOT NULL
);
`)
return db
}
static create(userName: string | null, nickName: string, avatar: Buffer | null): User {
return new User(
User.findAllByCondition(
'count = ?',
database.prepare(`INSERT INTO ${User.table_name} (id, username, registered_time, nickname, avatar, settings) VALUES (?, ?, ?, ?, ?)`).run(
crypto.randomUUID(),
userName,
Date.now(),
nickName,
avatar,
"{}"
).lastInsertRowid
)
)
}
private static findAllByCondition(condition: string, ...args: unknown[]): UserBean[] {
return database.prepare(`SELECT * FROM ${User.table_name} WHERE ${condition}`).all(...args)
}
private static checkLengthOrThrow(array: Array, leng: number, errMsg: string): Array {
if (array.length != leng) throw new Error(errMsg)
return array
}
static findById(id: string): User {
return new User(checkLengthOrThrow(User.findAllByCondition('id = ?', id), 1, `找不到用户 ID 为 ${id} 的用户`)[0])
}
static findByUserName(userName: string): User {
return new User(checkLengthOrThrow(User.findAllByCondition('username = ?', userName), 1, `找不到用户名为 ${userName} 的用户`)[0])
}
declare bean: UserBean
constructor(bean: UserBean) {
this.bean = bean
}
/* 一切的基础都是 count ID */
private setAttr(key: string, value: unknown): void {
User.database.prepare(`UPDATE ${User.table_name} SET ${key} = ? WHERE count = ?`).run(value, this.bean.count)
this.bean[key] = value
}
getUserName(): string {
return this.bean.username
}
setUserName(userName: string): void {
this.setAttr("username", userName)
}
getNickName(): string {
return this.bean.nickname
}
setNickName(nickName: string): void {
this.setAttr("nickname", nickName)
}
getAvatar(): Uint8Array {
return this.bean.avatar
}
setAvatar(avatar: Buffer): void {
this.setAttr("avatar", avatar)
}
/* getSettings(): Settings {
}
static Settings = class {
}
static SettingsBean = interface {
} */
}

8
src/data/UserBean.ts Normal file
View File

@@ -0,0 +1,8 @@
export default interface UserBean {
count: number,
username: string,
registered_time: number,
nickname: string,
avatar: Uint8Array,
settings: string,
}

47
src/data/_user.ts Normal file
View File

@@ -0,0 +1,47 @@
import { DatabaseSync } from "node:sqlite"
import fs from 'node:fs/promises'
await fs.mkdir('data', { recursive: true })
const db = new DatabaseSync("data/users.db")
const TABEL_NAME = "Users"
// 初始化表格
db.exec(
`
CREATE TABLE IF NOT EXISTS ${TABEL_NAME} (
/* 伺服器中 ID */ id INTEGER PRIMARY KEY AUTOINCREMENT,
/* 用戶名, 可選 */ username TEXT,
/* 姓名 */ nickname TEXT NOT NULL,
/* 头像, 可选 */ avatar BLOB
);
`,
)
// 插入测试数据
db.prepare(
`
INSERT INTO ${TABEL_NAME} (username, nickname, avatar) VALUES (?, ?, ?);
`,
).run("SisterWen", "文姐", await fs.readFile('in.webp'))
let rows = db.prepare(`SELECT id, username, nickname, avatar FROM ${TABEL_NAME}`).all();
for (const row of rows) {
console.log(row)
}
// 更新用户名
// 用户名要合规, 以免导致 SQL 注入!
db.prepare(
`
UPDATE ${TABEL_NAME} SET username = '${ "Sister_Wen" }' WHERE ${ "username" } = ${ "'SisterWen'" };
`,
).run()
rows = db.prepare(`SELECT id, username, nickname, avatar FROM ${TABEL_NAME}`).all();
for (const row of rows) {
console.log(row)
await fs.writeFile('out.webp', row.avatar)
}
db.close()

0
src/main.ts Normal file
View File

42
src/main_test.ts Normal file
View File

@@ -0,0 +1,42 @@
import { DatabaseSync } from "node:sqlite"
import fs from 'node:fs/promises'
await fs.mkdir('data', { recursive: true })
const db = new DatabaseSync("data/users.db")
const TABEL_NAME = "Users"
// 初始化表格
db.exec(
`
CREATE TABLE IF NOT EXISTS ${TABEL_NAME} (
/* 伺服器中 ID */ id INTEGER PRIMARY KEY AUTOINCREMENT,
/* 用戶名, 可選 */ username TEXT,
/* 姓名 */ nickname TEXT NOT NULL,
/* 头像, 可选 */ avatar BLOB
);
`,
)
// 插入测试数据
db.prepare(
`
INSERT INTO ${TABEL_NAME} (username, nickname, avatar) VALUES (?, ?, ?);
`,
).run("SisterWen", "文姐", null)
let rows = db.prepare(`SELECT id, username, nickname, avatar FROM ${TABEL_NAME}`).all();
for (const row of rows) {
console.log(row)
}
// 更新用户名
// 用户名要合规, 以免导致 SQL 注入!
db.prepare(`UPDATE ${TABEL_NAME} SET username = ? WHERE id = ?`).run("文姐", 1)
rows = db.prepare(`SELECT id, username, nickname, avatar FROM ${TABEL_NAME} WHERE username = ?`).all("文姐")
for (const row of rows) {
console.log(row)
}
db.close()