xinyin025/99AI
1
1
Fork 0
mirror of https://github.com/vastxie/99AI.git synced 2025-11-14 04:33:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

NineAI 2.4.2

Browse Source
This commit is contained in:
vastxie
2024-01-17 09:22:28 +08:00
commit bdc48207fc
636 changed files with 41864 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
dist/modules/mj/dto/mjDraw.dto.js vendored Normal file
View File

@@ -0,0 +1,26 @@
"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.MjDrawDto = void 0;
const class_validator_1 = require("class-validator");
const swagger_1 = require("@nestjs/swagger");
class MjDrawDto {
}
__decorate([
(0, swagger_1.ApiProperty)({
example: 'close-up polaroid photo, of a little joyful cute panda, in the forest, sun rays coming, photographic, sharp focus, depth of field, soft lighting, heigh quality, 24mm, Nikon Z FX',
description: '绘画提示词！',
required: true,
}),
(0, class_validator_1.IsDefined)({ message: '绘画提示词是必传参数！' }),
__metadata("design:type", String)
], MjDrawDto.prototype, "prompt", void 0);
exports.MjDrawDto = MjDrawDto;

27
dist/modules/mj/dto/mjEnlargeImg.dto.js vendored Normal file
View File

@@ -0,0 +1,27 @@
"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.MjEnlargeImgDto = void 0;
const class_validator_1 = require("class-validator");
const swagger_1 = require("@nestjs/swagger");
class MjEnlargeImgDto {
}
__decorate([
(0, swagger_1.ApiProperty)({ example: '1105361939590287360', description: '当前大图的message_id、四张的这种才存在有效的！', required: true }),
(0, class_validator_1.IsDefined)({ message: '图片的message_id是必传的' }),
__metadata("design:type", String)
], MjEnlargeImgDto.prototype, "message_id", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 1, description: '图片的orderId是必传的 表示放大图片的第几张！', required: true }),
(0, class_validator_1.IsDefined)({ message: '图片固体顺序id是必传的！' }),
__metadata("design:type", Number)
], MjEnlargeImgDto.prototype, "orderId", void 0);
exports.MjEnlargeImgDto = MjEnlargeImgDto;

27
dist/modules/mj/dto/mjTransform.dto.js vendored Normal file
View File

@@ -0,0 +1,27 @@
"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.MjTransformImgDto = void 0;
const class_validator_1 = require("class-validator");
const swagger_1 = require("@nestjs/swagger");
class MjTransformImgDto {
}
__decorate([
(0, swagger_1.ApiProperty)({ example: '1105361939590287360', description: '当前大图的message_id、四张的这种才存在有效的！', required: true }),
(0, class_validator_1.IsDefined)({ message: '图片的message_id是必传的' }),
__metadata("design:type", String)
], MjTransformImgDto.prototype, "message_id", void 0);
__decorate([
(0, swagger_1.ApiProperty)({ example: 1, description: '图片的orderId是必传的 表示放大图片的第几张！', required: true }),
(0, class_validator_1.IsDefined)({ message: '图片固体顺序id是必传的！' }),
__metadata("design:type", Number)
], MjTransformImgDto.prototype, "orderId", void 0);
exports.MjTransformImgDto = MjTransformImgDto;

75
dist/modules/mj/mj.controller.js vendored Normal file
View File

@@ -0,0 +1,75 @@
"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);
};
var __param = (this && this.__param) || function (paramIndex, decorator) {
return function (target, key) { decorator(target, key, paramIndex); }
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.MjController = void 0;
const common_1 = require("@nestjs/common");
const mj_service_1 = require("./mj.service");
const swagger_1 = require("@nestjs/swagger");
const mjDraw_dto_1 = require("./dto/mjDraw.dto");
const mjEnlargeImg_dto_1 = require("./dto/mjEnlargeImg.dto");
const jwtAuth_guard_1 = require("../../common/auth/jwtAuth.guard");
const mjTransform_dto_1 = require("./dto/mjTransform.dto");
let MjController = class MjController {
constructor(mjService) {
this.mjService = mjService;
}
draw(body, req) {
return this.mjService.draw(body, req);
}
upscaleSingleImg(body, req) {
return this.mjService.upscaleSingleImg(body, req);
}
variationSingleImg(body, req) {
return this.mjService.variationSingleImg(body, req);
}
};
__decorate([
(0, common_1.Post)('draw'),
(0, swagger_1.ApiOperation)({ summary: '绘制mj图片' }),
(0, common_1.UseGuards)(jwtAuth_guard_1.JwtAuthGuard),
(0, swagger_1.ApiBearerAuth)(),
__param(0, (0, common_1.Body)()),
__param(1, (0, common_1.Req)()),
__metadata("design:type", Function),
__metadata("design:paramtypes", [mjDraw_dto_1.MjDrawDto, Object]),
__metadata("design:returntype", void 0)
], MjController.prototype, "draw", null);
__decorate([
(0, common_1.Post)('upscaleSingleImg'),
(0, swagger_1.ApiOperation)({ summary: '放大单张图片' }),
(0, common_1.UseGuards)(jwtAuth_guard_1.JwtAuthGuard),
(0, swagger_1.ApiBearerAuth)(),
__param(0, (0, common_1.Body)()),
__param(1, (0, common_1.Req)()),
__metadata("design:type", Function),
__metadata("design:paramtypes", [mjEnlargeImg_dto_1.MjEnlargeImgDto, Object]),
__metadata("design:returntype", void 0)
], MjController.prototype, "upscaleSingleImg", null);
__decorate([
(0, common_1.Post)('variationSingleImg'),
(0, swagger_1.ApiOperation)({ summary: '变体单张图片' }),
(0, common_1.UseGuards)(jwtAuth_guard_1.JwtAuthGuard),
(0, swagger_1.ApiBearerAuth)(),
__param(0, (0, common_1.Body)()),
__param(1, (0, common_1.Req)()),
__metadata("design:type", Function),
__metadata("design:paramtypes", [mjTransform_dto_1.MjTransformImgDto, Object]),
__metadata("design:returntype", void 0)
], MjController.prototype, "variationSingleImg", null);
MjController = __decorate([
(0, swagger_1.ApiTags)('mj'),
(0, common_1.Controller)('mj'),
__metadata("design:paramtypes", [mj_service_1.MjService])
], MjController);
exports.MjController = MjController;

27
dist/modules/mj/mj.module.js vendored Normal file
View File

@@ -0,0 +1,27 @@
"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;
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.MjModule = void 0;
const chatLog_entity_1 = require("./../chatLog/chatLog.entity");
const common_1 = require("@nestjs/common");
const mj_service_1 = require("./mj.service");
const mj_controller_1 = require("./mj.controller");
const typeorm_1 = require("@nestjs/typeorm");
const balance_entity_1 = require("../userBalance/balance.entity");
let MjModule = class MjModule {
};
MjModule = __decorate([
(0, common_1.Global)(),
(0, common_1.Module)({
imports: [typeorm_1.TypeOrmModule.forFeature([chatLog_entity_1.ChatLogEntity, balance_entity_1.BalanceEntity])],
providers: [mj_service_1.MjService],
controllers: [mj_controller_1.MjController],
exports: [mj_service_1.MjService],
})
], MjModule);
exports.MjModule = MjModule;

575
dist/modules/mj/mj.service.js vendored Normal file
View File

@@ -0,0 +1,575 @@
"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);
};
var __param = (this && this.__param) || function (paramIndex, decorator) {
return function (target, key) { decorator(target, key, paramIndex); }
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.MjService = void 0;
const globalConfig_service_1 = require("./../globalConfig/globalConfig.service");
const upload_service_1 = require("./../upload/upload.service");
const common_1 = require("@nestjs/common");
const axios_1 = require("axios");
const chatLog_service_1 = require("../chatLog/chatLog.service");
const balance_constant_1 = require("../../common/constants/balance.constant");
const utils_1 = require("../../common/utils");
const chatLog_entity_1 = require("../chatLog/chatLog.entity");
const typeorm_1 = require("typeorm");
const typeorm_2 = require("@nestjs/typeorm");
const balance_entity_1 = require("../userBalance/balance.entity");
const fanyi_service_1 = require("../fanyi/fanyi.service");
const badwords_service_1 = require("../badwords/badwords.service");
let MjService = class MjService {
constructor(chatLogEntity, balanceEntity, uploadService, chatLogService, globalConfigService, fanyiService, badwordsService) {
this.chatLogEntity = chatLogEntity;
this.balanceEntity = balanceEntity;
this.uploadService = uploadService;
this.chatLogService = chatLogService;
this.globalConfigService = globalConfigService;
this.fanyiService = fanyiService;
this.badwordsService = badwordsService;
this.rateLimits = {};
this.drawWorking = [];
this.enlargeWorking = [];
this.queueCount = 0;
this.freeQueueUsers = {};
}
async mjDraw(data) {
const { jobId, prompt, startTime, userId } = data;
console.log('绘画任务开始', 'mjservice');
await new Promise((resolve) => setTimeout(resolve, 5000));
return { a: 1, b: 2 };
}
async draw(body, req) {
await this.checkAuth(req);
await this.badwordsService.checkBadWords(body.prompt, req.user.id);
const basicPrompt = body.prompt;
let fyPrompt = body.prompt;
const { baiduFanyiAppId, baiduFanyiSecret } = await this.globalConfigService.getConfigs(['baiduFanyiAppId', 'baiduFanyiSecret']);
if (baiduFanyiAppId && baiduFanyiSecret) {
fyPrompt = await this.fanyiService.convertToEnglish(basicPrompt);
}
const randomId = `[${(0, utils_1.createRandomUid)()}]`;
const prompt = `${randomId} ${fyPrompt}`;
console.log('randomId: ', randomId);
console.log('prompt --------> ', prompt);
const isWorking = this.drawWorking.find((item) => item.includes(body.prompt));
if (isWorking) {
throw new common_1.HttpException('当前提示词已经在任务队列中了、请勿重复提交。。。', common_1.HttpStatus.BAD_REQUEST);
}
if (this.queueCount >= 3) {
throw new common_1.HttpException('当前绘图任务满载、请排队等候、队列任务完成后即可开始您的任务...', common_1.HttpStatus.BAD_REQUEST);
}
await this.checkRateLimit(req);
this.queueCount++;
console.log(`开始请求用户${req.user.id} 队列+1: `, this.queueCount);
try {
const historyDraw = await this.chatLogEntity.find({ where: { prompt: (0, typeorm_1.Like)(`%${prompt}%`) } });
const histroyMessageIds = historyDraw.map((item) => item.message_id);
this.drawWorking.push(prompt);
let drawDetail;
const sendRes = await this.sendDrawInteractions(prompt, histroyMessageIds, randomId);
if (sendRes) {
console.log(`历史中存在当前图片、直接获取！`);
drawDetail = sendRes;
}
else {
drawDetail = await this.pollForResult(prompt, histroyMessageIds, randomId);
}
this.queueCount--;
this.queueCount < 0 && (this.queueCount = 0);
console.log('绘制图片任务结束 队列-1: ', this.queueCount);
const { id, content, channel_id, attachments = [], timestamp } = drawDetail;
if (!attachments.length || !attachments[0].url) {
throw new common_1.HttpException('绘画失败', common_1.HttpStatus.BAD_REQUEST);
}
const { filename, url, width, height, size } = attachments[0];
console.log('拿到了远程地址: ', url);
const mjNotSaveImg = this.globalConfigService.getConfigs(['mjNotSaveImg']);
let cosUrl = '';
if (!Number(mjNotSaveImg) || Number(mjNotSaveImg) === 0) {
cosUrl = await this.uploadService.uploadFileFromUrl({ filename, url });
console.log('存入图片完成: ', cosUrl);
}
const logInfo = {
curIp: (0, utils_1.getClientIp)(req),
userId: req.user.id,
type: balance_constant_1.DeductionKey.PAINT_TYPE,
prompt,
answer: cosUrl,
model: 'mj',
extend: this.removeEmoji(JSON.stringify(drawDetail)),
message_id: id,
variationId: id,
upscaleId: id,
group: 1,
isSaveImg: !Number(mjNotSaveImg) || Number(mjNotSaveImg) === 0,
fileInfo: JSON.stringify({ width, height, size, filename, cosUrl }),
};
await this.chatLogService.saveChatLog(logInfo);
await this.deductBalance(req);
this.drawWorking = this.drawWorking.filter((item) => item !== body.prompt);
return cosUrl;
}
catch (error) {
this.queueCount--;
this.queueCount < 0 && (this.queueCount = 0);
console.log('绘制图片任务异常中断 队列-1: ', this.queueCount);
this.drawWorking = this.drawWorking.filter((item) => item !== body.prompt);
throw new common_1.HttpException(error.response, common_1.HttpStatus.BAD_REQUEST);
}
}
async upscaleSingleImg(body, req) {
if (this.queueCount >= 3) {
throw new common_1.HttpException('当前绘图任务满载、请排队等候、队列任务完成后即可开始您的任务...', common_1.HttpStatus.BAD_REQUEST);
}
this.queueCount++;
console.log(`用户${req.user.id}开始请求放大图片 队列+1: `, this.queueCount);
const { message_id, orderId } = body;
try {
const historyLog = await this.chatLogEntity.findOne({ where: { message_id } });
if (!historyLog) {
throw new common_1.HttpException('历史记录中不存在当前图片、请确认您放大的图片是否存在', common_1.HttpStatus.BAD_REQUEST);
}
const isAreadlyEnlarge = await this.chatLogEntity.findOne({ where: { upscaleId: message_id, action: 'enlarge', orderId } });
if (isAreadlyEnlarge) {
throw new common_1.HttpException('当前图片已经放大过了、请勿重复放大!', common_1.HttpStatus.BAD_REQUEST);
}
const { prompt, extend } = historyLog;
let historyDetailDrawInfo = null;
try {
historyDetailDrawInfo = JSON.parse(extend);
}
catch (error) {
historyDetailDrawInfo = [];
}
const { components = [] } = historyDetailDrawInfo;
if (!components.length) {
throw new common_1.HttpException('当前图片没有绘画信息、无法放大!', common_1.HttpStatus.BAD_REQUEST);
}
const currentImgComponent = components[0]['components'][orderId - 1];
const { custom_id } = currentImgComponent;
console.log('放大custom_id: ', custom_id);
const params = { message_id, custom_id, prompt, orderId };
await this.sendSmInteractions(params);
console.log('发送放大指令成功');
const historyDraw = await this.chatLogEntity.find({ where: { prompt: (0, typeorm_1.Like)(`%${prompt}%`) } });
const histroyMessageIds = historyDraw.map((item) => item.message_id);
console.log('历史这些id已经被获取过了 不能拿了: ', histroyMessageIds);
const enlargeImgInfo = await this.pollForUpscaleResult(params, histroyMessageIds);
this.queueCount--;
this.queueCount < 0 && (this.queueCount = 0);
console.log('放大图片任务结束 队列-1: ', this.queueCount);
const { id, content, channel_id, attachments = [], timestamp } = enlargeImgInfo;
if (!attachments.length || !attachments[0].url) {
throw new common_1.HttpException('放大当前图片失败', common_1.HttpStatus.BAD_REQUEST);
}
const { filename, url, width, height, size } = attachments[0];
const mjNotSaveImg = this.globalConfigService.getConfigs(['mjNotSaveImg']);
let cosUrl = '';
if (!Number(mjNotSaveImg) || Number(mjNotSaveImg) === 0) {
cosUrl = await this.uploadService.uploadFileFromUrl({ filename, url });
console.log('存入图片完成: ', cosUrl);
}
const logInfo = {
curIp: (0, utils_1.getClientIp)(req),
userId: req.user.id,
type: balance_constant_1.DeductionKey.PAINT_TYPE,
prompt,
answer: cosUrl,
model: 'mj',
extend: this.removeEmoji(JSON.stringify(enlargeImgInfo)),
message_id,
upscaleId: id,
variationId: id,
action: 'enlarge',
orderId: params.orderId,
isSaveImg: !Number(mjNotSaveImg) || Number(mjNotSaveImg) === 0,
fileInfo: JSON.stringify({ width, height, size, filename, cosUrl }),
};
await this.chatLogService.saveChatLog(logInfo);
return cosUrl;
}
catch (error) {
console.log('error: ', error);
this.queueCount--;
this.queueCount < 0 && (this.queueCount = 0);
console.log('放大图片任务异常中断 队列-1: ', this.queueCount);
throw new common_1.HttpException(error.response, common_1.HttpStatus.BAD_REQUEST);
}
}
async variationSingleImg(body, req) {
if (this.queueCount >= 3) {
throw new common_1.HttpException('当前绘图任务满载、请排队等候、队列任务完成后即可开始您的任务...', common_1.HttpStatus.BAD_REQUEST);
}
await this.checkAuth(req);
await this.checkRateLimit(req);
this.queueCount++;
console.log(`用户${req.user.id}开始请求变换图片 队列+1: `, this.queueCount);
const { message_id, orderId } = body;
try {
const historyLog = await this.chatLogEntity.findOne({ where: { message_id } });
if (!historyLog) {
throw new common_1.HttpException('历史记录中不存在当前图片、请确认您需要变换的图片是否存在', common_1.HttpStatus.BAD_REQUEST);
}
const { prompt, extend } = historyLog;
let historyDetailDrawInfo = null;
try {
historyDetailDrawInfo = JSON.parse(extend);
}
catch (error) {
historyDetailDrawInfo = [];
}
const { components = [] } = historyDetailDrawInfo;
if (!components.length) {
throw new common_1.HttpException('当前图片没有绘画信息、无法变体!', common_1.HttpStatus.BAD_REQUEST);
}
const currentImgComponent = components[1]['components'][orderId - 1];
const { custom_id } = currentImgComponent;
const historyVariationLog = await this.chatLogEntity.find({ where: { variationId: (0, typeorm_1.Not)((0, typeorm_1.IsNull)()), prompt: (0, typeorm_1.Like)(`%${prompt}%`) } });
const historyVariationIds = historyVariationLog.map((item) => item.variationId);
const params = { message_id, custom_id, prompt, orderId };
await this.sendSmInteractions(params);
const variationImgInfo = await this.pollForVariationResult(params, historyVariationIds);
this.queueCount--;
this.queueCount < 0 && (this.queueCount = 0);
console.log('变换图片任务结束 队列-1: ', this.queueCount);
const { id, content, channel_id, attachments = [], timestamp } = variationImgInfo;
if (!attachments.length || !attachments[0].url) {
throw new common_1.HttpException('变换当前图片失败', common_1.HttpStatus.BAD_REQUEST);
}
const { filename, url, width, height, size } = attachments[0];
const mjNotSaveImg = this.globalConfigService.getConfigs(['mjNotSaveImg']);
let cosUrl = '';
if (!Number(mjNotSaveImg) || Number(mjNotSaveImg) === 0) {
cosUrl = await this.uploadService.uploadFileFromUrl({ filename, url });
console.log('存入图片完成: ', cosUrl);
}
const logInfo = {
curIp: (0, utils_1.getClientIp)(req),
userId: req.user.id,
type: balance_constant_1.DeductionKey.PAINT_TYPE,
prompt,
answer: cosUrl,
model: 'mj',
group: 1,
extend: this.removeEmoji(JSON.stringify(variationImgInfo)),
message_id: id,
upscaleId: id,
variationId: id,
action: 'enlarge',
orderId: params.orderId,
isSaveImg: !Number(mjNotSaveImg) || Number(mjNotSaveImg) === 0,
fileInfo: JSON.stringify({ width, height, size, filename, cosUrl }),
};
await this.chatLogService.saveChatLog(logInfo);
return cosUrl;
}
catch (error) {
console.log('error: ', error);
this.queueCount--;
this.queueCount < 0 && (this.queueCount = 0);
console.log('变化图片任务异常中断 队列-1: ', this.queueCount);
throw new common_1.HttpException(error.response, common_1.HttpStatus.BAD_REQUEST);
}
}
async sendSmInteractions(params) {
const { message_id, custom_id } = params;
const { application_id, guild_id, channel_id, session_id, version, id, authorization, mjProxy } = await this.getMjDefaultParams();
const url = mjProxy == 1 ? `http://172.247.48.137:8000/mj/draw` : 'https://discord.com/api/v9/interactions';
const headers = { authorization };
const body = {
type: 3,
guild_id,
channel_id,
message_flags: 0,
message_id,
application_id,
session_id,
data: {
component_type: 2,
custom_id,
},
};
try {
await axios_1.default.post(url, body, { headers });
console.log('绘图指令完成');
}
catch (error) {
console.log('error: ', error);
throw new common_1.HttpException('放大单张图片请求失败...', common_1.HttpStatus.BAD_REQUEST);
}
}
async pollForUpscaleResult(params, histroyMessageIds) {
const { message_id, custom_id, prompt, orderId } = params;
let enlargeImgDetail = null;
let pollingCount = 0;
while (!enlargeImgDetail && pollingCount < 10) {
try {
const startTime = Date.now();
const messageList = await this.queryMessageList();
console.log(`第 ${pollingCount + 1} 次开始查询 => 当前查询结果：${messageList.length}`);
if (messageList && messageList.length) {
enlargeImgDetail = await this.findCurrentEnlargeImgResult(messageList, params, histroyMessageIds);
}
const elapsedTime = Date.now() - startTime;
const nextPollingDelay = 3000;
await this.sleep(Math.max(nextPollingDelay - elapsedTime, 0));
pollingCount++;
}
catch (error) {
console.error(`查询期间出现错误：${error.message}`);
}
}
return enlargeImgDetail;
}
async pollForVariationResult(params, historyVariationIds) {
const { message_id, custom_id, prompt, orderId } = params;
console.log('开始轮询单张变换图片结果');
let variationImgDetail = null;
let pollingCount = 0;
while (!variationImgDetail && pollingCount < 10) {
try {
console.log(`第 ${pollingCount + 1} 次开始查询[变换图片]`);
const startTime = Date.now();
const messageList = await this.queryMessageList();
if (messageList && messageList.length) {
variationImgDetail = await this.findCurrentVariationImgResult(messageList, params, historyVariationIds);
}
const elapsedTime = Date.now() - startTime;
const nextPollingDelay = 8000;
await this.sleep(Math.max(nextPollingDelay - elapsedTime, 0));
pollingCount++;
}
catch (error) {
console.error(`查询期间出现错误：${error.message}`);
}
}
if (!variationImgDetail) {
throw new common_1.HttpException('变换当前图片超时！', common_1.HttpStatus.BAD_REQUEST);
}
return variationImgDetail;
}
async findCurrentEnlargeImgResult(messageList, params, histroyMessageIds) {
const { message_id, custom_id, prompt, orderId } = params;
const randomId = prompt.substring(0, 12);
console.log('本次放大图片的id: ', randomId);
const enlargeImgDetail = messageList.find((item) => {
const { content } = item;
if (!this.extractContent(content))
return false;
const { prompt, order } = this.extractContent(content);
return prompt.includes(randomId) && params.orderId === order && !histroyMessageIds.includes(item.id);
});
return enlargeImgDetail;
}
async findCurrentVariationImgResult(messageList, params, historyVariationIds) {
const { message_id, custom_id, prompt, orderId } = params;
const randomId = prompt.substring(0, 12);
const variationImgDetail = messageList.find((item) => {
const { content } = item;
const promptMatch = content.match(/\*\*(.+?)\*\*/);
const prompt = promptMatch ? promptMatch[1] : '';
if (!prompt)
return false;
return prompt.includes(randomId) && !historyVariationIds.includes(item.id);
});
return variationImgDetail;
}
async sendDrawInteractions(prompt, histroyMessageIds, randomId) {
const messageList = await this.queryMessageList();
const drawDetail = await this.findCurrentPromptResult(messageList, randomId, histroyMessageIds);
if (drawDetail) {
console.log('有历史信息之间返回: ', drawDetail);
return drawDetail;
}
const { application_id, guild_id, channel_id, session_id, version, id, authorization, mjProxy } = await this.getMjDefaultParams();
const payloadJson = {
type: 2,
application_id,
guild_id,
channel_id,
session_id,
data: { version, id, name: 'imagine', type: 1, options: [{ type: 3, name: 'prompt', value: prompt }], attachments: [] },
};
try {
const url = mjProxy == 1 ? `http://172.247.48.137:8000/mj/draw` : 'https://discord.com/api/v9/interactions';
const headers = { authorization };
const res = await axios_1.default.post(url, payloadJson, { headers });
console.log('发送绘画指令结果: ', res.data);
return false;
}
catch (error) {
console.log('axios: ', error);
throw new common_1.HttpException('绘画请求失败、当前使用人数过多、请稍后试试吧、排队中...', common_1.HttpStatus.BAD_REQUEST);
}
}
async pollForResult(prompt, histroyMessageIds, randomId) {
console.log('开始查询绘画结果轮询');
const startTime = Date.now();
try {
const MAX_POLLING_COUNT = 13;
const SHORT_INTERVAL = 12000;
const LONG_INTERVAL = 5000;
const TIME_THRESHOLD = 60 * 1000;
let pollingCount = 0;
let isLongInterval = false;
let drawDetail = null;
while (!drawDetail && pollingCount < MAX_POLLING_COUNT) {
console.log(`第 ${pollingCount + 1} 次开始查询`);
if (Date.now() - startTime >= TIME_THRESHOLD) {
isLongInterval = true;
}
await this.sleep(isLongInterval ? LONG_INTERVAL : SHORT_INTERVAL);
const messageList = await this.queryMessageList();
drawDetail = await this.findCurrentPromptResult(messageList, randomId, histroyMessageIds);
pollingCount++;
}
if (!drawDetail) {
throw new common_1.HttpException('绘画超时，请稍后再试！', common_1.HttpStatus.BAD_REQUEST);
}
const endTime = Date.now();
console.log(`本次绘图耗时: ${Math.floor((endTime - startTime) / 1000)} S`);
return drawDetail;
}
catch (err) {
console.error(err.message);
throw new common_1.HttpException('网络连接失败，请稍后再试！', common_1.HttpStatus.INTERNAL_SERVER_ERROR);
}
}
async findCurrentPromptResult(data, randomId, histroyMessageIds) {
if (!data || !data.length)
return;
console.log('本次比对的随机ID: ', randomId);
const matchingItem = data.find((item) => {
const { attachments = [], content, edited_timestamp } = item;
return content.includes(randomId) && attachments.length > 0 && !edited_timestamp && !histroyMessageIds.includes(item.id);
});
return matchingItem || null;
}
async queryMessageList() {
try {
const { application_id, guild_id, channel_id, session_id, version, id, authorization, mjProxy } = await this.getMjDefaultParams();
const url = mjProxy == 1
? `http://172.247.48.137:8000/mj/list?channel_id=${channel_id}`
: `https://discord.com/api/v9/channels/${channel_id}/messages?limit=50`;
const headers = { authorization };
const response = await axios_1.default.get(url, { headers });
return response.data;
}
catch (error) {
console.log('axios get: ', error);
throw new common_1.HttpException('查询绘制结果失败...', common_1.HttpStatus.BAD_REQUEST);
}
}
async sleep(ms) {
return new Promise((resolve) => setTimeout(resolve, ms));
}
extractContent(str) {
const promptMatch = str.match(/\*\*(.+?)\*\*/);
const orderMatch = str.match(/- Image #(\d+)/);
if (!promptMatch || !orderMatch) {
return null;
}
const prompt = promptMatch[1];
const order = parseInt(orderMatch[1]);
return { prompt, order };
}
async getMjDefaultParams() {
const configs = await this.globalConfigService.getConfigs([
'mjId',
'mjApplicationId',
'mjGuildId',
'mjChannelId',
'mjSessionId',
'mjVersion',
'mjAuthorization',
'mjRateLimit',
'mjProxy',
]);
const params = {
application_id: configs.mjApplicationId,
guild_id: configs.mjGuildId,
channel_id: configs.mjChannelId,
session_id: configs.mjSessionId,
version: configs.mjVersion,
id: configs.mjId,
authorization: configs.mjAuthorization,
mjRateLimit: configs.mjRateLimit,
mjProxy: configs.mjProxy || 0,
};
return params;
}
removeEmoji(str) {
const regex = /[\uD800-\uDBFF][\uDC00-\uDFFF]/g;
return str.replace(regex, '');
}
async checkAuth(req) {
const m = await this.balanceEntity.findOne({ where: { userId: req.user.id } });
const { id, balance } = m;
if (!balance || (m === null || m === void 0 ? void 0 : m.balance) < 1) {
throw new common_1.HttpException('您当前暂无MJ绘画余额！！！', common_1.HttpStatus.BAD_REQUEST);
}
}
async checkFree(req) {
const { id, role } = req.user;
if (!this.freeQueueUsers[id]) {
this.freeQueueUsers[id] = 1;
}
else {
this.freeQueueUsers[id] = this.freeQueueUsers[id] + 1;
}
console.log(`当前用户${id}使用的次数：`, this.freeQueueUsers[id]);
}
async checkRateLimit(req) {
const { id, role } = req.user;
if (['admin', 'super'].includes(role))
return true;
const { mjRateLimit } = await this.getMjDefaultParams();
if (this.rateLimits[id]) {
const val = this.rateLimits[id];
if (val > Date.now()) {
console.log(`当前用户 ${id} 请求过于频繁！`);
throw new common_1.HttpException(`由于速率限制、当前普通用户限制为${mjRateLimit}秒请求一次、请合理使用！`, common_1.HttpStatus.BAD_REQUEST);
}
else {
this.rateLimits[id] = Date.now() + Number(mjRateLimit) * 1000;
}
}
else {
const timeSpace = Date.now();
this.rateLimits[id] = timeSpace + 1000 * Number(mjRateLimit);
}
}
async deductBalance(req) {
await this.balanceEntity
.createQueryBuilder()
.update(balance_entity_1.BalanceEntity)
.set({ balance: () => 'balance - 1' })
.where('userId = :userId', { userId: req.user.id })
.execute();
}
async test() {
return 1;
}
};
MjService = __decorate([
(0, common_1.Injectable)(),
__param(0, (0, typeorm_2.InjectRepository)(chatLog_entity_1.ChatLogEntity)),
__param(1, (0, typeorm_2.InjectRepository)(balance_entity_1.BalanceEntity)),
__metadata("design:paramtypes", [typeorm_1.Repository,
typeorm_1.Repository,
upload_service_1.UploadService,
chatLog_service_1.ChatLogService,
globalConfig_service_1.GlobalConfigService,
fanyi_service_1.FanyiService,
badwords_service_1.BadwordsService])
], MjService);
exports.MjService = MjService;