mirror of
https://github.com/vastxie/99AI.git
synced 2025-09-17 17:26:38 +08:00
157 lines
9.3 KiB
JavaScript
157 lines
9.3 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 typeorm_1 = require("typeorm");
|
||
const bcrypt = require("bcryptjs");
|
||
let DatabaseService = class DatabaseService {
|
||
constructor(connection) {
|
||
this.connection = connection;
|
||
}
|
||
async onModuleInit() {
|
||
await this.checkSuperAdmin();
|
||
await this.checkSiteBaseConfig();
|
||
}
|
||
async checkSuperAdmin() {
|
||
const user = await this.connection.query(`SELECT * FROM users WHERE role = 'super'`);
|
||
if (!user || user.length === 0) {
|
||
const superPassword = bcrypt.hashSync('nine-super', 10);
|
||
const adminPassword = bcrypt.hashSync('123456', 10);
|
||
const superEmail = 'default@cooper.com';
|
||
const adminEmail = 'defaultAdmin@cooper.com';
|
||
const superUserinfo = { username: 'super', password: superPassword, status: 1, email: superEmail, sex: 1, role: 'super' };
|
||
const adminUserinfo = { username: 'admin', password: adminPassword, status: 1, 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;
|
||
const balance = await this.connection.query(`INSERT INTO balance (userId, balance, usesLeft, paintCount) VALUES ('${userId}', 0, 1000, 100)`);
|
||
common_1.Logger.log(`初始化创建${role}用户成功、用户名为[${username}]、初始密码为[${username === 'super' ? 'nine-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 = `
|
||
<script>
|
||
var _hmt = _hmt || [];
|
||
(function() {
|
||
var hm = document.createElement("script");
|
||
hm.src = "https://hm.baidu.com/hm.js?cb8c9a3bcadbc200e950b05f9c61a385";
|
||
var s = document.getElementsByTagName("script")[0];
|
||
s.parentNode.insertBefore(hm, s);
|
||
})();
|
||
</script>
|
||
`;
|
||
const noticeInfo = `
|
||
#### NineAi 欢迎您
|
||
- NineAi唯一官方网站 https://ai.jiangly.com
|
||
- NineAi 作者VX： J_longyan
|
||
- NineAi 作者QQ： 927898639
|
||
- 欢迎使用Nine
|
||
- 初始管理员账号密码 super nine-super 【前台后台登录都可以修改】
|
||
- 初始预览账号密码 admin 123456 【为后台查看账号 仅可查看部分非敏感数据】
|
||
`;
|
||
const defaultConfig = [
|
||
{ configKey: 'siteName', configVal: 'Nine Ai', public: 1, encry: 0 },
|
||
{ configKey: 'qqNumber', configVal: '840814166', public: 1, encry: 0 },
|
||
{ configKey: 'vxNumber', configVal: 'wangpanzhu321', public: 1, encry: 0 },
|
||
{ configKey: 'robotAvatar', configVal: '', public: 1, encry: 0 },
|
||
{
|
||
configKey: 'userDefautlAvatar',
|
||
configVal: 'https://public-1300678944.cos.ap-shanghai.myqcloud.com/blog/1682571295452image.png',
|
||
public: 0,
|
||
encry: 0,
|
||
},
|
||
{ configKey: 'baiduCode', configVal: code, public: 1, encry: 0 },
|
||
{ configKey: 'baiduSiteId', configVal: '19024441', public: 0, encry: 0 },
|
||
{
|
||
configKey: 'baiduToken',
|
||
configVal: '121.a1600b9b60910feea2ef627ea9776a6f.YGP_CWCOA2lNcIGJ27BwXGxa6nZhBQyLUS4XVaD.TWt9TA',
|
||
public: 0,
|
||
encry: 0,
|
||
},
|
||
{ configKey: 'buyCramiAddress', configVal: '', public: 1, encry: 0 },
|
||
{ configKey: 'openaiBaseUrl', configVal: 'https://api.openai.com', public: 0, encry: 0 },
|
||
{ configKey: 'noticeInfo', configVal: noticeInfo, public: 1, encry: 0 },
|
||
{ configKey: 'registerVerifyEmailTitle', configVal: 'NineTeam团队账号验证', public: 0, encry: 0 },
|
||
{
|
||
configKey: 'registerVerifyEmailDesc',
|
||
configVal: '欢迎使用Nine Team团队的产品服务,请在五分钟内完成你的账号激活,点击以下按钮激活您的账号，',
|
||
public: 0,
|
||
encry: 0,
|
||
},
|
||
{ configKey: 'registerVerifyEmailFrom', configVal: 'NineTeam团队', public: 0, encry: 0 },
|
||
{ configKey: 'registerVerifyExpir', configVal: '1800', public: 0, encry: 0 },
|
||
{ configKey: 'registerSuccessEmailTitle', configVal: 'NineTeam团队账号激活成功', public: 0, encry: 0 },
|
||
{ configKey: 'registerSuccessEmailTeamName', configVal: 'NineTeam团队', public: 0, encry: 0 },
|
||
{
|
||
configKey: 'registerSuccessEmaileAppend',
|
||
configVal: ',请妥善保管您的账号，我们将为您赠送50次对话额度和5次绘画额度、祝您使用愉快',
|
||
public: 0,
|
||
encry: 0,
|
||
},
|
||
{ configKey: 'registerFailEmailTitle', configVal: 'NineTeam账号激活失败', public: 0, encry: 0 },
|
||
{ configKey: 'registerFailEmailTeamName', configVal: 'NineTeam团队', public: 0, 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: '20', public: 1, encry: 0 },
|
||
{ configKey: 'firstRregisterSendModel4Count', configVal: '2', public: 1, encry: 0 },
|
||
{ configKey: 'firstRregisterSendDrawMjCount', configVal: '3', public: 1, encry: 0 },
|
||
{ configKey: 'inviteSendStatus', configVal: '1', public: 1, encry: 0 },
|
||
{ configKey: 'inviteGiveSendModel3Count', configVal: '30', public: 1, encry: 0 },
|
||
{ configKey: 'inviteGiveSendModel4Count', configVal: '3', public: 1, encry: 0 },
|
||
{ configKey: 'inviteGiveSendDrawMjCount', configVal: '1', public: 1, encry: 0 },
|
||
{ configKey: 'invitedGuestSendModel3Count', configVal: '10', public: 1, encry: 0 },
|
||
{ configKey: 'invitedGuestSendModel4Count', configVal: '1', public: 1, encry: 0 },
|
||
{ configKey: 'invitedGuestSendDrawMjCount', configVal: '1', 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);
|
||
}
|
||
}
|
||
};
|
||
DatabaseService = __decorate([
|
||
(0, common_1.Injectable)(),
|
||
__metadata("design:paramtypes", [typeorm_1.Connection])
|
||
], DatabaseService);
|
||
exports.DatabaseService = DatabaseService;
|