package ai

import (
	"context"

	"gitlab.guxuan.icu/jinshan_community/internal/config"

	"github.com/gin-gonic/gin"
	"gitlab.guxuan.icu/jinshan_community/internal/mods/ai/api"
	"gitlab.guxuan.icu/jinshan_community/internal/mods/ai/schema"
	"gorm.io/gorm"
)

type Ai struct {
	DB           *gorm.DB
	AiRequestAPI *api.AiRequest
}

func (a *Ai) AutoMigrate(ctx context.Context) error {
	return a.DB.AutoMigrate(new(schema.AiRequest))
}

func (a *Ai) Init(ctx context.Context) error {
	if config.C.Storage.DB.AutoMigrate {
		if err := a.AutoMigrate(ctx); err != nil {
			return err
		}
	}
	return nil
}

func (a *Ai) RegisterV1Routers(ctx context.Context, v1 *gin.RouterGroup) error {
	aiRequest := v1.Group("ai-requests")
	{
		aiRequest.GET("", a.AiRequestAPI.Query)
		aiRequest.GET(":id", a.AiRequestAPI.Get)
		aiRequest.POST("", a.AiRequestAPI.Create)
		aiRequest.PUT(":id", a.AiRequestAPI.Update)
		aiRequest.DELETE(":id", a.AiRequestAPI.Delete)
	}
	return nil
}

func (a *Ai) Release(ctx context.Context) error {
	return nil
}