fix: keep Web UI running after wrapped command exits

- Web UI now stays running after the command finishes
- User can view results and press Ctrl+C to exit
- Bump version to 0.0.11

Author: echoVic

cmd/bar/update.go

@@ -16,7 +16,7 @@ const (
 	repoOwner      = "echoVic"
 	repoName       = "blade-agent-runtime"
 	installURL     = "https://echovic.github.io/blade-agent-runtime/install.sh"
-	currentVersion = "0.0.10"
+	currentVersion = "0.0.11"
 )
 
 func updateCmd() *cobra.Command {

cmd/bar/wrap.go

@@ -21,6 +21,7 @@ import (
 func wrapCmd() *cobra.Command {
 	var noUI bool
 	var uiPort int
+	var uiServer *web.Server
 
 	cmd := &cobra.Command{
 		Use:   "wrap -- <command> [args...]",
@@ -45,13 +46,12 @@ When the command exits, BAR records a step with the diff of all changes.`,
 			// Start UI by default (unless --no-ui is set)
 			if !noUI {
 				addr := fmt.Sprintf(":%d", uiPort)
-				server := web.NewServer(addr, app.TaskManager, app.BarDir)
+				uiServer = web.NewServer(addr, app.TaskManager, app.BarDir)
 				go func() {
-					if err := server.Start(); err != nil {
+					if err := uiServer.Start(); err != nil {
 						app.Logger.Error("Web UI failed: %v", err)
 					}
 				}()
-				defer server.Stop()
 
 				// Wait a bit for server to start
 				time.Sleep(500 * time.Millisecond)
@@ -165,6 +165,19 @@ When the command exits, BAR records a step with the diff of all changes.`,
 			app.Logger.Info("Step %s recorded", stepID)
 			app.Logger.Info("Files changed: %d (+%d, -%d)", diffResult.Files, diffResult.Additions, diffResult.Deletions)
 
+			// If UI is running, wait for user to close it
+			if uiServer != nil {
+				app.Logger.Info("")
+				app.Logger.Info("Web UI still running. Press Ctrl+C to exit.")
+
+				waitChan := make(chan os.Signal, 1)
+				signal.Notify(waitChan, syscall.SIGINT, syscall.SIGTERM)
+				<-waitChan
+
+				uiServer.Stop()
+				app.Logger.Info("Web UI stopped.")
+			}
+
 			return nil
 		},
 	}