package ssh

import (
	"context"
	"gitee.com/kristas/booting-go/framework/logging"
	"golang.org/x/crypto/ssh"
	"net"
)

type Config struct {
	IP       string `json:"ip" yaml:"ip"`
	Username string `json:"username" yaml:"username"`
	Password string `json:"password" yaml:"password"`
}

func BuildSSHClient(config Config) *Client {
	client, err := DialWithPasswd(config.IP, config.Username, config.Password)
	if err != nil {
		panic(err)
	}
	//sshSuccessEcho, err := client.Cmd(`echo "SSH login success!"`).Output()
	//if err != nil {
	//	panic(err)
	//}
	//log.Println(string(sshSuccessEcho))
	return client
}

var log logging.Logger

type Dialer struct {
	Client *ssh.Client
	_      *context.Context
}

func NewDialer(client *Client) *Dialer {
	return &Dialer{Client: client.Client}
}

func (r *Dialer) Dial(context context.Context, addr string) (net.Conn, error) {
	log = logging.GetLog()
	log.Infof("[SSH] Dial [%s]", addr)
	return r.Client.Dial("tcp", addr)
}