Merge pull request #66 from sonroyaalmerol/remove-passthrough-proxy

refactoring sftp servers and ensure graceful exits
sonroyaalmerol · Jan 13, 2025 · d7798c1 · d7798c1
2 parents 24218dd + 06abe6b
commit d7798c1
Show file tree

Hide file tree

Showing 5 changed files with 434 additions and 252 deletions.
diff --git a/cmd/windows_agent/service.go b/cmd/windows_agent/service.go
@@ -186,7 +186,7 @@ func (p *agentService) run() {
 		for _, drive := range drives {
 			driveLetters = append(driveLetters, drive)
 
-			err := sftp.InitializeSFTPConfig(drive)
+			err := sftp.InitializeSFTPConfig(p.ctx, drive)
 			if err != nil {
 				logger.Errorf("Unable to initialize SFTP config: %v", err)
 				return

diff --git a/internal/agent/controllers/ws.go b/internal/agent/controllers/ws.go
@@ -16,13 +16,7 @@ import (
 var sftpSessions sync.Map
 
 func WSHandler(ctx context.Context, c *websocket.Conn, m websockets.Message, infoChan chan string, errChan chan string) {
-	if m.Type == "ping" {
-		response := websockets.Message{
-			Type:    "ping",
-			Content: "pong",
-		}
-		c.WriteJSON(response)
-	} else if m.Type == "backup_start" {
+	if m.Type == "backup_start" {
 		infoChan <- fmt.Sprintf("Received backup request for drive %s.", m.Content)
 
 		snapshot, err := snapshots.Snapshot(m.Content)