Update main.go

Update code and add Pty session to script to better interact with arista eos.

Author: twr14152

arista/ssh_client/main.go

@@ -12,30 +12,37 @@ import (
 func commands() map[string][]string {
 	var count int
 	var host string
+
 	fmt.Print("Number of hosts: ")
 	fmt.Scanf("%d", &count)
+
 	hostCmds := make(map[string][]string)
-	fmt.Printf("\nEnter host and ssh port being used. (Ex. x.x.x.x:22 or hostname:22)\n\n\n")
+
+	fmt.Printf("\nEnter host and ssh port being used. (Ex. x.x.x.x:22 or hostname:22)\n\n")
+
 	for i := 1; i <= count; i++ {
 		fmt.Print("Enter host: ")
 		fmt.Scanf("%s", &host)
 		fmt.Println()
+
 		reader := bufio.NewReader(os.Stdin)
 		fmt.Print("cmds: ")
 		cmds, err := reader.ReadString('\n')
 		if err != nil {
-			log.Fatal("You screwed something up: ", err)
+			log.Fatal(err)
 		}
+
 		s := strings.Split(cmds, ",")
-		fmt.Println()
 		hostCmds[host] = s
 	}
+
 	return hostCmds
 }
 
 func main() {
-	user := "arista"
-	pass := "arista"
+	user := "admin"
+	pass := "admin"
+
 	hostData := commands()
 
 	interactiveAuth := ssh.KeyboardInteractive(
@@ -44,44 +51,63 @@ func main() {
 			for i := range answers {
 				answers[i] = pass
 			}
-
 			return answers, nil
 		},
 	)
 
-	for host, commands := range hostData {
+	for host, cmds := range hostData {
+		fmt.Println("++++++++++++++++++++++++++++++++")
+		fmt.Println("Connected to:", host)
+		fmt.Println("++++++++++++++++++++++++++++++++")
+
 		config := &ssh.ClientConfig{
-			HostKeyCallback: ssh.InsecureIgnoreHostKey(),
 			User:            user,
 			Auth:            []ssh.AuthMethod{interactiveAuth},
+			HostKeyCallback: ssh.InsecureIgnoreHostKey(),
 		}
-		fmt.Println("++++++++++++++++++++++++++++++++")
-		fmt.Println("Connected to: ", host)
-		fmt.Println("++++++++++++++++++++++++++++++++")
+
 		conn, err := ssh.Dial("tcp", host, config)
 		if err != nil {
-			log.Fatal("Failed to dial: ", err)
+			log.Fatal("Failed to dial:", err)
 		}
 		defer conn.Close()
+
 		sess, err := conn.NewSession()
 		if err != nil {
-			log.Fatal("Failed to create session: ", err)
+			log.Fatal("Failed to create session:", err)
 		}
 		defer sess.Close()
-		stdin, err := sess.StdinPipe()
-		if err != nil {
-			log.Fatal("Failed to connect to remote devices stdin: ", err)
+		//
+        // Updated section
+		// EOS requires a PTY for CLI 
+		//
+		modes := ssh.TerminalModes{
+			ssh.ECHO: 0,
 		}
-		sess.Stdout = os.Stdout
-		sess.Stderr = os.Stderr
-		sess.Shell()
-		fmt.Fprintf(stdin, "term len 0\n")
-		for _, v := range commands {
-			fmt.Fprintf(stdin, "%s\n", v)
+
+		if err := sess.RequestPty("vt100", 80, 40, modes); err != nil {
+			log.Fatal("PTY request failed:", err)
 		}
-		stdin.Close()
-		sess.Wait()
+
+		// Build buffer for commands
+		var cmdBuffer strings.Builder
+		cmdBuffer.WriteString("term len 0\n")
+
+		for _, v := range cmds {
+			cmd := strings.TrimSpace(v)
+			if cmd == "" {
+				continue
+			}
+			cmdBuffer.WriteString(cmd + "\n")
+		}
+
+		// Run commands and print output
+		output, err := sess.CombinedOutput(cmdBuffer.String())
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		fmt.Print(string(output))
 	}
 }
 
-