feature: Raw connection handling

Author: dogukanoksuz

app/routes/index.go

@@ -6,7 +6,21 @@ import (
 )
 
 func Install(app *fiber.App) {
-	app.Post("/extensionRunner", handlers.ExtensionRunner)
-	app.Post("/commandRunner", handlers.CommandRunner)
+	// extension
+	app.Post("/run", handlers.ExtensionRunner)
+
+	// command
+	app.Post("/runCommand", handlers.CommandRunner)
+	app.Post("/runOutsideCommand", handlers.OutsideCommandRunner)
+
+	// tunnel
 	app.Post("/openTunnel", handlers.OpenTunnel)
+	app.Post("/keepTunnelAlive", handlers.KeepTunnelAlive)
+
+	// file
+	app.Post("/getFile", handlers.GetFile)
+	app.Post("/putFile", handlers.PutFile)
+
+	// script
+	app.Post("/runScript", handlers.ScriptRunner)
 }

internal/bridge/connection.go

@@ -28,6 +28,19 @@ func (p *Pool) Set(userID, serverID string, session *Session) {
 	Connections[userID+serverID] = session
 }
 
+func (p *Pool) GetRaw(userID, remoteHost, username string) (*Session, error) {
+	if conn, ok := Connections[userID+remoteHost+username]; ok {
+		conn.LastConnection = time.Now()
+		return conn, nil
+	} else {
+		return nil, errors.New("connection does not exist")
+	}
+}
+
+func (p *Pool) SetRaw(userID, remoteHost, username string, session *Session) {
+	Connections[userID+remoteHost+username] = session
+}
+
 func (p *Pool) Delete(key string) {
 	mutex.Lock()
 	defer mutex.Unlock()