package ssh

import (
	"fmt"
	"io/ioutil"

	"golang.org/x/crypto/ssh"
)

type SSHConfig struct {
	User       string
	Host       string
	Port       string
	Password   string
	PrivateKey string
}

func ConnectSSH(config SSHConfig) error {
	var auth ssh.AuthMethod

	if config.PrivateKey != "" {
		key, err := ioutil.ReadFile(config.PrivateKey)
		if err != nil {
			return fmt.Errorf("failed to read private key: %v", err)
		}
		signer, err := ssh.ParsePrivateKey(key)
		if err != nil {
			return fmt.Errorf("failed to parse private key: %v", err)
		}
		auth = ssh.PublicKeys(signer)
	} else {
		auth = ssh.Password(config.Password)
	}

	clientConfig := &ssh.ClientConfig{
		User:            config.User,
		Auth:            []ssh.AuthMethod{auth},
		HostKeyCallback: ssh.InsecureIgnoreHostKey(), // Use a proper callback for production!
	}

	client, err := ssh.Dial("tcp", fmt.Sprintf("%s:%s", config.Host, config.Port), clientConfig)
	if err != nil {
		return fmt.Errorf("failed to connect to SSH: %v", err)
	}
	defer client.Close()

	fmt.Println("Connected to SSH successfully!")
	return nil
}