package web

import (
	"context"
	"embed"
	"errors"
	"fmt"
	"net/http"
	"strings"
	"time"

	"github.com/LazyBachelor/LazyPM/internal/app"
	"github.com/LazyBachelor/LazyPM/internal/models"
	"github.com/LazyBachelor/LazyPM/internal/utils/browser"
	"github.com/LazyBachelor/LazyPM/pkg/web/handler"
	"github.com/LazyBachelor/LazyPM/pkg/web/server"
)

type Config = models.Config
type ValidationFeedback = models.ValidationFeedback

type Web struct {
	feedbackChan chan ValidationFeedback
	quitChan     chan bool
	submitChan   chan<- struct{}
}

func New() *Web {
	return &Web{}
}

//go:embed assets/*
var assets embed.FS

func (w *Web) Run(ctx context.Context, config Config) error {
	app, cleanup, err := app.New(ctx, config)
	if err != nil {
		return err
	}

	defer cleanup()

	httpServer := server.NewServer(server.Server{
		Address: config.WebAddress,
		Assets:  assets,
		App:     app,
	})

	fmt.Printf("Starting web server on %s...\n", config.WebAddress)

	address := config.WebAddress
	if !strings.Contains(address, "http") {
		address = "http://localhost" + address
	}

	err = browser.Open(address)
	if err != nil {
		return fmt.Errorf("failed to open browser: %w", err)
	}

	serverErr := make(chan error, 1)
	go func() {
		if err := httpServer.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
			serverErr <- err
		}
	}()

	if w.feedbackChan != nil {
		go func() {
			for feedback := range w.feedbackChan {
				handler.SetTaskFeedback(feedback)
			}
		}()
	}

	if w.submitChan != nil {
		handler.SetSubmitChan(w.submitChan)
	}

	select {
	case <-w.quitChan:
		fmt.Println("Task completed! Shutting down server...")
		shutdownCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
		defer cancel()
		return httpServer.Shutdown(shutdownCtx)
	case err := <-serverErr:
		return err
	case <-ctx.Done():
		return httpServer.Shutdown(context.Background())
	}
}

func (w *Web) SetChannels(feedbackChan chan ValidationFeedback, quitChan chan bool) {
	w.feedbackChan = feedbackChan
	w.quitChan = quitChan
}

func (w *Web) SetSubmitChan(submitChan chan<- struct{}) {
	w.submitChan = submitChan
}