feat(ui): add real-time updates #164 and refactor #126

glasskube · Feb 7, 2024 · b1e7709 · b1e7709
1 parent 56b8a84
commit b1e7709
Show file tree

Hide file tree

Showing 13 changed files with 905 additions and 196 deletions.
diff --git a/go.mod b/go.mod
@@ -9,6 +9,7 @@ require (
 	github.com/fluxcd/helm-controller/api v0.37.4
 	github.com/fluxcd/source-controller/api v1.2.4
 	github.com/go-logr/logr v1.4.1
+	github.com/gorilla/websocket v1.5.0
 	github.com/invopop/jsonschema v0.12.0
 	github.com/onsi/ginkgo/v2 v2.15.0
 	github.com/onsi/gomega v1.31.1

diff --git a/go.sum b/go.sum
@@ -63,6 +63,8 @@ github.com/google/pprof v0.0.0-20231229205709-960ae82b1e42 h1:dHLYa5D8/Ta0aLR2Xc
 github.com/google/pprof v0.0.0-20231229205709-960ae82b1e42/go.mod h1:czg5+yv1E0ZGTi6S6vVK1mke0fV+FaUhNGcd6VRS9Ik=
 github.com/google/uuid v1.5.0 h1:1p67kYwdtXjb0gL0BPiP1Av9wiZPo5A8z2cWkTZ+eyU=
 github.com/google/uuid v1.5.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/gorilla/websocket v1.5.0 h1:PPwGk2jz7EePpoHN/+ClbZu8SPxiqlu12wZP/3sWmnc=
+github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
 github.com/imdario/mergo v0.3.16 h1:wwQJbIsHYGMUyLSPrEq1CT16AhnhNJQ51+4fdHUnCl4=
 github.com/imdario/mergo v0.3.16/go.mod h1:WBLT9ZmE3lPoWsEzCh9LPo3TiwVN+ZKEjmz+hD27ysY=
 github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=

diff --git a/internal/web/components/installbutton.go b/internal/web/components/installbutton.go
@@ -0,0 +1,43 @@
+package components
+
+import (
+	"fmt"
+	"html/template"
+	"io"
+	"os"
+
+	"github.com/glasskube/glasskube/pkg/client"
+	"github.com/glasskube/glasskube/pkg/list"
+)
+
+func GetButtonId(pkgName string) string {
+	return fmt.Sprintf("install-%v", pkgName)
+}
+
+func GetSwap(buttonId string) string {
+	return fmt.Sprintf("outerHTML:#%s", buttonId)
+}
+
+func RenderInstallButton(w io.Writer, tmpl *template.Template, pkgName string, status *client.PackageStatus) {
+	buttonId := GetButtonId(pkgName)
+	err := tmpl.ExecuteTemplate(w, "installbutton", &map[string]any{
+		"ButtonId":    buttonId,
+		"Swap":        GetSwap(buttonId),
+		"PackageName": pkgName,
+		"Status":      status,
+	})
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "An error occurred rendering the install button for %v: \n%v\n"+
+			"This is most likely a BUG!", pkgName, err)
+	}
+}
+
+func ToInstallButtonInput(pkgTeaser list.PackageTeaserWithStatus) map[string]any {
+	buttonId := GetButtonId(pkgTeaser.PackageName)
+	return map[string]any{
+		"ButtonId":    buttonId,
+		"Swap":        GetSwap(buttonId),
+		"PackageName": pkgTeaser.PackageName,
+		"Status":      pkgTeaser.Status,
+	}
+}
diff --git a/internal/web/hub.go b/internal/web/hub.go
@@ -0,0 +1,45 @@
+package web
+
+// Hub maintains the set of active clients and broadcasts messages to the clients.
+type Hub struct {
+	// Inbound messages from the clients.
+	Broadcast chan []byte
+
+	// Register requests from the clients.
+	Register chan *WsClient
+
+	// Unregister requests from clients.
+	Unregister chan *WsClient
+
+	// Registered clients.
+	clients map[*WsClient]bool
+}
+
+// NewHub creates new Hub
+func NewHub() *Hub {
+	return &Hub{
+		Broadcast:  make(chan []byte),
+		Register:   make(chan *WsClient),
+		Unregister: make(chan *WsClient),
+		clients:    make(map[*WsClient]bool),
+	}
+}
+
+// Run handles communication operations with Hub
+func (h *Hub) Run() {
+	for {
+		select {
+		case client := <-h.Register:
+			h.clients[client] = true
+		case client := <-h.Unregister:
+			if _, ok := h.clients[client]; ok {
+				delete(h.clients, client)
+				close(client.send)
+			}
+		case message := <-h.Broadcast:
+			for client := range h.clients {
+				client.send <- message
+			}
+		}
+	}
+}
diff --git a/internal/web/root/static/js/htmx-1.9.10.min.js b/internal/web/root/static/js/htmx-1.9.10.min.js