Fixes #83

Author: omani

cmd/main.go

@@ -28,44 +28,52 @@ import (
 )
 
 var (
-	version   = "develop"
-	gitCommit = "unknown"
-)
+	version       = "develop"
+	gitCommit     = "unknown"
+	sshConfigFile string
 
-func main() {
-	log, err := logger.New("LAZYSSH")
-	if err != nil {
-		fmt.Println(err)
-		os.Exit(1)
-	}
+	rootCmd = &cobra.Command{
+		Use:   ui.AppName,
+		Short: "Lazy SSH server picker TUI",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			log, err := logger.New("LAZYSSH")
+			if err != nil {
+				fmt.Println(err)
+				os.Exit(1)
+			}
 
-	//nolint:errcheck // log.Sync may return an error which is safe to ignore here
-	defer log.Sync()
+			//nolint:errcheck // log.Sync may return an error which is safe to ignore here
+			defer log.Sync()
 
-	home, err := os.UserHomeDir()
-	if err != nil {
-		log.Errorw("failed to get user home directory", "error", err)
-		//nolint:gocritic // exitAfterDefer: ensure immediate exit on unrecoverable error
-		os.Exit(1)
-	}
-	sshConfigFile := filepath.Join(home, ".ssh", "config")
-	metaDataFile := filepath.Join(home, ".lazyssh", "metadata.json")
+			home, err := os.UserHomeDir()
+			if err != nil {
+				log.Errorw("failed to get user home directory", "error", err)
+				// nolint:gocritic // exitAfterDefer: ensure immediate exit on unrecoverable error
+				os.Exit(1)
+			}
 
-	serverRepo := ssh_config_file.NewRepository(log, sshConfigFile, metaDataFile)
-	serverService := services.NewServerService(log, serverRepo)
-	tui := ui.NewTUI(log, serverService, version, gitCommit)
+			if sshConfigFile == "" {
+				sshConfigFile = filepath.Join(home, ".ssh", "config")
+			}
+			metaDataFile := filepath.Join(home, ".lazyssh", "metadata.json")
+			serverRepo := ssh_config_file.NewRepository(log, sshConfigFile, metaDataFile)
+			serverService := services.NewServerService(log, serverRepo)
+			tui := ui.NewTUI(log, serverService, version, gitCommit)
 
-	rootCmd := &cobra.Command{
-		Use:   ui.AppName,
-		Short: "Lazy SSH server picker TUI",
-		RunE: func(cmd *cobra.Command, args []string) error {
 			return tui.Run()
 		},
 	}
-	rootCmd.SilenceUsage = true
+)
 
+func main() {
 	if err := rootCmd.Execute(); err != nil {
 		_, _ = fmt.Fprintln(os.Stderr, err)
 		os.Exit(1)
 	}
 }
+
+func init() {
+	rootCmd.PersistentFlags().StringVar(&sshConfigFile, "sshconfig", "", "path to ssh config file (default: ~/.ssh/config)")
+
+	rootCmd.SilenceUsage = true
+}