remove the snapshot to file loop

Author: hugodutka

cmd/server/server.go

@@ -76,12 +76,11 @@ var ServerCmd = &cobra.Command{
 			logger.Error("Failed to parse agent type", "error", err)
 			os.Exit(1)
 		}
-		process, cleanup, err := httpapi.SetupProcess(ctx, agent, argsToPass[1:]...)
+		process, err := httpapi.SetupProcess(ctx, agent, argsToPass[1:]...)
 		if err != nil {
 			logger.Error("Failed to setup process", "error", err)
 			os.Exit(1)
 		}
-		defer cleanup()
 		srv := httpapi.NewServer(ctx, agentType, process, port)
 		logger.Info("Starting server on port", "port", port)
 		go func() {

lib/httpapi/setup.go

@@ -5,7 +5,6 @@ import (
 	"fmt"
 	"os"
 	"os/signal"
-	"path/filepath"
 	"strings"
 	"syscall"
 	"time"
@@ -14,7 +13,7 @@ import (
 	"github.com/coder/agentapi/lib/termexec"
 )
 
-func SetupProcess(ctx context.Context, program string, programArgs ...string) (*termexec.Process, func(), error) {
+func SetupProcess(ctx context.Context, program string, programArgs ...string) (*termexec.Process, error) {
 	logger := logctx.From(ctx)
 
 	logger.Info(fmt.Sprintf("Running: %s %s", program, strings.Join(programArgs, " ")))
@@ -30,38 +29,6 @@ func SetupProcess(ctx context.Context, program string, programArgs ...string) (*
 		os.Exit(1)
 	}
 
-	snapshotFilePath := filepath.Join(".", fmt.Sprintf("snapshot-%s.log", time.Now().Format("20060102-150405")))
-	snapshotFile, err := os.Create(snapshotFilePath)
-	if err != nil {
-		logger.Error(fmt.Sprintf("Error creating snapshot file: %v", err))
-		os.Exit(1)
-	}
-	cleanup := func() {
-		if err := snapshotFile.Close(); err != nil {
-			logger.Error(fmt.Sprintf("Error closing snapshot file: %v", err))
-		}
-	}
-
-	go func() {
-		logger.Info("Starting snapshot loop")
-		for {
-			time.Sleep(100 * time.Millisecond)
-			// Truncate file and seek to beginning to overwrite completely
-			if err := snapshotFile.Truncate(0); err != nil {
-				logger.Error("Error truncating snapshot file", "error", err)
-				os.Exit(1)
-			}
-			if _, err := snapshotFile.Seek(0, 0); err != nil {
-				logger.Error("Error seeking in snapshot file", "error", err)
-				os.Exit(1)
-			}
-			if _, err := snapshotFile.Write([]byte(process.ReadScreen())); err != nil {
-				logger.Error("Error writing snapshot to file", "error", err)
-				os.Exit(1)
-			}
-		}
-	}()
-
 	// Handle SIGINT (Ctrl+C) and send it to the process
 	signalCh := make(chan os.Signal, 1)
 	signal.Notify(signalCh, os.Interrupt, syscall.SIGTERM)
@@ -72,5 +39,5 @@ func SetupProcess(ctx context.Context, program string, programArgs ...string) (*
 		}
 	}()
 
-	return process, cleanup, nil
+	return process, nil
 }