ai_english/serve/internal/services/study_plan_service.go

package services

import (
	"errors"
	"time"

	"github.com/Nanqipro/YunQue-Tech-Projects/ai_english_learning/serve/internal/models"
	"gorm.io/gorm"
)

// StudyPlanService 学习计划服务
type StudyPlanService struct {
	db *gorm.DB
}

func NewStudyPlanService(db *gorm.DB) *StudyPlanService {
	return &StudyPlanService{db: db}
}

// CreateStudyPlan 创建学习计划
func (s *StudyPlanService) CreateStudyPlan(userID int64, planName string, description *string, dailyGoal int, 
	bookID *string, startDate time.Time, endDate *time.Time, remindTime *string, remindDays *string) (*models.StudyPlan, error) {
	
	// 验证参数
	if dailyGoal < 1 || dailyGoal > 200 {
		return nil, errors.New("每日目标必须在1-200之间")
	}
	
	// 如果指定了词汇书，获取总单词数
	totalWords := 0
	if bookID != nil && *bookID != "" {
		var count int64
		s.db.Model(&models.VocabularyBookWord{}).Where("book_id = ?", *bookID).Count(&count)
		totalWords = int(count)
	}
	
	plan := &models.StudyPlan{
		UserID:          userID,
		BookID:          bookID,
		PlanName:        planName,
		Description:     description,
		DailyGoal:       dailyGoal,
		TotalWords:      totalWords,
		LearnedWords:    0,
		StartDate:       startDate,
		EndDate:         endDate,
		Status:          "active",
		RemindTime:      remindTime,
		RemindDays:      remindDays,
		IsRemindEnabled: remindTime != nil && remindDays != nil,
		StreakDays:      0,
	}
	
	if err := s.db.Create(plan).Error; err != nil {
		return nil, err
	}
	
	return plan, nil
}

// GetUserStudyPlans 获取用户的学习计划列表
func (s *StudyPlanService) GetUserStudyPlans(userID int64, status string) ([]models.StudyPlan, error) {
	var plans []models.StudyPlan
	query := s.db.Where("user_id = ?", userID)
	
	if status != "" && status != "all" {
		query = query.Where("status = ?", status)
	}
	
	err := query.Order("created_at DESC").Find(&plans).Error
	return plans, err
}

// GetStudyPlanByID 获取学习计划详情
func (s *StudyPlanService) GetStudyPlanByID(planID, userID int64) (*models.StudyPlan, error) {
	var plan models.StudyPlan
	err := s.db.Where("id = ? AND user_id = ?", planID, userID).First(&plan).Error
	if err != nil {
		return nil, err
	}
	return &plan, nil
}

// UpdateStudyPlan 更新学习计划
func (s *StudyPlanService) UpdateStudyPlan(planID, userID int64, updates map[string]interface{}) (*models.StudyPlan, error) {
	var plan models.StudyPlan
	if err := s.db.Where("id = ? AND user_id = ?", planID, userID).First(&plan).Error; err != nil {
		return nil, err
	}
	
	// 验证dailyGoal
	if dailyGoal, ok := updates["daily_goal"].(int); ok {
		if dailyGoal < 1 || dailyGoal > 200 {
			return nil, errors.New("每日目标必须在1-200之间")
		}
	}
	
	if err := s.db.Model(&plan).Updates(updates).Error; err != nil {
		return nil, err
	}
	
	return &plan, nil
}

// DeleteStudyPlan 删除学习计划（软删除）
func (s *StudyPlanService) DeleteStudyPlan(planID, userID int64) error {
	result := s.db.Where("id = ? AND user_id = ?", planID, userID).Delete(&models.StudyPlan{})
	if result.Error != nil {
		return result.Error
	}
	if result.RowsAffected == 0 {
		return errors.New("计划不存在或无权删除")
	}
	return nil
}

// UpdatePlanStatus 更新计划状态
func (s *StudyPlanService) UpdatePlanStatus(planID, userID int64, status string) error {
	validStatuses := map[string]bool{
		"active":    true,
		"paused":    true,
		"completed": true,
		"cancelled": true,
	}
	
	if !validStatuses[status] {
		return errors.New("无效的状态")
	}
	
	updates := map[string]interface{}{
		"status": status,
	}
	
	if status == "completed" {
		now := time.Now()
		updates["completed_at"] = now
	}
	
	result := s.db.Model(&models.StudyPlan{}).
		Where("id = ? AND user_id = ?", planID, userID).
		Updates(updates)
	
	if result.Error != nil {
		return result.Error
	}
	if result.RowsAffected == 0 {
		return errors.New("计划不存在或无权修改")
	}
	
	return nil
}

// RecordStudyProgress 记录学习进度
func (s *StudyPlanService) RecordStudyProgress(planID, userID int64, wordsStudied, studyDuration int) error {
	var plan models.StudyPlan
	if err := s.db.Where("id = ? AND user_id = ?", planID, userID).First(&plan).Error; err != nil {
		return err
	}
	
	today := time.Now().Truncate(24 * time.Hour)
	
	// 检查今天是否已有记录
	var record models.StudyPlanRecord
	err := s.db.Where("plan_id = ? AND user_id = ? AND study_date = ?", planID, userID, today).First(&record).Error
	
	if err == gorm.ErrRecordNotFound {
		// 创建新记录
		record = models.StudyPlanRecord{
			PlanID:        planID,
			UserID:        userID,
			StudyDate:     today,
			WordsStudied:  wordsStudied,
			GoalCompleted: wordsStudied >= plan.DailyGoal,
			StudyDuration: studyDuration,
		}
		if err := s.db.Create(&record).Error; err != nil {
			return err
		}
	} else if err == nil {
		// 更新现有记录
		record.WordsStudied += wordsStudied
		record.StudyDuration += studyDuration
		record.GoalCompleted = record.WordsStudied >= plan.DailyGoal
		if err := s.db.Save(&record).Error; err != nil {
			return err
		}
	} else {
		return err
	}
	
	// 更新计划的已学单词数
	plan.LearnedWords += wordsStudied
	
	// 更新连续学习天数
	if plan.LastStudyDate != nil {
		lastDate := plan.LastStudyDate.Truncate(24 * time.Hour)
		yesterday := today.AddDate(0, 0, -1)
		
		if lastDate.Equal(yesterday) {
			plan.StreakDays++
		} else if !lastDate.Equal(today) {
			plan.StreakDays = 1
		}
	} else {
		plan.StreakDays = 1
	}
	
	plan.LastStudyDate = &today
	
	// 检查是否完成计划
	if plan.TotalWords > 0 && plan.LearnedWords >= plan.TotalWords {
		plan.Status = "completed"
		now := time.Now()
		plan.CompletedAt = &now
	}
	
	return s.db.Save(&plan).Error
}

// GetStudyPlanStatistics 获取学习计划统计
func (s *StudyPlanService) GetStudyPlanStatistics(planID, userID int64) (map[string]interface{}, error) {
	var plan models.StudyPlan
	if err := s.db.Where("id = ? AND user_id = ?", planID, userID).First(&plan).Error; err != nil {
		return nil, err
	}
	
	// 统计总学习天数
	var totalStudyDays int64
	s.db.Model(&models.StudyPlanRecord{}).
		Where("plan_id = ? AND user_id = ?", planID, userID).
		Count(&totalStudyDays)
	
	// 统计完成目标的天数
	var completedDays int64
	s.db.Model(&models.StudyPlanRecord{}).
		Where("plan_id = ? AND user_id = ? AND goal_completed = ?", planID, userID, true).
		Count(&completedDays)
	
	// 计算平均每日学习单词数
	var avgWords float64
	s.db.Model(&models.StudyPlanRecord{}).
		Select("AVG(words_studied) as avg_words").
		Where("plan_id = ? AND user_id = ?", planID, userID).
		Scan(&avgWords)
	
	// 计算完成率
	completionRate := 0.0
	if plan.TotalWords > 0 {
		completionRate = float64(plan.LearnedWords) / float64(plan.TotalWords) * 100
	}
	
	// 获取最近7天的学习记录
	var recentRecords []models.StudyPlanRecord
	s.db.Where("plan_id = ? AND user_id = ?", planID, userID).
		Order("study_date DESC").
		Limit(7).
		Find(&recentRecords)
	
	return map[string]interface{}{
		"plan":             plan,
		"total_study_days": totalStudyDays,
		"completed_days":   completedDays,
		"avg_words":        avgWords,
		"completion_rate":  completionRate,
		"recent_records":   recentRecords,
		"streak_days":      plan.StreakDays,
	}, nil
}

// GetTodayStudyPlans 获取今日需要执行的学习计划
func (s *StudyPlanService) GetTodayStudyPlans(userID int64) ([]map[string]interface{}, error) {
	var plans []models.StudyPlan
	today := time.Now().Truncate(24 * time.Hour)
	weekday := int(time.Now().Weekday())
	if weekday == 0 {
		weekday = 7 // 将周日从0改为7
	}
	
	// 查找活跃的计划
	err := s.db.Where("user_id = ? AND status = ? AND start_date <= ?", userID, "active", today).
		Find(&plans).Error
	
	if err != nil {
		return nil, err
	}
	
	result := make([]map[string]interface{}, 0)
	
	for _, plan := range plans {
		// 检查今天是否已完成
		var record models.StudyPlanRecord
		err := s.db.Where("plan_id = ? AND user_id = ? AND study_date = ?", plan.ID, userID, today).
			First(&record).Error
		
		todayCompleted := err == nil && record.GoalCompleted
		todayProgress := 0
		if err == nil {
			todayProgress = record.WordsStudied
		}
		
		// 检查是否需要提醒
		needRemind := false
		if plan.IsRemindEnabled && plan.RemindDays != nil {
			// 简单检查：这里可以根据RemindDays判断今天是否需要提醒
			needRemind = true
		}
		
		result = append(result, map[string]interface{}{
			"plan":            plan,
			"today_completed": todayCompleted,
			"today_progress":  todayProgress,
			"need_remind":     needRemind,
		})
	}
	
	return result, nil
}