new-api/middleware/task_plugin_test.go

package middleware

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"io"
	"mime/multipart"
	"net/http"
	"net/http/httptest"
	"net/textproto"
	"strings"
	"testing"

	"github.com/QuantumNous/new-api/common"
	"github.com/QuantumNous/new-api/constant"
	"github.com/QuantumNous/new-api/dto"
	appI18n "github.com/QuantumNous/new-api/i18n"
	"github.com/QuantumNous/new-api/model"
	"github.com/QuantumNous/new-api/pkg/jsplugin"
	builtinplugins "github.com/QuantumNous/new-api/plugins"
	"github.com/QuantumNous/new-api/service"
	"github.com/gin-gonic/gin"
	"github.com/glebarez/sqlite"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"gorm.io/gorm"
)

const genericTaskPluginSource = `
export const meta = {apiVersion: 1, key: "generic-entry-test", name: "Generic", version: "1.0.0", author: {name: "Test"}, models: ["doc"], fetchMode: "per_task"};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
`

func TestPrepareTaskPluginSubmitRejectsMissingModel(t *testing.T) {
	_, err := jsplugin.DefaultRegistry.Register(genericTaskPluginSource, jsplugin.Options{})
	require.NoError(t, err)
	t.Cleanup(func() { jsplugin.DefaultRegistry.Unregister("generic-entry-test") })
	recorder := httptest.NewRecorder()
	c, _ := gin.CreateTestContext(recorder)
	c.Params = gin.Params{{Key: "key", Value: "generic-entry-test"}}
	c.Request = httptest.NewRequest(http.MethodPost, "/v1/tasks/generic-entry-test", strings.NewReader(`{"prompt":"x"}`))
	c.Request.Header.Set("Content-Type", "application/json")
	PrepareTaskPluginSubmit()(c)
	assert.Equal(t, http.StatusBadRequest, recorder.Code)
	assert.Contains(t, recorder.Body.String(), "model is required")
}

func TestPrepareTaskPluginRouteUsesCanonicalContextAndResolvedSubmit(t *testing.T) {
	plugin := compileTaskRoutePlugin(t, `
export const meta = {
  apiVersion: 1, key: "route-submit-test", name: "Submit", version: "1.0.0",
  author: {name: "Test"},
  models: ["resolved-model"], fetchMode: "per_task",
  routes: [{method: "POST", path: "/vendor/jobs/:category", type: "submit", action: "static-action", decode: "decodeJob", render: "jobCreated"}],
};
export const native = {
decodeJob: function(ctx) {
  if (ctx.path !== "/vendor/jobs/video" || ctx.method !== "POST") throw new Error("bad path");
  if (ctx.params.category !== "video") throw new Error("bad params");
  if (ctx.query.tag.length !== 2 || ctx.query.tag[0] !== "first" || ctx.query.tag[1] !== "second") throw new Error("bad query");
  if (ctx.body.kind !== "json" || !Array.isArray(ctx.body.value) || ctx.body.value[0] !== "prompt") throw new Error("bad body");
	return {kind: "submit", model: "resolved-model", action: "resolved-action", requestBody: {prompt: "normalized"}};
},
jobCreated: function(ctx, task) { return task; },
};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
`)

	router := gin.New()
	reachedSubmit := false
	router.POST("/vendor/jobs/:category", pinTaskPluginRoute(plugin, 0), PrepareTaskPluginRoute(), func(c *gin.Context) {
		reachedSubmit = true
		requestContext, ok := c.MustGet(jsplugin.ContextKeyRouteRequest).(jsplugin.RouteRequestContext)
		require.True(t, ok)
		assert.Equal(t, map[string]any{"prompt": "normalized"}, requestContext.RequestBody)
		assert.Equal(t, "resolved-model", c.GetString("resolved_task_model"))
		assert.Equal(t, "resolved-action", c.GetString("task_action"))
		assert.Equal(t, "route-submit-test", c.GetString("expected_task_plugin_key"))
		assert.Equal(t, "route-submit-test", c.GetString("task_plugin_key"))
		assert.Equal(t, "route-submit-test", c.GetString("platform"))
		c.Status(http.StatusNoContent)
	})
	request := httptest.NewRequest(http.MethodPost, "/vendor/jobs/video?tag=first&tag=second", strings.NewReader(`["prompt",2]`))
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.True(t, reachedSubmit)
	assert.Equal(t, http.StatusNoContent, recorder.Code)
}

// Native routes and the generic submit entry hand a plugin declaring
// json-order@1 the client's member order and keep its decoded requestBody as
// ordered JSON text beside the Go value; other plugins see sorted members.
func TestTaskPluginEntriesKeepJSONOrderForDeclaringPlugins(t *testing.T) {
	source := `
export const meta = {
  apiVersion: 1, key: "route-order-test", name: "Order", version: "1.0.0", author: {name: "Test"},
  models: ["m"], fetchMode: "per_task", requiredCapabilities: ["json-order@1"],
  routes: [{method: "POST", path: "/vendor/eval", type: "submit", decode: "decodeEval", render: "evalDone"}],
};
export const native = {
decodeEval: function(ctx) {
  const state = ctx.body.value.state;
  return {kind: "submit", model: ctx.body.value.model, requestBody: {keys: Object.keys(state), state: state}};
},
evalDone: function(ctx, task) { return task; },
};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
`
	cases := []struct {
		name     string
		source   string
		wantText string
		wantKeys []any
	}{
		{name: "declared", source: source, wantText: `{"keys":["zeta","alpha"],"state":{"zeta":1,"alpha":2}}`, wantKeys: []any{"zeta", "alpha"}},
		{name: "undeclared", source: strings.Replace(source, `requiredCapabilities: ["json-order@1"],`, "", 1), wantKeys: []any{"alpha", "zeta"}},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			plugin := compileTaskRoutePlugin(t, tc.source)
			router := gin.New()
			reached := false
			router.POST("/vendor/eval", pinTaskPluginRoute(plugin, 0), PrepareTaskPluginRoute(), func(c *gin.Context) {
				reached = true
				request, ok := c.MustGet("task_request").(map[string]any)
				require.True(t, ok)
				assert.Equal(t, tc.wantKeys, request["keys"])
				text, exists := c.Get(jsplugin.ContextKeyRequestBodyText)
				if tc.wantText == "" {
					assert.False(t, exists)
				} else {
					assert.Equal(t, json.RawMessage(tc.wantText), text)
				}
				c.Status(http.StatusNoContent)
			})
			request := httptest.NewRequest(http.MethodPost, "/vendor/eval", strings.NewReader(`{"model":"m","state":{"zeta":1,"alpha":2}}`))
			request.Header.Set("Content-Type", "application/json")
			recorder := httptest.NewRecorder()

			router.ServeHTTP(recorder, request)

			assert.True(t, reached, recorder.Body.String())
		})
	}

	t.Run("generic submit entry", func(t *testing.T) {
		declaring := strings.Replace(genericTaskPluginSource, `fetchMode: "per_task"`, `fetchMode: "per_task", requiredCapabilities: ["json-order@1"]`, 1)
		_, err := jsplugin.DefaultRegistry.Register(declaring, jsplugin.Options{})
		require.NoError(t, err)
		t.Cleanup(func() { jsplugin.DefaultRegistry.Unregister("generic-entry-test") })
		recorder := httptest.NewRecorder()
		c, _ := gin.CreateTestContext(recorder)
		c.Params = gin.Params{{Key: "key", Value: "generic-entry-test"}}
		c.Request = httptest.NewRequest(http.MethodPost, "/v1/tasks/generic-entry-test", strings.NewReader(`{"model":"doc","zeta":1,"alpha":2}`))
		c.Request.Header.Set("Content-Type", "application/json")

		PrepareTaskPluginSubmit()(c)

		require.False(t, c.IsAborted(), recorder.Body.String())
		text, exists := c.Get(jsplugin.ContextKeyRequestBodyText)
		require.True(t, exists)
		assert.Equal(t, json.RawMessage(`{"model":"doc","zeta":1,"alpha":2}`), text)
	})
}

func TestPrepareTaskPluginNativeRouteRejectsMultipartBeforeDecoder(t *testing.T) {
	plugin := compileTaskRoutePlugin(t, `
export const meta = {
  apiVersion: 1, key: "route-multipart-test", name: "Multipart", version: "1.0.0",
  author: {name: "Test"},
  models: ["multipart-model"], fetchMode: "per_task",
  routes: [{method: "POST", path: "/vendor/uploads", type: "submit", decode: "decodeUpload", render: "created"}],
};
export const native = {decodeUpload: function() { throw new Error("decoder must not run"); }, created: function(ctx, task) { return task; }};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
`)
	var body bytes.Buffer
	writer := multipart.NewWriter(&body)
	require.NoError(t, writer.WriteField("caption", "hello"))
	require.NoError(t, writer.WriteField("tag", "one"))
	require.NoError(t, writer.WriteField("tag", "two"))
	file, err := writer.CreateFormFile("media", "clip.bin")
	require.NoError(t, err)
	_, err = file.Write([]byte("opaque-file"))
	require.NoError(t, err)
	require.NoError(t, writer.Close())

	router := gin.New()
	router.POST("/vendor/uploads", pinTaskPluginRoute(plugin, 0), PrepareTaskPluginRoute(), func(c *gin.Context) {
		c.Status(http.StatusNoContent)
	})
	request := httptest.NewRequest(http.MethodPost, "/vendor/uploads", bytes.NewReader(body.Bytes()))
	request.Header.Set("Content-Type", writer.FormDataContentType())
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.Equal(t, http.StatusUnsupportedMediaType, recorder.Code)
}

func TestPrepareTaskPluginRouteModelScope(t *testing.T) {
	pluginSource := `
export const meta = {
  apiVersion: 1, key: "route-model-scope-test", name: "Scoped", version: "1.0.0",
  author: {name: "Test"},
  models: ["gpt-5.5", "gpt-5.6"], fetchMode: "per_task",
  routes: [{method: "POST", path: "/vendor/batch", type: "submit", models: ["gpt-5.5"], decode: "decodeBatch", render: "batchCreated"}],
};
export const native = {
  decodeBatch: function(ctx) { return {kind: "submit", model: ctx.body.value.model, requestBody: ctx.body.value}; },
  batchCreated: function(ctx, task) { return task; },
};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
`
	tests := []struct {
		name         string
		body         string
		wantStatus   int
		wantResolved bool
	}{
		{name: "listed model passes to decode", body: `{"model":"gpt-5.5","input":"x"}`, wantStatus: http.StatusNoContent, wantResolved: true},
		{name: "unlisted model rejected before JS", body: `{"model":"gpt-5.6","input":"x"}`, wantStatus: http.StatusBadRequest},
		{name: "missing model rejected before JS", body: `{"input":"x"}`, wantStatus: http.StatusBadRequest},
		{name: "non-string model rejected before JS", body: `{"model":7,"input":"x"}`, wantStatus: http.StatusBadRequest},
	}
	for _, testCase := range tests {
		t.Run(testCase.name, func(t *testing.T) {
			decodeRan := false
			source := pluginSource
			if !testCase.wantResolved {
				// The core invariant: rejected requests never reach the JS engine.
				source = strings.Replace(source,
					`decodeBatch: function(ctx) { return {kind: "submit", model: ctx.body.value.model, requestBody: ctx.body.value}; },`,
					`decodeBatch: function() { throw new Error("decoder must not run"); },`, 1)
			}
			plugin := compileTaskRoutePlugin(t, source)
			router := gin.New()
			router.POST("/vendor/batch", pinTaskPluginRoute(plugin, 0), PrepareTaskPluginRoute(), func(c *gin.Context) {
				decodeRan = true
				assert.Equal(t, "gpt-5.5", c.GetString("resolved_task_model"))
				c.Status(http.StatusNoContent)
			})
			request := httptest.NewRequest(http.MethodPost, "/vendor/batch", strings.NewReader(testCase.body))
			request.Header.Set("Content-Type", "application/json")
			recorder := httptest.NewRecorder()

			router.ServeHTTP(recorder, request)

			assert.Equal(t, testCase.wantStatus, recorder.Code)
			assert.Equal(t, testCase.wantResolved, decodeRan)
		})
	}
}

func TestPrepareTaskPluginRouteRejectsResolvedModelOutsideRouteScope(t *testing.T) {
	plugin := compileTaskRoutePlugin(t, `
export const meta = {
  apiVersion: 1, key: "route-resolved-scope-test", name: "Scoped", version: "1.0.0",
  author: {name: "Test"},
  models: ["gpt-5.5", "gpt-5.6"], fetchMode: "per_task",
  routes: [{method: "POST", path: "/vendor/batch", type: "submit", models: ["gpt-5.5"], decode: "decodeBatch", render: "batchCreated"}],
};
export const native = {
  decodeBatch: function() { return {kind: "submit", model: "gpt-5.6", requestBody: {model: "gpt-5.6"}}; },
  batchCreated: function(ctx, task) { return task; },
};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
`)
	reached := false
	router := gin.New()
	router.POST("/vendor/batch", pinTaskPluginRoute(plugin, 0), PrepareTaskPluginRoute(), func(c *gin.Context) {
		reached = true
		c.Status(http.StatusNoContent)
	})
	request := httptest.NewRequest(http.MethodPost, "/vendor/batch", strings.NewReader(`{"model":"gpt-5.5"}`))
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.Equal(t, http.StatusBadRequest, recorder.Code)
	assert.False(t, reached, "decode may run, but an out-of-scope resolved model must not continue into submit")
}

func TestPrepareTaskPluginRouteWithoutModelScopeIsUnrestricted(t *testing.T) {
	plugin := compileTaskRoutePlugin(t, `
export const meta = {
  apiVersion: 1, key: "route-unscoped-test", name: "Unscoped", version: "1.0.0",
  author: {name: "Test"},
  models: ["gpt-5.5", "gpt-5.6"], fetchMode: "per_task",
  routes: [{method: "POST", path: "/vendor/batch", type: "submit", decode: "decodeBatch", render: "batchCreated"}],
};
export const native = {
  decodeBatch: function(ctx) { return {kind: "submit", model: ctx.body.value.model, requestBody: ctx.body.value}; },
  batchCreated: function(ctx, task) { return task; },
};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
`)
	router := gin.New()
	router.POST("/vendor/batch", pinTaskPluginRoute(plugin, 0), PrepareTaskPluginRoute(), func(c *gin.Context) {
		c.Status(http.StatusNoContent)
	})
	request := httptest.NewRequest(http.MethodPost, "/vendor/batch", strings.NewReader(`{"model":"gpt-5.6"}`))
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.Equal(t, http.StatusNoContent, recorder.Code)
}

func TestBuildTaskPluginRouteRequestBodyUnion(t *testing.T) {
	tests := []struct {
		name        string
		contentType string
		body        string
		kind        jsplugin.BodyKind
		assertBody  func(*testing.T, map[string]any)
	}{
		{name: "none", kind: jsplugin.BodyNone},
		{name: "json", contentType: "application/problem+json; charset=utf-8", body: `{"model":"m"}`, kind: jsplugin.BodyJSON, assertBody: func(t *testing.T, body map[string]any) {
			assert.Equal(t, map[string]any{"model": "m"}, body["value"])
		}},
		{name: "form preserves repeated values", contentType: "application/x-www-form-urlencoded", body: "tag=one&tag=two", kind: jsplugin.BodyForm, assertBody: func(t *testing.T, body map[string]any) {
			assert.Equal(t, []string{"one", "two"}, body["fields"].(map[string][]string)["tag"])
		}},
	}
	for _, testCase := range tests {
		t.Run(testCase.name, func(t *testing.T) {
			c, _ := gin.CreateTestContext(httptest.NewRecorder())
			c.Request = httptest.NewRequest(http.MethodPost, "/body", strings.NewReader(testCase.body))
			if testCase.contentType != "" {
				c.Request.Header.Set("Content-Type", testCase.contentType)
			}
			requestContext, err := buildTaskPluginRouteRequest(c)
			require.NoError(t, err)
			decodedBody := requestContext.Body.(map[string]any)
			assert.Equal(t, string(testCase.kind), decodedBody["kind"])
			if testCase.assertBody != nil {
				testCase.assertBody(t, decodedBody)
			}
		})
	}

	var multipartBody bytes.Buffer
	writer := multipart.NewWriter(&multipartBody)
	require.NoError(t, writer.WriteField("tag", "one"))
	require.NoError(t, writer.WriteField("tag", "two"))
	file, err := writer.CreateFormFile("input", "image.png")
	require.NoError(t, err)
	_, err = file.Write([]byte("file bytes stay in Go"))
	require.NoError(t, err)
	require.NoError(t, writer.Close())
	c, _ := gin.CreateTestContext(httptest.NewRecorder())
	c.Request = httptest.NewRequest(http.MethodPost, "/body", bytes.NewReader(multipartBody.Bytes()))
	c.Request.Header.Set("Content-Type", writer.FormDataContentType())
	requestContext, err := buildTaskPluginRouteRequest(c)
	require.NoError(t, err)
	decodedBody := requestContext.Body.(map[string]any)
	assert.Equal(t, string(jsplugin.BodyMultipart), decodedBody["kind"])
	assert.Equal(t, []string{"one", "two"}, decodedBody["fields"].(map[string][]string)["tag"])
	files := decodedBody["files"].([]map[string]any)
	require.Len(t, files, 1)
	assert.Equal(t, "image.png", files[0]["filename"])
	assert.NotContains(t, fmt.Sprint(files[0]), "file bytes stay in Go")
}

func TestBuildTaskPluginRouteRequestRejectsUnsafeBodies(t *testing.T) {
	tests := []struct {
		name        string
		contentType string
		body        []byte
		errorText   string
	}{
		{name: "invalid json UTF-8", contentType: "application/json", body: []byte{'{', '"', 'x', '"', ':', '"', 0xff, '"', '}'}, errorText: "valid UTF-8"},
		{name: "invalid form UTF-8", contentType: "application/x-www-form-urlencoded", body: []byte("x=%FF"), errorText: "valid UTF-8"},
		{name: "too many repeated form fields", contentType: "application/x-www-form-urlencoded", body: []byte(strings.Repeat("x=v&", maxTaskPluginFormFields) + "x=v"), errorText: "exceeds 256 fields"},
		{name: "oversized form field", contentType: "application/x-www-form-urlencoded", body: []byte("x=" + strings.Repeat("a", maxTaskPluginFieldValueBytes+1)), errorText: "exceeds 1048576 bytes"},
		{name: "missing multipart boundary", contentType: "multipart/form-data", body: []byte("body"), errorText: "boundary is required"},
	}
	for _, testCase := range tests {
		t.Run(testCase.name, func(t *testing.T) {
			c, _ := gin.CreateTestContext(httptest.NewRecorder())
			c.Request = httptest.NewRequest(http.MethodPost, "/body", bytes.NewReader(testCase.body))
			c.Request.Header.Set("Content-Type", testCase.contentType)
			_, err := buildTaskPluginRouteRequest(c)
			require.ErrorContains(t, err, testCase.errorText)
		})
	}

	t.Run("conflicting Content-Type", func(t *testing.T) {
		c, _ := gin.CreateTestContext(httptest.NewRecorder())
		c.Request = httptest.NewRequest(http.MethodPost, "/body", strings.NewReader(`{}`))
		c.Request.Header["Content-Type"] = []string{"application/json", "application/x-www-form-urlencoded"}
		_, err := buildTaskPluginRouteRequest(c)
		require.ErrorContains(t, err, "conflicting Content-Type")
	})

	t.Run("oversized total body", func(t *testing.T) {
		previous := constant.MaxRequestBodyMB
		constant.MaxRequestBodyMB = 1
		t.Cleanup(func() { constant.MaxRequestBodyMB = previous })
		c, _ := gin.CreateTestContext(httptest.NewRecorder())
		c.Request = httptest.NewRequest(http.MethodPost, "/body", strings.NewReader(strings.Repeat(" ", (1<<20)+1)))
		c.Request.Header.Set("Content-Type", "application/json")
		_, err := buildTaskPluginRouteRequest(c)
		require.ErrorContains(t, err, "request body exceeds 1 MB")
	})

	multipartCases := []struct {
		name      string
		build     func(*testing.T, *multipart.Writer)
		errorText string
	}{
		{name: "too many parts", build: func(t *testing.T, writer *multipart.Writer) {
			for i := 0; i <= maxTaskPluginMultipartParts; i++ {
				require.NoError(t, writer.WriteField("field", "value"))
			}
		}, errorText: "exceeds 256 parts"},
		{name: "too many files", build: func(t *testing.T, writer *multipart.Writer) {
			for i := 0; i <= maxTaskPluginFiles; i++ {
				_, err := writer.CreateFormFile("file", fmt.Sprintf("%d.bin", i))
				require.NoError(t, err)
			}
		}, errorText: "exceeds 32 files"},
		{name: "invalid UTF-8 field", build: func(t *testing.T, writer *multipart.Writer) {
			part, err := writer.CreateFormField("field")
			require.NoError(t, err)
			_, err = part.Write([]byte{0xff})
			require.NoError(t, err)
		}, errorText: "valid UTF-8"},
		{name: "oversized multipart field", build: func(t *testing.T, writer *multipart.Writer) {
			part, err := writer.CreateFormField("field")
			require.NoError(t, err)
			_, err = part.Write([]byte(strings.Repeat("a", maxTaskPluginFieldValueBytes+1)))
			require.NoError(t, err)
		}, errorText: "exceeds 1048576 bytes"},
		{name: "nested multipart", build: func(t *testing.T, writer *multipart.Writer) {
			header := make(textproto.MIMEHeader)
			header.Set("Content-Disposition", `form-data; name="nested"`)
			header.Set("Content-Type", "multipart/mixed; boundary=inner")
			_, err := writer.CreatePart(header)
			require.NoError(t, err)
		}, errorText: "nested multipart"},
		{name: "invalid UTF-8 filename", build: func(t *testing.T, writer *multipart.Writer) {
			header := make(textproto.MIMEHeader)
			header.Set("Content-Disposition", "form-data; name=\"file\"; filename=\""+string([]byte{0xff})+"\"")
			_, err := writer.CreatePart(header)
			require.NoError(t, err)
		}, errorText: "invalid multipart"},
		{name: "injected disposition name", build: func(t *testing.T, writer *multipart.Writer) {
			header := make(textproto.MIMEHeader)
			header.Set("Content-Disposition", "form-data; name=\"prompt"+"\r\n"+"X-Injected: yes\"")
			part, err := writer.CreatePart(header)
			require.NoError(t, err)
			_, err = part.Write([]byte("hello"))
			require.NoError(t, err)
		}, errorText: "invalid multipart field name"},
		{name: "injected disposition filename", build: func(t *testing.T, writer *multipart.Writer) {
			header := make(textproto.MIMEHeader)
			header.Set("Content-Disposition", "form-data; name=\"file\"; filename=\"safe.png"+"\r\n"+"X-Injected: yes\"")
			part, err := writer.CreatePart(header)
			require.NoError(t, err)
			_, err = part.Write([]byte("file"))
			require.NoError(t, err)
		}, errorText: "invalid multipart"},
	}
	for _, testCase := range multipartCases {
		t.Run(testCase.name, func(t *testing.T) {
			var body bytes.Buffer
			writer := multipart.NewWriter(&body)
			testCase.build(t, writer)
			require.NoError(t, writer.Close())
			c, _ := gin.CreateTestContext(httptest.NewRecorder())
			c.Request = httptest.NewRequest(http.MethodPost, "/body", bytes.NewReader(body.Bytes()))
			c.Request.Header.Set("Content-Type", writer.FormDataContentType())
			_, err := buildTaskPluginRouteRequest(c)
			require.ErrorContains(t, err, testCase.errorText)
		})
	}

	t.Run("oversized multipart file", func(t *testing.T) {
		previous := constant.MaxFileDownloadMB
		constant.MaxFileDownloadMB = 1
		t.Cleanup(func() { constant.MaxFileDownloadMB = previous })
		var body bytes.Buffer
		writer := multipart.NewWriter(&body)
		file, err := writer.CreateFormFile("file", "large.bin")
		require.NoError(t, err)
		_, err = file.Write(bytes.Repeat([]byte{'x'}, (1<<20)+1))
		require.NoError(t, err)
		require.NoError(t, writer.Close())
		c, _ := gin.CreateTestContext(httptest.NewRecorder())
		c.Request = httptest.NewRequest(http.MethodPost, "/body", bytes.NewReader(body.Bytes()))
		c.Request.Header.Set("Content-Type", writer.FormDataContentType())
		_, err = buildTaskPluginRouteRequest(c)
		require.ErrorContains(t, err, "multipart file exceeds 1 MB")
	})
}

func TestPrepareTaskPluginEndpointPinsGenerationBeforeParseAndDistribution(t *testing.T) {
	const key = "endpoint-pin-test"
	firstSource := taskProtocolPluginSource(
		key,
		"1.0.0",
		`["claimed-model"]`,
		"/v1/responses",
		`if (ctx.protocol !== "openai_responses" || ctx.operation !== "create" || ctx.model !== "claimed-model" || ctx.path !== "/v1/responses" || ctx.stream !== false) throw new Error("bad context");
		 if (ctx.query.trace[0] !== "one" || ctx.requestBody.prompt !== "hello") throw new Error("bad request");
		 ctx.requestBody.prompt = "plugin-local-mutation";
		 return {model: ctx.model, action: "first-action", requestBody: {prompt: "normalized"}};`,
	)
	first, err := jsplugin.DefaultRegistry.Register(firstSource, jsplugin.Options{})
	require.NoError(t, err)
	t.Cleanup(func() { require.NoError(t, jsplugin.DefaultRegistry.Unregister(key)) })

	router := gin.New()
	reachedDistribution := false
	router.POST(
		"/v1/responses",
		PinTaskPluginEndpoint(),
		func(c *gin.Context) {
			_, updateErr := jsplugin.DefaultRegistry.Register(taskProtocolPluginSource(
				key,
				"2.0.0",
				`["claimed-model"]`,
				"/v1/responses",
				`return {model: ctx.model, action: "second-action"};`,
			), jsplugin.Options{})
			require.NoError(t, updateErr)
			c.Next()
		},
		PrepareTaskPluginEndpoint(),
		func(c *gin.Context) {
			reachedDistribution = true
			pinned := c.MustGet(jsplugin.ContextKeyPinnedEndpoint).(jsplugin.PinnedEndpoint)
			assert.Same(t, first, pinned.Plugin)
			assert.Equal(t, "claimed-model", pinned.Model)
			assert.Equal(t, "first-action", c.GetString("task_action"))
			assert.Equal(t, "claimed-model", c.GetString("resolved_task_model"))
			assert.Equal(t, key, c.GetString("expected_task_plugin_key"))
			assert.Equal(t, key, c.GetString("platform"))

			protocolRequest := c.MustGet(jsplugin.ContextKeyProtocolRequest).(jsplugin.ProtocolRequestContext)
			assert.Equal(t, map[string]any{"kind": "json", "value": map[string]any{"model": "claimed-model", "prompt": "hello"}}, protocolRequest.Body)
			assert.False(t, protocolRequest.Stream)
			routeRequest := c.MustGet(jsplugin.ContextKeyRouteRequest).(jsplugin.RouteRequestContext)
			assert.Equal(t, map[string]any{"prompt": "normalized"}, routeRequest.RequestBody)

			current, found := jsplugin.DefaultRegistry.Get(key)
			require.True(t, found)
			assert.NotSame(t, first, current)
			c.Status(http.StatusNoContent)
		},
	)
	request := httptest.NewRequest(
		http.MethodPost,
		"/v1/responses?trace=one",
		strings.NewReader(`{"model":"claimed-model","prompt":"hello"}`),
	)
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.True(t, reachedDistribution)
	assert.Equal(t, http.StatusNoContent, recorder.Code)
}

func TestPrepareTaskPluginEndpointClientDisconnectDoesNotCancelParseHook(t *testing.T) {
	const key = "endpoint-detached-parse-test"
	_, err := jsplugin.DefaultRegistry.Register(taskProtocolPluginSource(
		key,
		"1.0.0",
		`["claimed-model"]`,
		"/v1/responses",
		`return {model: "claimed-model"};`,
	), jsplugin.Options{})
	require.NoError(t, err)
	t.Cleanup(func() { require.NoError(t, jsplugin.DefaultRegistry.Unregister(key)) })

	router := gin.New()
	reachedDistribution := false
	router.POST(
		"/v1/responses",
		PinTaskPluginEndpoint(),
		func(c *gin.Context) {
			requestContext, cancel := context.WithCancel(c.Request.Context())
			cancel()
			c.Request = c.Request.WithContext(requestContext)
			c.Next()
		},
		PrepareTaskPluginEndpoint(),
		func(c *gin.Context) {
			reachedDistribution = true
			assert.Equal(t, "claimed-model", c.GetString("resolved_task_model"))
			c.Status(http.StatusNoContent)
		},
	)
	request := httptest.NewRequest(
		http.MethodPost,
		"/v1/responses",
		strings.NewReader(`{"model":"claimed-model"}`),
	)
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.True(t, reachedDistribution)
	assert.Equal(t, http.StatusNoContent, recorder.Code)
}

func TestPrepareTaskPluginEndpointUsesStrictOriginalStreamFlag(t *testing.T) {
	const key = "endpoint-stream-test"
	_, err := jsplugin.DefaultRegistry.Register(taskProtocolPluginSource(
		key,
		"1.0.0",
		`["claimed-model"]`,
		"/v1/responses",
		`if (ctx.stream !== true) throw new Error("stream mode was not preserved");
		 return {model: "claimed-model", requestBody: {stream: false}};`,
	), jsplugin.Options{})
	require.NoError(t, err)
	t.Cleanup(func() { require.NoError(t, jsplugin.DefaultRegistry.Unregister(key)) })

	router := gin.New()
	reachedDistribution := false
	router.POST(
		"/v1/responses",
		PinTaskPluginEndpoint(),
		PrepareTaskPluginEndpoint(),
		func(c *gin.Context) {
			reachedDistribution = true
			protocolRequest := c.MustGet(jsplugin.ContextKeyProtocolRequest).(jsplugin.ProtocolRequestContext)
			assert.True(t, protocolRequest.Stream)
			normalizedRequest := c.MustGet(jsplugin.ContextKeyRouteRequest).(jsplugin.RouteRequestContext)
			assert.Equal(t, map[string]any{"stream": false}, normalizedRequest.RequestBody)
			c.Status(http.StatusNoContent)
		},
	)
	request := httptest.NewRequest(
		http.MethodPost,
		"/v1/responses",
		strings.NewReader(`{"model":"claimed-model","stream":true}`),
	)
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.True(t, reachedDistribution)
	assert.Equal(t, http.StatusNoContent, recorder.Code)

	for _, invalid := range []string{`null`, `"true"`, `1`, `"yes"`} {
		request = httptest.NewRequest(
			http.MethodPost,
			"/v1/responses",
			strings.NewReader(`{"model":"claimed-model","stream":`+invalid+`}`),
		)
		request.Header.Set("Content-Type", "application/json")
		recorder = httptest.NewRecorder()
		router.ServeHTTP(recorder, request)
		assert.Equal(t, http.StatusBadRequest, recorder.Code)
	}
}

func TestTaskPluginEndpointMissPreservesOrdinaryRequestBody(t *testing.T) {
	const key = "endpoint-miss-test"
	_, err := jsplugin.DefaultRegistry.Register(taskProtocolPluginSource(
		key,
		"1.0.0",
		`["claimed-model"]`,
		"/v1/responses",
		`return {model: "claimed-model"};`,
	), jsplugin.Options{})
	require.NoError(t, err)
	t.Cleanup(func() { require.NoError(t, jsplugin.DefaultRegistry.Unregister(key)) })

	router := gin.New()
	router.POST(
		"/v1/responses",
		PinTaskPluginEndpoint(),
		PrepareTaskPluginEndpoint(),
		func(c *gin.Context) {
			_, pinned := c.Get(jsplugin.ContextKeyPinnedEndpoint)
			assert.False(t, pinned)
			storage, storageErr := common.GetBodyStorage(c)
			require.NoError(t, storageErr)
			raw, bytesErr := storage.Bytes()
			require.NoError(t, bytesErr)
			assert.Equal(t, []byte(`{"model":"ordinary-model","input":"hello"}`), raw)
			c.Status(http.StatusNoContent)
		},
	)
	request := httptest.NewRequest(
		http.MethodPost,
		"/v1/responses",
		strings.NewReader(`{"model":"ordinary-model","input":"hello"}`),
	)
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.Equal(t, http.StatusNoContent, recorder.Code)
}

func TestTaskPluginEndpointUsesOneCanonicalModelForDuplicateJSONKeys(t *testing.T) {
	require.NoError(t, appI18n.Init())
	const key = "endpoint-duplicate-model-test"
	_, err := jsplugin.DefaultRegistry.Register(taskProtocolPluginSource(
		key,
		"1.0.0",
		`["claimed-model"]`,
		"/v1/responses",
		`if (ctx.requestBody.model !== "claimed-model") throw new Error("noncanonical model");
		 return {model: ctx.requestBody.model};`,
	), jsplugin.Options{})
	require.NoError(t, err)
	t.Cleanup(func() { require.NoError(t, jsplugin.DefaultRegistry.Unregister(key)) })

	tests := []struct {
		name string
		body string
	}{
		{
			name: "claimed model first",
			body: `{"model":"claimed-model","model":"ordinary-model"}`,
		},
		{
			name: "claimed model second",
			body: `{"model":"ordinary-model","model":"claimed-model"}`,
		},
	}
	for _, testCase := range tests {
		t.Run(testCase.name, func(t *testing.T) {
			reachedDownstream := false
			router := gin.New()
			router.POST(
				"/v1/responses",
				PinTaskPluginEndpoint(),
				PrepareTaskPluginEndpoint(),
				func(c *gin.Context) {
					reachedDownstream = true
					c.Status(http.StatusNoContent)
				},
			)
			request := httptest.NewRequest(http.MethodPost, "/v1/responses", strings.NewReader(testCase.body))
			request.Header.Set("Content-Type", "application/json")
			recorder := httptest.NewRecorder()

			router.ServeHTTP(recorder, request)

			assert.False(t, reachedDownstream)
			assert.Equal(t, http.StatusBadRequest, recorder.Code)
		})
	}
}

func TestTaskPluginEndpointOnlyPreservesConditionalMiddlewareSemantics(t *testing.T) {
	tests := []struct {
		name               string
		claimed            bool
		wrappedAborts      bool
		expectedWrapped    int
		expectedDownstream bool
		expectedStatus     int
	}{
		{
			name:               "unclaimed skips wrapper",
			expectedDownstream: true,
			expectedStatus:     http.StatusNoContent,
		},
		{
			name:               "claimed invokes wrapper",
			claimed:            true,
			expectedWrapped:    1,
			expectedDownstream: true,
			expectedStatus:     http.StatusNoContent,
		},
		{
			name:            "claimed wrapper aborts",
			claimed:         true,
			wrappedAborts:   true,
			expectedWrapped: 1,
			expectedStatus:  http.StatusTooManyRequests,
		},
	}
	for _, testCase := range tests {
		t.Run(testCase.name, func(t *testing.T) {
			wrappedCalls := 0
			reachedDownstream := false
			router := gin.New()
			router.POST(
				"/v1/videos",
				func(c *gin.Context) {
					if testCase.claimed {
						c.Set(jsplugin.ContextKeyPinnedEndpoint, jsplugin.PinnedEndpoint{})
					}
					c.Next()
				},
				TaskPluginEndpointOnly(func(c *gin.Context) {
					wrappedCalls++
					if testCase.wrappedAborts {
						c.AbortWithStatus(http.StatusTooManyRequests)
						return
					}
					c.Next()
				}),
				func(c *gin.Context) {
					reachedDownstream = true
					c.Status(http.StatusNoContent)
				},
			)
			recorder := httptest.NewRecorder()

			router.ServeHTTP(recorder, httptest.NewRequest(http.MethodPost, "/v1/videos", nil))

			assert.Equal(t, testCase.expectedWrapped, wrappedCalls)
			assert.Equal(t, testCase.expectedDownstream, reachedDownstream)
			assert.Equal(t, testCase.expectedStatus, recorder.Code)
		})
	}
}

func TestPrepareTaskPluginEndpointRejectsModelDriftBeforeDistribution(t *testing.T) {
	const key = "endpoint-drift-test"
	_, err := jsplugin.DefaultRegistry.Register(taskProtocolPluginSource(
		key,
		"1.0.0",
		`["claimed-model"]`,
		"/v1/responses",
		`return {model: "outside-model"};`,
	), jsplugin.Options{})
	require.NoError(t, err)
	t.Cleanup(func() { require.NoError(t, jsplugin.DefaultRegistry.Unregister(key)) })

	reachedDistribution := false
	router := gin.New()
	router.POST(
		"/v1/responses",
		PinTaskPluginEndpoint(),
		PrepareTaskPluginEndpoint(),
		func(c *gin.Context) {
			reachedDistribution = true
			c.Status(http.StatusNoContent)
		},
	)
	request := httptest.NewRequest(
		http.MethodPost,
		"/v1/responses",
		strings.NewReader(`{"model":"claimed-model"}`),
	)
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.False(t, reachedDistribution)
	assert.Equal(t, http.StatusBadRequest, recorder.Code)
	var payload map[string]any
	require.NoError(t, common.Unmarshal(recorder.Body.Bytes(), &payload))
	errObj, _ := payload["error"].(map[string]any)
	require.NotNil(t, errObj)
	assert.Contains(t, fmt.Sprint(errObj["message"]), `model "outside-model" is not served by this plugin`)
}

func TestPrepareTaskPluginEndpointAcceptsRegisteredVideoMultipartBody(t *testing.T) {
	const key = "endpoint-multipart-test"
	_, err := jsplugin.DefaultRegistry.Register(taskProtocolPluginSource(
		key,
		"1.0.0",
		`["video-model"]`,
		"/v1/videos",
		`if (ctx.body.kind !== "multipart" || ctx.body.fields.prompt[0] !== "hello") throw new Error("bad prompt");
		 if (ctx.body.files.length !== 1 || ctx.body.files[0].field !== "input_reference") throw new Error("bad file ref");
		 return {model: ctx.model, requestBody: {prompt: ctx.body.fields.prompt[0]}};`,
	), jsplugin.Options{})
	require.NoError(t, err)
	t.Cleanup(func() { require.NoError(t, jsplugin.DefaultRegistry.Unregister(key)) })

	var body bytes.Buffer
	writer := multipart.NewWriter(&body)
	require.NoError(t, writer.WriteField("model", "video-model"))
	require.NoError(t, writer.WriteField("prompt", "hello"))
	file, err := writer.CreateFormFile("input_reference", "reference.bin")
	require.NoError(t, err)
	_, err = file.Write([]byte("opaque-video-reference"))
	require.NoError(t, err)
	require.NoError(t, writer.Close())

	router := gin.New()
	reachedDistribution := false
	router.POST(
		"/v1/videos",
		PinTaskPluginEndpoint(),
		PrepareTaskPluginEndpoint(),
		func(c *gin.Context) {
			reachedDistribution = true
			c.Status(http.StatusNoContent)
		},
	)
	request := httptest.NewRequest(http.MethodPost, "/v1/videos", bytes.NewReader(body.Bytes()))
	request.Header.Set("Content-Type", writer.FormDataContentType())
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.True(t, reachedDistribution)
	assert.Equal(t, http.StatusNoContent, recorder.Code)
}

// The OpenAI Images edits endpoint accepts multipart uploads. Every file of a
// repeated image[] field is exposed to the decoder with its own ref, and an
// unclaimed model on the shared endpoint still reaches the ordinary relay.
func TestPrepareTaskPluginEndpointExposesIndexedRefsForRepeatedImageFiles(t *testing.T) {
	const key = "endpoint-image-edits-test"
	_, err := jsplugin.DefaultRegistry.Register(taskProtocolPluginSource(
		key,
		"1.0.0",
		`["image-edit-model"]`,
		"/v1/images/edits",
		`if (ctx.body.kind !== "multipart" || ctx.body.fields.prompt[0] !== "watercolor") throw new Error("bad prompt");
		 return {model: ctx.model, action: "image_to_image", requestBody: {prompt: ctx.body.fields.prompt[0], refs: ctx.body.files.map(function(file) { return file.ref; })}};`,
	), jsplugin.Options{})
	require.NoError(t, err)
	t.Cleanup(func() { require.NoError(t, jsplugin.DefaultRegistry.Unregister(key)) })

	var body bytes.Buffer
	writer := multipart.NewWriter(&body)
	require.NoError(t, writer.WriteField("model", "image-edit-model"))
	require.NoError(t, writer.WriteField("prompt", "watercolor"))
	for _, name := range []string{"first.png", "second.png"} {
		file, createErr := writer.CreateFormFile("image[]", name)
		require.NoError(t, createErr)
		_, err = file.Write([]byte(name))
		require.NoError(t, err)
	}
	require.NoError(t, writer.Close())

	router := gin.New()
	var pinnedProtocol string
	var taskRequest any
	router.POST("/v1/images/edits", PinTaskPluginEndpoint(), PrepareTaskPluginEndpoint(), func(c *gin.Context) {
		if pinned, ok := c.MustGet(jsplugin.ContextKeyPinnedEndpoint).(jsplugin.PinnedEndpoint); ok {
			pinnedProtocol = pinned.Protocol
		}
		taskRequest, _ = c.Get("task_request")
		c.Status(http.StatusNoContent)
	})
	request := httptest.NewRequest(http.MethodPost, "/v1/images/edits", bytes.NewReader(body.Bytes()))
	request.Header.Set("Content-Type", writer.FormDataContentType())
	recorder := httptest.NewRecorder()
	router.ServeHTTP(recorder, request)

	require.Equal(t, http.StatusNoContent, recorder.Code, recorder.Body.String())
	assert.Equal(t, jsplugin.ProtocolOpenAIImage, pinnedProtocol)
	decoded, ok := taskRequest.(map[string]any)
	require.True(t, ok)
	assert.Equal(t, []any{"request_file:image[]", "request_file:image[]#1"}, decoded["refs"])

	unclaimed := gin.New()
	reachedRelay := false
	unclaimed.POST("/v1/images/generations", PinTaskPluginEndpoint(), PrepareTaskPluginEndpoint(), func(c *gin.Context) {
		_, pinned := c.Get(jsplugin.ContextKeyPinnedEndpoint)
		reachedRelay = !pinned
		c.Status(http.StatusNoContent)
	})
	plain := httptest.NewRequest(http.MethodPost, "/v1/images/generations", strings.NewReader(`{"model":"dall-e-3","prompt":"a cat"}`))
	plain.Header.Set("Content-Type", "application/json")
	recorder = httptest.NewRecorder()
	unclaimed.ServeHTTP(recorder, plain)
	assert.Equal(t, http.StatusNoContent, recorder.Code)
	assert.True(t, reachedRelay, "an unclaimed image model keeps the built-in relay")
}

func TestVideoGenerationsIsNotClaimedByOpenAIVideoProtocol(t *testing.T) {
	const key = "endpoint-video-gen-test"
	_, err := jsplugin.DefaultRegistry.Register(taskProtocolPluginSource(
		key,
		"1.0.0",
		`["generation-model"]`,
		"/v1/video/generations",
		`return {model: ctx.requestBody.model, action: "generate"};`,
	), jsplugin.Options{})
	require.NoError(t, err)
	t.Cleanup(func() { require.NoError(t, jsplugin.DefaultRegistry.Unregister(key)) })

	router := gin.New()
	reachedDistribution := false
	router.POST(
		"/v1/video/generations",
		PinTaskPluginEndpoint(),
		PrepareTaskPluginEndpoint(),
		func(c *gin.Context) {
			reachedDistribution = true
			c.Status(http.StatusNoContent)
		},
	)
	request := httptest.NewRequest(
		http.MethodPost,
		"/v1/video/generations",
		strings.NewReader(`{"model":"generation-model"}`),
	)
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.True(t, reachedDistribution)
	assert.Equal(t, http.StatusNoContent, recorder.Code)
}

func TestPrepareTaskPluginRouteKeepsRawBinaryOpaque(t *testing.T) {
	plugin := compileTaskRoutePlugin(t, `
export const meta = {
  apiVersion: 1, key: "route-binary-test", name: "Binary", version: "1.0.0",
  author: {name: "Test"},
  models: ["binary-model"], fetchMode: "per_task",
  routes: [{method: "POST", path: "/vendor/binary", type: "submit", decode: "decodeBinary", render: "created"}],
};
export const native = {decodeBinary: function() { throw new Error("decoder must not run"); }, created: function(ctx, task) { return task; }};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
`)
	router := gin.New()
	router.POST("/vendor/binary", pinTaskPluginRoute(plugin, 0), PrepareTaskPluginRoute(), func(c *gin.Context) {
		body, err := io.ReadAll(c.Request.Body)
		require.NoError(t, err)
		assert.Equal(t, []byte{0, 1, 2, 3}, body)
		c.Status(http.StatusNoContent)
	})
	request := httptest.NewRequest(http.MethodPost, "/vendor/binary", bytes.NewReader([]byte{0, 1, 2, 3}))
	request.Header.Set("Content-Type", "application/octet-stream")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.Equal(t, http.StatusUnsupportedMediaType, recorder.Code)
}

func TestPrepareTaskPluginDynamicQueryPreservesOrderAndSkipsNextHandlers(t *testing.T) {
	setupTaskPluginRouteDB(t)
	plugin := compileTaskRoutePlugin(t, `
export const meta = {
  apiVersion: 1, key: "route-query-test", name: "Query", version: "1.0.0",
  author: {name: "Test"},
  models: ["query-model"], fetchMode: "per_task",
  routes: [{method: "POST", path: "/vendor/query", type: "dynamic", decode: "decodeQuery", render: "renderQuery"}],
};
export const native = {
  decodeQuery: function() { return {kind: "query", taskIds: ["task-b", "task-a", "task-b"]}; },
  renderQuery: function(ctx, tasks) {
    return {ids: tasks.map(function(task) { return task.task_id; }), keys: Object.keys(tasks[0]).sort(), data: tasks[0].data};
  },
};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
`)
	insertTaskPluginRouteTask(t, &model.Task{
		TaskID: "task-a", UserId: 7, Platform: constant.TaskPlatform("route-query-test"),
		Status: model.TaskStatusSuccess, Progress: "100%", CreatedAt: 10,
	})
	taskData, err := common.Marshal(map[string]any{
		"task_id": "private-upstream-id",
		"nested":  map[string]any{"url": "https://upstream.invalid/tasks/private-upstream-id"},
	})
	require.NoError(t, err)
	insertTaskPluginRouteTask(t, &model.Task{
		TaskID: "task-b", UserId: 7, Platform: constant.TaskPlatform("route-query-test"),
		Status: model.TaskStatusInProgress, Progress: "50%", CreatedAt: 20, Data: taskData,
		ChannelId: 999, Quota: 12345, PrivateData: model.TaskPrivateData{UpstreamTaskID: "private-upstream-id", Key: "secret"},
	})

	nextHandlerCalled := false
	router := gin.New()
	router.POST("/vendor/query",
		pinTaskPluginRoute(plugin, 0),
		func(c *gin.Context) {
			c.Set("id", 7)
			c.Next()
		},
		PrepareTaskPluginRoute(),
		func(c *gin.Context) {
			nextHandlerCalled = true
			c.Status(http.StatusTeapot)
		},
	)
	recorder := httptest.NewRecorder()
	request := httptest.NewRequest(http.MethodPost, "/vendor/query", strings.NewReader(`{}`))
	request.Header.Set("Content-Type", "application/json")
	router.ServeHTTP(recorder, request)

	assert.Equal(t, http.StatusOK, recorder.Code)
	assert.False(t, nextHandlerCalled)
	var response map[string]any
	require.NoError(t, common.Unmarshal(recorder.Body.Bytes(), &response))
	assert.Equal(t, []any{"task-b", "task-a", "task-b"}, response["ids"])
	data, ok := response["data"].(map[string]any)
	require.True(t, ok)
	assert.Equal(t, "task-b", data["task_id"])
	nested, ok := data["nested"].(map[string]any)
	require.True(t, ok)
	assert.Equal(t, "https://upstream.invalid/tasks/private-upstream-id", nested["url"])
	keys, ok := response["keys"].([]any)
	require.True(t, ok)
	assert.NotContains(t, keys, "user_id")
	assert.NotContains(t, keys, "channel_id")
	assert.NotContains(t, keys, "quota")
	assert.NotContains(t, keys, "private_data")
	assert.NotContains(t, recorder.Body.String(), "secret")
}

func TestPrepareTaskPluginDynamicDecoderRejectsRendererField(t *testing.T) {
	plugin := compileTaskRoutePlugin(t, `
export const meta = {apiVersion:1,key:"dynamic-renderer",name:"Dynamic",version:"1.0.0",author:{name:"Test"},models:["model"],fetchMode:"per_task",routes:[{method:"POST",path:"/vendor/query",type:"dynamic",decode:"decode",render:"show"}]};
export const native = {decode:function(){return {kind:"query",taskIds:[],renderer:"legacy"};},show:function(){return {};}};
export function buildSubmitRequest(){return {}} export function parseSubmitResponse(){return {taskId:"one"}} export function buildQueryRequest(){return {}} export function parseTaskResult(){return {status:"SUCCESS"}}
`)
	router := gin.New()
	reached := false
	router.POST("/vendor/query", pinTaskPluginRoute(plugin, 0), PrepareTaskPluginRoute(), func(c *gin.Context) { reached = true })
	request := httptest.NewRequest(http.MethodPost, "/vendor/query", strings.NewReader(`{}`))
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.False(t, reached)
	assert.Equal(t, http.StatusBadRequest, recorder.Code)
}

func TestPrepareTaskPluginSubmitDecoderRejectsRendererField(t *testing.T) {
	plugin := compileTaskRoutePlugin(t, `
export const meta = {apiVersion:1,key:"submit-renderer",name:"Submit",version:"1.0.0",author:{name:"Test"},models:["model"],fetchMode:"per_task",routes:[{method:"POST",path:"/vendor/submit",type:"submit",decode:"decode",render:"created"}]};
export const native = {decode:function(){return {kind:"submit",model:"model",requestBody:{},renderer:"legacy"};},created:function(){return {};}};
export function buildSubmitRequest(){return {}} export function parseSubmitResponse(){return {taskId:"one"}} export function buildQueryRequest(){return {}} export function parseTaskResult(){return {status:"SUCCESS"}}
`)
	router := gin.New()
	reached := false
	router.POST("/vendor/submit", pinTaskPluginRoute(plugin, 0), PrepareTaskPluginRoute(), func(c *gin.Context) { reached = true })
	request := httptest.NewRequest(http.MethodPost, "/vendor/submit", strings.NewReader(`{}`))
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.False(t, reached)
	assert.Equal(t, http.StatusBadRequest, recorder.Code)
}

func TestPrepareTaskPluginStaticQueryHidesTaskExistenceAndSanitizesErrors(t *testing.T) {
	setupTaskPluginRouteDB(t)
	plugin := compileTaskRoutePlugin(t, `
export const meta = {
  apiVersion: 1, key: "route-static-query-test", name: "Static Query", version: "1.0.0",
  author: {name: "Test"},
  channelTypes: [651], models: ["query-model"], fetchMode: "per_task",
  routes: [{method: "GET", path: "/vendor/jobs/:id", type: "query", taskIdParam: "id", render: "status"}],
};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
export const native = {status: function(ctx, task) { return {id: task.task_id}; }, error: function(ctx, error) {
	return {vendor_error: {code: error.code, message: error.message, status: error.httpStatus, retryable: error.retryable}};
}};
`)
	insertTaskPluginRouteTask(t, &model.Task{
		TaskID: "foreign-task", UserId: 8, Platform: constant.TaskPlatform("route-static-query-test"),
	})
	insertTaskPluginRouteTask(t, &model.Task{
		TaskID: "wrong-platform", UserId: 7, Platform: constant.TaskPlatform("another-plugin"),
	})
	insertTaskPluginRouteTask(t, &model.Task{
		TaskID: "wrong-legacy-platform", UserId: 7, Platform: constant.TaskPlatform("652"),
	})
	insertTaskPluginRouteTask(t, &model.Task{
		TaskID: "legacy-task", UserId: 7, Platform: constant.TaskPlatform("651"),
	})
	// Submitted through a route with retainResult: false; owned by the caller
	// but not retrievable, so it must be indistinguishable from a missing task.
	insertTaskPluginRouteTask(t, &model.Task{
		TaskID: "discarded-task", UserId: 7, Platform: constant.TaskPlatform("route-static-query-test"),
		Status: model.TaskStatusSuccess, PrivateData: model.TaskPrivateData{ResultDiscarded: true},
	})

	router := gin.New()
	router.GET("/vendor/jobs/:id",
		pinTaskPluginRoute(plugin, 0),
		func(c *gin.Context) {
			c.Set("id", 7)
			c.Next()
		},
		PrepareTaskPluginRoute(),
	)

	legacyRecorder := httptest.NewRecorder()
	router.ServeHTTP(legacyRecorder, httptest.NewRequest(http.MethodGet, "/vendor/jobs/legacy-task", nil))
	assert.Equal(t, http.StatusOK, legacyRecorder.Code)
	assert.JSONEq(t, `{"id":"legacy-task"}`, legacyRecorder.Body.String())

	var firstBody string
	for _, taskID := range []string{"missing-task", "foreign-task", "wrong-platform", "wrong-legacy-platform", "discarded-task"} {
		recorder := httptest.NewRecorder()
		router.ServeHTTP(recorder, httptest.NewRequest(http.MethodGet, "/vendor/jobs/"+taskID, nil))
		assert.Equal(t, http.StatusNotFound, recorder.Code)
		if firstBody == "" {
			firstBody = recorder.Body.String()
		} else {
			assert.JSONEq(t, firstBody, recorder.Body.String())
		}
		assert.Contains(t, recorder.Body.String(), `"code":"task_not_found"`)
		assert.NotContains(t, recorder.Body.String(), taskID)
	}
}

func TestRespondTaskPluginErrorNeverExposesInternalDetails(t *testing.T) {
	plugin := compileTaskRoutePlugin(t, `
export const meta = {
  apiVersion: 1, key: "route-error-test", name: "Error", version: "1.0.0",
  author: {name: "Test"},
  models: ["error-model"], fetchMode: "per_task",
};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
export const native = {error: function(ctx, error) {
	return {path: ctx.path, code: error.code, message: error.message, status: error.httpStatus, retryable: error.retryable, keys: Object.keys(error).sort()};
}};
`)
	recorder := httptest.NewRecorder()
	c, _ := gin.CreateTestContext(recorder)
	c.Request = httptest.NewRequest(http.MethodPost, "/vendor/failure", strings.NewReader(`{"credential":"do-not-copy"}`))
	c.Set(jsplugin.ContextKeyPinnedRoute, jsplugin.PinnedRoute{Plugin: plugin})
	c.Set(jsplugin.ContextKeyRouteRequest, jsplugin.RouteRequestContext{
		Path: "/vendor/failure", Method: http.MethodPost, Params: map[string]string{}, Query: map[string][]string{},
	})

	handled := RespondTaskPluginError(c, &dto.TaskError{
		Code:       "upstream_credential_failure",
		Message:    "https://user:[email protected]?token=secret",
		Data:       map[string]any{"authorization": "Bearer secret"},
		StatusCode: http.StatusBadGateway,
		Error:      assert.AnError,
	})

	assert.True(t, handled)
	assert.Equal(t, http.StatusBadGateway, recorder.Code)
	assert.JSONEq(t, `{
	  "path": "/vendor/failure",
	  "code": "server_error",
	  "message": "Task request failed",
	  "status": 502,
	  "retryable": true,
	  "keys": ["code", "httpStatus", "message", "requestId", "retryable"]
	}`, recorder.Body.String())
	assert.NotContains(t, recorder.Body.String(), "upstream.invalid")
	assert.NotContains(t, recorder.Body.String(), "secret")
	assert.NotContains(t, recorder.Body.String(), "password")
}

func TestTaskPluginErrorFallbackIsSanitized(t *testing.T) {
	for _, testCase := range []struct {
		name       string
		renderHook string
	}{
		{name: "missing hook"},
		{name: "throwing hook", renderHook: `export const native = {error: function() { throw new Error("renderer secret"); }};`},
	} {
		t.Run(testCase.name, func(t *testing.T) {
			plugin := compileTaskRoutePlugin(t, genericTaskPluginSource+"\n"+testCase.renderHook)
			recorder := httptest.NewRecorder()
			c, _ := gin.CreateTestContext(recorder)
			c.Request = httptest.NewRequest(http.MethodPost, "/vendor/failure", nil)
			c.Set(common.RequestIdKey, "fallback-req")
			c.Set(jsplugin.ContextKeyPinnedRoute, jsplugin.PinnedRoute{Plugin: plugin})
			c.Set(jsplugin.ContextKeyRouteRequest, jsplugin.RouteRequestContext{
				Path: "/vendor/failure", Method: http.MethodPost,
				Params: map[string]string{}, Query: map[string][]string{},
			})

			abortWithOpenAiMessage(c, http.StatusBadGateway, "https://user:[email protected]?token=secret")

			assert.Equal(t, http.StatusBadGateway, recorder.Code)
			assert.JSONEq(t, `{"code":"server_error","message":"Task request failed (request id: fallback-req)","data":null}`, recorder.Body.String())
			assert.NotContains(t, recorder.Body.String(), "upstream.invalid")
			assert.NotContains(t, recorder.Body.String(), "renderer secret")
			assert.NotContains(t, recorder.Body.String(), "password")
		})
	}
}

func TestResolvedTaskPluginIDsEnforcesPublicQueryContract(t *testing.T) {
	valid, ok := resolvedTaskPluginIDs([]any{"task-a", "task-b", "task-a"})
	require.True(t, ok)
	assert.Equal(t, []string{"task-a", "task-b", "task-a"}, valid)

	empty, ok := resolvedTaskPluginIDs([]any{})
	require.True(t, ok)
	assert.Empty(t, empty)

	tooMany := make([]any, 101)
	for index := range tooMany {
		tooMany[index] = "task"
	}
	for _, invalid := range []any{
		[]any{"task-a", ""},
		[]any{"task-a", "   "},
		[]any{"task-a", 2},
		tooMany,
		"task-a",
	} {
		_, ok = resolvedTaskPluginIDs(invalid)
		assert.False(t, ok)
	}
}

func TestSunoFetchEmptyIDsReturnsSuccessfulEmptyArray(t *testing.T) {
	source, err := builtinplugins.Source("sunoapi")
	require.NoError(t, err)
	plugin := compileTaskRoutePlugin(t, source)

	nextHandlerCalled := false
	router := gin.New()
	router.POST(
		"/suno/fetch",
		pinTaskPluginRoute(plugin, 1),
		PrepareTaskPluginRoute(),
		func(c *gin.Context) {
			nextHandlerCalled = true
			c.Status(http.StatusTeapot)
		},
	)
	request := httptest.NewRequest(http.MethodPost, "/suno/fetch", strings.NewReader(`{"ids":[]}`))
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.False(t, nextHandlerCalled)
	assert.Equal(t, http.StatusOK, recorder.Code)
	assert.JSONEq(t, `{"code":"success","message":"","data":[]}`, recorder.Body.String())
}

func TestPrepareTaskPluginRouteSurfacesDecodeHookMessage(t *testing.T) {
	plugin := compileTaskRoutePlugin(t, `
export const meta = {
  apiVersion: 1, key: "route-decode-detail-test", name: "Decode", version: "1.0.0",
  author: {name: "Test"},
  models: ["detail-model"], fetchMode: "per_task",
  routes: [{method: "POST", path: "/vendor/jobs", type: "submit", decode: "createTask", render: "created"}],
};
export const native = {createTask: function() { throw new Error("model is required"); }, created: function(ctx, task) { return task; }};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
`)
	router := gin.New()
	router.POST("/vendor/jobs", pinTaskPluginRoute(plugin, 0), PrepareTaskPluginRoute(), func(c *gin.Context) {
		c.Status(http.StatusNoContent)
	})
	request := httptest.NewRequest(http.MethodPost, "/vendor/jobs", strings.NewReader(`{"prompt":"x"}`))
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.Equal(t, http.StatusBadRequest, recorder.Code)
	var body dto.TaskError
	require.NoError(t, common.Unmarshal(recorder.Body.Bytes(), &body))
	assert.Equal(t, "invalid_request", body.Code)
	assert.Equal(t, "model is required", body.Message)
}

func TestPrepareTaskPluginRouteNativeErrorReceivesHookMessage(t *testing.T) {
	plugin := compileTaskRoutePlugin(t, `
export const meta = {
  apiVersion: 1, key: "route-error-detail-test", name: "Error", version: "1.0.0",
  author: {name: "Test"},
  models: ["detail-model"], fetchMode: "per_task",
  routes: [{method: "POST", path: "/vendor/jobs", type: "submit", decode: "createTask", render: "created"}],
};
export const native = {
  createTask: function() { throw new Error("model is required"); },
  created: function(ctx, task) { return task; },
  error: function(ctx, error) { return {code: error.code, message: error.message, requestId: error.requestId}; },
};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
`)
	router := gin.New()
	router.Use(func(c *gin.Context) {
		c.Set(common.RequestIdKey, "native-error-req")
		c.Next()
	})
	router.POST("/vendor/jobs", pinTaskPluginRoute(plugin, 0), PrepareTaskPluginRoute(), func(c *gin.Context) {
		c.Status(http.StatusNoContent)
	})
	request := httptest.NewRequest(http.MethodPost, "/vendor/jobs", strings.NewReader(`{"prompt":"x"}`))
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.Equal(t, http.StatusBadRequest, recorder.Code)
	assert.JSONEq(t, `{"code":"invalid_request","message":"model is required","requestId":"native-error-req"}`, recorder.Body.String())
}

func TestPrepareTaskPluginRouteRejectsNonObjectResultWithFixedMessage(t *testing.T) {
	plugin := compileTaskRoutePlugin(t, `
export const meta = {
  apiVersion: 1, key: "route-result-object-test", name: "Result", version: "1.0.0",
  author: {name: "Test"},
  models: ["detail-model"], fetchMode: "per_task",
  routes: [{method: "POST", path: "/vendor/jobs", type: "submit", decode: "createTask", render: "created"}],
};
export const native = {createTask: function() { return "not-an-object"; }, created: function(ctx, task) { return task; }};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
`)
	router := gin.New()
	router.POST("/vendor/jobs", pinTaskPluginRoute(plugin, 0), PrepareTaskPluginRoute(), func(c *gin.Context) {
		c.Status(http.StatusNoContent)
	})
	request := httptest.NewRequest(http.MethodPost, "/vendor/jobs", strings.NewReader(`{"model":"detail-model"}`))
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.Equal(t, http.StatusBadRequest, recorder.Code)
	var body dto.TaskError
	require.NoError(t, common.Unmarshal(recorder.Body.Bytes(), &body))
	assert.Equal(t, "invalid_request", body.Code)
	assert.Equal(t, "plugin returned an invalid route result", body.Message)
}

func TestPrepareTaskPluginRouteSurfacesRequestDecodeDetail(t *testing.T) {
	plugin := compileTaskRoutePlugin(t, `
export const meta = {
  apiVersion: 1, key: "route-decode-body-test", name: "Decode", version: "1.0.0",
  author: {name: "Test"},
  models: ["detail-model"], fetchMode: "per_task",
  routes: [{method: "POST", path: "/vendor/jobs", type: "submit", decode: "createTask", render: "created"}],
};
export const native = {createTask: function() { throw new Error("decoder must not run"); }, created: function(ctx, task) { return task; }};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
`)
	router := gin.New()
	router.POST("/vendor/jobs", pinTaskPluginRoute(plugin, 0), PrepareTaskPluginRoute(), func(c *gin.Context) {
		c.Status(http.StatusNoContent)
	})
	request := httptest.NewRequest(http.MethodPost, "/vendor/jobs", strings.NewReader(`{}`))
	request.Header["Content-Type"] = []string{"application/json", "application/x-www-form-urlencoded"}
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.Equal(t, http.StatusBadRequest, recorder.Code)
	var body dto.TaskError
	require.NoError(t, common.Unmarshal(recorder.Body.Bytes(), &body))
	assert.Equal(t, "invalid_request", body.Code)
	assert.Contains(t, body.Message, "conflicting Content-Type")
}

func TestSanitizedTaskPluginErrorIgnoresDetailOn5xx(t *testing.T) {
	got := sanitizedTaskPluginError(http.StatusInternalServerError, "database secret")
	assert.Equal(t, "server_error", got.Code)
	assert.Equal(t, "Task request failed", got.Message)
	assert.Equal(t, http.StatusInternalServerError, got.HTTPStatus)

	got = sanitizedTaskPluginError(http.StatusBadGateway, "https://user:[email protected]")
	assert.Equal(t, "server_error", got.Code)
	assert.Equal(t, "Task request failed", got.Message)
}

func TestTaskPluginErrorFallbackMessageIncludesRequestID(t *testing.T) {
	plugin := compileTaskRoutePlugin(t, genericTaskPluginSource)
	recorder := httptest.NewRecorder()
	c, _ := gin.CreateTestContext(recorder)
	c.Request = httptest.NewRequest(http.MethodPost, "/vendor/failure", nil)
	c.Set(common.RequestIdKey, "req-fallback-1")
	c.Set(jsplugin.ContextKeyPinnedRoute, jsplugin.PinnedRoute{Plugin: plugin})
	c.Set(jsplugin.ContextKeyRouteRequest, jsplugin.RouteRequestContext{
		Path: "/vendor/failure", Method: http.MethodPost,
		Params: map[string]string{}, Query: map[string][]string{},
	})

	abortTaskPluginRouteErrorDetail(c, http.StatusBadRequest, "model is required")

	assert.Equal(t, http.StatusBadRequest, recorder.Code)
	var body dto.TaskError
	require.NoError(t, common.Unmarshal(recorder.Body.Bytes(), &body))
	assert.Equal(t, "invalid_request", body.Code)
	assert.Equal(t, "model is required (request id: req-fallback-1)", body.Message)
	assert.True(t, strings.HasSuffix(body.Message, "(request id: req-fallback-1)"))
}

func TestPrepareTaskPluginEndpointSurfacesDecodeHookMessage(t *testing.T) {
	const key = "endpoint-decode-detail-test"
	_, err := jsplugin.DefaultRegistry.Register(taskProtocolPluginSource(
		key,
		"1.0.0",
		`["claimed-model"]`,
		"/v1/responses",
		`throw new Error("model is required");`,
	), jsplugin.Options{})
	require.NoError(t, err)
	t.Cleanup(func() { require.NoError(t, jsplugin.DefaultRegistry.Unregister(key)) })

	router := gin.New()
	router.POST("/v1/responses", PinTaskPluginEndpoint(), PrepareTaskPluginEndpoint(), func(c *gin.Context) {
		c.Status(http.StatusNoContent)
	})
	request := httptest.NewRequest(http.MethodPost, "/v1/responses", strings.NewReader(`{"model":"claimed-model"}`))
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()

	router.ServeHTTP(recorder, request)

	assert.Equal(t, http.StatusBadRequest, recorder.Code)
	assert.Contains(t, recorder.Body.String(), "model is required")
	assert.NotContains(t, recorder.Body.String(), "Invalid task protocol request")
}

func TestPinTaskPluginEndpointRejectsUnsupportedRequestForms(t *testing.T) {
	setupTaskPluginRouteDB(t)
	tests := []struct {
		name     string
		key      string
		supports string
		hooks    string
		body     string
		message  string
	}{
		{
			name:     "stream against sync and background",
			key:      "form-gate-final-only",
			supports: `["sync", "background"]`,
			hooks:    `renderFinal: function() { return {}; }`,
			body:     `{"model":"form-gate-model","stream":true}`,
			message:  `Streaming is not supported for this model. Set "stream": false, or use "background": true and retrieve the response later.`,
		},
		{
			name:     "sync against stream only",
			key:      "form-gate-stream-only",
			supports: `["stream"]`,
			hooks:    `renderEvents: function() { return {events: [], done: false}; }`,
			body:     `{"model":"form-gate-model"}`,
			message:  `Synchronous non-streaming requests are not supported for this model. Set "stream": true.`,
		},
		{
			name:     "background against stream and sync",
			key:      "form-gate-no-background",
			supports: `["stream", "sync"]`,
			hooks:    `renderEvents: function() { return {events: [], done: false}; }, renderFinal: function() { return {}; }`,
			body:     `{"model":"form-gate-model","background":true}`,
			message:  `Background mode is not supported for this model. Remove "background": true.`,
		},
		{
			name:     "background plus stream reports stream first",
			key:      "form-gate-no-stream",
			supports: `["sync", "background"]`,
			hooks:    `renderFinal: function() { return {}; }`,
			body:     `{"model":"form-gate-model","stream":true,"background":true}`,
			message:  `Streaming is not supported for this model. Set "stream": false, or use "background": true and retrieve the response later.`,
		},
	}
	for _, testCase := range tests {
		t.Run(testCase.name, func(t *testing.T) {
			_, err := jsplugin.DefaultRegistry.Register(taskResponsesPluginSource(
				testCase.key, 0, `["form-gate-model"]`, testCase.supports, testCase.hooks, `return {model: ctx.model};`,
			), jsplugin.Options{})
			require.NoError(t, err)
			t.Cleanup(func() { require.NoError(t, jsplugin.DefaultRegistry.Unregister(testCase.key)) })

			reachedPrepare := false
			quotaConsumed := false
			router := gin.New()
			router.POST("/v1/responses", PinTaskPluginEndpoint(), PrepareTaskPluginEndpoint(), func(c *gin.Context) {
				reachedPrepare = true
				quotaConsumed = true
				require.NoError(t, model.DB.Create(&model.Task{TaskID: "should-not-exist", UserId: 1}).Error)
				c.Status(http.StatusNoContent)
			})
			request := httptest.NewRequest(http.MethodPost, "/v1/responses", strings.NewReader(testCase.body))
			request.Header.Set("Content-Type", "application/json")
			recorder := httptest.NewRecorder()
			router.ServeHTTP(recorder, request)

			assert.Equal(t, http.StatusBadRequest, recorder.Code)
			var payload map[string]any
			require.NoError(t, common.Unmarshal(recorder.Body.Bytes(), &payload))
			errObj, ok := payload["error"].(map[string]any)
			require.True(t, ok)
			message, _ := errObj["message"].(string)
			assert.Contains(t, message, testCase.message)
			assert.False(t, reachedPrepare)
			assert.False(t, quotaConsumed)
			var count int64
			require.NoError(t, model.DB.Model(&model.Task{}).Count(&count).Error)
			assert.Zero(t, count)
		})
	}
}

func TestPinTaskPluginEndpointMalformedStreamStillFailsInPrepare(t *testing.T) {
	const key = "form-gate-malformed-stream"
	_, err := jsplugin.DefaultRegistry.Register(taskResponsesPluginSource(
		key, 0, `["form-gate-bool-model"]`, `["sync", "background"]`,
		`renderFinal: function() { return {}; }`,
		`return {model: ctx.model};`,
	), jsplugin.Options{})
	require.NoError(t, err)
	t.Cleanup(func() { require.NoError(t, jsplugin.DefaultRegistry.Unregister(key)) })

	reachedNext := false
	router := gin.New()
	router.POST("/v1/responses", PinTaskPluginEndpoint(), PrepareTaskPluginEndpoint(), func(c *gin.Context) {
		reachedNext = true
		c.Status(http.StatusNoContent)
	})
	request := httptest.NewRequest(http.MethodPost, "/v1/responses", strings.NewReader(`{"model":"form-gate-bool-model","stream":"yes"}`))
	request.Header.Set("Content-Type", "application/json")
	recorder := httptest.NewRecorder()
	router.ServeHTTP(recorder, request)

	assert.Equal(t, http.StatusBadRequest, recorder.Code)
	assert.Contains(t, recorder.Body.String(), "stream must be a boolean")
	assert.False(t, reachedNext)
}

func TestPinTaskPluginEndpointMovesParserToSurvivingSharedCandidate(t *testing.T) {
	streamOnly, err := jsplugin.DefaultRegistry.Register(taskResponsesPluginSource(
		"alpha-stream", constant.ChannelTypeReplicate, `["shared-form-model"]`, `["stream"]`,
		`renderEvents: function() { return {events: [], done: false}; }`,
		`return {model: ctx.model, action: "stream-parser"};`,
	), jsplugin.Options{})
	require.NoError(t, err)
	t.Cleanup(func() { require.NoError(t, jsplugin.DefaultRegistry.Unregister("alpha-stream")) })
	full, err := jsplugin.DefaultRegistry.Register(taskResponsesPluginSource(
		"bravo-full", constant.ChannelTypeCodex, `["shared-form-model"]`, `["stream", "sync", "background"]`,
		`renderEvents: function() { return {events: [], done: false}; }, renderFinal: function() { return {}; }`,
		`return {model: ctx.model, action: "full-parser"};`,
	), jsplugin.Options{})
	require.NoError(t, err)
	t.Cleanup(func() { require.NoError(t, jsplugin.DefaultRegistry.Unregister("bravo-full")) })

	generation := jsplugin.DefaultRegistry.Generation()
	unfiltered := generation.LookupEndpointCandidates("POST", "/v1/responses", "shared-form-model")
	require.Len(t, unfiltered, 2)
	assert.Equal(t, "alpha-stream", unfiltered[0].Plugin.Meta.Key)

	t.Run("sync moves pin to second candidate", func(t *testing.T) {
		var pinned jsplugin.PinnedEndpoint
		var action string
		router := gin.New()
		router.POST("/v1/responses", PinTaskPluginEndpoint(), PrepareTaskPluginEndpoint(), func(c *gin.Context) {
			pinned = c.MustGet(jsplugin.ContextKeyPinnedEndpoint).(jsplugin.PinnedEndpoint)
			action = c.GetString("task_action")
			c.Status(http.StatusNoContent)
		})
		request := httptest.NewRequest(http.MethodPost, "/v1/responses", strings.NewReader(`{"model":"shared-form-model"}`))
		request.Header.Set("Content-Type", "application/json")
		recorder := httptest.NewRecorder()
		router.ServeHTTP(recorder, request)

		assert.Equal(t, http.StatusNoContent, recorder.Code)
		assert.Same(t, full, pinned.Plugin)
		require.Len(t, pinned.Candidates, 1)
		assert.Same(t, full, pinned.Candidates[0].Plugin)
		assert.Equal(t, "full-parser", action)
	})

	t.Run("stream keeps first candidate", func(t *testing.T) {
		var pinned jsplugin.PinnedEndpoint
		var action string
		router := gin.New()
		router.POST("/v1/responses", PinTaskPluginEndpoint(), PrepareTaskPluginEndpoint(), func(c *gin.Context) {
			pinned = c.MustGet(jsplugin.ContextKeyPinnedEndpoint).(jsplugin.PinnedEndpoint)
			action = c.GetString("task_action")
			c.Status(http.StatusNoContent)
		})
		request := httptest.NewRequest(http.MethodPost, "/v1/responses", strings.NewReader(`{"model":"shared-form-model","stream":true}`))
		request.Header.Set("Content-Type", "application/json")
		recorder := httptest.NewRecorder()
		router.ServeHTTP(recorder, request)

		assert.Equal(t, http.StatusNoContent, recorder.Code)
		assert.Same(t, streamOnly, pinned.Plugin)
		require.Len(t, pinned.Candidates, 2)
		assert.Same(t, streamOnly, pinned.Candidates[0].Plugin)
		assert.Equal(t, "stream-parser", action)
	})
}

func compileTaskRoutePlugin(t *testing.T, source string) *jsplugin.LoadedPlugin {
	t.Helper()
	plugin, err := jsplugin.CompilePlugin(source, jsplugin.Options{})
	require.NoError(t, err)
	return plugin
}

func taskResponsesPluginSource(key string, channelType int, models, supports, hooks, parseRequestBody string) string {
	channelField := ""
	if channelType > 0 {
		channelField = fmt.Sprintf("channelTypes: [%d],", channelType)
	}
	return fmt.Sprintf(`
export const meta = {
  apiVersion: 1,
  key: %q,
  name: %q,
  version: "1.0.0",
  author: {name: "Test"},
  %s
  models: %s,
  fetchMode: "per_task",
  protocols: [{name: "openai_responses", supports: %s}],
};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
export function listArtifacts() { return []; }
export function buildContentRequest() { throw new Error("artifact_not_found"); }
export const protocols = {openai_responses: {
  decodeRequest: function(ctx) {
    ctx.requestBody = ctx.body.value;
    const decode = function() { %s };
    const result = decode();
    if (!result.kind) result.kind = "submit";
    return result;
  },
  %s
}};
`, key, key, channelField, models, supports, parseRequestBody, hooks)
}

func taskProtocolPluginSource(key, version, models, endpoint, parseRequestBody string) string {
	protocol := "openai_responses"
	protocolClaim := `{name: "openai_responses", supports: ["stream", "sync", "background"]}`
	presenters := `renderEvents: function() { return {events: [], done: false}; }, renderFinal: function() { return {}; },`
	if endpoint == "/v1/videos" {
		protocol = "openai_video"
		protocolClaim = `"openai_video"`
		presenters = `render: function() { return {}; },`
	}
	if endpoint == "/v1/images/edits" || endpoint == "/v1/images/generations" {
		protocol = "openai_image"
		protocolClaim = `"openai_image"`
		presenters = `render: function() { return {data: []}; },`
	}
	return fmt.Sprintf(`
export const meta = {
  apiVersion: 1,
  key: %q,
  name: %q,
  version: %q,
  author: {name: "Test"},
  models: %s,
  fetchMode: "per_task",
  protocols: [%s],
};
export function buildSubmitRequest() { return {url: "https://example.com"}; }
export function parseSubmitResponse() { return {taskId: "one"}; }
export function buildQueryRequest() { return {url: "https://example.com"}; }
export function parseTaskResult() { return {status: "SUCCESS"}; }
export function listArtifacts() { return []; }
export function buildContentRequest() { throw new Error("artifact_not_found"); }
export const protocols = {
  %s: {
    decodeRequest: function(ctx) {
      ctx.requestBody = ctx.body.value;
      if (ctx.body.kind === "multipart") {
        ctx.requestBody = {};
        for (const key of Object.keys(ctx.body.fields || {})) ctx.requestBody[key] = ctx.body.fields[key][0];
      }
      const decode = function() { %s };
      const result = decode();
      if (!result.kind) result.kind = "submit";
      return result;
    },
		%s
  },
};
`, key, key, version, models, protocolClaim, protocol, parseRequestBody, presenters)
}

func pinTaskPluginRoute(plugin *jsplugin.LoadedPlugin, routeIndex int) gin.HandlerFunc {
	return func(c *gin.Context) {
		c.Set(jsplugin.ContextKeyPinnedRoute, jsplugin.PinnedRoute{Plugin: plugin, Route: plugin.Meta.Routes[routeIndex]})
		c.Next()
	}
}

func setupTaskPluginRouteDB(t *testing.T) {
	t.Helper()
	previousDB := model.DB
	previousType := common.MainDatabaseType()
	database, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
	require.NoError(t, err)
	require.NoError(t, database.AutoMigrate(&model.Task{}))
	model.DB = database
	common.SetMainDatabaseType(common.DatabaseTypeSQLite)
	t.Cleanup(func() {
		model.DB = previousDB
		common.SetMainDatabaseType(previousType)
	})
}

func insertTaskPluginRouteTask(t *testing.T, task *model.Task) {
	t.Helper()
	require.NoError(t, model.DB.Create(task).Error)
}

func TestPrepareTaskPluginEndpointFiltersEachSharedCandidate(t *testing.T) {
	for _, tc := range []struct {
		name, alpha, beta string
		wantKeys          []string
		wantError         string
	}{
		{name: "first decoder rejects", alpha: `throw new Error("alpha only accepts 720p")`, beta: `return {model:ctx.model,action:"beta",requestBody:{resolution:"1080p"}}`, wantKeys: []string{"decode-beta"}},
		{name: "second decoder rejects", alpha: `return {model:ctx.model,action:"alpha"}`, beta: `throw new Error("beta rejects")`, wantKeys: []string{"decode-alpha"}},
		{name: "both decoders accept", alpha: `return {model:ctx.model,action:"alpha"}`, beta: `return {model:ctx.model,action:"beta"}`, wantKeys: []string{"decode-alpha", "decode-beta"}},
		{name: "all decoders reject", alpha: `throw new Error("alpha rejects first")`, beta: `throw new Error("beta rejects second")`, wantError: "decode-alpha: alpha rejects first; decode-beta: beta rejects second"},
		{name: "duplicate failures are grouped", alpha: `throw new Error("unsupported resolution")`, beta: `throw new Error("unsupported resolution")`, wantError: "decode-alpha, decode-beta: unsupported resolution"},
		{name: "invalid result is excluded", alpha: `return {kind:"query",model:ctx.model}`, beta: `return {model:ctx.model,action:"beta"}`, wantKeys: []string{"decode-beta"}},
		{name: "rewritten model is excluded", alpha: `return {model:"another-model"}`, beta: `return {model:ctx.model,action:"beta"}`, wantKeys: []string{"decode-beta"}},
	} {
		t.Run(tc.name, func(t *testing.T) {
			for _, spec := range []struct{ key, decode string }{{"decode-alpha", tc.alpha}, {"decode-beta", tc.beta}} {
				_, err := jsplugin.DefaultRegistry.Register(taskResponsesPluginSource(spec.key, 0, `["decode-shared-model"]`, `["sync"]`, `renderFinal:function(){return {};}`, spec.decode), jsplugin.Options{})
				require.NoError(t, err)
				t.Cleanup(func() { require.NoError(t, jsplugin.DefaultRegistry.Unregister(spec.key)) })
			}
			var gotKeys []string
			router := gin.New()
			router.POST("/v1/responses", PinTaskPluginEndpoint(), PrepareTaskPluginEndpoint(), func(c *gin.Context) {
				pinned := c.MustGet(jsplugin.ContextKeyPinnedEndpoint).(jsplugin.PinnedEndpoint)
				for _, candidate := range pinned.Candidates {
					gotKeys = append(gotKeys, candidate.Plugin.Meta.Key)
				}
				assert.Equal(t, tc.wantKeys[0], pinned.Plugin.Meta.Key)
				assert.Equal(t, tc.wantKeys[0], c.GetString("task_plugin_key"))
				assert.Same(t, pinned.Plugin, c.MustGet(jsplugin.ContextKeyPinnedPlugin).(jsplugin.PinnedPlugin).Plugin)
				assert.Equal(t, tc.wantKeys, service.GetChannelConstraints(c).Filters[0].TaskPluginKeys)
				c.Status(http.StatusNoContent)
			})
			request := httptest.NewRequest(http.MethodPost, "/v1/responses", strings.NewReader(`{"model":"decode-shared-model","resolution":"1080p"}`))
			request.Header.Set("Content-Type", "application/json")
			recorder := httptest.NewRecorder()
			router.ServeHTTP(recorder, request)
			if tc.wantError != "" {
				assert.Equal(t, http.StatusBadRequest, recorder.Code)
				assert.Contains(t, recorder.Body.String(), tc.wantError)
				assert.Empty(t, gotKeys)
			} else {
				assert.Equal(t, http.StatusNoContent, recorder.Code, recorder.Body.String())
				assert.Equal(t, tc.wantKeys, gotKeys)
			}
		})
	}
}