draft: terminal

backend/go.mod

@@ -20,4 +20,6 @@ require (
 	github.com/aws/aws-sdk-go-v2/service/internal/checksum v1.3.18 // indirect
 	github.com/aws/aws-sdk-go-v2/service/internal/presigned-url v1.11.18 // indirect
 	github.com/aws/aws-sdk-go-v2/service/internal/s3shared v1.17.16 // indirect
+	github.com/coder/websocket v1.8.12 // indirect
+	golang.org/x/net v0.28.0 // indirect
 )

backend/go.sum

@@ -22,6 +22,8 @@ github.com/aws/aws-sdk-go-v2/service/s3 v1.59.0 h1:Cso4Ev/XauMVsbwdhYEoxg8rxZWw4
 github.com/aws/aws-sdk-go-v2/service/s3 v1.59.0/go.mod h1:BSPI0EfnYUuNHPS0uqIo5VrRwzie+Fp+YhQOUs16sKI=
 github.com/aws/smithy-go v1.20.4 h1:2HK1zBdPgRbjFOHlfeQZfpC4r72MOb9bZkiFwggKO+4=
 github.com/aws/smithy-go v1.20.4/go.mod h1:irrKGvNn1InZwb2d7fkIRNucdfwR8R+Ts3wxYa/cJHg=
+github.com/coder/websocket v1.8.12 h1:5bUXkEPPIbewrnkU8LTCLVaxi4N4J8ahufH2vlo4NAo=
+github.com/coder/websocket v1.8.12/go.mod h1:LNVeNrXQZfe5qhS9ALED3uA+l5pPqvwXg3CKoDBB2gs=
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
@@ -40,6 +42,8 @@ github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO
 github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
 github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
 github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+golang.org/x/net v0.28.0 h1:a9JDOJc5GMUJ0+UDqmLT86WiEy7iWyIhz8gz8E4e5hE=
+golang.org/x/net v0.28.0/go.mod h1:yqtgsTWOOnlGLG9GFRrK3++bGOUEkNBoHZc8MEDWPNg=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=

backend/main.go

@@ -20,6 +20,8 @@ func main() {
 	}
 
 	http.Handle("/api/", http.StripPrefix("/api", router.HandleApiRouter()))
+	http.Handle("/ws/", http.StripPrefix("/ws", router.HandleWebsocket()))
+
 	ui.ServeUI()
 
 	host := utils.GetEnv("HOST", "0.0.0.0")

backend/router/router.go

@@ -1,6 +1,8 @@
 package router
 
-import "net/http"
+import (
+	"net/http"
+)
 
 func HandleApiRouter() *http.ServeMux {
 	router := http.NewServeMux()
@@ -21,3 +23,11 @@ func HandleApiRouter() *http.ServeMux {
 
 	return router
 }
+
+func HandleWebsocket() *http.ServeMux {
+	router := http.NewServeMux()
+
+	router.HandleFunc("/terminal", TerminalHandler)
+
+	return router
+}

backend/router/terminal.go

@@ -0,0 +1,114 @@
+package router
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"os"
+	"os/exec"
+
+	"github.com/coder/websocket"
+)
+
+type Terminal struct {
+	ws         *websocket.Conn
+	remoteAddr string
+	cmd        *exec.Cmd
+}
+
+func (c *Terminal) NewSession(w http.ResponseWriter, r *http.Request) {
+	cwd, err := os.Getwd()
+	if err != nil {
+		fmt.Printf("Failed to get current working directory: %v\n", err)
+	}
+
+	cmd := &exec.Cmd{
+		// Path: "/bin/bash",
+		Path: "/usr/bin/fish",
+		Dir:  cwd,
+	}
+	c.cmd = cmd
+
+	stdin, err := cmd.StdinPipe()
+	if err != nil {
+		fmt.Printf("Failed to create stdin pipe: %v\n", err)
+		return
+	}
+
+	stdout, err := cmd.StdoutPipe()
+	if err != nil {
+		fmt.Printf("Failed to create stdout pipe: %v\n", err)
+		return
+	}
+	cmd.Stderr = cmd.Stdout
+	cmd.Stdout = os.Stdout
+
+	if err := cmd.Start(); err != nil {
+		fmt.Printf("Failed to start command: %v\n", err)
+		return
+	}
+
+	go func() {
+		// if err := cmd.Wait(); err != nil {
+		// 	fmt.Printf("Shell exited with error: %v\n", err)
+		// }
+		// c.ws.CloseNow()
+	}()
+
+	go func() {
+		buf := make([]byte, 1024)
+		for {
+			n, err := stdout.Read(buf)
+			if err != nil {
+				return
+			}
+
+			fmt.Printf("Data from stdout: %s\n", string(buf[:n]))
+
+			err = c.ws.Write(context.Background(), websocket.MessageText, buf[:n])
+			if err != nil {
+				fmt.Printf("Failed to write data to websocket: %v\n", err)
+				return
+			}
+
+		}
+	}()
+
+	for {
+		_, data, err := c.ws.Read(context.Background())
+		if err != nil {
+			return
+		}
+
+		fmt.Printf("Data from websocket: %s\n", string(data))
+
+		if _, err := stdin.Write(data); err != nil {
+			fmt.Printf("Failed to write data to stdin: %v\n", err)
+			continue
+		}
+	}
+}
+
+func (c *Terminal) Close() {
+	if c.cmd != nil {
+		c.cmd.Process.Kill()
+	}
+
+	fmt.Println("Terminal session closed:", c.remoteAddr)
+	c.ws.Close(websocket.StatusInternalError, "the connection is closed")
+}
+
+func TerminalHandler(w http.ResponseWriter, r *http.Request) {
+	fmt.Printf("New terminal session: %s\n", r.RemoteAddr)
+
+	c, err := websocket.Accept(w, r, nil)
+	if err != nil {
+		fmt.Printf("Failed to accept websocket: %v\n", err)
+		return
+	}
+
+	term := &Terminal{ws: c, remoteAddr: r.RemoteAddr}
+	defer term.Close()
+
+	term.NewSession(w, r)
+}