mirror of
https://github.com/coder/coder.git
synced 2026-06-02 20:48:20 +00:00
Ethan
3a9080fff6
Workspace-agent logs emitted while serving chatd-driven requests were not correlated with the originating chat, making agent logs hard to attribute to the corresponding/originating chat. This adds agent-side chat context middleware that parses `Coder-Chat-Id` once, enriches agent access logs and structured handler/background logs, and adds a chatd bridge log when chat headers are attached to an agent connection. Closes CODAGT-324
36 lines
898 B
Go
36 lines
898 B
Go
package agentchat
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
|
|
"github.com/google/uuid"
|
|
|
|
"github.com/coder/coder/v2/codersdk/workspacesdk"
|
|
)
|
|
|
|
// extractContext reads chat identity headers from the request.
|
|
// Returns zero values if headers are absent (non-chat request).
|
|
func extractContext(r *http.Request) (chatID uuid.UUID, ancestorIDs []uuid.UUID, ok bool) {
|
|
raw := r.Header.Get(workspacesdk.CoderChatIDHeader)
|
|
if raw == "" {
|
|
return uuid.Nil, nil, false
|
|
}
|
|
chatID, err := uuid.Parse(raw)
|
|
if err != nil {
|
|
return uuid.Nil, nil, false
|
|
}
|
|
rawAncestors := r.Header.Get(workspacesdk.CoderAncestorChatIDsHeader)
|
|
if rawAncestors != "" {
|
|
var ids []string
|
|
if err := json.Unmarshal([]byte(rawAncestors), &ids); err == nil {
|
|
for _, s := range ids {
|
|
if id, err := uuid.Parse(s); err == nil {
|
|
ancestorIDs = append(ancestorIDs, id)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return chatID, ancestorIDs, true
|
|
}
|