initial
1. 修复 ignore 2. 推翻旧后端代码全部重写, 使用 SQLite
This commit is contained in:
2
.gitignore
vendored
2
.gitignore
vendored
@@ -1,4 +1,4 @@
|
||||
# 配置文件
|
||||
thewhitesilk_config.json
|
||||
# **默认**数据目录
|
||||
data/
|
||||
thewhitesilk_data/
|
||||
|
||||
17
src/config.ts
Normal file
17
src/config.ts
Normal 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
64
src/data/Chat.ts
Normal 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
3
src/data/ChatBean.ts
Normal file
@@ -0,0 +1,3 @@
|
||||
export default interface ChatBean {
|
||||
id: string
|
||||
}
|
||||
101
src/data/User.ts
Normal file
101
src/data/User.ts
Normal 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
8
src/data/UserBean.ts
Normal 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
47
src/data/_user.ts
Normal 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
0
src/main.ts
Normal file
42
src/main_test.ts
Normal file
42
src/main_test.ts
Normal 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()
|
||||
Reference in New Issue
Block a user