feat: custom url scheme

Author: kermanx

.gitignore

@@ -10,3 +10,4 @@ frontend/public/release
 .eslintcache
 sdk/node/README.md
 **/.vitepress/cache
+.fuse_hidden*

service/discovery/file.go

@@ -12,7 +12,7 @@ import (
 func getFilePath() string {
 	if runtime.GOOS == "windows" {
 		localAppData := os.Getenv("LOCALAPPDATA")
-		return filepath.Join(localAppData, "UnitedToken", "service.json")
+		return filepath.Join(localAppData, "UniToken", "service.json")
 	}
 	home := os.Getenv("HOME")
 	return filepath.Join(home, ".local", "share", "uni-token", "service.json")

service/go.mod

@@ -12,6 +12,7 @@ require (
 	github.com/kardianos/service v1.2.4
 	github.com/pkg/browser v0.0.0-20240102092130-5ac0b6a4141c
 	go.etcd.io/bbolt v1.4.2
+	golang.org/x/sys v0.35.0
 )
 
 require (
@@ -37,7 +38,6 @@ require (
 	golang.org/x/arch v0.18.0 // indirect
 	golang.org/x/crypto v0.39.0 // indirect
 	golang.org/x/net v0.41.0 // indirect
-	golang.org/x/sys v0.34.0 // indirect
 	golang.org/x/text v0.26.0 // indirect
 	google.golang.org/protobuf v1.36.6 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect

service/go.sum

@@ -90,8 +90,8 @@ golang.org/x/sync v0.15.0 h1:KWH3jNZsfyT6xfAfKiz6MRNmd46ByHDYaZ7KSkCtdW8=
 golang.org/x/sync v0.15.0/go.mod h1:1dzgHSNfp02xaA81J2MS99Qcpr2w7fw1gpm99rleRqA=
 golang.org/x/sys v0.1.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.34.0 h1:H5Y5sJ2L2JRdyv7ROF1he/lPdvFsd0mJHFw2ThKHxLA=
-golang.org/x/sys v0.34.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
+golang.org/x/sys v0.35.0 h1:vz1N37gP5bs89s7He8XuIYXpyY0+QlsKmzipCbUtyxI=
+golang.org/x/sys v0.35.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
 golang.org/x/text v0.26.0 h1:P42AVeLghgTYr4+xUnTRKDMqpar+PtX7KWuNQL21L8M=
 golang.org/x/text v0.26.0/go.mod h1:QK15LZJUUQVJxhz7wXgxSy/CJaTFjd0G+YLonydOVQA=
 google.golang.org/protobuf v1.36.6 h1:z1NpPI8ku2WgiWnf+t9wTPsn6eP1L7ksHUlkfLvd9xY=

service/logic/url_scheme/linux.go

@@ -0,0 +1,69 @@
+//go:build linux
+
+package urlScheme
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"path/filepath"
+)
+
+// Stub for Windows function when building on Linux
+func registerURLSchemeWindows(options UrlSchemeRegisterOption) error {
+	return fmt.Errorf("windows URL scheme registration not available on linux")
+}
+
+// registerURLSchemeForLinux registers URL scheme on Linux
+func registerURLSchemeLinux(options UrlSchemeRegisterOption) error {
+	homeDir, err := os.UserHomeDir()
+	if err != nil {
+		return fmt.Errorf("failed to get user home directory: %w", err)
+	}
+
+	// .desktop files are usually placed in ~/.local/share/applications/
+	desktopDir := filepath.Join(homeDir, ".local", "share", "applications")
+	if _, err := os.Stat(desktopDir); os.IsNotExist(err) {
+		if err := os.MkdirAll(desktopDir, 0755); err != nil {
+			return fmt.Errorf("failed to create directory %s: %w", desktopDir, err)
+		}
+	}
+
+	// Define .desktop file name and path
+	desktopFileName := fmt.Sprintf("%s.desktop", options.AppName)
+	desktopFilePath := filepath.Join(desktopDir, desktopFileName)
+
+	// Get current executable path
+	exePath, err := os.Executable()
+	if err != nil {
+		return fmt.Errorf("failed to get executable path: %w", err)
+	}
+
+	// Write .desktop file content
+	content := fmt.Sprintf(`[Desktop Entry]
+Name=%s
+Comment=Opens %s:// links
+Exec=%s url %%u
+Terminal=false
+Type=Application
+MimeType=x-scheme-handler/%s;
+`, options.AppName, options.Scheme, exePath, options.Scheme)
+
+	if err := os.WriteFile(desktopFilePath, []byte(content), 0644); err != nil {
+		return fmt.Errorf("failed to write .desktop file: %w", err)
+	}
+	// Use xdg-mime command to register MIME type
+	// This tells the system which application handles 'x-scheme-handler/my-app'
+	cmd := exec.Command("xdg-mime", "default", desktopFileName, fmt.Sprintf("x-scheme-handler/%s", options.Scheme))
+	if output, err := cmd.CombinedOutput(); err != nil {
+		return fmt.Errorf("xdg-mime execution failed: %w\nOutput: %s", err, output)
+	}
+
+	// Some desktop environments may need to update desktop database, but xdg-mime default is usually sufficient
+	// cmd = exec.Command("update-desktop-database", desktopDir)
+	// if output, err := cmd.CombinedOutput(); err != nil {
+	// 	fmt.Printf("⚠️ Warning: update-desktop-database may fail or be unnecessary: %w\nOutput: %s\n", err, output)
+	// }
+
+	return nil
+}

service/logic/url_scheme/register.go

@@ -0,0 +1,24 @@
+package urlScheme
+
+import (
+	"fmt"
+	"runtime"
+)
+
+type UrlSchemeRegisterOption struct {
+	Scheme  string
+	AppName string
+}
+
+func RegisterURLScheme(options UrlSchemeRegisterOption) error {
+	switch runtime.GOOS {
+	case "windows":
+		return registerURLSchemeWindows(options)
+	case "linux":
+		return registerURLSchemeLinux(options)
+	// case "darwin":
+	// 	return registerURLSchemeMacOS(options)
+	default:
+		return fmt.Errorf("unsupported OS: %s", runtime.GOOS)
+	}
+}

service/logic/url_scheme/stub.go

@@ -0,0 +1,13 @@
+//go:build !linux && !windows
+
+package urlScheme
+
+import "fmt"
+
+func registerURLSchemeWindows(options UrlSchemeRegisterOption) error {
+	return fmt.Errorf("Windows URL scheme registration not implemented on this platform")
+}
+
+func registerURLSchemeLinux(options UrlSchemeRegisterOption) error {
+	return fmt.Errorf("Linux URL scheme registration not implemented on this platform")
+}

service/logic/url_scheme/windows.go

@@ -0,0 +1,66 @@
+//go:build windows
+
+package urlScheme
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"golang.org/x/sys/windows/registry"
+)
+
+// Stub for Linux function when building on Windows
+func registerURLSchemeLinux(options UrlSchemeRegisterOption) error {
+	return fmt.Errorf("Linux URL scheme registration not available on Windows")
+}
+
+// registerURLSchemeForWindows registers URL scheme on Windows
+func registerURLSchemeWindows(options UrlSchemeRegisterOption) error {
+	// Get current executable path
+	exePath, err := os.Executable()
+	if err != nil {
+		return fmt.Errorf("failed to get executable path: %w", err)
+	}
+	// Convert path to Windows-style slashes
+	exePath = filepath.ToSlash(exePath)
+
+	// Build registry path
+	// HKEY_CURRENT_USER is user-specific, can be modified without admin privileges
+	baseKey := "Software\\Classes\\" + options.Scheme
+	commandKey := baseKey + "\\shell\\open\\command"
+
+	// 1. Create or open main key (e.g.: HKEY_CURRENT_USER\Software\Classes\my-app)
+	k, _, err := registry.CreateKey(registry.CURRENT_USER, baseKey, registry.SET_VALUE)
+	if err != nil {
+		return fmt.Errorf("failed to create registry key %s: %w", baseKey, err)
+	}
+	defer k.Close()
+
+	// Set default value and "URL Protocol" flag
+	// @="URL: My App Protocol"
+	if err := k.SetStringValue("", "URL: "+strings.ToUpper(options.Scheme)+" Protocol"); err != nil {
+		return fmt.Errorf("failed to set default value for key %s: %w", baseKey, err)
+	}
+	// "URL Protocol"="" (empty string value, indicates this is a URL Protocol Handler)
+	if err := k.SetStringValue("URL Protocol", ""); err != nil {
+		return fmt.Errorf("failed to set URL Protocol value for key %s: %w", baseKey, err)
+	}
+
+	// 2. Create or open command key (e.g.: ...\my-app\shell\open\command)
+	cmdK, _, err := registry.CreateKey(registry.CURRENT_USER, commandKey, registry.SET_VALUE)
+	if err != nil {
+		return fmt.Errorf("failed to create registry key %s: %w", commandKey, err)
+	}
+	defer cmdK.Close()
+
+	// Set command key value to executable path and %1 (full URL placeholder)
+	// @="\"C:\\Path\\To\\Your\\MyApp.exe\" \"%1\""
+	commandValue := fmt.Sprintf(`"%s" "%%1"`, exePath) // Note: %% before %1 is for escaping
+	if err := cmdK.SetStringValue("", commandValue); err != nil {
+		return fmt.Errorf("failed to set command value for key %s: %w", commandKey, err)
+	}
+
+	return nil
+}

service/main.go

@@ -3,14 +3,14 @@ package main
 import (
 	"fmt"
 	"os"
-	"path/filepath"
 	"strings"
 	"time"
 
 	"github.com/kardianos/service"
 
 	"uni-token-service/discovery"
 	"uni-token-service/logic"
+	urlScheme "uni-token-service/logic/url_scheme"
 	"uni-token-service/server"
 	"uni-token-service/store"
 )
@@ -94,44 +94,23 @@ func main() {
 	}
 	command := os.Args[1]
 
-	if command == "sudo" {
-		command, err := filepath.Abs(os.Args[0])
-		if err != nil {
-			panic(err)
-		}
-		for _, arg := range os.Args[2:] {
-			command += " " + arg
-		}
-
-		res, err := logic.SudoExec(command, &logic.SudoOptions{
-			Name: serviceDisplayName,
-		})
-		if err != nil {
-			panic(err)
-		}
-		if res.Stdout != "" {
-			fmt.Println(res.Stdout)
-		}
-		if res.Stderr != "" {
-			fmt.Println(res.Stderr)
-		}
+	if command == "version" {
+		fmt.Printf("Service Version: %d\n", logic.GetVersion())
 		return
 	}
 
-	if command == "version" {
-		fmt.Printf("Service Version: %d\n", logic.GetVersion())
+	if command == "url" {
+		handleUrlScheme(os.Args[2])
 		return
 	}
 
 	if command == "setup" {
-		err = service.Control(s, "install")
-		if err != nil && !strings.Contains(err.Error(), "already exists") {
-			fmt.Println("Failed to install service:", err)
-		}
-		err = service.Control(s, "start")
-		if err != nil {
-			panic(err)
-		}
+		handleSetup()
+		return
+	}
+
+	if command == "setup-in-sudo" {
+		handleSetupInSudo(&s)
 		return
 	}
 
@@ -142,3 +121,61 @@ func main() {
 		panic(err)
 	}
 }
+
+func handleSetup() {
+	// Register URL scheme for the application
+	err := urlScheme.RegisterURLScheme(urlScheme.UrlSchemeRegisterOption{
+		Scheme:  "uni-token",
+		AppName: "UniToken",
+	})
+	if err != nil {
+		panic(err)
+	}
+
+	// Install and start the service
+	execPath, err := os.Executable()
+	if err != nil {
+		panic(err)
+	}
+	res, err := logic.SudoExec(
+		execPath+" setup-in-sudo",
+		&logic.SudoOptions{
+			Name: serviceDisplayName,
+		},
+	)
+	if err != nil {
+		panic(err)
+	}
+	if res.Stdout != "" {
+		fmt.Println(res.Stdout)
+	}
+	if res.Stderr != "" {
+		fmt.Println(res.Stderr)
+	}
+}
+
+func handleUrlScheme(url string) {
+	if !strings.HasPrefix(url, "uni-token://") {
+		fmt.Println("Invalid URL scheme. Expected 'uni-token://'.")
+		return
+	}
+	url = strings.TrimPrefix(url, "uni-token://")
+
+	switch url {
+	case "start":
+		handleSetup()
+	default:
+		fmt.Printf("Unknown URL action: %s\n", url)
+	}
+}
+
+func handleSetupInSudo(s *service.Service) {
+	err := service.Control(*s, "install")
+	if err != nil && !strings.Contains(err.Error(), "already exists") {
+		fmt.Println("Failed to install service:", err)
+	}
+	err = service.Control(*s, "start")
+	if err != nil {
+		panic(err)
+	}
+}

service/server/action.go

@@ -16,8 +16,8 @@ func SetupActionAPI(router *gin.Engine) {
 
 func handleCheck(c *gin.Context) {
 	c.JSON(http.StatusOK, gin.H{
-		"__united_token": true,
-		"version":        logic.GetVersion(),
+		"__uni_token": true,
+		"version":     logic.GetVersion(),
 	})
 }