package main

import (
	"bytes"
	"fmt"
	"github.com/ScriptRock/crypto/ssh"
	"time"
	"log"
)

func main() {

	credential := map[string]interface{}{
		"username": "xyz",
		"password": "xyz",
		"hostname": "xyz",
		"port": "22",
	}

	config := &ssh.ClientConfig{
		User: credential["username"].(string),
		Auth: []ssh.AuthMethod{
			ssh.Password(credential["password"].(string)),
		},
		HostKeyCallback: ssh.InsecureIgnoreHostKey(),
		Config: ssh.Config{
			Ciphers: ssh.AllSupportedCiphers(),
		},
	}

	client, err := ssh.Dial("tcp", credential["hostname"].(string)+":"+credential["port"].(string), config)
	if err != nil {
		log.Fatal(err)
	}
	defer client.Close()

	sess, err := client.NewSession()
	if err != nil {
		log.Fatal("Failed to create session: ", err)
	}
	defer sess.Close()

	stdin, err := sess.StdinPipe()
	if err != nil {
		log.Fatal(err)
	}

	var b bytes.Buffer
	sess.Stdout = &b
	sess.Stderr = &b

	err = sess.Shell()
	if err != nil {
		log.Fatal(err)
	}

	commands := []string{
		"cli",
		"show cli",
		"exit",
		"exit",
	}

	d := time.Now()

	for _, cmd := range commands {

		_, err = fmt.Fprintf(stdin, "%s\n", cmd)

		if err != nil {
			log.Fatal(err)
		}
	}

	err = sess.Wait()

	d = time.Now()

	fmt.Println(d)

	if err != nil {
		log.Fatal(err)
	}
	fmt.Println(b.String())

}