Reorganized directories and packages.

pkg/htmx/htmx.go

@@ -0,0 +1,82 @@
+package htmx
+
+import (
+	"net/http"
+
+	"github.com/labstack/echo/v4"
+)
+
+//  Headers (https://htmx.org/docs/#requests)
+const (
+	HeaderRequest            = "HX-Request"
+	HeaderBoosted            = "HX-Boosted"
+	HeaderTrigger            = "HX-Trigger"
+	HeaderTriggerName        = "HX-Trigger-Name"
+	HeaderTriggerAfterSwap   = "HX-Trigger-After-Swap"
+	HeaderTriggerAfterSettle = "HX-Trigger-After-Settle"
+	HeaderTarget             = "HX-Target"
+	HeaderPrompt             = "HX-Prompt"
+	HeaderPush               = "HX-Push"
+	HeaderRedirect           = "HX-Redirect"
+	HeaderRefresh            = "HX-Refresh"
+)
+
+type (
+	// Request contains data that HTMX provides during requests
+	Request struct {
+		Enabled     bool
+		Boosted     bool
+		Trigger     string
+		TriggerName string
+		Target      string
+		Prompt      string
+	}
+
+	// Response contain data that the server can communicate back to HTMX
+	Response struct {
+		Push               string
+		Redirect           string
+		Refresh            bool
+		Trigger            string
+		TriggerAfterSwap   string
+		TriggerAfterSettle string
+		NoContent          bool
+	}
+)
+
+// GetRequest extracts HTMX data from the request
+func GetRequest(ctx echo.Context) Request {
+	return Request{
+		Enabled:     ctx.Request().Header.Get(HeaderRequest) == "true",
+		Boosted:     ctx.Request().Header.Get(HeaderBoosted) == "true",
+		Trigger:     ctx.Request().Header.Get(HeaderTrigger),
+		TriggerName: ctx.Request().Header.Get(HeaderTriggerName),
+		Target:      ctx.Request().Header.Get(HeaderTarget),
+		Prompt:      ctx.Request().Header.Get(HeaderPrompt),
+	}
+}
+
+// Apply applies data from a Response to a server response
+func (r Response) Apply(ctx echo.Context) {
+	if r.Push != "" {
+		ctx.Response().Header().Set(HeaderPush, r.Push)
+	}
+	if r.Redirect != "" {
+		ctx.Response().Header().Set(HeaderRedirect, r.Redirect)
+	}
+	if r.Refresh {
+		ctx.Response().Header().Set(HeaderRefresh, "true")
+	}
+	if r.Trigger != "" {
+		ctx.Response().Header().Set(HeaderTrigger, r.Trigger)
+	}
+	if r.TriggerAfterSwap != "" {
+		ctx.Response().Header().Set(HeaderTriggerAfterSwap, r.TriggerAfterSwap)
+	}
+	if r.TriggerAfterSettle != "" {
+		ctx.Response().Header().Set(HeaderTriggerAfterSettle, r.TriggerAfterSettle)
+	}
+	if r.NoContent {
+		ctx.Response().Status = http.StatusNoContent
+	}
+}

pkg/htmx/htmx_test.go

@@ -0,0 +1,52 @@
+package htmx
+
+import (
+	"net/http"
+	"testing"
+
+	"github.com/mikestefanello/pagoda/pkg/tests"
+
+	"github.com/stretchr/testify/assert"
+
+	"github.com/labstack/echo/v4"
+)
+
+func TestSetRequest(t *testing.T) {
+	ctx, _ := tests.NewContext(echo.New(), "/")
+	ctx.Request().Header.Set(HeaderRequest, "true")
+	ctx.Request().Header.Set(HeaderBoosted, "true")
+	ctx.Request().Header.Set(HeaderTrigger, "a")
+	ctx.Request().Header.Set(HeaderTriggerName, "b")
+	ctx.Request().Header.Set(HeaderTarget, "c")
+	ctx.Request().Header.Set(HeaderPrompt, "d")
+
+	r := GetRequest(ctx)
+	assert.Equal(t, true, r.Enabled)
+	assert.Equal(t, true, r.Boosted)
+	assert.Equal(t, "a", r.Trigger)
+	assert.Equal(t, "b", r.TriggerName)
+	assert.Equal(t, "c", r.Target)
+	assert.Equal(t, "d", r.Prompt)
+}
+
+func TestResponse_Apply(t *testing.T) {
+	ctx, _ := tests.NewContext(echo.New(), "/")
+	r := Response{
+		Push:               "a",
+		Redirect:           "b",
+		Refresh:            true,
+		Trigger:            "c",
+		TriggerAfterSwap:   "d",
+		TriggerAfterSettle: "e",
+		NoContent:          true,
+	}
+	r.Apply(ctx)
+
+	assert.Equal(t, "a", ctx.Response().Header().Get(HeaderPush))
+	assert.Equal(t, "b", ctx.Response().Header().Get(HeaderRedirect))
+	assert.Equal(t, "true", ctx.Response().Header().Get(HeaderRefresh))
+	assert.Equal(t, "c", ctx.Response().Header().Get(HeaderTrigger))
+	assert.Equal(t, "d", ctx.Response().Header().Get(HeaderTriggerAfterSwap))
+	assert.Equal(t, "e", ctx.Response().Header().Get(HeaderTriggerAfterSettle))
+	assert.Equal(t, http.StatusNoContent, ctx.Response().Status)
+}