package main

import (
	"bufio"
	"flag"
	"fmt"
	"golang.org/x/crypto/ssh"
	"io"
	"log"
	"os"
)

var (
	user = flag.String("u", "lab_34_2018", "User name")
	pass   = flag.String("pass", "0000", "User password")
	host = flag.String("h", "localhost", "Host")
	port = flag.Int("p", 2222, "Port")
)


func main() {
	flag.Parse()

	//задаю конфиг
	config := &ssh.ClientConfig{
		User: *user,
		Auth: []ssh.AuthMethod{
			ssh.Password(*pass),
		},
	}
	config.HostKeyCallback = ssh.InsecureIgnoreHostKey()

	//задаю адрес
	addr := fmt.Sprintf("%s:%d", *host, *port)
	client, err := ssh.Dial("tcp", addr, config)
	if err != nil {
		panic(err)
	}

	//создаю сессию
	session, err := client.NewSession()
	if err != nil {
		panic(err)
	}

	//настройка обмена сообщения
	modes := ssh.TerminalModes{
		ssh.ECHO:  0,
		ssh.IGNCR: 1,
	}
	stdin, err := session.StdinPipe()
	if err != nil {
		fmt.Errorf("Unable to setup stdin for session: %v", err)
	}
	go io.Copy(stdin, os.Stdin)

	stdout, err := session.StdoutPipe()
	if err != nil {
		fmt.Errorf("Unable to setup stdout for session: %v", err)
	}
	go io.Copy(os.Stdout, stdout)

	stderr, err := session.StderrPipe()
	if err != nil {
		fmt.Errorf("Unable to setup stderr for session: %v", err)
	}
	go io.Copy(os.Stderr, stderr)


	// Создаю вертуальный терминал для работы команд
	if err := session.RequestPty("xterm", 80, 40, modes); err != nil {
		log.Fatalf("request for pseudo terminal failed: %s", err)
	}

	// Start remote shell
	if err := session.Shell(); err != nil {
		log.Fatalf("failed to start shell: %s", err)
	}

	// Handle control + C
	//c := make(chan os.Signal, 1)
	//signal.Notify(c, os.Interrupt)
	//go func() {
	//  for {
	//      <-c
	//      fmt.Println("^C")
	//      fmt.Fprint(stdin, "\n")
	//      //fmt.Fprint(in, '\t')
	//  }
	//}()



	// Считывание и передача команды
	for {
		reader := bufio.NewReader(os.Stdin)
		str, _ := reader.ReadString('\n')
		fmt.Fprint(stdin, str)
	}

}