datastar/site/routes_examples_signals_cha...

62 lines
1.8 KiB
Go

package site
import (
"net/http"
"github.com/go-chi/chi/v5"
"github.com/gorilla/sessions"
datastar "github.com/starfederation/datastar/sdk/go"
)
func setupExamplesSignalsChanged(examplesRouter chi.Router, signals sessions.Store) error {
sessionKey := "datastar-on-signals-changed-example"
currentTotalUpdates := func(w http.ResponseWriter, r *http.Request) (*sessions.Session, int, error) {
session, err := signals.Get(r, sessionKey)
if err != nil || len(session.Values) == 0 {
session.Values["updated"] = 0
if err := session.Save(r, w); err != nil {
return nil, 0, err
}
}
total := session.Values["updated"].(int)
return session, total, nil
}
saveAndRenderTotalUpdates := func(w http.ResponseWriter, r *http.Request, session *sessions.Session, totalUpdates int) {
session.Values["updated"] = totalUpdates
if err := session.Save(r, w); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
}
datastar.NewSSE(w, r).MergeFragmentf(
`<div id="from_server">Lifetime server updates %d</div>`,
totalUpdates,
)
}
examplesRouter.Route("/signals_changed/updates", func(signalsChangedUpdatesRouter chi.Router) {
signalsChangedUpdatesRouter.Post("/", func(w http.ResponseWriter, r *http.Request) {
session, totalUpdates, err := currentTotalUpdates(w, r)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
totalUpdates++
saveAndRenderTotalUpdates(w, r, session, totalUpdates)
})
signalsChangedUpdatesRouter.Delete("/", func(w http.ResponseWriter, r *http.Request) {
session, _, err := currentTotalUpdates(w, r)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
saveAndRenderTotalUpdates(w, r, session, 0)
})
})
return nil
}