Function cmdObserve

cmd/inject-cli/main.go:861–906

----- observe command -----

(c *apiClient, sessionID string)

// ----- observe command -----

func cmdObserve(c *apiClient, sessionID string) {
	sess, err := c.getSession(sessionID)
	if err != nil {
		fmt.Printf("  error: %v\n", err)
		return
	}

	wsURI := c.observeWsURL(sessionID)
	header := http.Header{}
	if c.secret != "" {
		header.Set("Authorization", "Bearer "+c.secret)
	}
	conn, _, wsErr := websocket.DefaultDialer.Dial(wsURI, header)
	if wsErr != nil {
		fmt.Printf("  ws connect failed: %v\n", wsErr)
		return
	}
	defer conn.Close()

	agentName, _ := formatAgent(sess.UserAgent)
	fmt.Printf("\n  \033[35mObserving\033[0m session \033[36m%s\033[0m (%s) — press Ctrl+C to stop\n\n", sessionID, agentName)

	// Use a signal channel so Ctrl+C stops observe without killing the process.
	sigCh := make(chan os.Signal, 1)
	signal.Notify(sigCh, os.Interrupt)
	defer signal.Stop(sigCh)

	done := make(chan struct{})
	go func() {
		defer close(done)
		for {
			var event ObserveEvent
			if err := conn.ReadJSON(&event); err != nil {
				return
			}
			printObserveEvent(&event)
		}
	}()

	select {
	case <-sigCh:
		fmt.Println("\n  (stopped observing)")
	case <-done:
		fmt.Println("\n  (observe connection closed)")
	}
}

func printObserveEvent(event *ObserveEvent) {
	ts := event.Timestamp.Format("15:04:05")