141 lines
3.7 KiB
TypeScript
141 lines
3.7 KiB
TypeScript
import mkfileIfNotExist from "./mkfileIfNotExist"
|
|
import DB from 'better-sqlite3'
|
|
import fs from 'fs'
|
|
|
|
let checked = false
|
|
|
|
/**
|
|
* 检查并完善userData的目录结构，若不存在密钥则生成随机的密钥，初始化数据库
|
|
* @returns {void}
|
|
*/
|
|
function checkUserData() {
|
|
if (checked) return
|
|
checked = true
|
|
// console.log('check userData')
|
|
const dirList = [
|
|
'./userData/',
|
|
'./userData/db/',
|
|
'./userData/imgBed/',
|
|
]
|
|
dirList.forEach(mkdirIfNotExist)
|
|
|
|
mkfileIfNotExist('./userData/key.txt', new Array(2048).fill(0).map(() => Math.floor(Math.random() * 16).toString(16)).join(''))
|
|
|
|
createDB()
|
|
|
|
}
|
|
|
|
function createDB() {
|
|
createUserDB();
|
|
createPostsDB();
|
|
createImgBedDB();
|
|
}
|
|
|
|
|
|
const createUserDB = () => {
|
|
const userDB = new DB('./userData/db/user.db');
|
|
userDB.exec(`
|
|
CREATE TABLE IF NOT EXISTS user (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
username TEXT NOT NULL UNIQUE,
|
|
password TEXT NOT NULL,
|
|
createTime INTEGER NOT NULL,
|
|
lastLoginTime INTEGER NOT NULL
|
|
);
|
|
`);
|
|
userDB.close();
|
|
}
|
|
|
|
const createPostsDB = () => {
|
|
const postsDB = new DB('./userData/db/posts.db');
|
|
postsDB.exec(`
|
|
CREATE TABLE IF NOT EXISTS posts (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
title TEXT NOT NULL,
|
|
date INTEGER NOT NULL,
|
|
summary TEXT NOT NULL,
|
|
type TEXT NOT NULL,
|
|
url TEXT NOT NULL,
|
|
tag_id INTEGER NOT NULL,
|
|
FOREIGN KEY(tag_id) REFERENCES tags(id)
|
|
);
|
|
`);
|
|
postsDB.exec(`
|
|
CREATE TABLE IF NOT EXISTS postContent (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
content TEXT NOT NULL
|
|
);
|
|
`);
|
|
postsDB.exec(`
|
|
CREATE TABLE IF NOT EXISTS tags (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
tag1 TEXT,
|
|
tag2 TEXT,
|
|
tag3 TEXT,
|
|
tag4 TEXT,
|
|
tag5 TEXT,
|
|
tag6 TEXT,
|
|
tag7 TEXT,
|
|
tag8 TEXT,
|
|
tag9 TEXT,
|
|
tag10 TEXT,
|
|
tag11 TEXT,
|
|
tag12 TEXT,
|
|
tag13 TEXT,
|
|
tag14 TEXT,
|
|
tag15 TEXT
|
|
);
|
|
`);
|
|
postsDB.close();
|
|
}
|
|
|
|
const createImgBedDB = () => {
|
|
const imgBedDB = new DB('./userData/db/imgBed.db');
|
|
imgBedDB.exec(`
|
|
CREATE TABLE IF NOT EXISTS imgBed (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
name TEXT NOT NULL,
|
|
hash TEXT NOT NULL,
|
|
size INTEGER NOT NULL,
|
|
expire INTEGER NOT NULL
|
|
);
|
|
`);
|
|
imgBedDB.close();
|
|
setInterval(cleanExpiredFiles_imgBed, 1000 * 30 * 60 );
|
|
// cleanExpiredFiles_imgBed()
|
|
}
|
|
|
|
const cleanExpiredFiles_imgBed = async () => {
|
|
try {
|
|
const db = new DB('./userData/db/imgBed.db');
|
|
const currentTime = Date.now();
|
|
// 从数据库中查询过期文件的哈希
|
|
const expiredFiles = db.prepare(`
|
|
SELECT hash FROM imgBed WHERE expire <?
|
|
`).all(currentTime) as {
|
|
hash: string;
|
|
}[];
|
|
db.close();
|
|
|
|
for (const file of expiredFiles) {
|
|
const hash = file.hash;
|
|
const filePath = `./userData/imgBed/${hash}`;
|
|
// 检查文件是否存在
|
|
if (fs.existsSync(filePath)) {
|
|
// 删除文件
|
|
fs.unlinkSync(filePath);
|
|
}
|
|
}
|
|
|
|
// 从数据库中删除过期文件的记录
|
|
const db2 = new DB('./userData/db/imgBed.db');
|
|
db2.prepare(`
|
|
DELETE FROM imgBed WHERE expire <?
|
|
`).run(currentTime);
|
|
db2.close();
|
|
} catch (error) {
|
|
console.error('Error cleaning expired files:', error);
|
|
}
|
|
};
|
|
|
|
export default checkUserData |