coder/aibridge/context/context.go

package context

import (
	"context"

	"github.com/coder/coder/v2/aibridge/recorder"
)

type (
	actorContextKey struct{}
)

type Actor struct {
	ID       string
	Metadata recorder.Metadata
}

func AsActor(ctx context.Context, actorID string, metadata recorder.Metadata) context.Context {
	return context.WithValue(ctx, actorContextKey{}, &Actor{ID: actorID, Metadata: metadata})
}

func ActorFromContext(ctx context.Context) *Actor {
	a, ok := ctx.Value(actorContextKey{}).(*Actor)
	if !ok {
		return nil
	}

	return a
}

// ActorIDFromContext safely extracts the actor ID from the context.
// Returns an empty string if no actor is found.
func ActorIDFromContext(ctx context.Context) string {
	if actor := ActorFromContext(ctx); actor != nil {
		return actor.ID
	}
	return ""
}