使用manssh轻松管理Golang中的SSH别名配置

manssh是一个用Go编写的轻量级命令行工具，可以帮助开发者轻松管理SSH别名配置，简化SSH连接过程。下面我将详细介绍如何使用manssh以及如何集成到你的Golang项目中。

manssh简介

manssh的主要功能是帮助用户管理~/.ssh/config文件中的SSH别名配置，提供简单的命令行界面来添加、删除、列出和搜索SSH连接配置。

安装manssh

go get -u github.com/xwjdsh/manssh

基本使用方法

1. 添加SSH别名

manssh add my_server -H 192.168.1.100 -u root -p 22

2. 列出所有配置

manssh list

3. 搜索配置

manssh search my

4. 删除配置

manssh delete my_server

Golang集成示例

以下是如何在Golang程序中使用manssh的示例代码：

package main

import (
	"fmt"
	"os"
	"os/exec"
)

func main() {
	// 添加SSH配置
	addSSHConfig("my_server", "192.168.1.100", "root", 22)
	
	// 列出所有配置
	listSSHConfigs()
	
	// 搜索配置
	searchSSHConfigs("my")
	
	// 删除配置
	deleteSSHConfig("my_server")
}

func addSSHConfig(name, host, user string, port int) {
	cmd := exec.Command("manssh", "add", name, "-H", host, "-u", user, "-p", fmt.Sprintf("%d", port))
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr
	if err := cmd.Run(); err != nil {
		fmt.Printf("添加SSH配置失败: %v\n", err)
	}
}

func listSSHConfigs() {
	cmd := exec.Command("manssh", "list")
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr
	if err := cmd.Run(); err != nil {
		fmt.Printf("列出SSH配置失败: %v\n", err)
	}
}

func searchSSHConfigs(keyword string) {
	cmd := exec.Command("manssh", "search", keyword)
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr
	if err := cmd.Run(); err != nil {
		fmt.Printf("搜索SSH配置失败: %v\n", err)
	}
}

func deleteSSHConfig(name string) {
	cmd := exec.Command("manssh", "delete", name)
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr
	if err := cmd.Run(); err != nil {
		fmt.Printf("删除SSH配置失败: %v\n", err)
	}
}

高级用法：直接操作SSH配置

如果你不想依赖manssh命令行工具，也可以直接在Golang中操作SSH配置文件：

package main

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"
)

const sshConfigPath = "~/.ssh/config"

type SSHConfig struct {
	Host     string
	HostName string
	User     string
	Port     int
}

func main() {
	// 添加配置
	config := SSHConfig{
		Host:     "my_server",
		HostName: "192.168.1.100",
		User:     "root",
		Port:     22,
	}
	addSSHConfig(config)
	
	// 列出配置
	configs := listSSHConfigs()
	fmt.Println("所有SSH配置:")
	for _, c := range configs {
		fmt.Printf("%s -> %s@%s:%d\n", c.Host, c.User, c.HostName, c.Port)
	}
}

func addSSHConfig(config SSHConfig) error {
	configLine := fmt.Sprintf(`
Host %s
    HostName %s
    User %s
    Port %d
`, config.Host, config.HostName, config.User, config.Port)
	
	file, err := os.OpenFile(expandPath(sshConfigPath), os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0600)
	if err != nil {
		return err
	}
	defer file.Close()
	
	_, err = file.WriteString(configLine)
	return err
}

func listSSHConfigs() []SSHConfig {
	var configs []SSHConfig
	content, err := os.ReadFile(expandPath(sshConfigPath))
	if err != nil {
		return configs
	}
	
	lines := strings.Split(string(content), "\n")
	var currentConfig *SSHConfig
	
	for _, line := range lines {
		line = strings.TrimSpace(line)
		if strings.HasPrefix(line, "Host ") {
			if currentConfig != nil {
				configs = append(configs, *currentConfig)
			}
			currentConfig = &SSHConfig{Host: strings.TrimSpace(line[5:])}
		} else if strings.HasPrefix(line, "HostName ") && currentConfig != nil {
			currentConfig.HostName = strings.TrimSpace(line[9:])
		} else if strings.HasPrefix(line, "User ") && currentConfig != nil {
			currentConfig.User = strings.TrimSpace(line[5:])
		} else if strings.HasPrefix(line, "Port ") && currentConfig != nil {
			fmt.Sscanf(strings.TrimSpace(line[5:]), "%d", &currentConfig.Port)
		}
	}
	
	if currentConfig != nil {
		configs = append(configs, *currentConfig)
	}
	
	return configs
}

func expandPath(path string) string {
	if strings.HasPrefix(path, "~/") {
		home, _ := os.UserHomeDir()
		return filepath.Join(home, path[2:])
	}
	return path
}

总结

manssh是一个简单实用的工具，可以帮助开发者高效管理SSH配置。无论是通过命令行直接使用，还是集成到Golang项目中，都能显著提升工作效率。对于需要频繁连接多个SSH服务器的开发者来说，这是一个非常值得尝试的工具。

如果你需要更复杂的功能，也可以基于上述代码进行扩展，比如添加配置验证、模板支持等功能。