Files
waste-go/cmd/daemon/main.go
Fredrik Johansson 77830a3b3f feat: WebSocket IPC endpoint for web UI
Add --ws-port flag to waste-daemon; when set, starts a WebSocket IPC
server alongside the existing TCP one. The web UI connects to this from
the browser. Uses nhooyr.io/websocket with localhost-only origin policy.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-06-22 21:57:10 +02:00

106 lines
3.1 KiB
Go

// waste-daemon: the local peer process.
// Run one of these on each friend's machine.
package main
import (
"flag"
"fmt"
"log"
"os"
"github.com/waste-go/internal/crypto"
"github.com/waste-go/internal/ipc"
"github.com/waste-go/internal/invite"
"github.com/waste-go/internal/netmgr"
)
func main() {
dataDir := flag.String("data-dir", "~/.waste", "path to identity/config directory")
alias := flag.String("alias", "anon", "display name shown to peers (advisory only)")
ipcPort := flag.Int("ipc-port", 17337, "port for local IPC (UI connects here)")
wsPort := flag.Int("ws-port", 0, "port for WebSocket IPC (web UI); 0 = disabled")
anchorURL := flag.String("anchor", "", "anchor WebSocket URL, e.g. ws://your-vps:17339/ws")
shareDir := flag.String("share-dir", "", "directory to share with peers on the network")
joinInvite := flag.String("join", "", "waste: invite string — sets anchor URL and auto-joins the network on startup")
importBackup := flag.String("import-identity", "", "path to a yaw-key-backup-1 JSON file to import")
importPassword := flag.String("import-passphrase", "", "passphrase for --import-identity")
flag.Parse()
dir := expandHome(*dataDir)
// --import-identity: decrypt backup and write identity.json, then exit.
if *importBackup != "" {
if *importPassword == "" {
log.Fatal("--import-passphrase is required with --import-identity")
}
raw, err := os.ReadFile(*importBackup)
if err != nil {
log.Fatalf("import-identity: read file: %v", err)
}
imported, err := crypto.ImportIdentity(raw, *importPassword)
if err != nil {
log.Fatalf("import-identity: %v", err)
}
if err := os.MkdirAll(dir, 0700); err != nil {
log.Fatalf("import-identity: mkdir: %v", err)
}
if err := crypto.SaveIdentity(dir, imported); err != nil {
log.Fatalf("import-identity: save: %v", err)
}
fmt.Printf("identity imported: %s\n", imported.PeerID())
return
}
id, err := crypto.LoadOrCreate(dir, *alias)
if err != nil {
log.Fatalf("identity: %v", err)
}
log.Printf("daemon: local peer id: %s alias: %s", id.PeerID().Short(), id.Alias)
// --join overrides/sets the anchor URL and triggers an auto-join.
var autoJoinNetwork string
if *joinInvite != "" {
inv, err := invite.Decode(*joinInvite)
if err != nil {
log.Fatalf("invalid invite: %v", err)
}
*anchorURL = inv.Anchor
autoJoinNetwork = inv.Network
log.Printf("daemon: invite decoded — anchor=%s network=%s", inv.Anchor, inv.Network)
}
mgr := netmgr.New(netmgr.Config{
MasterIdentity: id,
StoreDir: dir,
AnchorURL: *anchorURL,
ShareDir: expandHome(*shareDir),
})
if autoJoinNetwork != "" {
if _, err := mgr.Join(autoJoinNetwork, ""); err != nil {
log.Fatalf("auto-join: %v", err)
}
}
if *wsPort != 0 {
go func() {
if err := ipc.RunWS(mgr, *wsPort); err != nil {
log.Fatalf("ipc ws: %v", err)
}
}()
}
if err := ipc.Run(mgr, *ipcPort); err != nil {
log.Fatalf("ipc: %v", err)
}
}
func expandHome(path string) string {
if len(path) >= 2 && path[:2] == "~/" {
if home, err := os.UserHomeDir(); err == nil {
return home + path[1:]
}
}
return path
}