package http

import (
	"context"
	"net/http"

	"forge.cadoles.com/cadoles/bouncer/internal/rule"
)

type contextKey string

const (
	contextKeyRequest  contextKey = "request"
	contextKeyResponse contextKey = "response"
)

func WithRequest(ctx context.Context, r *http.Request) context.Context {
	return context.WithValue(ctx, contextKeyRequest, r)
}

func WithResponse(ctx context.Context, r *http.Response) context.Context {
	return context.WithValue(ctx, contextKeyResponse, r)
}

func ctxRequest(ctx context.Context) (*http.Request, bool) {
	return rule.Context[*http.Request](ctx, contextKeyRequest)
}

func ctxResponse(ctx context.Context) (*http.Response, bool) {
	return rule.Context[*http.Response](ctx, contextKeyResponse)
}