rebuild branch

sdk/cliproxy/pipeline/context.go

@@ -0,0 +1,64 @@
+package pipeline
+
+import (
+	"context"
+	"net/http"
+
+	cliproxyauth "github.com/router-for-me/CLIProxyAPI/v6/sdk/cliproxy/auth"
+	cliproxyexecutor "github.com/router-for-me/CLIProxyAPI/v6/sdk/cliproxy/executor"
+	sdktranslator "github.com/router-for-me/CLIProxyAPI/v6/sdk/translator"
+)
+
+// Context encapsulates execution state shared across middleware, translators, and executors.
+type Context struct {
+	// Request encapsulates the provider facing request payload.
+	Request cliproxyexecutor.Request
+	// Options carries execution flags (streaming, headers, etc.).
+	Options cliproxyexecutor.Options
+	// Auth references the credential selected for execution.
+	Auth *cliproxyauth.Auth
+	// Translator represents the pipeline responsible for schema adaptation.
+	Translator *sdktranslator.Pipeline
+	// HTTPClient allows middleware to customise the outbound transport per request.
+	HTTPClient *http.Client
+}
+
+// Hook captures middleware callbacks around execution.
+type Hook interface {
+	BeforeExecute(ctx context.Context, execCtx *Context)
+	AfterExecute(ctx context.Context, execCtx *Context, resp cliproxyexecutor.Response, err error)
+	OnStreamChunk(ctx context.Context, execCtx *Context, chunk cliproxyexecutor.StreamChunk)
+}
+
+// HookFunc aggregates optional hook implementations.
+type HookFunc struct {
+	Before func(context.Context, *Context)
+	After  func(context.Context, *Context, cliproxyexecutor.Response, error)
+	Stream func(context.Context, *Context, cliproxyexecutor.StreamChunk)
+}
+
+// BeforeExecute implements Hook.
+func (h HookFunc) BeforeExecute(ctx context.Context, execCtx *Context) {
+	if h.Before != nil {
+		h.Before(ctx, execCtx)
+	}
+}
+
+// AfterExecute implements Hook.
+func (h HookFunc) AfterExecute(ctx context.Context, execCtx *Context, resp cliproxyexecutor.Response, err error) {
+	if h.After != nil {
+		h.After(ctx, execCtx, resp, err)
+	}
+}
+
+// OnStreamChunk implements Hook.
+func (h HookFunc) OnStreamChunk(ctx context.Context, execCtx *Context, chunk cliproxyexecutor.StreamChunk) {
+	if h.Stream != nil {
+		h.Stream(ctx, execCtx, chunk)
+	}
+}
+
+// RoundTripperProvider allows injection of custom HTTP transports per auth entry.
+type RoundTripperProvider interface {
+	RoundTripperFor(auth *cliproxyauth.Auth) http.RoundTripper
+}