package site

import (
	"net/http"

	"github.com/go-chi/chi/v5"
	"github.com/gorilla/sessions"
	datastar "github.com/starfederation/datastar/sdk/go"
)

func setupExamplesOnLoad(examplesRouter chi.Router, signals sessions.Store) error {

	sessionKey := "datastar-on-load-example"
	examplesRouter.Post("/on_load/data", func(w http.ResponseWriter, r *http.Request) {
		session, err := signals.Get(r, sessionKey)

		if err != nil || len(session.Values) == 0 {
			session.Values["foo"] = "bar"
			session.Values["baz"] = 42
			if err := session.Save(r, w); err != nil {
				http.Error(w, err.Error(), http.StatusInternalServerError)
			}
		}

		// You can comment out the below block and still persist the session
		sse := datastar.NewSSE(w, r)
		sse.MergeFragmentTempl(onLoadView(sessionKey, session.Values))
	})

	return nil
}