feat: ssh as shell command (#67)

* feat(ssh): add action to run shell commands over a remote server

Signed-off-by: mudler <mudler@localai.io>

* fix(github): correctly get content

Signed-off-by: mudler <mudler@localai.io>

---------

Signed-off-by: mudler <mudler@localai.io>

services/actions.go

@@ -33,6 +33,7 @@ const (
 	ActionGenerateImage                  = "generate_image"
 	ActionCounter                        = "counter"
 	ActionCallAgents                     = "call_agents"
+	ActionShellcommand                   = "shell-command"
 )
 
 var AvailableActions = []string{
@@ -55,6 +56,7 @@ var AvailableActions = []string{
 	ActionTwitterPost,
 	ActionCounter,
 	ActionCallAgents,
+	ActionShellcommand,
 }
 
 func Actions(a *state.AgentConfig) func(ctx context.Context, pool *state.AgentPool) []agent.Action {
@@ -112,6 +114,8 @@ func Actions(a *state.AgentConfig) func(ctx context.Context, pool *state.AgentPo
 				allActions = append(allActions, actions.NewCounter(config))
 			case ActionCallAgents:
 				allActions = append(allActions, actions.NewCallAgent(config, pool))
+			case ActionShellcommand:
+				allActions = append(allActions, actions.NewShell(config))
 			}
 		}
 

services/actions/githubrepositorygetcontent.go

@@ -60,9 +60,9 @@ func (g *GithubRepositoryGetContent) Run(ctx context.Context, params action.Acti
 		return action.ActionResult{Result: resultString}, err
 	}
 
-	var content string
+	content, err := fileContent.GetContent()
-	if fileContent.Content != nil {
+	if err != nil {
-		content = *fileContent.Content
+		return action.ActionResult{}, err
 	}
 
 	return action.ActionResult{Result: fmt.Sprintf("File %s\nContent:%s\n", result.Path, content)}, err

services/actions/githubrepositoryreadme.go

@@ -43,9 +43,9 @@ func (g *GithubRepositoryREADME) Run(ctx context.Context, params action.ActionPa
 		return action.ActionResult{Result: resultString}, err
 	}
 
-	var content string
+	content, err := fileContent.GetContent()
-	if fileContent.Content != nil {
+	if err != nil {
-		content = *fileContent.Content
+		return action.ActionResult{}, err
 	}
 
 	return action.ActionResult{Result: content}, err

services/actions/shell.go

@@ -0,0 +1,103 @@
+package actions
+
+import (
+	"context"
+	"fmt"
+	"log"
+
+	"github.com/mudler/LocalAgent/core/action"
+	"github.com/sashabaranov/go-openai/jsonschema"
+	"golang.org/x/crypto/ssh"
+)
+
+func NewShell(config map[string]string) *ShellAction {
+	return &ShellAction{
+		privateKey: config["privateKey"],
+	}
+}
+
+type ShellAction struct {
+	privateKey string
+}
+
+func (a *ShellAction) Run(ctx context.Context, params action.ActionParams) (action.ActionResult, error) {
+	result := struct {
+		Command string `json:"command"`
+		Host    string `json:"host"`
+		User    string `json:"user"`
+	}{}
+	err := params.Unmarshal(&result)
+	if err != nil {
+		fmt.Printf("error: %v", err)
+
+		return action.ActionResult{}, err
+	}
+
+	output, err := sshCommand(a.privateKey, result.Command, result.User, result.Host)
+	if err != nil {
+		return action.ActionResult{}, err
+	}
+
+	return action.ActionResult{Result: output}, nil
+}
+
+func (a *ShellAction) Definition() action.ActionDefinition {
+	return action.ActionDefinition{
+		Name:        "shell",
+		Description: "Run a shell command on a remote server.",
+		Properties: map[string]jsonschema.Definition{
+			"command": {
+				Type:        jsonschema.String,
+				Description: "The command to run on the remote server.",
+			},
+			"host": {
+				Type:        jsonschema.String,
+				Description: "The host of the remote server. e.g. ip:port",
+			},
+			"user": {
+				Type:        jsonschema.String,
+				Description: "The user to connect to the remote server.",
+			},
+		},
+		Required: []string{"command", "host", "user"},
+	}
+}
+
+func sshCommand(privateKey, command, user, host string) (string, error) {
+	// Create signer from private key string
+	key, err := ssh.ParsePrivateKey([]byte(privateKey))
+	if err != nil {
+		log.Fatalf("failed to parse private key: %v", err)
+	}
+
+	// SSH client configuration
+	config := &ssh.ClientConfig{
+		User: user,
+		Auth: []ssh.AuthMethod{
+			ssh.PublicKeys(key),
+		},
+		HostKeyCallback: ssh.InsecureIgnoreHostKey(),
+	}
+
+	// Connect to SSH server
+	client, err := ssh.Dial("tcp", host, config)
+	if err != nil {
+		return "", fmt.Errorf("failed to dial: %v", err)
+	}
+	defer client.Close()
+
+	// Open a new session
+	session, err := client.NewSession()
+	if err != nil {
+		return "", fmt.Errorf("failed to create session: %v", err)
+	}
+	defer session.Close()
+
+	// Run a command
+	output, err := session.CombinedOutput(command)
+	if err != nil {
+		return "", fmt.Errorf("failed to run: %v", err)
+	}
+
+	return string(output), nil
+}