package plugins

import (
	"io"
	"strings"
	"time"
	"ymgb/database"
	"ymgb/openai"
	"ymgb/utils/ai"

	"git.nix13.pw/scuroneko/laniakea"
)

func RegisterAi() *laniakea.Plugin[database.Context] {
	p := laniakea.NewPlugin[database.Context]("AI")
	p.AddCommand(p.NewCommand(gptTest, "gpt").SkipCommandAutoGen())
	//p.AddCommand(p.NewCommand(gptTest, "gpt2").SkipCommandAutoGen())
	return p
}

func gptTest(ctx *laniakea.MsgContext, _ *database.Context) {
	q := strings.Join(ctx.Args, " ")
	m := ctx.Answer("Генерация запущена")
	api := openai.NewOpenAIAPI(ai.GPTBaseUrl, "", "gpt-5.2")
	resp, err := api.CreateCompletionStream([]openai.Message{}, q, 1.0)
	if err != nil {
		m.Delete()
		ctx.Error(err)
		return
	}

	draft := ctx.NewDraft()
	for r, err := range resp {
		if err == io.EOF {
			break
		}
		if m != nil {
			m.Delete()
			m = nil
		}
		if err != nil {
			ctx.Error(err)
			return
		}

		if len(r.Choices) == 0 {
			continue
		}
		content := r.Choices[0].Delta.Content
		if content == "" {
			continue
		}
		err = draft.Push(content)
		if err != nil {
			ctx.Error(err)
			//draft.Flush()
			break
		}
		time.Sleep(time.Millisecond * 250)
	}
	err = draft.Flush()
	if err != nil {
		ctx.Error(err)
	}
}

//func gpt(ctx *laniakea.MsgContext, db *database.Context) {
//	q := strings.Join(ctx.Args, " ")
//	api := openai.NewOpenAIAPI(ai.GPTBaseUrl, "", "anthropic/claude-sonnet-4")
//	defer api.Close()
//
//	aiRedisRep := red.NewAiRepository(db)
//	chatId, err := aiRedisRep.GetOrCreateChatId(ctx.FromID)
//	if err != nil {
//		ctx.Error(err)
//		return
//	}
//	history, err := mdb.GetGptChatHistory(db, chatId)
//	if err != nil {
//		ctx.Error(err)
//		return
//	}
//	aiHistory := make([]openai.Message, len(history))
//	for _, m := range history {
//		aiHistory = append(aiHistory, openai.Message{
//			Role:    m.Role,
//			Content: m.Message,
//		})
//	}
//
//	m := ctx.Answer("Генерация запущена...")
//	res, err := api.CreateCompletion(aiHistory, q, 1.0)
//	if err != nil {
//		ctx.Error(err)
//		return
//	}
//	answer := res.Choices[0].Message.Content
//	m.Delete()
//	err = mdb.UpdateGptChatHistory(db, chatId, "user", q)
//	if err != nil {
//		ctx.Error(err)
//	}
//	err = mdb.UpdateGptChatHistory(db, chatId, "assistant", answer)
//	if err != nil {
//		ctx.Error(err)
//	}
//
//	ctx.Answer(answer)
//}