feat: user password
This commit is contained in:
@@ -10,6 +10,7 @@ import UserBean from './UserBean.ts'
|
|||||||
|
|
||||||
import FileManager from './FileManager.ts'
|
import FileManager from './FileManager.ts'
|
||||||
import { SQLInputValue } from "node:sqlite";
|
import { SQLInputValue } from "node:sqlite";
|
||||||
|
import DataWrongError from "../api/DataWrongError.ts";
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* User.ts - Wrapper and manager
|
* User.ts - Wrapper and manager
|
||||||
@@ -19,12 +20,13 @@ import { SQLInputValue } from "node:sqlite";
|
|||||||
export default class User {
|
export default class User {
|
||||||
static table_name: string = "Users"
|
static table_name: string = "Users"
|
||||||
private static database: DatabaseSync = User.init()
|
private static database: DatabaseSync = User.init()
|
||||||
private static init(): DatabaseSync {
|
private static init() {
|
||||||
const db: DatabaseSync = new DatabaseSync(path.join(config.data_path, User.table_name + '.db'))
|
const db: DatabaseSync = new DatabaseSync(path.join(config.data_path, User.table_name + '.db'))
|
||||||
db.exec(`
|
db.exec(`
|
||||||
CREATE TABLE IF NOT EXISTS ${User.table_name} (
|
CREATE TABLE IF NOT EXISTS ${User.table_name} (
|
||||||
/* 序号 */ count INTEGER PRIMARY KEY AUTOINCREMENT,
|
/* 序号 */ count INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
/* 用户 ID, 哈希 */ id TEXT,
|
/* 用户 ID, UUID */ id TEXT,
|
||||||
|
/* 密碼, 哈希 */ password TEXT,
|
||||||
/* 注册时间, 时间戳 */ registered_time INT8 NOT NULL,
|
/* 注册时间, 时间戳 */ registered_time INT8 NOT NULL,
|
||||||
/* 用戶名, 可選 */ username TEXT,
|
/* 用戶名, 可選 */ username TEXT,
|
||||||
/* 昵称 */ nickname TEXT NOT NULL,
|
/* 昵称 */ nickname TEXT NOT NULL,
|
||||||
@@ -35,29 +37,32 @@ export default class User {
|
|||||||
return db
|
return db
|
||||||
}
|
}
|
||||||
|
|
||||||
static createWithUserNameChecked(userName: string | null, nickName: string, avatar: Buffer | null): User {
|
static createWithUserNameChecked(userName: string | null, password: string, nickName: string, avatar: Buffer | null) {
|
||||||
if (userName && User.findAllBeansByCondition('username = ?', userName).length > 0)
|
if (userName && User.findAllBeansByCondition('username = ?', userName).length > 0)
|
||||||
throw new Error(`用户名 ${userName} 已存在`)
|
throw new DataWrongError(`用户名 ${userName} 已存在`)
|
||||||
return User.create(
|
return User.create(
|
||||||
userName,
|
userName,
|
||||||
|
password,
|
||||||
nickName,
|
nickName,
|
||||||
avatar
|
avatar
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
static create(userName: string | null, nickName: string, avatar: Buffer | null): User {
|
static create(userName: string | null, password: string, nickName: string, avatar: Buffer | null) {
|
||||||
const user = new User(
|
const user = new User(
|
||||||
User.findAllBeansByCondition(
|
User.findAllBeansByCondition(
|
||||||
'count = ?',
|
'count = ?',
|
||||||
User.database.prepare(`INSERT INTO ${User.table_name} (
|
User.database.prepare(`INSERT INTO ${User.table_name} (
|
||||||
id,
|
id,
|
||||||
|
password,
|
||||||
registered_time,
|
registered_time,
|
||||||
username,
|
username,
|
||||||
nickname,
|
nickname,
|
||||||
avatar_file_hash,
|
avatar_file_hash,
|
||||||
settings
|
settings
|
||||||
) VALUES (?, ?, ?, ?, ?, ?);`).run(
|
) VALUES (?, ?, ?, ?, ?, ?, ?);`).run(
|
||||||
crypto.randomUUID(),
|
crypto.randomUUID(),
|
||||||
|
password,
|
||||||
Date.now(),
|
Date.now(),
|
||||||
userName,
|
userName,
|
||||||
nickName,
|
nickName,
|
||||||
@@ -73,18 +78,18 @@ export default class User {
|
|||||||
private static findAllBeansByCondition(condition: string, ...args: SQLInputValue[]): UserBean[] {
|
private static findAllBeansByCondition(condition: string, ...args: SQLInputValue[]): UserBean[] {
|
||||||
return User.database.prepare(`SELECT * FROM ${User.table_name} WHERE ${condition};`).all(...args) as unknown as UserBean[]
|
return User.database.prepare(`SELECT * FROM ${User.table_name} WHERE ${condition};`).all(...args) as unknown as UserBean[]
|
||||||
}
|
}
|
||||||
static findById(id: string): User {
|
static findById(id: string) {
|
||||||
const beans = User.findAllBeansByCondition('id = ?', id)
|
const beans = User.findAllBeansByCondition('id = ?', id)
|
||||||
if (beans.length == 0)
|
if (beans.length == 0)
|
||||||
throw new Error(`找不到用户 ID 为 ${id} 的用户`)
|
return null
|
||||||
else if (beans.length > 1)
|
else if (beans.length > 1)
|
||||||
console.error(chalk.red(`警告: 查询 id = ${id} 时, 查询到多个相同用户 ID 的用户`))
|
console.error(chalk.red(`警告: 查询 id = ${id} 时, 查询到多个相同用户 ID 的用户`))
|
||||||
return new User(beans[0])
|
return new User(beans[0])
|
||||||
}
|
}
|
||||||
static findByUserName(userName: string): User {
|
static findByUserName(userName: string) {
|
||||||
const beans = User.findAllBeansByCondition('username = ?', userName)
|
const beans = User.findAllBeansByCondition('username = ?', userName)
|
||||||
if (beans.length == 0)
|
if (beans.length == 0)
|
||||||
throw new Error(`找不到用户名为 ${userName} 的用户`)
|
return null
|
||||||
else if (beans.length > 1)
|
else if (beans.length > 1)
|
||||||
console.error(chalk.red(`警告: 查询 username = ${userName} 时, 查询到多个相同用户名的用户`))
|
console.error(chalk.red(`警告: 查询 username = ${userName} 时, 查询到多个相同用户名的用户`))
|
||||||
return new User(beans[0])
|
return new User(beans[0])
|
||||||
@@ -95,26 +100,32 @@ export default class User {
|
|||||||
this.bean = bean
|
this.bean = bean
|
||||||
}
|
}
|
||||||
/* 一切的基础都是 count ID */
|
/* 一切的基础都是 count ID */
|
||||||
private setAttr(key: string, value: unknown): void {
|
private setAttr(key: string, value: unknown) {
|
||||||
User.database.prepare(`UPDATE ${User.table_name} SET ${key} = ? WHERE count = ?`).run(value, this.bean.count)
|
User.database.prepare(`UPDATE ${User.table_name} SET ${key} = ? WHERE count = ?`).run(value, this.bean.count)
|
||||||
this.bean[key] = value
|
this.bean[key] = value
|
||||||
}
|
}
|
||||||
getUserName(): string | null {
|
getUserName(): string | null {
|
||||||
return this.bean.username
|
return this.bean.username
|
||||||
}
|
}
|
||||||
setUserName(userName: string): void {
|
setUserName(userName: string) {
|
||||||
this.setAttr("username", userName)
|
this.setAttr("username", userName)
|
||||||
}
|
}
|
||||||
getNickName(): string {
|
getNickName(): string {
|
||||||
return this.bean.nickname
|
return this.bean.nickname
|
||||||
}
|
}
|
||||||
setNickName(nickName: string): void {
|
setNickName(nickName: string) {
|
||||||
this.setAttr("nickname", nickName)
|
this.setAttr("nickname", nickName)
|
||||||
}
|
}
|
||||||
|
getPassword(): string {
|
||||||
|
return this.bean.password
|
||||||
|
}
|
||||||
|
setPassword(password: string) {
|
||||||
|
this.setAttr("password", password)
|
||||||
|
}
|
||||||
getAvatar(): Buffer | null {
|
getAvatar(): Buffer | null {
|
||||||
return FileManager.findByHash(this.bean.avatar_file_hash)?.readSync()
|
return FileManager.findByHash(this.bean.avatar_file_hash)?.readSync()
|
||||||
}
|
}
|
||||||
setAvatar(avatar: Buffer): void {
|
setAvatar(avatar: Buffer) {
|
||||||
this.setAttr("avatar_file_hash", FileManager.uploadFile(`avatar_user_${this.bean.count}`, avatar).getHash())
|
this.setAttr("avatar_file_hash", FileManager.uploadFile(`avatar_user_${this.bean.count}`, avatar).getHash())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
export default class UserBean {
|
export default class UserBean {
|
||||||
declare count: number
|
declare count: number
|
||||||
declare id: string
|
declare id: string
|
||||||
|
declare password: string
|
||||||
declare username: string | null
|
declare username: string | null
|
||||||
declare registered_time: number
|
declare registered_time: number
|
||||||
declare nickname: string
|
declare nickname: string
|
||||||
|
|||||||
Reference in New Issue
Block a user