Initial commit

internal/ssh/client.go

@@ -0,0 +1,89 @@
+package ssh
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/pkg/sftp"
+	"golang.org/x/crypto/ssh"
+)
+
+// ClientConfig holds the configuration for SSH connections
+type ClientConfig struct {
+	Host     string
+	Port     string
+	Username string
+	Password string
+	Timeout  time.Duration
+}
+
+// NewClientConfig creates a new SSH client configuration
+func NewClientConfig(host, port, username, password string) *ClientConfig {
+	return &ClientConfig{
+		Host:     host,
+		Port:     port,
+		Username: username,
+		Password: password,
+		Timeout:  30 * time.Second,
+	}
+}
+
+// CreateSSHClient creates and returns an SSH client using the provided configuration
+func CreateSSHClient(config *ClientConfig) (*ssh.Client, error) {
+	if config == nil {
+		return nil, fmt.Errorf("SSH config cannot be nil")
+	}
+
+	sshConfig := &ssh.ClientConfig{
+		User: config.Username,
+		Auth: []ssh.AuthMethod{
+			ssh.Password(config.Password),
+		},
+		HostKeyCallback: ssh.InsecureIgnoreHostKey(),
+		Timeout:         config.Timeout,
+	}
+
+	client, err := ssh.Dial("tcp", config.Host+":"+config.Port, sshConfig)
+	if err != nil {
+		return nil, fmt.Errorf("failed to connect to SSH server: %w", err)
+	}
+
+	return client, nil
+}
+
+// CreateSFTPClient creates and returns an SFTP client using the provided configuration
+func CreateSFTPClient(config *ClientConfig) (*sftp.Client, error) {
+	if config == nil {
+		return nil, fmt.Errorf("SFTP config cannot be nil")
+	}
+
+	// Create SSH client
+	sshClient, err := CreateSSHClient(config)
+	if err != nil {
+		return nil, err
+	}
+
+	// Create SFTP client
+	sftpClient, err := sftp.NewClient(sshClient)
+	if err != nil {
+		sshClient.Close() // Close the SSH client in case of failure
+		return nil, fmt.Errorf("failed to create SFTP client: %w", err)
+	}
+
+	return sftpClient, nil
+}
+
+// ExecuteCommand executes a command on the remote server
+func ExecuteCommand(client *ssh.Client, command string) error {
+	session, err := client.NewSession()
+	if err != nil {
+		return fmt.Errorf("session creation failed: %w", err)
+	}
+	defer session.Close()
+
+	if err := session.Run(command); err != nil {
+		return fmt.Errorf("command '%s' failed: %w", command, err)
+	}
+
+	return nil
+}