package home

import (
	"context"
	"fmt"
	"html/template"
	"net/http"

	"github.com/FAU-CDI/wisski-distillery/internal/dis/component"
	"github.com/FAU-CDI/wisski-distillery/internal/dis/component/control/static/custom"
	"github.com/FAU-CDI/wisski-distillery/internal/dis/component/instances"
	"github.com/FAU-CDI/wisski-distillery/pkg/lazy"
)

type Home struct {
	component.Base
	Dependencies struct {
		Instances *instances.Instances
		Custom    *custom.Custom
	}

	redirect      lazy.Lazy[*Redirect]
	instanceNames lazy.Lazy[map[string]struct{}]
	homeBytes     lazy.Lazy[[]byte]
	homeTemplate  lazy.Lazy[*template.Template]
}

var (
	_ component.Routeable = (*Home)(nil)
)

func (*Home) Routes() component.Routes {
	return component.Routes{
		Paths: []string{"/"},
		CSRF:  false,
	}
}

func (home *Home) HandleRoute(ctx context.Context, route string) (http.Handler, error) {
	return home, nil
}

func (home *Home) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	slug, ok := home.Config.SlugFromHost(r.Host)
	switch {
	case !ok:
		http.NotFound(w, r)
	case slug != "":
		home.serveWissKI(w, slug, r)
	default:
		home.serveRoot(w, r)
	}
}

func (home *Home) serveRoot(w http.ResponseWriter, r *http.Request) {
	// not the root url => server the fallback
	if !(r.URL.Path == "" || r.URL.Path == "/") {
		home.redirect.Get(nil).ServeHTTP(w, r)
		return
	}

	w.Header().Set("Content-Type", "text/html")
	w.WriteHeader(http.StatusAccepted)
	w.Write(home.homeBytes.Get(nil))
}

func (home *Home) serveWissKI(w http.ResponseWriter, slug string, r *http.Request) {
	if _, ok := home.instanceNames.Get(nil)[slug]; !ok {
		// Get(nil) guaranteed to work by precondition
		w.WriteHeader(http.StatusNotFound)
		fmt.Fprintf(w, "WissKI %q not found\n", slug)
		return
	}

	w.WriteHeader(http.StatusBadGateway)
	fmt.Fprintf(w, "WissKI %q is currently offline\n", slug)
}