"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.MidjourneyController = void 0; const adminAuth_guard_1 = require("../../common/auth/adminAuth.guard"); const jwtAuth_guard_1 = require("../../common/auth/jwtAuth.guard"); const superAuth_guard_1 = require("../../common/auth/superAuth.guard"); const common_1 = require("@nestjs/common"); const swagger_1 = require("@nestjs/swagger"); const axios_1 = require("axios"); const getList_dto_1 = require("./dto/getList.dto"); const mjDraw_dto_1 = require("./dto/mjDraw.dto"); const midjourney_service_1 = require("./midjourney.service"); let MidjourneyController = class MidjourneyController { constructor(midjourneyService) { this.midjourneyService = midjourneyService; } async getDrawList(req, params) { return await this.midjourneyService.getDrawList(req, params); } async getList(params) { return await this.midjourneyService.getList(params); } async getFullPrompt(id) { return await this.midjourneyService.getFullPrompt(id); } async getAdminDrawList(req, params) { return await this.midjourneyService.getAdminDrawList(req, params); } async recDraw(params) { return await this.midjourneyService.recDraw(params); } async download(url, res) { const response = await axios_1.default.get(url, { responseType: 'arraybuffer' }); const buffer = Buffer.from(response.data, 'binary'); res.set({ 'Content-Type': 'image/png' }); res.send(buffer); } async deleteDraw(id, req) { return await this.midjourneyService.deleteDraw(id, req); } async delLog(req, body) { return await this.midjourneyService.delLog(req, body); } async setPrompt(req, body) { return await this.midjourneyService.setPrompt(req, body); } async delPrompt(req, body) { return await this.midjourneyService.delPrompt(req, body); } async queryPrompt() { return await this.midjourneyService.queryPrompt(); } async proxyImg(params) { return await this.midjourneyService.proxyImg(params); } async mjDraw(body, req) { return await this.midjourneyService.addMjDrawQueue(body, req); } }; __decorate([ (0, common_1.Get)('drawList'), (0, swagger_1.ApiOperation)({ summary: '获取我的绘画列表' }), (0, common_1.UseGuards)(jwtAuth_guard_1.JwtAuthGuard), (0, swagger_1.ApiBearerAuth)(), __param(0, (0, common_1.Req)()), __param(1, (0, common_1.Query)()), __metadata("design:type", Function), __metadata("design:paramtypes", [Object, Object]), __metadata("design:returntype", Promise) ], MidjourneyController.prototype, "getDrawList", null); __decorate([ (0, common_1.Get)('getList'), (0, swagger_1.ApiOperation)({ summary: '获取绘画列表' }), __param(0, (0, common_1.Query)()), __metadata("design:type", Function), __metadata("design:paramtypes", [getList_dto_1.GetListDto]), __metadata("design:returntype", Promise) ], MidjourneyController.prototype, "getList", null); __decorate([ (0, common_1.Get)('getFullPrompt'), (0, swagger_1.ApiOperation)({ summary: '获取绘画列表' }), __param(0, (0, common_1.Query)('id')), __metadata("design:type", Function), __metadata("design:paramtypes", [Number]), __metadata("design:returntype", Promise) ], MidjourneyController.prototype, "getFullPrompt", null); __decorate([ (0, common_1.Get)('adminDrawList'), (0, swagger_1.ApiOperation)({ summary: '管理端获取绘画列表' }), (0, common_1.UseGuards)(adminAuth_guard_1.AdminAuthGuard), (0, swagger_1.ApiBearerAuth)(), __param(0, (0, common_1.Req)()), __param(1, (0, common_1.Query)()), __metadata("design:type", Function), __metadata("design:paramtypes", [Object, getList_dto_1.GetListDto]), __metadata("design:returntype", Promise) ], MidjourneyController.prototype, "getAdminDrawList", null); __decorate([ (0, common_1.Post)('rec'), (0, swagger_1.ApiOperation)({ summary: '推荐图片' }), (0, common_1.UseGuards)(superAuth_guard_1.SuperAuthGuard), (0, swagger_1.ApiBearerAuth)(), __param(0, (0, common_1.Body)()), __metadata("design:type", Function), __metadata("design:paramtypes", [Object]), __metadata("design:returntype", Promise) ], MidjourneyController.prototype, "recDraw", null); __decorate([ (0, common_1.Post)('download'), (0, swagger_1.ApiOperation)({ summary: '下载绘画' }), __param(0, (0, common_1.Body)('url')), __param(1, (0, common_1.Res)()), __metadata("design:type", Function), __metadata("design:paramtypes", [String, Object]), __metadata("design:returntype", Promise) ], MidjourneyController.prototype, "download", null); __decorate([ (0, common_1.Post)('delete'), (0, swagger_1.ApiOperation)({ summary: '删除绘画' }), (0, common_1.UseGuards)(jwtAuth_guard_1.JwtAuthGuard), (0, swagger_1.ApiBearerAuth)(), __param(0, (0, common_1.Body)('id')), __param(1, (0, common_1.Req)()), __metadata("design:type", Function), __metadata("design:paramtypes", [Number, Object]), __metadata("design:returntype", Promise) ], MidjourneyController.prototype, "deleteDraw", null); __decorate([ (0, common_1.Post)('del'), (0, swagger_1.ApiOperation)({ summary: '删除log' }), (0, common_1.UseGuards)(superAuth_guard_1.SuperAuthGuard), (0, swagger_1.ApiBearerAuth)(), __param(0, (0, common_1.Req)()), __param(1, (0, common_1.Body)()), __metadata("design:type", Function), __metadata("design:paramtypes", [Object, Object]), __metadata("design:returntype", Promise) ], MidjourneyController.prototype, "delLog", null); __decorate([ (0, common_1.Post)('setPrompt'), (0, swagger_1.ApiOperation)({ summary: '添加修改prompt提示词' }), (0, common_1.UseGuards)(superAuth_guard_1.SuperAuthGuard), (0, swagger_1.ApiBearerAuth)(), __param(0, (0, common_1.Req)()), __param(1, (0, common_1.Body)()), __metadata("design:type", Function), __metadata("design:paramtypes", [Object, Object]), __metadata("design:returntype", Promise) ], MidjourneyController.prototype, "setPrompt", null); __decorate([ (0, common_1.Post)('delPrompt'), (0, swagger_1.ApiOperation)({ summary: '添加修改prompt提示词' }), (0, common_1.UseGuards)(superAuth_guard_1.SuperAuthGuard), (0, swagger_1.ApiBearerAuth)(), __param(0, (0, common_1.Req)()), __param(1, (0, common_1.Body)()), __metadata("design:type", Function), __metadata("design:paramtypes", [Object, Object]), __metadata("design:returntype", Promise) ], MidjourneyController.prototype, "delPrompt", null); __decorate([ (0, common_1.Get)('queryPrompts'), (0, swagger_1.ApiOperation)({ summary: '查询prompt列表' }), __metadata("design:type", Function), __metadata("design:paramtypes", []), __metadata("design:returntype", Promise) ], MidjourneyController.prototype, "queryPrompt", null); __decorate([ (0, common_1.Get)('proxy'), (0, swagger_1.ApiOperation)({ summary: '代理图片' }), (0, swagger_1.ApiBearerAuth)(), __param(0, (0, common_1.Query)()), __metadata("design:type", Function), __metadata("design:paramtypes", [Object]), __metadata("design:returntype", Promise) ], MidjourneyController.prototype, "proxyImg", null); __decorate([ (0, common_1.Post)('addMjDrawQueue'), (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", [mjDraw_dto_1.MjDrawDto, Object]), __metadata("design:returntype", Promise) ], MidjourneyController.prototype, "mjDraw", null); MidjourneyController = __decorate([ (0, common_1.Controller)('midjourney'), __metadata("design:paramtypes", [midjourney_service_1.MidjourneyService]) ], MidjourneyController); exports.MidjourneyController = MidjourneyController;