ai_english/serve/internal/database/seed.go

package database

import (
	"log"
	"time"

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

// stringPtr 返回字符串指针
func stringPtr(s string) *string {
	return &s
}

// SeedData 初始化种子数据
func SeedData(db *gorm.DB) error {
	log.Println("开始初始化种子数据...")

	// 检查是否已有数据
	var userCount int64
	db.Model(&models.User{}).Count(&userCount)
	if userCount > 0 {
		log.Println("数据库已有数据，跳过种子数据初始化")
		return nil
	}

	// 创建词汇分类
	if err := createVocabularyCategories(db); err != nil {
		return err
	}

	// 创建示例词汇
	if err := createSampleVocabularies(db); err != nil {
		return err
	}

	// 创建测试用户
	if err := createTestUsers(db); err != nil {
		return err
	}

	// 创建听力材料
	if err := createListeningMaterials(db); err != nil {
		return err
	}

	// 创建阅读材料
	if err := createReadingMaterials(db); err != nil {
		return err
	}

	// 创建写作提示
	if err := createWritingPrompts(db); err != nil {
		return err
	}

	// 创建口语场景
	if err := createSpeakingScenarios(db); err != nil {
		return err
	}

	log.Println("种子数据初始化完成")
	return nil
}

// createVocabularyCategories 创建词汇分类
func createVocabularyCategories(db *gorm.DB) error {
	categories := []models.VocabularyCategory{
		{
			ID:          utils.GenerateUUID(),
			Name:        "日常生活",
			Description: stringPtr("日常生活中常用的词汇"),
			Level:       "beginner",
			IsActive:    true,
			CreatedAt:   time.Now(),
			UpdatedAt:   time.Now(),
		},
		{
			ID:          utils.GenerateUUID(),
			Name:        "商务英语",
			Description: stringPtr("商务场景中使用的专业词汇"),
			Level:       "intermediate",
			IsActive:    true,
			CreatedAt:   time.Now(),
			UpdatedAt:   time.Now(),
		},
		{
			ID:          utils.GenerateUUID(),
			Name:        "学术英语",
			Description: stringPtr("学术研究和论文写作中的词汇"),
			Level:       "advanced",
			IsActive:    true,
			CreatedAt:   time.Now(),
			UpdatedAt:   time.Now(),
		},
		{
			ID:          utils.GenerateUUID(),
			Name:        "旅游英语",
			Description: stringPtr("旅游出行相关的实用词汇"),
			Level:       "beginner",
			IsActive:    true,
			CreatedAt:   time.Now(),
			UpdatedAt:   time.Now(),
		},
		{
			ID:          utils.GenerateUUID(),
			Name:        "科技英语",
			Description: stringPtr("科技和互联网相关词汇"),
			Level:       "intermediate",
			IsActive:    true,
			CreatedAt:   time.Now(),
			UpdatedAt:   time.Now(),
		},
	}

	for _, category := range categories {
		if err := db.Create(&category).Error; err != nil {
			return err
		}
	}

	log.Println("词汇分类创建完成")
	return nil
}

// createSampleVocabularies 创建示例词汇
func createSampleVocabularies(db *gorm.DB) error {
	// 获取第一个分类ID
	var category models.VocabularyCategory
	if err := db.First(&category).Error; err != nil {
		return err
	}

	vocabularies := []struct {
		Word        string
		Phonetic    string
		Level       string
		Frequency   int
		Definitions []models.VocabularyDefinition
		Examples    []models.VocabularyExample
	}{
		{
			Word:      "hello",
			Phonetic:  "/həˈloʊ/",
			Level:     "beginner",
			Frequency: 100,
			Definitions: []models.VocabularyDefinition{
				{
					PartOfSpeech: "interjection",
					Definition:   "used as a greeting or to begin a phone conversation",
					Translation:  "你好",
					SortOrder:    0,
					CreatedAt:    time.Now(),
				},
			},
			Examples: []models.VocabularyExample{
				{
					Example:     "Hello, how are you?",
					Translation: "你好，你好吗？",
					SortOrder:   0,
					CreatedAt:   time.Now(),
				},
			},
		},
		{
			Word:      "world",
			Phonetic:  "/wɜːrld/",
			Level:     "beginner",
			Frequency: 95,
			Definitions: []models.VocabularyDefinition{
				{
					PartOfSpeech: "noun",
					Definition:   "the earth, together with all of its countries and peoples",
					Translation:  "世界",
					SortOrder:    0,
					CreatedAt:    time.Now(),
				},
			},
			Examples: []models.VocabularyExample{
				{
					Example:     "The world is a beautiful place.",
					Translation: "世界是一个美丽的地方。",
					SortOrder:   0,
					CreatedAt:   time.Now(),
				},
			},
		},
		{
			Word:      "learn",
			Phonetic:  "/lɜːrn/",
			Level:     "beginner",
			Frequency: 90,
			Definitions: []models.VocabularyDefinition{
				{
					PartOfSpeech: "verb",
					Definition:   "acquire knowledge of or skill in something",
					Translation:  "学习",
					SortOrder:    0,
					CreatedAt:    time.Now(),
				},
			},
			Examples: []models.VocabularyExample{
				{
					Example:     "I want to learn English.",
					Translation: "我想学习英语。",
					SortOrder:   0,
					CreatedAt:   time.Now(),
				},
			},
		},
		{
			Word:      "study",
			Phonetic:  "/ˈstʌdi/",
			Level:     "beginner",
			Frequency: 85,
			Definitions: []models.VocabularyDefinition{
				{
					PartOfSpeech: "verb",
					Definition:   "devote time and attention to acquiring knowledge",
					Translation:  "学习，研究",
					SortOrder:    0,
					CreatedAt:    time.Now(),
				},
			},
			Examples: []models.VocabularyExample{
				{
					Example:     "She studies hard every day.",
					Translation: "她每天都努力学习。",
					SortOrder:   0,
					CreatedAt:   time.Now(),
				},
			},
		},
		{
			Word:      "practice",
			Phonetic:  "/ˈpræktɪs/",
			Level:     "intermediate",
			Frequency: 80,
			Definitions: []models.VocabularyDefinition{
				{
					PartOfSpeech: "verb",
					Definition:   "perform an activity repeatedly to improve one's skill",
					Translation:  "练习",
					SortOrder:    0,
					CreatedAt:    time.Now(),
				},
			},
			Examples: []models.VocabularyExample{
				{
					Example:     "Practice makes perfect.",
					Translation: "熟能生巧。",
					SortOrder:   0,
					CreatedAt:   time.Now(),
				},
			},
		},
	}

	for _, vocabData := range vocabularies {
		// 检查词汇是否已存在
		var existingVocab models.Vocabulary
		if err := db.Where("word = ?", vocabData.Word).First(&existingVocab).Error; err == nil {
			// 词汇已存在，跳过
			log.Printf("词汇 '%s' 已存在，跳过创建", vocabData.Word)
			continue
		}

		// 创建词汇
		vocab := models.Vocabulary{
			Word:      vocabData.Word,
			Phonetic:  &vocabData.Phonetic,
			Level:     vocabData.Level,
			Frequency: vocabData.Frequency,
			IsActive:  true,
			CreatedAt: time.Now(),
			UpdatedAt: time.Now(),
		}

		if err := db.Create(&vocab).Error; err != nil {
			return err
		}

		// 关联分类
		if err := db.Model(&vocab).Association("Categories").Append(&category); err != nil {
			return err
		}

		// 创建定义
		for _, def := range vocabData.Definitions {
			def.VocabularyID = vocab.ID
			if err := db.Create(&def).Error; err != nil {
				return err
			}
		}

		// 创建例句
		for _, example := range vocabData.Examples {
			example.VocabularyID = vocab.ID
			if err := db.Create(&example).Error; err != nil {
				return err
			}
		}
	}

	log.Println("示例词汇创建完成")
	return nil
}

// createTestUsers 创建测试用户
func createTestUsers(db *gorm.DB) error {
	users := []models.User{
		{
			ID:            0, // 让数据库自动生成
			Username:      "testuser",
			Email:         "test@example.com",
			PasswordHash:  "$2a$10$92IXUNpkjO0rOQ5byMi.Ye4oKoEa3Ro9llC/.og/at2.uheWG/igi", // password
			Nickname:      stringPtr("测试用户"),
			Avatar:        stringPtr("https://via.placeholder.com/150"),
			Gender:        stringPtr("other"),
			BirthDate:     nil,
			Location:      stringPtr("北京"),
			Bio:           stringPtr("这是一个测试用户"),
			EmailVerified: true,
			Status:        "active",
			CreatedAt:     time.Now(),
			UpdatedAt:     time.Now(),
		},
		{
			ID:            0, // 让数据库自动生成
			Username:      "demo",
			Email:         "demo@example.com",
			PasswordHash:  "$2a$10$92IXUNpkjO0rOQ5byMi.Ye4oKoEa3Ro9llC/.og/at2.uheWG/igi", // password
			Nickname:      stringPtr("演示用户"),
			Avatar:        stringPtr("https://via.placeholder.com/150"),
			Gender:        stringPtr("other"),
			BirthDate:     nil,
			Location:      stringPtr("上海"),
			Bio:           stringPtr("这是一个演示用户"),
			EmailVerified: true,
			Status:        "active",
			CreatedAt:     time.Now(),
			UpdatedAt:     time.Now(),
		},
	}

	for _, user := range users {
		if err := db.Create(&user).Error; err != nil {
			return err
		}

		// 创建用户偏好设置
        preference := models.UserPreference{
            ID:                   0, // 让数据库自动生成
            UserID:               user.ID,
            DailyGoal:            30,
            WeeklyGoal:           210,
            ReminderEnabled:      true,
            ReminderTime:         stringPtr("09:00:00"),
            DifficultyLevel:      "intermediate",
            LearningMode:         "casual",
            PreferredTopics:      stringPtr("[\"vocabulary\", \"listening\"]"),
            NotificationSettings: stringPtr("{\"email\": true, \"push\": true}"),
            PrivacySettings:      stringPtr("{\"profile_public\": false}"),
            CreatedAt:            time.Now(),
            UpdatedAt:            time.Now(),
        }
		if err := db.Create(&preference).Error; err != nil {
			return err
		}
	}

	log.Println("测试用户创建完成")
	return nil
}

// createListeningMaterials 创建听力材料
func createListeningMaterials(db *gorm.DB) error {
	materials := []models.ListeningMaterial{
		{
			ID:          utils.GenerateUUID(),
			Title:       "Daily Conversation",
			Description: stringPtr("Basic daily conversation practice"),
			Level:       "beginner",
			Duration:    180, // 3 minutes
			AudioURL:    "https://example.com/audio/daily-conversation.mp3",
			Transcript:  stringPtr("A: Hello, how are you today? B: I'm fine, thank you. How about you?"),
			IsActive:    true,
			CreatedAt:   time.Now(),
			UpdatedAt:   time.Now(),
		},
		{
			ID:          utils.GenerateUUID(),
			Title:       "Business Meeting",
			Description: stringPtr("Business meeting discussion"),
			Level:       "intermediate",
			Duration:    300, // 5 minutes
			AudioURL:    "https://example.com/audio/business-meeting.mp3",
			Transcript:  stringPtr("Let's discuss the quarterly report and our future plans."),
			IsActive:    true,
			CreatedAt:   time.Now(),
			UpdatedAt:   time.Now(),
		},
	}

	for _, material := range materials {
		if err := db.Create(&material).Error; err != nil {
			return err
		}
	}

	log.Println("听力材料创建完成")
	return nil
}

// intPtr 返回整数指针
func intPtr(i int) *int {
	return &i
}

// createReadingMaterials 创建阅读材料
func createReadingMaterials(db *gorm.DB) error {
	materials := []models.ReadingMaterial{
		{
			ID:          utils.GenerateUUID(),
			Title:       "The Benefits of Reading",
			Content:     "Reading is one of the most beneficial activities for the human mind. It improves vocabulary, enhances critical thinking, and provides entertainment.",
			Level:       "beginner",
			WordCount:   25,
			
			Summary:     stringPtr("这是一篇关于阅读益处的文章"),
			Source:      stringPtr("Education Weekly"),
			Author:      stringPtr("Reading Expert"),
			Tags:        stringPtr("[\"reading\", \"education\"]"),
			IsActive:    true,
			CreatedAt:   time.Now(),
			UpdatedAt:   time.Now(),
		},
		{
			ID:          utils.GenerateUUID(),
			Title:       "Climate Change and Technology",
			Content:     "Climate change represents one of the most pressing challenges of our time. Technology plays a crucial role in both contributing to and solving environmental problems.",
			Level:       "intermediate",
			WordCount:   30,
			
			Summary:     stringPtr("这是一篇关于气候变化与科技的文章"),
			Source:      stringPtr("Science Today"),
			Author:      stringPtr("Climate Researcher"),
			Tags:        stringPtr("[\"climate\", \"technology\"]"),
			IsActive:    true,
			CreatedAt:   time.Now(),
			UpdatedAt:   time.Now(),
		},
	}

	for _, material := range materials {
		if err := db.Create(&material).Error; err != nil {
			return err
		}
	}

	log.Println("阅读材料创建完成")
	return nil
}

// createWritingPrompts 创建写作提示
func createWritingPrompts(db *gorm.DB) error {
	prompts := []models.WritingPrompt{
		{
			ID:           utils.GenerateUUID(),
			Title:        "My Daily Routine",
			Prompt:       "Describe your daily routine from morning to evening. Include what you do, when you do it, and why.",
			Level:        "beginner",
			MinWords:     intPtr(100),
            MaxWords:     intPtr(200),
            TimeLimit:    intPtr(1800), // 30 minutes
			Instructions: stringPtr("Write a clear and descriptive essay"),
			Tags:         stringPtr("[\"daily\", \"routine\"]"),
			SampleAnswer: stringPtr("Every morning, I wake up at 7 AM and start my day..."),
			Rubric:       stringPtr("{\"grammar\": 25, \"vocabulary\": 25, \"coherence\": 25, \"content\": 25}"),
			IsActive:     true,
			CreatedAt:    time.Now(),
			UpdatedAt:    time.Now(),
		},
		{
			ID:           utils.GenerateUUID(),
			Title:        "The Impact of Social Media",
			Prompt:       "Discuss the positive and negative impacts of social media on modern society. Provide specific examples and your personal opinion.",
			Level:        "intermediate",
			MinWords:     intPtr(250),
            MaxWords:     intPtr(400),
            TimeLimit:    intPtr(2700), // 45 minutes
			Instructions: stringPtr("Write a balanced argumentative essay"),
			Tags:         stringPtr("[\"social media\", \"society\"]"),
			SampleAnswer: stringPtr("Social media has transformed how we communicate..."),
			Rubric:       stringPtr("{\"grammar\": 30, \"vocabulary\": 20, \"coherence\": 25, \"content\": 25}"),
			IsActive:     true,
			CreatedAt:    time.Now(),
			UpdatedAt:    time.Now(),
		},
	}

	for _, prompt := range prompts {
		if err := db.Create(&prompt).Error; err != nil {
			return err
		}
	}

	log.Println("写作提示创建完成")
	return nil
}

// createSpeakingScenarios 创建口语场景
func createSpeakingScenarios(db *gorm.DB) error {
	scenarios := []models.SpeakingScenario{
		{
			ID:          utils.GenerateUUID(),
			Title:       "Restaurant Ordering",
			Description: "You are at a restaurant and want to order food. The waiter will take your order.",
			Level:       "beginner",
			Context:     stringPtr("You are at a restaurant with friends"),
			Tags:        stringPtr("[\"restaurant\", \"food\"]"),
			Dialogue:    stringPtr("[{\"speaker\": \"waiter\", \"text\": \"Good evening, welcome to our restaurant!\"}]"),
			KeyPhrases:  stringPtr("[\"I'd like to order\", \"Could I have\", \"The bill, please\"]"),
			IsActive:    true,
			CreatedAt:   time.Now(),
			UpdatedAt:   time.Now(),
		},
		{
			ID:          utils.GenerateUUID(),
			Title:       "Job Interview",
			Description: "You are in a job interview for a position you really want. Answer the interviewer's questions confidently.",
			Level:       "intermediate",
			Context:     stringPtr("You are applying for a job"),
			Tags:        stringPtr("[\"interview\", \"job\"]"),
			Dialogue:    stringPtr("[{\"speaker\": \"interviewer\", \"text\": \"Tell me about yourself\"}]"),
			KeyPhrases:  stringPtr("[\"I have experience in\", \"My strengths are\", \"I'm interested in\"]"),
			IsActive:    true,
			CreatedAt:   time.Now(),
			UpdatedAt:   time.Now(),
		},
	}

	for _, scenario := range scenarios {
		if err := db.Create(&scenario).Error; err != nil {
			return err
		}
	}

	log.Println("口语场景创建完成")
	return nil
}