mirror of
https://github.com/ikechan8370/chatgpt-plugin.git
synced 2025-12-16 21:37:11 +00:00
feat: init v3
This commit is contained in:
ikechan8370
parent
d6cb085c40
commit
531986b2dc
284 changed files with 618 additions and 405179 deletions
358
models/storage.js
Normal file
358
models/storage.js
Normal file
|
|
@ -0,0 +1,358 @@
|
|||
// storage.js written by sonnet
|
||||
import { Low } from 'lowdb'
|
||||
import { JSONFile } from 'lowdb/node'
|
||||
import path from 'path'
|
||||
import fs from 'fs'
|
||||
import ChatGPTConfig from '../config/config.js'
|
||||
|
||||
/**
|
||||
* 基于 LowDB 的简单存储类，提供 CRUD 和条件查询功能
|
||||
*/
|
||||
export class LowDBStorage {
|
||||
/**
|
||||
* 创建一个新的存储实例
|
||||
* @param {Object} options 配置选项
|
||||
* @param {string} options.filename 数据文件名称
|
||||
* @param {string} options.directory 数据目录，默认为当前目录下的 data 文件夹
|
||||
*/
|
||||
constructor (options = {}) {
|
||||
const { filename = 'db.json', directory = path.join(process.cwd(), 'data') } = options
|
||||
|
||||
// 确保目录存在
|
||||
if (!fs.existsSync(directory)) {
|
||||
fs.mkdirSync(directory, { recursive: true })
|
||||
}
|
||||
|
||||
this.filePath = path.join(directory, filename)
|
||||
this.adapter = new JSONFile(this.filePath)
|
||||
this.db = new Low(this.adapter)
|
||||
this.initialized = false
|
||||
}
|
||||
|
||||
/**
|
||||
* 初始化存储
|
||||
* @returns {Promise<LowDBStorage>} 当前存储实例
|
||||
*/
|
||||
async init () {
|
||||
// 读取数据文件，如果不存在则创建默认结构
|
||||
await this.db.read()
|
||||
this.db.data ||= { collections: {} }
|
||||
await this.db.write()
|
||||
|
||||
this.initialized = true
|
||||
return this
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取或创建一个集合
|
||||
* @param {string} name 集合名称
|
||||
* @returns {LowDBCollection} 集合实例
|
||||
*/
|
||||
collection (name) {
|
||||
this._checkInit()
|
||||
|
||||
// 确保集合存在
|
||||
if (!this.db.data.collections[name]) {
|
||||
this.db.data.collections[name] = []
|
||||
this.db.write()
|
||||
}
|
||||
|
||||
return new LowDBCollection(this, name)
|
||||
}
|
||||
|
||||
/**
|
||||
* 列出所有集合名称
|
||||
* @returns {string[]} 集合名称列表
|
||||
*/
|
||||
listCollections () {
|
||||
this._checkInit()
|
||||
return Object.keys(this.db.data.collections)
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除一个集合
|
||||
* @param {string} name 要删除的集合名称
|
||||
* @returns {Promise<boolean>} 是否成功删除
|
||||
*/
|
||||
async dropCollection (name) {
|
||||
this._checkInit()
|
||||
|
||||
if (this.db.data.collections[name]) {
|
||||
delete this.db.data.collections[name]
|
||||
await this.db.write()
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查存储是否已初始化
|
||||
* @private
|
||||
*/
|
||||
_checkInit () {
|
||||
if (!this.initialized) {
|
||||
throw new Error('存储尚未初始化，请先调用 init() 方法')
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 集合类，提供对特定数据集合的操作
|
||||
*/
|
||||
export class LowDBCollection {
|
||||
/**
|
||||
* 创建一个集合实例
|
||||
* @param {LowDBStorage} storage 所属存储实例
|
||||
* @param {string} name 集合名称
|
||||
*/
|
||||
constructor (storage, name) {
|
||||
this.storage = storage
|
||||
this.name = name
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取集合数据引用
|
||||
* @private
|
||||
*/
|
||||
get _collection () {
|
||||
return this.storage.db.data.collections[this.name]
|
||||
}
|
||||
|
||||
/**
|
||||
* 保存数据到存储
|
||||
* @private
|
||||
*/
|
||||
async _save () {
|
||||
return this.storage.db.write()
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成唯一ID
|
||||
* @private
|
||||
*/
|
||||
_generateId () {
|
||||
return Date.now().toString(36) + Math.random().toString(36).substring(2, 15)
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建新文档
|
||||
* @param {Object} doc 要插入的文档
|
||||
* @returns {Promise<Object>} 插入的文档（带ID）
|
||||
*/
|
||||
async insert (doc) {
|
||||
// 生成唯一ID，如果没有提供
|
||||
if (!doc.id) {
|
||||
doc.id = this._generateId()
|
||||
}
|
||||
|
||||
// 加上时间戳
|
||||
if (!doc.createdAt) {
|
||||
doc.createdAt = new Date().toISOString()
|
||||
}
|
||||
|
||||
doc.updatedAt = new Date().toISOString()
|
||||
|
||||
// 添加到集合
|
||||
this._collection.push(doc)
|
||||
await this._save()
|
||||
|
||||
return doc
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量插入多个文档
|
||||
* @param {Object[]} docs 要插入的文档数组
|
||||
* @returns {Promise<Object[]>} 插入的文档（带ID）
|
||||
*/
|
||||
async insertMany (docs) {
|
||||
const inserted = []
|
||||
|
||||
for (const doc of docs) {
|
||||
inserted.push(await this.insert(doc))
|
||||
}
|
||||
|
||||
return inserted
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据ID查找单个文档
|
||||
* @param {string} id 文档ID
|
||||
* @returns {Promise<Object|null>} 查找到的文档或null
|
||||
*/
|
||||
async findById (id) {
|
||||
return this._collection.find(doc => doc.id === id) || null
|
||||
}
|
||||
|
||||
/**
|
||||
* 返回集合中的所有文档
|
||||
* @returns {Promise<Object[]>} 文档数组
|
||||
*/
|
||||
async findAll () {
|
||||
return [...this._collection]
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据条件查找文档
|
||||
* @param {Object} query 查询条件（字段等值匹配）
|
||||
* @returns {Promise<Object[]>} 匹配的文档数组
|
||||
*/
|
||||
async find (query = {}) {
|
||||
return this._collection.filter(doc => {
|
||||
for (const key in query) {
|
||||
const value = query[key]
|
||||
|
||||
// 处理嵌套属性 (例如 user.profile.name)
|
||||
if (key.includes('.')) {
|
||||
const parts = key.split('.')
|
||||
let current = doc
|
||||
|
||||
for (let i = 0; i < parts.length; i++) {
|
||||
if (current === undefined || current === null) return false
|
||||
current = current[parts[i]]
|
||||
}
|
||||
|
||||
if (current !== value) return false
|
||||
} else if (doc[key] !== value) {
|
||||
return false
|
||||
}
|
||||
}
|
||||
return true
|
||||
})
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据条件查找单个文档
|
||||
* @param {Object} query 查询条件
|
||||
* @returns {Promise<Object|null>} 第一个匹配的文档或null
|
||||
*/
|
||||
async findOne (query = {}) {
|
||||
const results = await this.find(query)
|
||||
return results.length > 0 ? results[0] : null
|
||||
}
|
||||
|
||||
/**
|
||||
* 使用自定义函数进行高级查询
|
||||
* @param {Function} filterFn 过滤函数
|
||||
* @returns {Promise<Object[]>} 匹配的文档数组
|
||||
*/
|
||||
async findWhere (filterFn) {
|
||||
return this._collection.filter(filterFn)
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据ID更新文档
|
||||
* @param {string} id 文档ID
|
||||
* @param {Object} updates 要更新的字段
|
||||
* @returns {Promise<Object|null>} 更新后的文档或null
|
||||
*/
|
||||
async updateById (id, updates) {
|
||||
const index = this._collection.findIndex(doc => doc.id === id)
|
||||
|
||||
if (index === -1) return null
|
||||
|
||||
// 防止覆盖ID
|
||||
const { id: _, ...safeUpdates } = updates
|
||||
|
||||
// 更新文档
|
||||
const updatedDoc = {
|
||||
...this._collection[index],
|
||||
...safeUpdates,
|
||||
updatedAt: new Date().toISOString()
|
||||
}
|
||||
|
||||
this._collection[index] = updatedDoc
|
||||
await this._save()
|
||||
|
||||
return updatedDoc
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据条件更新文档
|
||||
* @param {Object} query 查询条件
|
||||
* @param {Object} updates 要更新的字段
|
||||
* @returns {Promise<number>} 更新的文档数量
|
||||
*/
|
||||
async update (query, updates) {
|
||||
const matches = await this.find(query)
|
||||
let updated = 0
|
||||
|
||||
for (const doc of matches) {
|
||||
await this.updateById(doc.id, updates)
|
||||
updated++
|
||||
}
|
||||
|
||||
return updated
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据ID删除文档
|
||||
* @param {string} id 文档ID
|
||||
* @returns {Promise<boolean>} 是否成功删除
|
||||
*/
|
||||
async deleteById (id) {
|
||||
const index = this._collection.findIndex(doc => doc.id === id)
|
||||
|
||||
if (index === -1) return false
|
||||
|
||||
this._collection.splice(index, 1)
|
||||
await this._save()
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据条件删除文档
|
||||
* @param {Object} query 查询条件
|
||||
* @returns {Promise<number>} 删除的文档数量
|
||||
*/
|
||||
async delete (query) {
|
||||
const before = this._collection.length
|
||||
|
||||
const remaining = this._collection.filter(doc => {
|
||||
for (const key in query) {
|
||||
if (doc[key] !== query[key]) {
|
||||
return true // 保留不匹配的
|
||||
}
|
||||
}
|
||||
return false // 删除匹配的
|
||||
})
|
||||
|
||||
this.storage.db.data.collections[this.name] = remaining
|
||||
await this._save()
|
||||
|
||||
return before - remaining.length
|
||||
}
|
||||
|
||||
/**
|
||||
* 清空集合中的所有文档
|
||||
* @returns {Promise<number>} 删除的文档数量
|
||||
*/
|
||||
async deleteAll () {
|
||||
const count = this._collection.length
|
||||
this.storage.db.data.collections[this.name] = []
|
||||
await this._save()
|
||||
return count
|
||||
}
|
||||
|
||||
/**
|
||||
* 返回集合中文档的数量
|
||||
* @returns {Promise<number>} 文档数量
|
||||
*/
|
||||
async count (query = {}) {
|
||||
if (Object.keys(query).length === 0) {
|
||||
return this._collection.length
|
||||
}
|
||||
|
||||
const matches = await this.find(query)
|
||||
return matches.length
|
||||
}
|
||||
}
|
||||
|
||||
const ChatGPTStorage = new LowDBStorage({
|
||||
filename: 'storage.json',
|
||||
directory: ChatGPTConfig.dataDir
|
||||
})
|
||||
|
||||
ChatGPTStorage.init()
|
||||
|
||||
export default ChatGPTStorage
|
||||
Loading…
Add table
Add a link
Reference in a new issue