logman/ssh.go

package main

import (
	"fmt"

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

// modes := ssh.TerminalModes{
// 	ssh.ECHO: 0,
// 	ssh.TTY_OP_ISPEED: 14400,
// 	ssh.TTY_OP_OSPEED: 14400,
// }

type SSHConnection struct {
	Username string
	Password string
	Port     int
	Hosts    []string
}

type SSHClients []*ssh.Client

// TODO: I've got two slices of the same type. To take one slice out, some refactoring is needed
type SSHSessions []*ssh.Session

type SSHTarFile []*ssh.Session

type SSHPush []*ssh.Client

type SSHSFTP []string

type SSHSuccess []error

func initializeConnection(config Configuration) (*SSHConnection, *ssh.ClientConfig) {
	sshConn := &SSHConnection{
		Username: config.Username,
		Password: config.Password,
		Port:     config.Port,
		Hosts:    config.Hosts,
	}

	sshConfig := &ssh.ClientConfig{
		User: sshConn.Username,
		Auth: []ssh.AuthMethod{
			ssh.Password(sshConn.Password),
		},
		HostKeyCallback: sshConn.getHostKeys(),
	}

	return sshConn, sshConfig
}

func (s SSHConnection) getHostKeys() ssh.HostKeyCallback {
	return ssh.InsecureIgnoreHostKey()
}

func (s SSHConnection) dialConnection(sshConfig *ssh.ClientConfig) SSHClients {
	clientConn := SSHClients{}

	for _, j := range s.Hosts {
		hostPort := fmt.Sprintf("%s:22", j)
		connection, err := ssh.Dial("tcp", hostPort, sshConfig)

		if err != nil {
			fmt.Errorf("Can't connect", err)
		}

		clientConn = append(clientConn, connection)
	}

	return clientConn
}

func (s SSHConnection) openSession(client SSHClients) SSHSessions {
	clientSessions := SSHSessions{}

	for _, j := range client {
		session, err := j.NewSession()

		if err != nil {
			fmt.Errorf("Can't open session", err)
		}

		clientSessions = append(clientSessions, session)
	}

	return clientSessions
}

func (s SSHConnection) executeSFTP(execute SSHClients) SSHSFTP {
	// execute order 66 lol
	sftp := SSHSFTP{}

	for _, j := range execute {
		// TODO: this is just a placeholder, change to the actual tarring executable
		getFile(j)

		err := gzipit("/root/"+filename, ".")

		if err != nil {
			fmt.Errorf("Cannot gzip file(s)", err)
		}

	}

	return sftp
}

func (s SSHConnection) gzipItUp() {
	// TODO: placeholder function??
}