diff options
Diffstat (limited to 'internal/cli/cli.go')
| -rw-r--r-- | internal/cli/cli.go | 42 |
1 files changed, 41 insertions, 1 deletions
diff --git a/internal/cli/cli.go b/internal/cli/cli.go index 66b5f79..0876d08 100644 --- a/internal/cli/cli.go +++ b/internal/cli/cli.go @@ -11,6 +11,7 @@ import ( "git.theodohertyfamily.com/tools/wg-wrap/internal/config" "git.theodohertyfamily.com/tools/wg-wrap/internal/namespace" "git.theodohertyfamily.com/tools/wg-wrap/internal/paths" + "git.theodohertyfamily.com/tools/wg-wrap/internal/wireguard" "git.theodohertyfamily.com/tools/wg-wrap/pkg/wgconf" ) @@ -120,6 +121,16 @@ func (a *App) Run() error { return a.ExecuteCommand(cfg) } + // Before bootstrapping, see if an active namespace/process for the profile exists. + // If yes, we can join it! + pm := a.getPathManager() + joined, err := namespace.JoinExistingNamespace(pm, cfg.Profile) + if err == nil && joined { + // We have joined the active namespace (user, mnt, net). + // We can now execute the command immediately in this context! + return a.ExecuteCommand(cfg) + } + if err := namespace.Bootstrap(); err != nil { return fmt.Errorf("bootstrap failed: %w", err) } @@ -154,7 +165,36 @@ func (a *App) ExecuteCommand(cfg *config.Config) error { }() fmt.Printf("Initializing WireGuard tunnel for profile %s...\n", cfg.Profile) - // TODO: Integrate with internal/wireguard to set up TUN and WG-Go + + // Parse the profile configuration + profilesDir := pm.ConfigDir() + profilePath := filepath.Join(profilesDir, cfg.Profile+".conf") + + // Create tunnel if the file exists + if _, err := os.Stat(profilePath); err == nil { + wgCfg, err := wgconf.Parse(profilePath) + if err != nil { + return fmt.Errorf("failed to parse profile %s: %w", cfg.Profile, err) + } + + // Start the WireGuard userspace device & routing table setup + tunnel, err := wireguard.StartTunnel(wgCfg) + if err != nil { + return fmt.Errorf("failed to start WireGuard tunnel: %w", err) + } + defer tunnel.Close() + + // Pin the namespace so others can join it + if err := namespace.PinNamespace(pm, cfg.Profile); err != nil { + fmt.Printf("warning: failed to pin namespace: %v\n", err) + } + } else { + // If profile is not default or it was explicitly requested but doesn't exist, we error + if cfg.Profile != "default" { + return fmt.Errorf("profile %s not found: %w", cfg.Profile, err) + } + fmt.Printf("warning: default profile configuration not found. Executing command in bare isolation.\n") + } cmd := exec.Command(cfg.Command[0], cfg.Command[1:]...) cmd.Stdin = os.Stdin |
