Function sessionREPL

cmd/inject-cli/main.go:442–567

(c *apiClient, sess *SessionDetail)

}

func sessionREPL(c *apiClient, sess *SessionDetail) {
	agentName := sess.UserAgent
	if idx := strings.Index(agentName, " "); idx > 0 {
		agentName = agentName[:idx]
	}
	if idx := strings.Index(agentName, "/"); idx > 0 {
		agentName = agentName[:idx]
	}
	if len(agentName) > 15 {
		agentName = agentName[:12] + "..."
	}

	prompt := fmt.Sprintf("\033[36m[%s %s]\033[0m$ ", sess.ID, agentName)

	rl, err := readline.NewEx(&readline.Config{
		Prompt:          prompt,
		AutoComplete:    sessionCompleter(),
		InterruptPrompt: "^C",
		EOFPrompt:       "exit",
	})
	if err != nil {
		fmt.Printf("  readline init failed: %v\n", err)
		return
	}
	defer rl.Close()

	// WebSocket for result streaming.
	wsURI := c.wsURL(sess.ID)
	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 (results won't stream)\n", wsErr)
		conn = nil
	}

	var wsDone chan struct{}
	var resultMu sync.Mutex
	var latestResult *CommandResult

	if conn != nil {
		defer conn.Close()
		wsDone = make(chan struct{})
		go func() {
			defer close(wsDone)
			for {
				var result CommandResult
				if err := conn.ReadJSON(&result); err != nil {
					return
				}
				resultMu.Lock()
				latestResult = &result
				resultMu.Unlock()
				fmt.Printf("\n%s\n", result.Output)
				rl.Refresh()
			}