mirror of
https://github.com/vastxie/99AI.git
synced 2025-09-17 09:16:38 +08:00
280 lines
12 KiB
JavaScript
280 lines
12 KiB
JavaScript
"use strict";
|
|
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
|
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
};
|
|
var __metadata = (this && this.__metadata) || function (k, v) {
|
|
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
|
};
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.DatabaseService = void 0;
|
|
const common_1 = require("@nestjs/common");
|
|
const bcrypt = require("bcryptjs");
|
|
const typeorm_1 = require("typeorm");
|
|
let DatabaseService = class DatabaseService {
|
|
constructor(connection) {
|
|
this.connection = connection;
|
|
}
|
|
async onModuleInit() {
|
|
await this.checkSuperAdmin();
|
|
await this.checkSiteBaseConfig();
|
|
await this.createSystemReservedApps();
|
|
}
|
|
async checkSuperAdmin() {
|
|
const user = await this.connection.query(`SELECT * FROM users WHERE role = 'super'`);
|
|
if (!user || user.length === 0) {
|
|
const superPassword = bcrypt.hashSync('123456', 10);
|
|
const adminPassword = bcrypt.hashSync('123456', 10);
|
|
const superEmail = 'super';
|
|
const adminEmail = 'admin';
|
|
const superUserinfo = {
|
|
username: 'super',
|
|
password: superPassword,
|
|
status: 1,
|
|
email: superEmail,
|
|
sex: 1,
|
|
role: 'super',
|
|
};
|
|
const adminUserinfo = {
|
|
username: 'admin',
|
|
password: adminPassword,
|
|
status: 0,
|
|
email: adminEmail,
|
|
sex: 1,
|
|
role: 'admin',
|
|
};
|
|
await this.createDefaultUser(superUserinfo);
|
|
await this.createDefaultUser(adminUserinfo);
|
|
}
|
|
}
|
|
async createDefaultUser(userInfo) {
|
|
try {
|
|
const { username, password, status, email, role } = userInfo;
|
|
const user = await this.connection.query(`INSERT INTO users (username, password, status, email, role) VALUES ('${username}', '${password}', '${status}', '${email}', '${role}')`);
|
|
const userId = user.insertId;
|
|
await this.connection.query(`INSERT INTO balance (userId, balance, usesLeft, paintCount) VALUES ('${userId}', 0, 1000, 100)`);
|
|
common_1.Logger.log(`初始化创建${role}用户成功、用户名为[${username}]、初始密码为[${username === 'super' ? 'super' : '123456'}] ==============> 请注意查阅`, 'DatabaseService');
|
|
}
|
|
catch (error) {
|
|
console.log('error: ', error);
|
|
throw new common_1.HttpException('创建默认超级管理员失败！', common_1.HttpStatus.INTERNAL_SERVER_ERROR);
|
|
}
|
|
}
|
|
async checkSiteBaseConfig() {
|
|
const keys = [
|
|
'siteName',
|
|
'qqNumber',
|
|
'vxNumber',
|
|
'robotAvatar',
|
|
'userDefautlAvatar',
|
|
];
|
|
const result = await this.connection.query(`
|
|
SELECT COUNT(*) AS count FROM config WHERE \`configKey\` IN (${keys
|
|
.map((k) => `'${k}'`)
|
|
.join(',')})
|
|
`);
|
|
const count = parseInt(result[0].count);
|
|
if (count === 0) {
|
|
await this.createBaseSiteConfig();
|
|
}
|
|
}
|
|
async createBaseSiteConfig() {
|
|
try {
|
|
const code = ``;
|
|
const noticeInfo = `
|
|
#### AIWeb 欢迎您
|
|
- 欢迎使用 AIWeb
|
|
- 初始管理员账号密码 super 123456 【前台后台登录都可以修改】
|
|
- 初始预览账号密码 admin 123456 【为后台查看账号 仅可查看部分非敏感数据】
|
|
`;
|
|
const defaultConfig = [
|
|
{ configKey: 'siteName', configVal: '', public: 1, encry: 0 },
|
|
{ configKey: 'qqNumber', configVal: '', public: 1, encry: 0 },
|
|
{ configKey: 'vxNumber', configVal: '', public: 1, encry: 0 },
|
|
{ configKey: 'robotAvatar', configVal: '', public: 1, encry: 0 },
|
|
{
|
|
configKey: 'userDefautlAvatar',
|
|
configVal: '',
|
|
public: 0,
|
|
encry: 0,
|
|
},
|
|
{ configKey: 'baiduCode', configVal: code, public: 1, encry: 0 },
|
|
{ configKey: 'baiduSiteId', configVal: '', public: 0, encry: 0 },
|
|
{
|
|
configKey: 'baiduToken',
|
|
configVal: '',
|
|
public: 0,
|
|
encry: 0,
|
|
},
|
|
{ configKey: 'buyCramiAddress', configVal: '', public: 1, encry: 0 },
|
|
{
|
|
configKey: 'openaiBaseUrl',
|
|
configVal: 'https://api.lightai.io',
|
|
public: 0,
|
|
encry: 0,
|
|
},
|
|
{ configKey: 'openaiTimeout', configVal: '300', public: 0, encry: 0 },
|
|
{ configKey: 'openaiBaseKey', configVal: 'sk-', public: 0, encry: 0 },
|
|
{
|
|
configKey: 'mjTranslatePrompt',
|
|
configVal: `Translate any given phrase from any language into English. For instance, when I input '{可爱的熊猫}', you should output '{cute panda}', with no period at the end.`,
|
|
public: 0,
|
|
encry: 0,
|
|
},
|
|
{ configKey: 'noticeInfo', configVal: noticeInfo, public: 1, encry: 0 },
|
|
{
|
|
configKey: 'registerSendStatus',
|
|
configVal: '1',
|
|
public: 1,
|
|
encry: 0,
|
|
},
|
|
{
|
|
configKey: 'registerSendModel3Count',
|
|
configVal: '30',
|
|
public: 1,
|
|
encry: 0,
|
|
},
|
|
{
|
|
configKey: 'registerSendModel4Count',
|
|
configVal: '3',
|
|
public: 1,
|
|
encry: 0,
|
|
},
|
|
{
|
|
configKey: 'registerSendDrawMjCount',
|
|
configVal: '3',
|
|
public: 1,
|
|
encry: 0,
|
|
},
|
|
{
|
|
configKey: 'firstRegisterSendStatus',
|
|
configVal: '1',
|
|
public: 1,
|
|
encry: 0,
|
|
},
|
|
{
|
|
configKey: 'firstRegisterSendRank',
|
|
configVal: '500',
|
|
public: 1,
|
|
encry: 0,
|
|
},
|
|
{
|
|
configKey: 'firstRregisterSendModel3Count',
|
|
configVal: '10',
|
|
public: 1,
|
|
encry: 0,
|
|
},
|
|
{
|
|
configKey: 'firstRregisterSendModel4Count',
|
|
configVal: '10',
|
|
public: 1,
|
|
encry: 0,
|
|
},
|
|
{
|
|
configKey: 'firstRregisterSendDrawMjCount',
|
|
configVal: '10',
|
|
public: 1,
|
|
encry: 0,
|
|
},
|
|
{ configKey: 'inviteSendStatus', configVal: '1', public: 1, encry: 0 },
|
|
{
|
|
configKey: 'inviteGiveSendModel3Count',
|
|
configVal: '0',
|
|
public: 1,
|
|
encry: 0,
|
|
},
|
|
{
|
|
configKey: 'inviteGiveSendModel4Count',
|
|
configVal: '0',
|
|
public: 1,
|
|
encry: 0,
|
|
},
|
|
{
|
|
configKey: 'inviteGiveSendDrawMjCount',
|
|
configVal: '0',
|
|
public: 1,
|
|
encry: 0,
|
|
},
|
|
{
|
|
configKey: 'invitedGuestSendModel3Count',
|
|
configVal: '10',
|
|
public: 1,
|
|
encry: 0,
|
|
},
|
|
{
|
|
configKey: 'invitedGuestSendModel4Count',
|
|
configVal: '10',
|
|
public: 1,
|
|
encry: 0,
|
|
},
|
|
{
|
|
configKey: 'invitedGuestSendDrawMjCount',
|
|
configVal: '10',
|
|
public: 1,
|
|
encry: 0,
|
|
},
|
|
{ configKey: 'isVerifyEmail', configVal: '1', public: 1, encry: 0 },
|
|
];
|
|
const res = await this.connection.query(`INSERT INTO config (configKey, configVal, public, encry) VALUES ${defaultConfig
|
|
.map((d) => `('${d.configKey}', '${d.configVal.replace(/'/g, "\\'")}', '${d.public}', '${d.encry}')`)
|
|
.join(', ')}`);
|
|
common_1.Logger.log(`初始化网站配置信息成功、如您需要修改网站配置信息，请前往管理系统系统配置设置 ==============> 请注意查阅`, 'DatabaseService');
|
|
}
|
|
catch (error) {
|
|
console.log('error: ', error);
|
|
throw new common_1.HttpException('创建默认网站配置失败！', common_1.HttpStatus.INTERNAL_SERVER_ERROR);
|
|
}
|
|
}
|
|
async createSystemReservedApps() {
|
|
const systemApps = [
|
|
{
|
|
name: '提示词优化PromptOptimization',
|
|
catId: 9900,
|
|
des: 'PromptOptimization',
|
|
preset: "Translate any given phrase from any language into English. For instance, when I input '{可爱的熊猫}', you should output '{cute panda}', with no period at the end.",
|
|
appModel: 'gpt-3.5-turbo',
|
|
isFixedModel: 1,
|
|
isSystemReserved: 1,
|
|
},
|
|
{
|
|
name: '思维导图MindMap',
|
|
catId: 9900,
|
|
des: 'MindMap',
|
|
preset: '我希望你使用markdown格式回答我得问题、我的需求是得到一份markdown格式的大纲、尽量做的精细、层级多一点、不管我问你什么、都需要您回复我一个大纲出来、我想使用大纲做思维导图、除了大纲之外、不要无关内容和总结。',
|
|
appModel: 'gpt-3.5-turbo',
|
|
isFixedModel: 1,
|
|
isSystemReserved: 1,
|
|
},
|
|
];
|
|
try {
|
|
for (const app of systemApps) {
|
|
const result = await this.connection.query(`SELECT COUNT(*) AS count FROM app WHERE name = ? AND des = ? AND isSystemReserved = ?`, [app.name, app.des, app.isSystemReserved]);
|
|
const count = parseInt(result[0].count, 10);
|
|
if (count === 0) {
|
|
await this.connection.query(`INSERT INTO app (name, catId, des, preset, appModel, isFixedModel, isSystemReserved) VALUES (?, ?, ?, ?, ?, ?, ?)`, [
|
|
app.name,
|
|
app.catId,
|
|
app.des,
|
|
app.preset,
|
|
app.appModel,
|
|
app.isFixedModel,
|
|
app.isSystemReserved,
|
|
]);
|
|
common_1.Logger.log(`系统预留应用${app.name}创建成功`, 'DatabaseService');
|
|
}
|
|
}
|
|
}
|
|
catch (error) {
|
|
console.log('创建系统预留应用过程中出现错误: ', error);
|
|
throw new Error('创建系统预留应用失败！');
|
|
}
|
|
}
|
|
};
|
|
DatabaseService = __decorate([
|
|
(0, common_1.Injectable)(),
|
|
__metadata("design:paramtypes", [typeorm_1.Connection])
|
|
], DatabaseService);
|
|
exports.DatabaseService = DatabaseService;
|