create-dp-koa 1.0.0

package/template/src/controllers/logManagement.controller.ts

@@ -0,0 +1,396 @@
+import { Context } from 'koa'
+import { logger } from '@src/framework/utils/logger'
+import { BaseController } from '@src/controllers/base.controller'
+import fs from 'fs'
+import path from 'path'
+
+/**
+ * 日志管理控制器
+ * 提供日志查询、下载、清理等功能
+ */
+export class LogManagementController extends BaseController {
+  
+  /**
+   * 获取日志统计信息
+   */
+  async getLogStats(ctx: Context) {
+    try {
+      const logsDir = path.join(process.cwd(), 'logs')
+      const stats = {
+        totalFiles: 0,
+        totalSize: 0,
+        fileTypes: {} as Record<string, { count: number, size: number }>,
+        lastModified: null as Date | null
+      }
+
+      if (fs.existsSync(logsDir)) {
+        const files = fs.readdirSync(logsDir)
+        
+        for (const file of files) {
+          const filePath = path.join(logsDir, file)
+          const fileStat = fs.statSync(filePath)
+          
+          if (fileStat.isFile()) {
+            stats.totalFiles++
+            stats.totalSize += fileStat.size
+            
+            const ext = path.extname(file) || 'log'
+            if (!stats.fileTypes[ext]) {
+              stats.fileTypes[ext] = { count: 0, size: 0 }
+            }
+            stats.fileTypes[ext].count++
+            stats.fileTypes[ext].size += fileStat.size
+            
+            if (!stats.lastModified || fileStat.mtime > stats.lastModified) {
+              stats.lastModified = fileStat.mtime
+            }
+          }
+        }
+      }
+
+      logger.info('Log stats retrieved', { stats })
+      
+      ctx.body = {
+        code: 0,
+        data: stats,
+        message: '日志统计信息获取成功'
+      }
+    } catch (error) {
+      logger.error('Failed to get log stats', error as Error)
+      ctx.status = 500
+      ctx.body = {
+        code: 500,
+        message: '获取日志统计信息失败'
+      }
+    }
+  }
+
+  /**
+   * 获取日志文件列表
+   */
+  async getLogFiles(ctx: Context) {
+    try {
+      const logsDir = path.join(process.cwd(), 'logs')
+      const files: any[] = []
+
+      if (fs.existsSync(logsDir)) {
+        const fileList = fs.readdirSync(logsDir)
+        
+        for (const file of fileList) {
+          const filePath = path.join(logsDir, file)
+          const fileStat = fs.statSync(filePath)
+          
+          if (fileStat.isFile()) {
+            files.push({
+              name: file,
+              size: fileStat.size,
+              modified: fileStat.mtime,
+              created: fileStat.birthtime,
+              type: path.extname(file) || 'log'
+            })
+          }
+        }
+      }
+
+      // 按修改时间排序
+      files.sort((a, b) => b.modified.getTime() - a.modified.getTime())
+
+      logger.info('Log files list retrieved', { fileCount: files.length })
+      
+      ctx.body = {
+        code: 0,
+        data: files,
+        message: '日志文件列表获取成功'
+      }
+    } catch (error) {
+      logger.error('Failed to get log files', error as Error)
+      ctx.status = 500
+      ctx.body = {
+        code: 500,
+        message: '获取日志文件列表失败'
+      }
+    }
+  }
+
+  /**
+   * 下载日志文件
+   */
+  async downloadLogFile(ctx: Context) {
+    try {
+      const { filename } = ctx.params
+      
+      if (!filename || !filename.match(/^[a-zA-Z0-9._-]+$/)) {
+        ctx.status = 400
+        ctx.body = {
+          code: 400,
+          message: '无效的文件名'
+        }
+        return
+      }
+
+      const filePath = path.join(process.cwd(), 'logs', filename)
+      
+      if (!fs.existsSync(filePath)) {
+        ctx.status = 404
+        ctx.body = {
+          code: 404,
+          message: '日志文件不存在'
+        }
+        return
+      }
+
+      // 审计日志
+      logger.audit('LOG_DOWNLOAD', filename, {
+        userId: ctx.state.user?.id,
+        ip: ctx.ip,
+        userAgent: ctx.get('User-Agent')
+      })
+
+      ctx.set('Content-Type', 'application/octet-stream')
+      ctx.set('Content-Disposition', `attachment; filename="${filename}"`)
+      ctx.body = fs.createReadStream(filePath)
+      
+    } catch (error) {
+      logger.error('Failed to download log file', error as Error, {
+        filename: ctx.params.filename
+      })
+      ctx.status = 500
+      ctx.body = {
+        code: 500,
+        message: '下载日志文件失败'
+      }
+    }
+  }
+
+  /**
+   * 查看日志内容
+   */
+  async viewLogContent(ctx: Context) {
+    try {
+      const { filename } = ctx.params
+      const { lines = 100, offset = 0 } = ctx.query
+      
+      if (!filename || !filename.match(/^[a-zA-Z0-9._-]+$/)) {
+        ctx.status = 400
+        ctx.body = {
+          code: 400,
+          message: '无效的文件名'
+        }
+        return
+      }
+
+      const filePath = path.join(process.cwd(), 'logs', filename)
+      
+      if (!fs.existsSync(filePath)) {
+        ctx.status = 404
+        ctx.body = {
+          code: 404,
+          message: '日志文件不存在'
+        }
+        return
+      }
+
+      const content = fs.readFileSync(filePath, 'utf8')
+      const linesArray = content.split('\n')
+      const startLine = Math.max(0, Number(offset))
+      const endLine = Math.min(linesArray.length, startLine + Number(lines))
+      
+      const selectedLines = linesArray.slice(startLine, endLine)
+      
+      // 审计日志
+      logger.audit('LOG_VIEW', filename, {
+        userId: ctx.state.user?.id,
+        lines: Number(lines),
+        offset: Number(offset),
+        ip: ctx.ip
+      })
+
+      ctx.body = {
+        code: 0,
+        data: {
+          filename,
+          totalLines: linesArray.length,
+          selectedLines: selectedLines,
+          startLine,
+          endLine,
+          hasMore: endLine < linesArray.length
+        },
+        message: '日志内容获取成功'
+      }
+      
+    } catch (error) {
+      logger.error('Failed to view log content', error as Error, {
+        filename: ctx.params.filename
+      })
+      ctx.status = 500
+      ctx.body = {
+        code: 500,
+        message: '查看日志内容失败'
+      }
+    }
+  }
+
+  /**
+   * 清理过期日志
+   */
+  async cleanExpiredLogs(ctx: Context) {
+    try {
+      const { days = 30 } = ctx.request.body
+      const logsDir = path.join(process.cwd(), 'logs')
+      const cutoffDate = new Date()
+      cutoffDate.setDate(cutoffDate.getDate() - Number(days))
+      
+      let deletedCount = 0
+      let deletedSize = 0
+
+      if (fs.existsSync(logsDir)) {
+        const files = fs.readdirSync(logsDir)
+        
+        for (const file of files) {
+          const filePath = path.join(logsDir, file)
+          const fileStat = fs.statSync(filePath)
+          
+          if (fileStat.isFile() && fileStat.mtime < cutoffDate) {
+            deletedSize += fileStat.size
+            fs.unlinkSync(filePath)
+            deletedCount++
+          }
+        }
+      }
+
+      // 审计日志
+      logger.audit('LOG_CLEANUP', 'expired_logs', {
+        userId: ctx.state.user?.id,
+        days: Number(days),
+        deletedCount,
+        deletedSize,
+        ip: ctx.ip
+      })
+
+      ctx.body = {
+        code: 0,
+        data: {
+          deletedCount,
+          deletedSize,
+          cutoffDate: cutoffDate.toISOString()
+        },
+        message: `成功清理 ${deletedCount} 个过期日志文件`
+      }
+      
+    } catch (error) {
+      logger.error('Failed to clean expired logs', error as Error)
+      ctx.status = 500
+      ctx.body = {
+        code: 500,
+        message: '清理过期日志失败'
+      }
+    }
+  }
+
+  /**
+   * 设置日志级别
+   */
+  async setLogLevel(ctx: Context) {
+    try {
+      const { level, category = 'default' } = ctx.request.body
+      
+      const validLevels = ['trace', 'debug', 'info', 'warn', 'error', 'fatal']
+      if (!validLevels.includes(level)) {
+        ctx.status = 400
+        ctx.body = {
+          code: 400,
+          message: '无效的日志级别'
+        }
+        return
+      }
+
+      // 这里需要实现动态设置日志级别的逻辑
+      // 由于log4js的限制，可能需要重启应用才能生效
+      
+      // 审计日志
+      logger.audit('LOG_LEVEL_CHANGE', category, {
+        userId: ctx.state.user?.id,
+        oldLevel: 'unknown',
+        newLevel: level,
+        ip: ctx.ip
+      })
+
+      ctx.body = {
+        code: 0,
+        data: {
+          category,
+          level,
+          message: '日志级别设置成功，重启应用后生效'
+        },
+        message: '日志级别设置成功'
+      }
+      
+    } catch (error) {
+      logger.error('Failed to set log level', error as Error)
+      ctx.status = 500
+      ctx.body = {
+        code: 500,
+        message: '设置日志级别失败'
+      }
+    }
+  }
+
+  /**
+   * 获取实时日志流
+   */
+  async getLogStream(ctx: Context) {
+    try {
+      const { filename } = ctx.params
+      
+      if (!filename || !filename.match(/^[a-zA-Z0-9._-]+$/)) {
+        ctx.status = 400
+        ctx.body = {
+          code: 400,
+          message: '无效的文件名'
+        }
+        return
+      }
+
+      const filePath = path.join(process.cwd(), 'logs', filename)
+      
+      if (!fs.existsSync(filePath)) {
+        ctx.status = 404
+        ctx.body = {
+          code: 404,
+          message: '日志文件不存在'
+        }
+        return
+      }
+
+      // 设置SSE响应头
+      ctx.set({
+        'Content-Type': 'text/event-stream',
+        'Cache-Control': 'no-cache',
+        'Connection': 'keep-alive',
+        'Access-Control-Allow-Origin': '*',
+        'Access-Control-Allow-Headers': 'Cache-Control'
+      })
+
+      // 审计日志
+      logger.audit('LOG_STREAM_START', filename, {
+        userId: ctx.state.user?.id,
+        ip: ctx.ip
+      })
+
+      // 创建可读流
+      const stream = fs.createReadStream(filePath, { encoding: 'utf8' })
+      
+      ctx.body = stream
+      
+    } catch (error) {
+      logger.error('Failed to get log stream', error as Error, {
+        filename: ctx.params.filename
+      })
+      ctx.status = 500
+      ctx.body = {
+        code: 500,
+        message: '获取日志流失败'
+      }
+    }
+  }
+}

package/template/src/controllers/public/emailSend.controller.ts

@@ -0,0 +1,65 @@
+import { Body,Post, ResponseValidateIf, ResponseValidator } from "@src/framework/decorator/controller";
+import { BaseController, ControllerResponse } from "@src/controllers/base.controller";
+import { SendEmailDto, SendEmailResponseDto } from "@src/dto/controller/home/emailSend.controller.dto";
+import { Inject } from "dp-ioc2";
+import * as jwt from "jsonwebtoken";
+import { YtUserService } from "@src/service/ytUser.service";
+import { logger } from "@src/framework/utils/logger";
+import { AokEmailSender } from "@src/libs/aokEmailSender";
+
+const emailSender = new AokEmailSender(process.env.aok_email_api_key ?? "330b853af20ef6762f259545ef7b4c17")
+
+export class EmailSendController extends BaseController {
+
+    @Inject(YtUserService)
+    ytUserService: YtUserService;
+
+    /**
+     * 短信验证码邮件
+     * @param body 
+     * @returns 
+     */
+    @ResponseValidateIf(SendEmailResponseDto, (data) => data && data.data)
+    @Post()
+    async sendRegisterEmail(
+        @Body(SendEmailDto) body: SendEmailDto
+    ): Promise<ControllerResponse<SendEmailResponseDto>> {
+
+        // 1. 检查邮箱是否已注册
+        // const exist = await this.ytUserService.findByEmail(body.email);
+        // if (exist) {
+        //     return this.fail(-1, "该邮箱已注册");
+        // }
+
+        // 2. 生成随机验证码(4位数字)
+        const vcode = Math.floor(1000 + Math.random() * 9000).toString();
+
+        // 3. 设置验证码过期时间(5分钟后)
+        const expire = Date.now() + 5 * 60 * 1000;
+
+        // 4. 生成JWT token包含验证码和过期时间
+        const vcodeToken = jwt.sign(
+            { vcode, expire, email: body.email },
+            process.env.jwt_secret ?? "123456",
+            { expiresIn: "5m" }
+        );
+
+        // 5. 发送邮件验证码(模拟)
+        const result = await emailSender.sendEmail({
+            templateId: "E_106565864351",
+            templateData: {
+                vcode: String(vcode),
+                login_link: `
+                <a href="https://ytfx.yndaopu.com/login/${vcode}">快捷登录链接</a>
+                `
+            },
+            to: body.email,
+        })
+        // console.log(result)
+        // logger.info(`发送邮件验证码到${body.email}: ${vcode}`);
+        return this.success({
+            expire,
+            vcodeToken
+        }, "验证码发送成功");
+    }
+}

package/template/src/controllers/public/ytUserAuth.controller.ts

@@ -0,0 +1,174 @@
+import { Body, ControllerCache, Get, Post, Query, ResponseValidateIf } from "@src/framework/decorator/controller";
+import { BaseController, ControllerResponse } from "@src/controllers/base.controller";
+import { Inject } from "dp-ioc2";
+import { YtUserService } from "@src/service/ytUser.service";
+import jwt from 'jsonwebtoken';
+import md5 from 'md5';
+import svgCaptcha from 'svg-captcha';
+import { GetYtUserBasicInfoDto, GetYtUserBasicInfoResponseDto, YtUserLoginAutoDto, YtUserLoginDto } from "@src/dto/controller/public/ytUserAuth.controller.dto";
+import { YtUserEntity, YtUserStatusEnum, YtUserTypeEnum } from "@src/entity/ytUser.entity";
+import { CommonServiceResultCode } from "@src/framework/types/ServiceResult";
+import { logger } from "@src/framework/utils/logger";
+import { isDebug } from "@src/framework/utils/function";
+
+interface IvcodeInfo {
+    vcode: string
+    expire: number
+}
+
+export class YtUserAuthController extends BaseController {
+    @Inject(YtUserService)
+    ytUserService: YtUserService;
+
+    @Post()
+    async login(
+        @Body(YtUserLoginDto) body: YtUserLoginDto
+    ): Promise<ControllerResponse<string>> {
+
+        // 验证注册码
+        let vcodeInfo: IvcodeInfo | null = null;
+        try {
+            vcodeInfo = jwt.verify(body.vcodeToken, process.env.jwt_secret ?? "123456") as any;
+        } catch (err) {
+            return this.fail(-1, "验证码可能失效");
+        }
+
+        if (!vcodeInfo) {
+            return this.fail(-1, "验证码解析失败");
+        }
+
+        if (vcodeInfo?.expire < new Date().getTime()) {
+            return this.fail(-1, "验证码过期");
+        }
+
+        if (vcodeInfo.vcode !== body.vcode) {
+            return this.fail(-1, "验证码不正确");
+        }
+
+        // 1. 验证用户是否存在
+        let user = await this.ytUserService.findByEmail(body.email);
+        if (!user) {
+            const ytUser = new YtUserEntity();
+            ytUser.nickName = '用户' + body.email.split("@")[0].slice(0, 5);
+            ytUser.email = body.email;
+            ytUser.avatar = "https://ytfx.yndaopu.com/static/images/Avatar-1.png";
+            // ytUser.password = md5(body.password);
+            let result = await this.ytUserService.add(ytUser);
+            // if (result.code != CommonServiceResultCode.SUCCESS) {
+            //     logger.warn(`注册注册失败` + JSON.stringify(body), result.message);
+            //     return this.fail(-1, "注册失败");
+            // }
+            user = result
+        }
+
+        if (user.status == YtUserStatusEnum.FORBIDDEN) {
+            return this.fail(-1, "用户已被禁用");
+        }
+
+
+        // 3. 生成JWT token
+        const token = jwt.sign(
+            { userId: user.id, type: user.userType },
+            process.env.jwt_secret ?? "123456",
+            { expiresIn: "7d" }
+        );
+
+        return this.success(token, "登录成功");
+    }
+
+
+    /**
+     * 自动注册和登录
+     * @param body 
+     * @returns 
+     */
+    @Post()
+    async loginAuto(
+        @Body(YtUserLoginAutoDto) body: YtUserLoginAutoDto
+    ): Promise<ControllerResponse<string>> {
+
+        // 验证注册码
+        let vcodeInfo: IvcodeInfo | null = null;
+        try {
+            vcodeInfo = jwt.verify(body.vcodeToken, process.env.jwt_secret ?? "123456") as any;
+        } catch (err) {
+            return this.fail(-1, "验证码可能失效");
+        }
+
+        if (!vcodeInfo) {
+            return this.fail(-1, "验证码解析失败");
+        }
+
+        if (vcodeInfo?.expire < new Date().getTime()) {
+            return this.fail(-1, "验证码过期");
+        }
+
+        if (String(vcodeInfo.vcode).toLowerCase() !== String(body.vcode).toLowerCase()) {
+            return this.fail(-1, "验证码不正确");
+        }
+
+        let user = new YtUserEntity();
+        user.nickName = `新用户-${vcodeInfo.vcode}`;
+        user.email = `${vcodeInfo.vcode}@example.com`
+        user.avatar = "https://ytfx.yndaopu.com/static/images/Avatar-1.png";
+        user = await this.ytUserService.add(user)
+        // 3. 生成JWT token
+        const token = jwt.sign(
+            { userId: user.id, type: user.userType },
+            process.env.jwt_secret ?? "123456",
+            { expiresIn: "360d" }
+        );
+
+        return this.success(token, "登录成功");
+    }
+
+    @ControllerCache((query: GetYtUserBasicInfoDto) => `YtUserAuthController-getYtUserBasicInfo-${query.id}`, {
+        ttl: {
+            max: 60 * 5,
+            min: 30,
+        },
+        enable: !isDebug()
+    })
+    @ResponseValidateIf(GetYtUserBasicInfoResponseDto, (data) => data && data.data)
+    @Get()
+    async getYtUserBasicInfo(
+        @Query(GetYtUserBasicInfoDto) query: GetYtUserBasicInfoDto,
+    ): Promise<ControllerResponse<GetYtUserBasicInfoResponseDto | null>> {
+        let userResult = await this.ytUserService.getById(YtUserEntity, query.id);
+
+        if (!userResult) {
+            return this.fail(-1, "用户不存在");
+        }
+        return this.success({
+            id: userResult.id,
+            name: userResult.nickName,
+            avatar: userResult.avatar,
+        }, "获取成功");
+    }
+
+
+
+    /**
+     * 创建验证码
+     * @returns 
+     */
+    @Get()
+    async getCaptcha() {
+        const captcha = svgCaptcha.create({
+            size: 4,
+            ignoreChars: '0o1i',
+            noise: 2,
+            color: true
+        });
+        const token = jwt.sign(
+            { vcode: captcha.text, expire: new Date().getTime() + 1000 * 60 * 5 },
+            process.env.jwt_secret ?? "123456"
+        );
+        return {
+            svg: captcha.data,
+            token
+        };
+    }
+
+
+}