180 lines
3.9 KiB
JavaScript
180 lines
3.9 KiB
JavaScript
const sqlite3 = require('sqlite3').verbose()
|
|
const path = require('path')
|
|
const fs = require('fs')
|
|
const { app } = require('electron')
|
|
|
|
// 数据库路径
|
|
const dbPath = path.join(app.getPath('userData'), 'database.sqlite')
|
|
let db
|
|
|
|
// 初始化数据库
|
|
function initDatabase() {
|
|
const dbDir = path.dirname(dbPath)
|
|
if (!fs.existsSync(dbDir)) {
|
|
fs.mkdirSync(dbDir, { recursive: true })
|
|
}
|
|
|
|
db = new sqlite3.Database(dbPath)
|
|
|
|
// 创建表
|
|
db.serialize(() => {
|
|
db.run(`
|
|
CREATE TABLE IF NOT EXISTS files (
|
|
file_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
file_name TEXT NOT NULL,
|
|
file_ext TEXT,
|
|
file_cat TEXT,
|
|
file_tags TEXT,
|
|
file_desc TEXT,
|
|
file_created_time DATETIME,
|
|
file_updated_time DATETIME,
|
|
file_path TEXT NOT NULL UNIQUE,
|
|
is_directory BOOLEAN DEFAULT 0,
|
|
added_time DATETIME DEFAULT CURRENT_TIMESTAMP
|
|
)
|
|
`)
|
|
})
|
|
|
|
console.log('Database initialized at:', dbPath)
|
|
return db
|
|
}
|
|
|
|
// 添加文件记录
|
|
function addFile(fileData) {
|
|
return new Promise((resolve, reject) => {
|
|
const stmt = db.prepare(`
|
|
INSERT INTO files (
|
|
file_name, file_ext, file_cat, file_tags, file_desc,
|
|
file_created_time, file_updated_time, file_path, is_directory
|
|
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
`)
|
|
|
|
stmt.run(
|
|
fileData.fileName,
|
|
fileData.fileExt,
|
|
fileData.fileCat,
|
|
JSON.stringify(fileData.fileTags || []),
|
|
fileData.fileDesc,
|
|
fileData.fileCreatedTime,
|
|
fileData.fileUpdatedTime,
|
|
fileData.filePath,
|
|
fileData.isDirectory ? 1 : 0,
|
|
function (err) {
|
|
if (err) {
|
|
reject(err)
|
|
} else {
|
|
resolve(this.lastID)
|
|
}
|
|
},
|
|
)
|
|
|
|
stmt.finalize()
|
|
})
|
|
}
|
|
|
|
// 搜索文件
|
|
function searchFiles(criteria) {
|
|
return new Promise((resolve, reject) => {
|
|
let query = `SELECT * FROM files WHERE 1=1`
|
|
const params = []
|
|
|
|
if (criteria.fileName) {
|
|
query += ` AND file_name LIKE ?`
|
|
params.push(`%${criteria.fileName}%`)
|
|
}
|
|
|
|
if (criteria.fileCat) {
|
|
query += ` AND file_cat = ?`
|
|
params.push(criteria.fileCat)
|
|
}
|
|
|
|
if (criteria.fileTags && criteria.fileTags.length > 0) {
|
|
const tagConditions = criteria.fileTags.map(() => `file_tags LIKE ?`).join(' OR ')
|
|
query += ` AND (${tagConditions})`
|
|
criteria.fileTags.forEach((tag) => {
|
|
params.push(`%${tag}%`)
|
|
})
|
|
}
|
|
|
|
if (criteria.startDate) {
|
|
query += ` AND added_time >= ?`
|
|
params.push(criteria.startDate)
|
|
}
|
|
|
|
if (criteria.endDate) {
|
|
query += ` AND added_time <= ?`
|
|
params.push(criteria.endDate)
|
|
}
|
|
|
|
db.all(query, params, (err, rows) => {
|
|
if (err) {
|
|
reject(err)
|
|
} else {
|
|
// 解析JSON格式的标签
|
|
const results = rows.map((row) => ({
|
|
...row,
|
|
fileTags: JSON.parse(row.file_tags || '[]'),
|
|
}))
|
|
resolve(results)
|
|
}
|
|
})
|
|
})
|
|
}
|
|
|
|
// 获取统计数据
|
|
function getStats() {
|
|
return new Promise((resolve, reject) => {
|
|
const stats = {}
|
|
|
|
// 获取文件总数
|
|
db.get('SELECT COUNT(*) as count FROM files WHERE is_directory = 0', (err, row) => {
|
|
if (err) {
|
|
reject(err)
|
|
return
|
|
}
|
|
stats.fileCount = row.count
|
|
|
|
// 获取文件夹总数
|
|
db.get('SELECT COUNT(*) as count FROM files WHERE is_directory = 1', (err, row) => {
|
|
if (err) {
|
|
reject(err)
|
|
return
|
|
}
|
|
stats.dirCount = row.count
|
|
|
|
// 获取文件类型分布
|
|
db.all(
|
|
'SELECT file_ext, COUNT(*) as count FROM files WHERE is_directory = 0 GROUP BY file_ext',
|
|
(err, rows) => {
|
|
if (err) {
|
|
reject(err)
|
|
return
|
|
}
|
|
stats.fileTypes = rows
|
|
|
|
// 获取最近添加的文件
|
|
db.all('SELECT * FROM files ORDER BY added_time DESC LIMIT 10', (err, rows) => {
|
|
if (err) {
|
|
reject(err)
|
|
} else {
|
|
stats.recentFiles = rows.map((row) => ({
|
|
...row,
|
|
fileTags: JSON.parse(row.file_tags || '[]'),
|
|
}))
|
|
resolve(stats)
|
|
}
|
|
})
|
|
},
|
|
)
|
|
})
|
|
})
|
|
})
|
|
}
|
|
|
|
module.exports = {
|
|
initDatabase,
|
|
addFile,
|
|
searchFiles,
|
|
getStats,
|
|
}
|