Initial commit.

2026-06-11 00:08:57 +00:00 · 2025-11-28 02:50:25 +00:00
commit 22552aec99
41 changed files with 10626 additions and 0 deletions
@@ -0,0 +1,474 @@
+// Package installer handles installation and uninstallation of the lolcathost daemon.
+package installer
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"os/user"
+	"path/filepath"
+	"runtime"
+	"strconv"
+	"strings"
+
+	"github.com/lukaszraczylo/lolcathost/internal/config"
+)
+
+const (
+	// GroupName is the name of the lolcathost group.
+	GroupName = "lolcathost"
+	// GroupGID is the GID for the lolcathost group (macOS).
+	GroupGID = 850
+
+	// Paths
+	LogDir          = "/var/log/lolcathost"
+	BackupDir       = "/var/backups/lolcathost"
+	SocketPath      = "/var/run/lolcathost.sock"
+	LaunchDaemonDir = "/Library/LaunchDaemons"
+	SystemdDir      = "/etc/systemd/system"
+)
+
+// LaunchDaemonPlist is the macOS LaunchDaemon plist template.
+const LaunchDaemonPlist = `<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+    <key>Label</key>
+    <string>com.lolcathost.daemon</string>
+    <key>ProgramArguments</key>
+    <array>
+        <string>%s</string>
+        <string>--daemon</string>
+        <string>--config</string>
+        <string>/etc/lolcathost/config.yaml</string>
+    </array>
+    <key>RunAtLoad</key>
+    <true/>
+    <key>KeepAlive</key>
+    <true/>
+    <key>StandardOutPath</key>
+    <string>/var/log/lolcathost/daemon.log</string>
+    <key>StandardErrorPath</key>
+    <string>/var/log/lolcathost/daemon.err</string>
+</dict>
+</plist>
+`
+
+// SystemdUnit is the Linux systemd unit template.
+const SystemdUnit = `[Unit]
+Description=lolcathost - Dynamic Host Management Daemon
+After=network.target
+
+[Service]
+Type=simple
+ExecStart=%s --daemon --config /etc/lolcathost/config.yaml
+Restart=always
+RestartSec=5
+User=root
+Group=root
+
+[Install]
+WantedBy=multi-user.target
+`
+
+// Installer handles installation and uninstallation.
+type Installer struct {
+	binaryPath string
+	verbose    bool
+}
+
+// New creates a new installer.
+func New() (*Installer, error) {
+	binaryPath, err := os.Executable()
+	if err != nil {
+		return nil, fmt.Errorf("failed to get executable path: %w", err)
+	}
+
+	// Resolve symlinks
+	binaryPath, err = filepath.EvalSymlinks(binaryPath)
+	if err != nil {
+		return nil, fmt.Errorf("failed to resolve executable path: %w", err)
+	}
+
+	return &Installer{
+		binaryPath: binaryPath,
+		verbose:    true,
+	}, nil
+}
+
+// Install performs the full installation.
+func (i *Installer) Install() error {
+	if os.Geteuid() != 0 {
+		return fmt.Errorf("--install requires sudo")
+	}
+
+	i.log("Installing lolcathost...")
+
+	// Create group
+	if err := i.createGroup(); err != nil {
+		return fmt.Errorf("failed to create group: %w", err)
+	}
+
+	// Add current user to group
+	if err := i.addCurrentUserToGroup(); err != nil {
+		return fmt.Errorf("failed to add user to group: %w", err)
+	}
+
+	// Create directories
+	if err := i.createDirectories(); err != nil {
+		return fmt.Errorf("failed to create directories: %w", err)
+	}
+
+	// Create system config for daemon
+	if err := i.createSystemConfig(); err != nil {
+		return fmt.Errorf("failed to create system config: %w", err)
+	}
+
+	// Install service
+	if runtime.GOOS == "darwin" {
+		if err := i.installLaunchDaemon(); err != nil {
+			return fmt.Errorf("failed to install LaunchDaemon: %w", err)
+		}
+	} else if runtime.GOOS == "linux" {
+		if err := i.installSystemdService(); err != nil {
+			return fmt.Errorf("failed to install systemd service: %w", err)
+		}
+	}
+
+	// Create default config for the invoking user
+	if err := i.createDefaultConfig(); err != nil {
+		i.log("Warning: failed to create default config: %v", err)
+	}
+
+	i.log("")
+	i.log("✓ Installed successfully!")
+	i.log("")
+	i.log("Next steps:")
+	i.log("  1. Open a NEW terminal (for group membership to take effect)")
+	i.log("  2. Run 'lolcathost' to start the TUI")
+	i.log("")
+
+	return nil
+}
+
+// Uninstall removes the installation.
+func (i *Installer) Uninstall() error {
+	if os.Geteuid() != 0 {
+		return fmt.Errorf("--uninstall requires sudo")
+	}
+
+	i.log("Uninstalling lolcathost...")
+
+	// Stop and remove service
+	if runtime.GOOS == "darwin" {
+		i.uninstallLaunchDaemon()
+	} else if runtime.GOOS == "linux" {
+		i.uninstallSystemdService()
+	}
+
+	// Remove socket
+	os.Remove(SocketPath)
+
+	// Note: We don't remove the group, logs, or backups
+	// The user may want to keep these
+
+	i.log("")
+	i.log("✓ Uninstalled successfully!")
+	i.log("")
+	i.log("Note: Log files, backups, and the group were preserved.")
+	i.log("To fully remove, manually delete:")
+	i.log("  - /var/log/lolcathost/")
+	i.log("  - /var/backups/lolcathost/")
+	i.log("  - ~/.config/lolcathost/")
+	if runtime.GOOS == "darwin" {
+		i.log("  - Remove group: sudo dscl . -delete /Groups/%s", GroupName)
+	} else {
+		i.log("  - Remove group: sudo groupdel %s", GroupName)
+	}
+	i.log("")
+
+	return nil
+}
+
+func (i *Installer) log(format string, args ...any) {
+	if i.verbose {
+		fmt.Printf(format+"\n", args...)
+	}
+}
+
+func (i *Installer) createGroup() error {
+	switch runtime.GOOS {
+	case "darwin":
+		return i.createGroupDarwin()
+	case "linux":
+		return i.createGroupLinux()
+	default:
+		return fmt.Errorf("unsupported OS: %s", runtime.GOOS)
+	}
+}
+
+func (i *Installer) createGroupDarwin() error {
+	// Check if group exists
+	if _, err := exec.Command("dscl", ".", "-read", "/Groups/"+GroupName).Output(); err == nil {
+		i.log("  Group '%s' already exists", GroupName)
+		return nil
+	}
+
+	i.log("  Creating group '%s' (GID %d)...", GroupName, GroupGID)
+
+	// Create group
+	cmds := [][]string{
+		{"dscl", ".", "-create", "/Groups/" + GroupName},
+		{"dscl", ".", "-create", "/Groups/" + GroupName, "PrimaryGroupID", strconv.Itoa(GroupGID)},
+		{"dscl", ".", "-create", "/Groups/" + GroupName, "RealName", "lolcathost users"},
+	}
+
+	for _, args := range cmds {
+		if err := exec.Command(args[0], args[1:]...).Run(); err != nil {
+			return fmt.Errorf("command %v failed: %w", args, err)
+		}
+	}
+
+	return nil
+}
+
+func (i *Installer) createGroupLinux() error {
+	// Check if group exists
+	if _, err := exec.Command("getent", "group", GroupName).Output(); err == nil {
+		i.log("  Group '%s' already exists", GroupName)
+		return nil
+	}
+
+	i.log("  Creating group '%s'...", GroupName)
+
+	if err := exec.Command("groupadd", "-r", GroupName).Run(); err != nil {
+		return fmt.Errorf("groupadd failed: %w", err)
+	}
+
+	return nil
+}
+
+func (i *Installer) addCurrentUserToGroup() error {
+	// Get the real user (not root)
+	username := os.Getenv("SUDO_USER")
+	if username == "" {
+		// Fall back to current user
+		u, err := user.Current()
+		if err != nil {
+			return fmt.Errorf("failed to get current user: %w", err)
+		}
+		username = u.Username
+	}
+
+	if username == "root" {
+		i.log("  Skipping adding root to group")
+		return nil
+	}
+
+	switch runtime.GOOS {
+	case "darwin":
+		return i.addUserToGroupDarwin(username)
+	case "linux":
+		return i.addUserToGroupLinux(username)
+	default:
+		return fmt.Errorf("unsupported OS: %s", runtime.GOOS)
+	}
+}
+
+func (i *Installer) addUserToGroupDarwin(username string) error {
+	// Check if user is already in group
+	output, err := exec.Command("dscl", ".", "-read", "/Groups/"+GroupName, "GroupMembership").Output()
+	if err == nil && strings.Contains(string(output), username) {
+		i.log("  User '%s' already in group '%s'", username, GroupName)
+		return nil
+	}
+
+	i.log("  Adding user '%s' to group '%s'...", username, GroupName)
+
+	if err := exec.Command("dscl", ".", "-append", "/Groups/"+GroupName, "GroupMembership", username).Run(); err != nil {
+		return fmt.Errorf("failed to add user to group: %w", err)
+	}
+
+	return nil
+}
+
+func (i *Installer) addUserToGroupLinux(username string) error {
+	// Check if user is already in group
+	output, err := exec.Command("id", "-nG", username).Output()
+	if err == nil && strings.Contains(string(output), GroupName) {
+		i.log("  User '%s' already in group '%s'", username, GroupName)
+		return nil
+	}
+
+	i.log("  Adding user '%s' to group '%s'...", username, GroupName)
+
+	if err := exec.Command("usermod", "-aG", GroupName, username).Run(); err != nil {
+		return fmt.Errorf("failed to add user to group: %w", err)
+	}
+
+	return nil
+}
+
+func (i *Installer) createDirectories() error {
+	dirs := []string{LogDir, BackupDir, config.SystemConfigDir}
+
+	for _, dir := range dirs {
+		i.log("  Creating directory '%s'...", dir)
+		if err := os.MkdirAll(dir, 0755); err != nil {
+			return fmt.Errorf("failed to create %s: %w", dir, err)
+		}
+	}
+
+	return nil
+}
+
+func (i *Installer) createSystemConfig() error {
+	// Check if system config already exists
+	if _, err := os.Stat(config.SystemConfigPath); err == nil {
+		i.log("  System config already exists at %s", config.SystemConfigPath)
+		return nil
+	}
+
+	i.log("  Creating system config at %s...", config.SystemConfigPath)
+	return config.CreateDefault(config.SystemConfigPath)
+}
+
+func (i *Installer) installLaunchDaemon() error {
+	plistPath := filepath.Join(LaunchDaemonDir, "com.lolcathost.daemon.plist")
+	plistContent := fmt.Sprintf(LaunchDaemonPlist, i.binaryPath)
+
+	i.log("  Writing LaunchDaemon plist...")
+	if err := os.WriteFile(plistPath, []byte(plistContent), 0644); err != nil {
+		return fmt.Errorf("failed to write plist: %w", err)
+	}
+
+	// Unload if already loaded
+	exec.Command("launchctl", "bootout", "system/com.lolcathost.daemon").Run()
+
+	// Bootstrap the daemon
+	i.log("  Starting daemon...")
+	if err := exec.Command("launchctl", "bootstrap", "system", plistPath).Run(); err != nil {
+		return fmt.Errorf("failed to bootstrap daemon: %w", err)
+	}
+
+	return nil
+}
+
+func (i *Installer) uninstallLaunchDaemon() {
+	plistPath := filepath.Join(LaunchDaemonDir, "com.lolcathost.daemon.plist")
+
+	i.log("  Stopping daemon...")
+	exec.Command("launchctl", "bootout", "system/com.lolcathost.daemon").Run()
+
+	i.log("  Removing LaunchDaemon plist...")
+	os.Remove(plistPath)
+}
+
+func (i *Installer) installSystemdService() error {
+	unitPath := filepath.Join(SystemdDir, "lolcathost.service")
+	unitContent := fmt.Sprintf(SystemdUnit, i.binaryPath)
+
+	i.log("  Writing systemd unit...")
+	if err := os.WriteFile(unitPath, []byte(unitContent), 0644); err != nil {
+		return fmt.Errorf("failed to write unit file: %w", err)
+	}
+
+	// Reload systemd
+	i.log("  Reloading systemd...")
+	if err := exec.Command("systemctl", "daemon-reload").Run(); err != nil {
+		return fmt.Errorf("failed to reload systemd: %w", err)
+	}
+
+	// Enable and start the service
+	i.log("  Enabling and starting service...")
+	if err := exec.Command("systemctl", "enable", "--now", "lolcathost.service").Run(); err != nil {
+		return fmt.Errorf("failed to enable service: %w", err)
+	}
+
+	return nil
+}
+
+func (i *Installer) uninstallSystemdService() {
+	i.log("  Stopping and disabling service...")
+	exec.Command("systemctl", "disable", "--now", "lolcathost.service").Run()
+
+	i.log("  Removing systemd unit...")
+	os.Remove(filepath.Join(SystemdDir, "lolcathost.service"))
+
+	exec.Command("systemctl", "daemon-reload").Run()
+}
+
+func (i *Installer) createDefaultConfig() error {
+	// Get the real user's home directory
+	username := os.Getenv("SUDO_USER")
+	if username == "" {
+		return nil // Can't determine user
+	}
+
+	u, err := user.Lookup(username)
+	if err != nil {
+		return fmt.Errorf("failed to lookup user: %w", err)
+	}
+
+	configPath := filepath.Join(u.HomeDir, ".config", "lolcathost", "config.yaml")
+
+	// Check if config already exists
+	if _, err := os.Stat(configPath); err == nil {
+		i.log("  Config already exists at %s", configPath)
+		return nil
+	}
+
+	i.log("  Creating default config at %s...", configPath)
+
+	if err := config.CreateDefault(configPath); err != nil {
+		return err
+	}
+
+	// Change ownership to the real user
+	uid, _ := strconv.Atoi(u.Uid)
+	gid, _ := strconv.Atoi(u.Gid)
+
+	configDir := filepath.Dir(configPath)
+	os.Chown(configDir, uid, gid)
+	os.Chown(filepath.Dir(configDir), uid, gid)
+	os.Chown(configPath, uid, gid)
+
+	return nil
+}
+
+// CheckInstallation checks if the daemon is properly installed.
+func CheckInstallation() error {
+	// Check if socket exists
+	if _, err := os.Stat(SocketPath); os.IsNotExist(err) {
+		return fmt.Errorf("daemon not running (socket not found)")
+	}
+
+	// Check if user is in group
+	u, err := user.Current()
+	if err != nil {
+		return fmt.Errorf("failed to get current user: %w", err)
+	}
+
+	groups, err := u.GroupIds()
+	if err != nil {
+		return fmt.Errorf("failed to get user groups: %w", err)
+	}
+
+	inGroup := false
+	for _, gid := range groups {
+		g, err := user.LookupGroupId(gid)
+		if err != nil {
+			continue
+		}
+		if g.Name == GroupName {
+			inGroup = true
+			break
+		}
+	}
+
+	if !inGroup {
+		return fmt.Errorf("user '%s' is not in group '%s'. Run 'sudo lolcathost --install' and open a new terminal", u.Username, GroupName)
+	}
+
+	return nil
+}