119 lines
4.2 KiB
JavaScript
119 lines
4.2 KiB
JavaScript
|
||
let fspromises = require('fs').promises;
|
||
import { get, cloneDeep } from 'lodash';
|
||
import { define } from '../define';
|
||
|
||
export class DynamicSetting {
|
||
constructor(global) {
|
||
this.global = global;
|
||
}
|
||
|
||
|
||
/**
|
||
* 通过指定的类型，获取数据
|
||
* @param {*} type default：在代码中写死的 dynamic：用户自定义的 all：写死的和自定义的合并返回
|
||
* @param {*} mainType 主分类，gpt，tag这些
|
||
* @param {*} property 返回书信的名称 gpt_options，gpt_model_options，gpt_auto_inference
|
||
* @param {*} defaultSetting 默认的设置，用于合并数据
|
||
* @param {*} defaultData 默认数据，默认值为null
|
||
* @returns
|
||
*/
|
||
async getDataByTypeAndProperty(type, mainType, property, defaultSetting, defaultData = []) {
|
||
try {
|
||
let res = [];
|
||
// 获取自定义的GPT数据
|
||
let dynamic_setting = JSON.parse(await fspromises.readFile(define.dynamic_setting, 'utf-8'));
|
||
let tags = get(dynamic_setting, mainType, {});
|
||
let data = get(tags, property, defaultData);
|
||
|
||
if (type == "default") {
|
||
// res = get(this, property, defaultData);
|
||
} else if (type == "dynamic") {
|
||
res = data;
|
||
} else if (type == "all") {
|
||
res = defaultSetting.concat(data);
|
||
}
|
||
else {
|
||
throw new Error(`不存在的类型 : ${value}`);
|
||
}
|
||
return {
|
||
code: 1,
|
||
data: res
|
||
}
|
||
} catch (error) {
|
||
return {
|
||
code: 0,
|
||
message: error.toString()
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 保存 dynamic_setting 中主分支中指定属性的数据
|
||
* @param {*} value 要保存的数据
|
||
* @param {*} mainType 大的设置里面的主分支
|
||
* @param {*} property 对应的属性
|
||
* @param {*} callback 回调函数，可以对数据进行自定义的判断
|
||
*/
|
||
async saveDataByTypeAndProperty(value, mainType, property, callback = null) {
|
||
try {
|
||
// 获取自定义的GPT数据
|
||
let dynamic_setting = JSON.parse(await fspromises.readFile(define.dynamic_setting, 'utf-8'));
|
||
let tmp_gpt = get(dynamic_setting, mainType, {});
|
||
let gpt = tmp_gpt[property] ? tmp_gpt[property] : [];
|
||
if (callback) {
|
||
callback(gpt);
|
||
}
|
||
|
||
if (value.id) {
|
||
// 判断当前ID的数据是否存在，存在覆盖，不存在追加
|
||
let index = gpt.findIndex(item => item.id == value.id);
|
||
if (index < 0) {
|
||
gpt.push(value);
|
||
} else {
|
||
gpt[index] = value;
|
||
}
|
||
} else {
|
||
value.id = uuidv4();
|
||
gpt.push(value);
|
||
}
|
||
tmp_gpt[property] = gpt;
|
||
// 将修改后的数据保存
|
||
dynamic_setting[mainType] = tmp_gpt;
|
||
// 写入文件
|
||
await fspromises.writeFile(define.dynamic_setting, JSON.stringify(dynamic_setting));
|
||
|
||
} catch (error) {
|
||
throw error;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 删除指定的属性中的数据
|
||
* @param {*} id 要删除数据的ID
|
||
* @param {*} mainType 删除的主分支的类别
|
||
* @param {*} property 删除的属性中的数据
|
||
*/
|
||
async deleteDataByTypeAndProperty(id, mainType, property) {
|
||
try {
|
||
// 获取自定义的GPT数据
|
||
let dynamic_setting = JSON.parse(await fspromises.readFile(define.dynamic_setting, 'utf-8'));
|
||
let tmp_data = get(dynamic_setting, mainType, {});
|
||
let gpt = tmp_data[property] ? tmp_data[property] : [];
|
||
// 判断当前ID的数据是否存在，存在删除
|
||
let index = gpt.findIndex(item => item.id == id);
|
||
if (index >= 0) {
|
||
gpt.splice(index, 1);
|
||
}
|
||
// 将修改后的数据保存
|
||
dynamic_setting[mainType][property] = gpt;
|
||
// 写入文件
|
||
await fspromises.writeFile(define.dynamic_setting, JSON.stringify(dynamic_setting));
|
||
|
||
} catch (error) {
|
||
throw error;
|
||
}
|
||
}
|
||
|
||
|
||
} |