diff --git a/src/ai-coach/ai-coach.controller.ts b/src/ai-coach/ai-coach.controller.ts index 00ebd1d..a4c8427 100644 --- a/src/ai-coach/ai-coach.controller.ts +++ b/src/ai-coach/ai-coach.controller.ts @@ -1,4 +1,4 @@ -import { Body, Controller, Delete, Get, Param, Post, Query, Res, StreamableFile, UseGuards } from '@nestjs/common'; +import { Body, Controller, Delete, Get, HttpException, HttpStatus, Logger, Param, Post, Query, Res, StreamableFile, UseGuards } from '@nestjs/common'; import { ApiTags, ApiOperation, ApiBody, ApiQuery, ApiParam } from '@nestjs/swagger'; import { Response } from 'express'; import { JwtAuthGuard } from '../common/guards/jwt-auth.guard'; @@ -7,12 +7,17 @@ import { AccessTokenPayload } from '../users/services/apple-auth.service'; import { AiCoachService } from './ai-coach.service'; import { AiChatRequestDto, AiChatResponseDto, AiResponseDataDto } from './dto/ai-chat.dto'; import { PostureAssessmentRequestDto, PostureAssessmentResponseDto } from './dto/posture-assessment.dto'; +import { UsersService } from '../users/users.service'; @ApiTags('ai-coach') @Controller('ai-coach') @UseGuards(JwtAuthGuard) export class AiCoachController { - constructor(private readonly aiCoachService: AiCoachService) { } + private readonly logger = new Logger(AiCoachController.name); + constructor( + private readonly aiCoachService: AiCoachService, + private readonly usersService: UsersService, + ) { } @Post('chat') @ApiOperation({ summary: '流式大模型对话（普拉提教练）' }) @@ -26,6 +31,13 @@ export class AiCoachController { const stream = body.stream !== false; // 默认流式 const userContent = body.messages?.[body.messages.length - 1]?.content || ''; + // 判断用户是否有聊天次数 + const usageCount = await this.usersService.getUserUsageCount(userId); + if (usageCount <= 0) { + this.logger.error(`chat: ${userId} has no usage count`); + throw new HttpException('用户没有聊天次数', HttpStatus.FORBIDDEN); + } + // 创建或沿用会话ID，并保存用户消息 const { conversationId } = await this.aiCoachService.createOrAppendMessages({ userId, @@ -45,6 +57,8 @@ export class AiCoachController { confirmationData: body.confirmationData, }); + await this.usersService.deductUserUsageCount(userId); + // 检查是否返回结构化数据（如确认选项） // 结构化数据必须使用非流式模式返回 if (typeof result === 'object' && 'type' in result) { diff --git a/src/users/users.controller.ts b/src/users/users.controller.ts index 662e187..dbb8b04 100644 --- a/src/users/users.controller.ts +++ b/src/users/users.controller.ts @@ -317,99 +317,4 @@ export class UsersController { } } - /** - * 获取营养汇总分析 - */ - @UseGuards(JwtAuthGuard) - @Get('nutrition-summary') - @HttpCode(HttpStatus.OK) - @ApiOperation({ summary: '获取最近饮食的营养汇总分析' }) - @ApiQuery({ name: 'mealCount', required: false, description: '分析最近几顿饮食，默认10顿' }) - @ApiResponse({ status: 200, description: '成功获取营养汇总', type: DietAnalysisResponseDto }) - async getNutritionSummary( - @Query('mealCount') mealCount: string = '10', - @CurrentUser() user: AccessTokenPayload, - ): Promise { - this.logger.log(`获取营养汇总 - 用户ID: ${user.sub}, 分析${mealCount}顿饮食`); - - const count = Math.min(20, Math.max(1, parseInt(mealCount) || 10)); - const nutritionSummary = await this.usersService.getRecentNutritionSummary(user.sub, count); - - // 获取最近的饮食记录用于分析 - const recentRecords = await this.usersService.getDietHistory(user.sub, { limit: count }); - - // 简单的营养评分算法（可以后续优化） - const nutritionScore = this.calculateNutritionScore(nutritionSummary); - - // 生成基础建议（后续可以接入AI分析） - const recommendations = this.generateBasicRecommendations(nutritionSummary); - - return { - nutritionSummary, - recentRecords: recentRecords.records, - healthAnalysis: '基于您最近的饮食记录，我将为您提供个性化的营养分析和健康建议。', - nutritionScore, - recommendations, - }; - } - - /** - * 简单的营养评分算法 - */ - private calculateNutritionScore(summary: any): number { - let score = 50; // 基础分数 - - // 基于热量是否合理调整分数 - const dailyCalories = summary.totalCalories / (summary.recordCount / 3); // 假设一天3餐 - if (dailyCalories >= 1500 && dailyCalories <= 2500) score += 20; - else if (dailyCalories < 1200 || dailyCalories > 3000) score -= 20; - - // 基于蛋白质摄入调整分数 - const dailyProtein = summary.totalProtein / (summary.recordCount / 3); - if (dailyProtein >= 50 && dailyProtein <= 150) score += 15; - else if (dailyProtein < 30) score -= 15; - - // 基于膳食纤维调整分数 - const dailyFiber = summary.totalFiber / (summary.recordCount / 3); - if (dailyFiber >= 25) score += 15; - else if (dailyFiber < 10) score -= 10; - - return Math.max(0, Math.min(100, score)); - } - - /** - * 生成基础营养建议 - */ - private generateBasicRecommendations(summary: any): string[] { - const recommendations: string[] = []; - - const dailyCalories = summary.totalCalories / (summary.recordCount / 3); - const dailyProtein = summary.totalProtein / (summary.recordCount / 3); - const dailyFiber = summary.totalFiber / (summary.recordCount / 3); - const dailySodium = summary.totalSodium / (summary.recordCount / 3); - - if (dailyCalories < 1200) { - recommendations.push('您的日均热量摄入偏低，建议适当增加营养密度高的食物。'); - } else if (dailyCalories > 2500) { - recommendations.push('您的日均热量摄入偏高，建议控制portion size或选择低热量食物。'); - } - - if (dailyProtein < 50) { - recommendations.push('建议增加优质蛋白质摄入，如鸡胸肉、鱼类、豆制品等。'); - } - - if (dailyFiber < 25) { - recommendations.push('建议增加膳食纤维摄入，多吃蔬菜、水果和全谷物。'); - } - - if (dailySodium > 2000) { - recommendations.push('钠摄入偏高，建议减少加工食品和调味料的使用。'); - } - - if (recommendations.length === 0) { - recommendations.push('您的饮食结构相对均衡，继续保持良好的饮食习惯！'); - } - - return recommendations; - } } \ No newline at end of file diff --git a/src/users/users.service.ts b/src/users/users.service.ts index 68b4f00..76ef3fe 100644 --- a/src/users/users.service.ts +++ b/src/users/users.service.ts @@ -36,7 +36,7 @@ import { ActivityLogsService } from '../activity-logs/activity-logs.service'; import { ActivityActionType, ActivityEntityType } from '../activity-logs/models/activity-log.model'; import { CreateDietRecordDto, UpdateDietRecordDto, GetDietHistoryQueryDto, DietRecordResponseDto, DietHistoryResponseDto, NutritionSummaryDto } from './dto/diet-record.dto'; -const DEFAULT_FREE_USAGE_COUNT = 10; +const DEFAULT_FREE_USAGE_COUNT = 5; @Injectable() export class UsersService { @@ -126,6 +126,49 @@ export class UsersService { } } + /** + * @desc 获取用户剩余的聊天次数 + */ + async getUserUsageCount(userId: string): Promise { + try { + const user = await this.userModel.findOne({ where: { id: userId } }); + + if (!user) { + this.logger.log(`getUserUsageCount: ${userId} not found, return 0`); + return 0 + } + + if (user.isVip) { + // 会员用户无限次 + this.logger.log(`getUserUsageCount: ${userId} is vip, return 999`); + return 999 + } + + this.logger.log(`getUserUsageCount: ${userId} freeUsageCount: ${user.freeUsageCount}`); + + return user.freeUsageCount || 0; + } catch (error) { + this.logger.error(`getUserUsageCount error: ${error instanceof Error ? error.message : String(error)}`); + return 0 + } + } + + // 扣减用户免费次数 + async deductUserUsageCount(userId: string, count: number = 1): Promise { + try { + this.logger.log(`deductUserUsageCount: ${userId} deduct ${count} times`); + const user = await this.userModel.findOne({ where: { id: userId } }); + if (!user) { + throw new NotFoundException(`ID为${userId}的用户不存在`); + } + user.freeUsageCount -= count; + await user.save(); + } catch (error) { + this.logger.error(`deductUserUsageCount error: ${error instanceof Error ? error.message : String(error)}`); + throw error; + } + } + // 更新用户昵称、头像 async updateUser(updateUserDto: UpdateUserDto): Promise { @@ -453,8 +496,8 @@ export class UsersService { } /** - * 获取最近N顿饮食的营养汇总 - */ + * 获取最近N顿饮食的营养汇总 + */ async getRecentNutritionSummary(userId: string, mealCount: number = 10): Promise { const records = await this.userDietHistoryModel.findAll({ where: { userId, deleted: false },