feat: add frontend

Author: socoldkiller

Makefile

@@ -14,8 +14,7 @@ endif
 # PROTO_SOURCES = $(wildcard **/*.proto)
 GO_SOURCES = $(call rwildcard,,*.go)
 PROTO_SOURCES = $(call rwildcard,,*.proto)
-
-
+TARGET_DIR = headscale-ui
 build:
 	nix build
 
@@ -62,3 +61,12 @@ compress: build
 generate:
 	rm -rf gen
 	buf generate proto
+
+frontend:
+	@if not [ -d "$(TARGET_DIR)" ]; then \
+		git clone https://github.com/gurucomputing/headscale-ui.git "$(TARGET_DIR)"; \
+	fi
+	-rm -r hscontrol/build
+	bun install --cwd headscale-ui
+	bun run --cwd headscale-ui build
+	mv headscale-ui/build hscontrol

devbox.json

@@ -8,7 +8,9 @@
       "platforms": ["x86_64-darwin", "aarch64-darwin"]
     },
     "buf":           "1.48.0",
-    "protoc-gen-go": "1.35.2"
+    "protoc-gen-go": "1.35.2",
+    "bun":           "1.1.42",
+    "nodejs":        "18.8.0"
   },
   "shell": {
     "init_hook": [

devbox.lock

@@ -49,6 +49,54 @@
         }
       }
     },
+    "[email protected]": {
+      "last_modified": "2024-12-26T01:25:00Z",
+      "resolved": "github:NixOS/nixpkgs/16e046229f3b4f53257973a5532bcbb72457d2f2#bun",
+      "source": "devbox-search",
+      "version": "1.1.42",
+      "systems": {
+        "aarch64-darwin": {
+          "outputs": [
+            {
+              "name": "out",
+              "path": "/nix/store/5c73vw31nbji92hggrgjnfi33wzhbxk3-bun-1.1.42",
+              "default": true
+            }
+          ],
+          "store_path": "/nix/store/5c73vw31nbji92hggrgjnfi33wzhbxk3-bun-1.1.42"
+        },
+        "aarch64-linux": {
+          "outputs": [
+            {
+              "name": "out",
+              "path": "/nix/store/shm69afh0zrm5r0gkpkj9l0z8l6vkpmr-bun-1.1.42",
+              "default": true
+            }
+          ],
+          "store_path": "/nix/store/shm69afh0zrm5r0gkpkj9l0z8l6vkpmr-bun-1.1.42"
+        },
+        "x86_64-darwin": {
+          "outputs": [
+            {
+              "name": "out",
+              "path": "/nix/store/70nzsdr814rka1b3qaxz8b4hfk8r6wi9-bun-1.1.42",
+              "default": true
+            }
+          ],
+          "store_path": "/nix/store/70nzsdr814rka1b3qaxz8b4hfk8r6wi9-bun-1.1.42"
+        },
+        "x86_64-linux": {
+          "outputs": [
+            {
+              "name": "out",
+              "path": "/nix/store/ws04zldxbkg26d4qkwyr4km1zd26mach-bun-1.1.42",
+              "default": true
+            }
+          ],
+          "store_path": "/nix/store/ws04zldxbkg26d4qkwyr4km1zd26mach-bun-1.1.42"
+        }
+      }
+    },
     "darwin.apple_sdk.frameworks.IOKit": {
       "resolved": "github:NixOS/nixpkgs/75a52265bda7fd25e06e3a67dee3f0354e73243c#darwin.apple_sdk.frameworks.IOKit",
       "source": "nixpkg"
@@ -153,6 +201,71 @@
         }
       }
     },
+    "[email protected]": {
+      "last_modified": "2022-08-27T08:52:24Z",
+      "plugin_version": "0.0.2",
+      "resolved": "github:NixOS/nixpkgs/ed0fab06cc1ca9799e6dda30529c963b95c4dc2a#nodejs",
+      "source": "devbox-search",
+      "version": "18.8.0",
+      "systems": {
+        "aarch64-darwin": {
+          "outputs": [
+            {
+              "name": "out",
+              "path": "/nix/store/amvfx98dwsqjiclz570fkmc7pd9h27nj-nodejs-18.8.0",
+              "default": true
+            },
+            {
+              "name": "libv8",
+              "path": "/nix/store/xg73kcbyknfnz3wh9iz7pb8mcvmahyli-nodejs-18.8.0-libv8"
+            }
+          ],
+          "store_path": "/nix/store/amvfx98dwsqjiclz570fkmc7pd9h27nj-nodejs-18.8.0"
+        },
+        "aarch64-linux": {
+          "outputs": [
+            {
+              "name": "out",
+              "path": "/nix/store/771xwnjhlhyxq5nfh825b239q552fxnk-nodejs-18.8.0",
+              "default": true
+            },
+            {
+              "name": "libv8",
+              "path": "/nix/store/b6mwrm6y94pw4mkm8jn4q5p4fwkk69pa-nodejs-18.8.0-libv8"
+            }
+          ],
+          "store_path": "/nix/store/771xwnjhlhyxq5nfh825b239q552fxnk-nodejs-18.8.0"
+        },
+        "x86_64-darwin": {
+          "outputs": [
+            {
+              "name": "out",
+              "path": "/nix/store/0ws9319m3qpsafm0grigpmh60d2qkby9-nodejs-18.8.0",
+              "default": true
+            },
+            {
+              "name": "libv8",
+              "path": "/nix/store/02ddlysk5c29sg23gafzcmj62yjsiiq8-nodejs-18.8.0-libv8"
+            }
+          ],
+          "store_path": "/nix/store/0ws9319m3qpsafm0grigpmh60d2qkby9-nodejs-18.8.0"
+        },
+        "x86_64-linux": {
+          "outputs": [
+            {
+              "name": "out",
+              "path": "/nix/store/sndm8q6yhr6a8yy9zdd1sxni3lc8pcmw-nodejs-18.8.0",
+              "default": true
+            },
+            {
+              "name": "libv8",
+              "path": "/nix/store/fc199pzbq6l7bcgxadfljdy6addrrwv1-nodejs-18.8.0-libv8"
+            }
+          ],
+          "store_path": "/nix/store/sndm8q6yhr6a8yy9zdd1sxni3lc8pcmw-nodejs-18.8.0"
+        }
+      }
+    },
     "[email protected]": {
       "last_modified": "2024-12-03T12:40:06Z",
       "resolved": "github:NixOS/nixpkgs/566e53c2ad750c84f6d31f9ccb9d00f823165550#protoc-gen-go",

hscontrol/app.go

@@ -490,7 +490,14 @@ func (h *Headscale) createRouter(grpcMux *grpcRuntime.ServeMux) *mux.Router {
 	apiRouter.Use(h.httpAuthenticationMiddleware)
 	apiRouter.PathPrefix("/v1/").HandlerFunc(grpcMux.ServeHTTP)
 
-	router.PathPrefix("/").HandlerFunc(notFoundHandler)
+	if h.cfg.Frontend.Enabled {
+		if path := h.cfg.Frontend.Path; path != "" {
+			router.PathPrefix("/").Handler(SvelteKitHandler("/"))
+		}
+
+	} else {
+		router.PathPrefix("/").HandlerFunc(notFoundHandler)
+	}
 
 	return router
 }

hscontrol/svelte.go

@@ -0,0 +1,32 @@
+package hscontrol
+
+import (
+	"embed"
+	"errors"
+	"fmt"
+	"io/fs"
+	"net/http"
+	"os"
+	"strings"
+)
+
+//go:embed all:build
+var files embed.FS
+
+func SvelteKitHandler(path string) http.Handler {
+	fsys, err := fs.Sub(files, "build")
+	if err != nil {
+
+	}
+	filesystem := http.FS(fsys)
+
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		path := strings.TrimPrefix(r.URL.Path, path)
+		_, err := filesystem.Open(path)
+		if errors.Is(err, os.ErrNotExist) {
+			path = fmt.Sprintf("%s.html", path)
+		}
+		r.URL.Path = path
+		http.FileServer(filesystem).ServeHTTP(w, r)
+	})
+}

hscontrol/types/config.go

@@ -97,6 +97,8 @@ type Config struct {
 	Policy PolicyConfig
 
 	Tuning Tuning
+
+	Frontend FrontendConfig
 }
 
 type DNSConfig struct {
@@ -220,6 +222,12 @@ type PolicyConfig struct {
 	Mode PolicyMode
 }
 
+type FrontendConfig struct {
+	Enabled bool
+	URL     string
+	Path    string
+}
+
 func (p *PolicyConfig) IsEmpty() bool {
 	return p.Mode == PolicyModeFile && p.Path == ""
 }
@@ -323,6 +331,8 @@ func LoadConfig(path string, isFile bool) error {
 
 	viper.SetDefault("prefixes.allocation", string(IPAllocationStrategySequential))
 
+	viper.SetDefault("frontend.enabled", false)
+
 	if err := viper.ReadInConfig(); err != nil {
 		return fmt.Errorf("fatal error reading config file: %w", err)
 	}
@@ -980,6 +990,12 @@ func LoadServerConfig() (*Config, error) {
 				"tuning.node_mapsession_buffered_chan_size",
 			),
 		},
+
+		Frontend: FrontendConfig{
+			Enabled: viper.GetBool("frontend.enabled"),
+			URL:     viper.GetString("frontend.url"),
+			Path:    viper.GetString("frontend.path"),
+		},
 	}, nil
 }