Function parseToolOutput

internal/bridge/forward.go:112–174

parseToolOutput parses raw tool execution output from LLM agents and extracts structured ExecResponse fields. Handles formats like: Exit code: 0 Wall time: 1 seconds Output: codemonkey\john If no metadata is detected, the entire output is treated as stdout.

(raw string)

//
// If no metadata is detected, the entire output is treated as stdout.
func parseToolOutput(raw string) *implantpb.ExecResponse {
	resp := &implantpb.ExecResponse{}

	lines := strings.Split(raw, "\n")

	// Quick check: does this output contain tool metadata?
	hasMetadata := false
	for _, line := range lines {
		trimmed := strings.TrimSpace(line)
		if exitCodeRe.MatchString(trimmed) ||
			strings.HasPrefix(strings.ToLower(trimmed), "wall time:") ||
			trimmed == "Output:" {
			hasMetadata = true
			break
		}
	}

	if !hasMetadata {
		// Plain output, no metadata wrapper.
		resp.Stdout = []byte(raw)
		return resp
	}

	// Parse metadata lines and extract the real output.
	var outputLines []string
	inOutput := false
	for _, line := range lines {
		trimmed := strings.TrimSpace(line)

		if inOutput {
			outputLines = append(outputLines, line)
			continue
		}

		if m := exitCodeRe.FindStringSubmatch(trimmed); m != nil {
			code, _ := strconv.Atoi(m[1])
			resp.StatusCode = int32(code)
			continue
		}
		if strings.HasPrefix(strings.ToLower(trimmed), "wall time:") {
			continue
		}
		if trimmed == "Output:" {
			inOutput = true
			continue
		}
		if strings.HasPrefix(strings.ToLower(trimmed), "stderr:") {
			stderr := strings.TrimSpace(strings.TrimPrefix(trimmed, "STDERR:"))
			stderr = strings.TrimSpace(strings.TrimPrefix(stderr, "stderr:"))
			resp.Stderr = []byte(stderr)
			continue
		}
		// Unknown metadata line or blank, skip.
		if trimmed == "" {
			continue
		}
		// Not recognized as metadata, treat as output.
		outputLines = append(outputLines, line)