mirror of
https://github.com/Gu1llaum-3/sshm.git
synced 2025-09-04 09:46:32 +02:00
108 lines
2.6 KiB
Go
108 lines
2.6 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
"sshm/internal/config"
|
|
"sshm/internal/ui"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// version will be set at build time via -ldflags
|
|
var version = "dev"
|
|
|
|
var rootCmd = &cobra.Command{
|
|
Use: "sshm",
|
|
Short: "SSH Manager - A modern SSH connection manager",
|
|
Long: `SSH Manager (sshm) is a modern command-line tool for managing SSH connections.
|
|
It provides an interactive interface to browse and connect to your SSH hosts
|
|
configured in your ~/.ssh/config file.`,
|
|
Version: version,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
// If no arguments provided, run interactive mode
|
|
if len(args) == 0 {
|
|
runInteractiveMode()
|
|
return
|
|
}
|
|
|
|
// If a host name is provided, connect directly
|
|
hostName := args[0]
|
|
connectToHost(hostName)
|
|
},
|
|
}
|
|
|
|
func runInteractiveMode() {
|
|
// Parse SSH configurations
|
|
hosts, err := config.ParseSSHConfig()
|
|
if err != nil {
|
|
log.Fatalf("Error reading SSH config file: %v", err)
|
|
}
|
|
|
|
if len(hosts) == 0 {
|
|
fmt.Println("No SSH hosts found in your ~/.ssh/config file.")
|
|
fmt.Print("Would you like to add a new host now? [y/N]: ")
|
|
var response string
|
|
_, err := fmt.Scanln(&response)
|
|
if err == nil && (response == "y" || response == "Y") {
|
|
err := ui.RunAddForm("")
|
|
if err != nil {
|
|
fmt.Printf("Error adding host: %v\n", err)
|
|
}
|
|
// After adding, try to reload hosts and continue if any exist
|
|
hosts, err = config.ParseSSHConfig()
|
|
if err != nil || len(hosts) == 0 {
|
|
fmt.Println("No hosts available, exiting.")
|
|
os.Exit(1)
|
|
}
|
|
} else {
|
|
fmt.Println("No hosts available, exiting.")
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
// Run the interactive TUI
|
|
if err := ui.RunInteractiveMode(hosts); err != nil {
|
|
log.Fatalf("Error running interactive mode: %v", err)
|
|
}
|
|
}
|
|
|
|
func connectToHost(hostName string) {
|
|
// Parse SSH configurations to verify host exists
|
|
hosts, err := config.ParseSSHConfig()
|
|
if err != nil {
|
|
log.Fatalf("Error reading SSH config file: %v", err)
|
|
}
|
|
|
|
// Check if host exists
|
|
var hostFound bool
|
|
for _, host := range hosts {
|
|
if host.Name == hostName {
|
|
hostFound = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if !hostFound {
|
|
fmt.Printf("Error: Host '%s' not found in SSH configuration.\n", hostName)
|
|
fmt.Println("Use 'sshm' to see available hosts.")
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Connect to the host
|
|
fmt.Printf("Connecting to %s...\n", hostName)
|
|
// Note: In a real implementation, you'd use exec.Command here
|
|
// For now, just print the command that would be executed
|
|
fmt.Printf("ssh %s\n", hostName)
|
|
}
|
|
|
|
// Execute adds all child commands to the root command and sets flags appropriately.
|
|
func Execute() {
|
|
if err := rootCmd.Execute(); err != nil {
|
|
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|