85 lines
2.6 KiB
Go
85 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
|
|
"proxyctl/internal/app"
|
|
"proxyctl/internal/profile"
|
|
"proxyctl/internal/shell"
|
|
"proxyctl/internal/target"
|
|
"proxyctl/internal/tui"
|
|
)
|
|
|
|
func main() {
|
|
if err := run(os.Args[1:], os.Stdin, os.Stdout); err != nil {
|
|
fmt.Fprintln(os.Stderr, "proxyctl:", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func run(args []string, input io.Reader, output io.Writer) error {
|
|
executable, err := os.Executable()
|
|
if err != nil {
|
|
return fmt.Errorf("locate proxyctl executable: %w", err)
|
|
}
|
|
return runWithEnvironment(args, input, output, processEnvironment{}, shell.OSFileSystem{}, executable)
|
|
}
|
|
|
|
type processEnvironment struct{}
|
|
|
|
func (processEnvironment) CurrentUser() (profile.UserIdentity, error) {
|
|
return profile.SystemCurrentUser()
|
|
}
|
|
func (processEnvironment) LookupUser(name string) (profile.UserIdentity, error) {
|
|
return profile.SystemLookupUser(name)
|
|
}
|
|
func (processEnvironment) LookupEnv(name string) (string, bool) { return os.LookupEnv(name) }
|
|
|
|
func runWithEnvironment(args []string, input io.Reader, output io.Writer, environment profile.IdentityEnvironment, files shell.FileSystem, executable string) error {
|
|
profileConfiguration, err := profile.ResolveConfiguration(environment)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
shellConfiguration, err := shell.ResolveConfiguration(environment)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
shellTarget := shell.New(shellConfiguration, files, environment)
|
|
|
|
if len(args) != 0 {
|
|
if len(args) != 2 || args[0] != "shell" || (args[1] != "install" && args[1] != "uninstall") {
|
|
return fmt.Errorf("usage: proxyctl [shell install|shell uninstall]")
|
|
}
|
|
availability := shellTarget.Availability(context.Background())
|
|
if !availability.Available {
|
|
return fmt.Errorf("Shell Target unavailable: %s", availability.Evidence)
|
|
}
|
|
if args[1] == "install" {
|
|
if err := shellTarget.Install(context.Background(), executable); err != nil {
|
|
return err
|
|
}
|
|
fmt.Fprintln(output, "Shell integration installed; restart or source your Shell startup file once.")
|
|
return nil
|
|
}
|
|
if err := shellTarget.Uninstall(context.Background()); err != nil {
|
|
return err
|
|
}
|
|
fmt.Fprintln(output, "Shell integration uninstalled.")
|
|
return nil
|
|
}
|
|
|
|
store := profile.NewOwnedFileStoreAt(profileConfiguration.Path, profileConfiguration.Owner, profileConfiguration.TrustedRoot)
|
|
application, err := app.NewWithTargets(store, []target.Target{shellTarget})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
program := tea.NewProgram(tui.NewModel(application), tea.WithInput(input), tea.WithOutput(output))
|
|
_, err = program.Run()
|
|
return err
|
|
}
|