package svc

import (
	"healthapi/internal/config"
	"healthapi/internal/database"
	"healthapi/pkg/ai"

	"github.com/zeromicro/go-zero/core/logx"
	"gorm.io/gorm"
)

type ServiceContext struct {
	Config   config.Config
	DB       *gorm.DB
	AIClient ai.AIClient
}

func NewServiceContext(c config.Config) *ServiceContext {
	// 初始化数据库
	db, err := database.NewDB(c.Database)
	if err != nil {
		logx.Errorf("Failed to connect database: %v", err)
		panic(err)
	}

	// 初始化数据（问卷题库、测试用户）
	if err := database.SeedQuestionBank(db); err != nil {
		logx.Errorf("Failed to seed question bank: %v", err)
	}
	if err := database.SeedTestUser(db); err != nil {
		logx.Errorf("Failed to seed test user: %v", err)
	}

	// 初始化 AI 客户端
	aiClient := ai.NewAIClient(c.AI)

	return &ServiceContext{
		Config:   c,
		DB:       db,
		AIClient: aiClient,
	}
}