Initial commit
This commit is contained in:
89
internal/ssh/client.go
Normal file
89
internal/ssh/client.go
Normal file
@ -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
|
||||
}
|
Reference in New Issue
Block a user