package main

import (
	"fmt"
	"more_ssh/myssh01"
	"more_ssh/ssh"
	"more_ssh/util"

	"github.com/gin-gonic/gin"
)

func main() {
	r := gin.Default()
	r.Use(util.Cors()) //解决跨域问题
	r.GET("/myssh", myssh01.RunWebSSH)
	r.Run() //默认8080端口

}

package myssh01

import (
	"bytes"
	"fmt"
	"io"
	"net/http"
	"sync"
	"time"

	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
	"golang.org/x/crypto/ssh"
)

// 定义一个结构体 方便保存各种连接信息
type MySSH struct {
	Websocket *websocket.Conn
	Stdin     io.WriteCloser
	Stdout    *wsBufferWriter
	Session   *ssh.Session
}

// 定义一个wsBufferWriter 并且写入时候加锁 防止stdout跟stderr同时写入
type wsBufferWriter struct {
	buffer bytes.Buffer
	mu     sync.Mutex
}

//定义write方法， 防止stdout跟stderr同时写入
func (w *wsBufferWriter) Write(p []byte) (int, error) {
	w.mu.Lock()
	defer w.mu.Unlock()
	return w.buffer.Write(p)
}

// 程序入口
func RunWebSSH(c *gin.Context) {
	mySSH := &MySSH{}

	// 1. 升级请求websocket
	upGrader := websocket.Upgrader{
		ReadBufferSize:  1024,
		WriteBufferSize: 1024 * 1024 * 10,
		CheckOrigin: func(r *http.Request) bool {
			return true
		},
		Subprotocols: []string{"webssh"},
	}

	webcon, err := upGrader.Upgrade(c.Writer, c.Request, nil)
	if err != nil {
		fmt.Println("升级http 为websoket失败：", err)
	}
	mySSH.Websocket = webcon // 将websocket连接保存到对象中

	// 创建一个ssh的配置
	config := &ssh.ClientConfig{
		Timeout:         time.Second * 10, //ssh 连接time out 时间一秒钟, 如果ssh验证错误 会在一秒内返回
		User:            "root",
		HostKeyCallback: ssh.InsecureIgnoreHostKey(), //这个可以， 但是不够安全
		//HostKeyCallback: hostKeyCallBackFunc(h.Host),
		Auth: []ssh.AuthMethod{ssh.Password("more@123")},
	}

	// 创建一个客户端
	sshClient, err := ssh.Dial("tcp", "162.14.109.53:22", config)
	if err != nil {
		fmt.Println(err)
		return
	}

	session, err := sshClient.NewSession()
	if err != nil {
		fmt.Println(err)
		return
	}
	mySSH.Session = session

	// 保存输入流
	mySSH.Stdin, err = session.StdinPipe()
	if err != nil {
		fmt.Println(err)
		return
	}

	//保存ssh输出流
	sshOut := new(wsBufferWriter)
	session.Stdout = sshOut
	session.Stderr = sshOut
	mySSH.Stdout = sshOut

	modes := ssh.TerminalModes{
		ssh.ECHO:          1,     // disable echo
		ssh.TTY_OP_ISPEED: 14400, // input speed = 14.4kbaud
		ssh.TTY_OP_OSPEED: 14400, // output speed = 14.4kbaud
	}
	// Request pseudo terminal
	if err := session.RequestPty("xterm", 30, 120, modes); err != nil {
		fmt.Println("绑定pty失败：", err)
		return
	}

	session.Shell()

	//执行远程命令
	go Send2SSH(mySSH)
	go Send2Web(mySSH)

}

// 读取websocket数据，发送到ssh输入流中
func Send2SSH(mySSh *MySSH) {
	for {
		//read websocket msg  需要通过msgType 判断是传输类型
		_, wsData, err := mySSh.Websocket.ReadMessage()
		if err != nil {
			fmt.Println("读取websocket数据失败：", err)
			return
		}
		_, err = mySSh.Stdin.Write(wsData)
		if err != nil {
			fmt.Println("ssh发送数据失败：", err)
		}
		// fmt.Println("ssh发送数据：", string(wsData))
	}

}

// 读取ssh输出，发送到websocket中
func Send2Web(mySSh *MySSH) {
	for {
		if mySSh.Stdout.buffer.Len() > 0 {
			err := mySSh.Websocket.WriteMessage(websocket.TextMessage, mySSh.Stdout.buffer.Bytes())
			fmt.Printf(string(mySSh.Stdout.buffer.Bytes()))
			if err != nil {
				fmt.Println("websocket发送数据失败：", err)
			}
			mySSh.Stdout.buffer.Reset() //读完清空
		}

	}
}

<template>
    <div>
        <div ref="terminalBox" style="height: 100vh;"></div>
    </div>
</template>

<script setup>
import 'xterm/css/xterm.css'
import { Terminal } from 'xterm'
import { FitAddon } from 'xterm-addon-fit'
import { ref, onMounted } from 'vue'
import { useMessage } from "naive-ui";

const message = useMessage();
let terminalBox = ref(null)
let term
let socket

onMounted(() => {
    //创建一个客户端
    term = new Terminal({
        rendererType: 'canvas', //使用这个能解决vim不显示或乱码
        cursorBlink: true,
        cursorStyle: "bar",
    })
    // term.write
    // 将客户端挂载到dom上
    const fitAddon = new FitAddon()
    term.loadAddon(fitAddon)
    term.open(terminalBox.value)
    fitAddon.fit()

    // 创建socket连接
    term.write('正在连接...\r\n');
    socket = new WebSocket('ws://127.0.0.1:8080/myssh')

    socket.binaryType = "arraybuffer";

    // 打开socket监听事件的方法
    socket.onopen = function () {
        fitAddon.fit()
        term.onData(function (data) {
            // socket.send(JSON.stringify({ type: "stdin", data: data }))
            socket.send(data)
            console.log(data)
        });
        message.success("会话成功连接！")
    }
    socket.onclose = function () {
        term.writeln('连接关闭');
    }
    socket.onerror = function (err) {
        // console.log(err)
        term.writeln('读取数据异常：', err);
    }
    // 接收数据
    socket.onmessage = function (recv) {
        try {
            term.write(recv.data)
        } catch (e) {
            console.log('unsupport data', recv.data)
        }
    }

    window.addEventListener("resize", () => {
        fitAddon.fit()
    }, false)

})


</script>

<style scoped>
.upload {
    min-height: 100px;
}
</style>