gssh-agent 1.0.0

package/pkg/rpc/handler.go

@@ -0,0 +1,206 @@
+package rpc
+
+import (
+	"encoding/json"
+	"fmt"
+	"io"
+	"log"
+	"net"
+	"net/http"
+	"os"
+
+	"gssh/internal/protocol"
+	"gssh/internal/session"
+)
+
+// Handler handles JSON-RPC requests
+type Handler struct {
+	manager *session.Manager
+}
+
+// NewHandler creates a new RPC handler
+func NewHandler(manager *session.Manager) *Handler {
+	return &Handler{manager: manager}
+}
+
+// Handle processes a JSON-RPC request
+func (h *Handler) Handle(data []byte) ([]byte, error) {
+	var req protocol.Request
+	if err := json.Unmarshal(data, &req); err != nil {
+		return h.errorResponse(nil, -32700, "Parse error")
+	}
+
+	var result interface{}
+	var err error
+
+	switch req.Method {
+	case "connect":
+		var params protocol.ConnectParams
+		if err := json.Unmarshal(req.Params, &params); err != nil {
+			return h.errorResponse(req.ID, -32602, "Invalid params")
+		}
+		result, err = h.manager.Connect(params.User, params.Host, params.Port, params.Password, params.KeyPath)
+
+	case "disconnect":
+		var params protocol.DisconnectParams
+		if err := json.Unmarshal(req.Params, &params); err != nil {
+			return h.errorResponse(req.ID, -32602, "Invalid params")
+		}
+		err = h.manager.Disconnect(params.SessionID)
+
+	case "reconnect":
+		var params protocol.ReconnectParams
+		if err := json.Unmarshal(req.Params, &params); err != nil {
+			return h.errorResponse(req.ID, -32602, "Invalid params")
+		}
+		result, err = h.manager.Reconnect(params.SessionID)
+
+	case "exec":
+		var params protocol.ExecParams
+		if err := json.Unmarshal(req.Params, &params); err != nil {
+			return h.errorResponse(req.ID, -32602, "Invalid params")
+		}
+		result, err = h.manager.Exec(params.SessionID, params.Command)
+
+	case "list":
+		result = h.manager.List()
+
+	case "use":
+		var params protocol.UseParams
+		if err := json.Unmarshal(req.Params, &params); err != nil {
+			return h.errorResponse(req.ID, -32602, "Invalid params")
+		}
+		err = h.manager.Use(params.SessionID)
+
+	case "forward":
+		var params protocol.ForwardParams
+		if err := json.Unmarshal(req.Params, &params); err != nil {
+			return h.errorResponse(req.ID, -32602, "Invalid params")
+		}
+		result, err = h.manager.AddForward(params.SessionID, params.Type, params.Local, params.Remote)
+
+	case "forwards":
+		result = h.manager.ListForwards()
+
+	case "forward_close":
+		var params protocol.ForwardCloseParams
+		if err := json.Unmarshal(req.Params, &params); err != nil {
+			return h.errorResponse(req.ID, -32602, "Invalid params")
+		}
+		err = h.manager.CloseForward(params.ForwardID)
+
+	default:
+		return h.errorResponse(req.ID, -32601, fmt.Sprintf("Method not found: %s", req.Method))
+	}
+
+	if err != nil {
+		return h.errorResponse(req.ID, -32000, err.Error())
+	}
+
+	return h.successResponse(req.ID, result)
+}
+
+func (h *Handler) successResponse(id interface{}, result interface{}) ([]byte, error) {
+	resp := protocol.Response{
+		JSONRPC: "2.0",
+		Result:  result,
+		ID:      id,
+	}
+	return json.Marshal(resp)
+}
+
+func (h *Handler) errorResponse(id interface{}, code int, message string) ([]byte, error) {
+	resp := protocol.Response{
+		JSONRPC: "2.0",
+		Error: &protocol.Error{
+			Code:    code,
+			Message: message,
+		},
+		ID: id,
+	}
+	return json.Marshal(resp)
+}
+
+// ServeUnixSocket starts the RPC server on a Unix socket
+func ServeUnixSocket(manager *session.Manager, socketPath string) error {
+	// Remove existing socket file
+	os.Remove(socketPath)
+
+	listener, err := net.Listen("unix", socketPath)
+	if err != nil {
+		return fmt.Errorf("failed to listen on socket: %w", err)
+	}
+
+	// Set socket permissions
+	os.Chmod(socketPath, 0700)
+
+	handler := NewHandler(manager)
+
+	log.Printf("RPC server listening on %s", socketPath)
+
+	for {
+		conn, err := listener.Accept()
+		if err != nil {
+			log.Printf("Accept error: %v", err)
+			continue
+		}
+
+		go handleConnection(conn, handler)
+	}
+}
+
+func handleConnection(conn net.Conn, handler *Handler) {
+	defer conn.Close()
+
+	buf := make([]byte, 4096)
+	for {
+		n, err := conn.Read(buf)
+		if err == io.EOF {
+			return
+		}
+		if err != nil {
+			log.Printf("Read error: %v", err)
+			return
+		}
+
+		data := buf[:n]
+		response, err := handler.Handle(data)
+		if err != nil {
+			log.Printf("Handle error: %v", err)
+			continue
+		}
+
+		// Add newline to flush
+		response = append(response, '\n')
+		_, err = conn.Write(response)
+		if err != nil {
+			log.Printf("Write error: %v", err)
+		}
+	}
+}
+
+// HTTPHandler creates an HTTP handler for testing
+func HTTPHandler(manager *session.Manager) http.Handler {
+	handler := NewHandler(manager)
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if r.Method != http.MethodPost {
+			http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
+			return
+		}
+
+		data, err := io.ReadAll(r.Body)
+		if err != nil {
+			http.Error(w, "Read error", http.StatusBadRequest)
+			return
+		}
+
+		response, err := handler.Handle(data)
+		if err != nil {
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return
+		}
+
+		w.Header().Set("Content-Type", "application/json")
+		w.Write(response)
+	})
+}

package/skill.md

@@ -0,0 +1,96 @@
+# gssh 使用指南
+
+## 快速开始
+
+### 1. 安装
+
+```bash
+git clone <repository-url>
+cd gssh
+go build -o bin/daemon cmd/daemon/main.go
+go build -o bin/gssh cmd/gssh/main.go
+./homebrew/install.sh
+```
+
+### 2. 基本使用
+
+```bash
+# 连接 SSH
+gssh connect -u admin1 -h 192.168.1.100 -P 1234
+
+# 执行命令
+gssh exec "ls -la"
+
+# 列出 session
+gssh list
+```
+
+## 核心概念
+
+- **daemon**: 后台服务，管理 SSH 连接
+- **session**: 一个 SSH 连接会话
+- **默认 session**: 不指定 session ID 时使用的会话
+
+## 命令列表
+
+| 命令 | 说明 |
+|------|------|
+| `connect` | 建立新 SSH 连接 |
+| `exec` | 执行命令 |
+| `list` | 列出所有 session |
+| `use` | 切换默认 session |
+| `disconnect` | 断开 session |
+| `reconnect` | 重连 session |
+| `forward` | 端口转发 |
+| `forwards` | 列出转发 |
+| `forward-close` | 关闭转发 |
+
+## 典型工作流
+
+```bash
+# 1. 连接远程主机
+gssh connect -u admin -h 192.168.1.100 -P password
+
+# 2. 执行命令（使用默认 session）
+gssh exec "pwd"
+gssh exec "uname -a"
+
+# 3. 断开连接
+gssh disconnect
+
+# 4. 重连
+gssh reconnect -s <session-id>
+```
+
+## SSH 密钥认证
+
+```bash
+# 使用密钥连接
+gssh connect -u admin -h 192.168.1.100 -i ~/.ssh/id_rsa
+```
+
+## 端口转发
+
+```bash
+# 本地转发：localhost:8080 -> remote:80
+gssh forward -l 8080 -r 80
+```
+
+## 服务管理
+
+```bash
+# 启动
+brew services start gssh
+
+# 停止
+brew services stop gssh
+
+# 状态
+brew services list
+```
+
+## 注意事项
+
+- Agent 使用场景：无状态，每次请求独立完成
+- sudo 命令：需要配置 passwordless sudo 或使用密钥认证
+- Session 状态由 daemon 维护，CLI 只负责发送请求