51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/mirivlad/sshkeeper/internal/ssh"
|
|
)
|
|
|
|
var sshConfigCmd = &cobra.Command{
|
|
Use: "ssh-config",
|
|
Short: "OpenSSH config management",
|
|
}
|
|
|
|
var sshConfigGenerateCmd = &cobra.Command{
|
|
Use: "generate",
|
|
Short: "Generate OpenSSH config from server profiles",
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
servers, err := appDB.ListServers()
|
|
if err != nil {
|
|
return fmt.Errorf("list servers: %w", err)
|
|
}
|
|
|
|
if err := ssh.WriteConfig(servers); err != nil {
|
|
return fmt.Errorf("write config: %w", err)
|
|
}
|
|
|
|
home, _ := os.UserHomeDir()
|
|
fmt.Printf("Config written to: %s/.ssh/config.d/sshkeeper.conf\n", home)
|
|
return nil
|
|
},
|
|
}
|
|
|
|
var sshConfigInstallIncludeCmd = &cobra.Command{
|
|
Use: "install-include",
|
|
Short: "Add Include directive to ~/.ssh/config",
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
if err := ssh.InstallInclude(); err != nil {
|
|
return fmt.Errorf("install include: %w", err)
|
|
}
|
|
fmt.Println("Include directive added to ~/.ssh/config")
|
|
return nil
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
sshConfigCmd.AddCommand(sshConfigGenerateCmd)
|
|
sshConfigCmd.AddCommand(sshConfigInstallIncludeCmd)
|
|
}
|