chore: import upstream snapshot with attribution

2026-07-13 13:32:57 +08:00
commit cd420f9332
4811 changed files with 884702 additions and 0 deletions
@@ -0,0 +1,107 @@
+import type { ActionFunctionArgs, LoaderFunctionArgs } from "@remix-run/server-runtime";
+import { json } from "@remix-run/server-runtime";
+import { z } from "zod";
+import { prisma } from "~/db.server";
+import { requireAdminApiRequest } from "~/services/personalAccessToken.server";
+import { controlPlaneResolver } from "~/v3/runOpsMigration/controlPlaneResolver.server";
+import { engine } from "~/v3/runEngine.server";
+import { updateEnvConcurrencyLimits } from "~/v3/runQueue.server";
+
+const ParamsSchema = z.object({
+  environmentId: z.string(),
+});
+
+const RequestBodySchema = z.object({
+  envMaximumConcurrencyLimit: z.number(),
+  orgMaximumConcurrencyLimit: z.number(),
+});
+
+export async function action({ request, params }: ActionFunctionArgs) {
+  await requireAdminApiRequest(request);
+
+  const parsedParams = ParamsSchema.parse(params);
+
+  const rawBody = await request.json();
+  const body = RequestBodySchema.parse(rawBody);
+
+  const environment = await prisma.runtimeEnvironment.update({
+    where: {
+      id: parsedParams.environmentId,
+    },
+    data: {
+      maximumConcurrencyLimit: body.envMaximumConcurrencyLimit,
+      organization: {
+        update: {
+          data: {
+            maximumConcurrencyLimit: body.orgMaximumConcurrencyLimit,
+          },
+        },
+      },
+    },
+    include: {
+      organization: true,
+      project: true,
+    },
+  });
+
+  await updateEnvConcurrencyLimits(environment);
+
+  // Org max-concurrency changed too, which is embedded in every env of the org; invalidating
+  // the org drops the env/authEnv rows for all of them (including this env).
+  controlPlaneResolver.invalidateOrganization(environment.organizationId);
+
+  return json({ success: true });
+}
+
+const SearchParamsSchema = z.object({
+  queue: z.string().optional(),
+});
+
+export async function loader({ request, params }: LoaderFunctionArgs) {
+  await requireAdminApiRequest(request);
+
+  const parsedParams = ParamsSchema.parse(params);
+
+  const environment = await prisma.runtimeEnvironment.findFirst({
+    where: {
+      id: parsedParams.environmentId,
+    },
+    include: {
+      organization: true,
+      project: true,
+    },
+  });
+
+  if (!environment) {
+    return json({ error: "Environment not found" }, { status: 404 });
+  }
+
+  const requestUrl = new URL(request.url);
+  const searchParams = SearchParamsSchema.parse(
+    Object.fromEntries(requestUrl.searchParams.entries())
+  );
+
+  const concurrencyLimit = await engine.runQueue.getEnvConcurrencyLimit(environment);
+  const currentConcurrency = await engine.runQueue.currentConcurrencyOfEnvironment(environment);
+
+  if (searchParams.queue) {
+    const queueConcurrencyLimit = await engine.runQueue.getQueueConcurrencyLimit(
+      environment,
+      searchParams.queue
+    );
+    const queueCurrentConcurrency = await engine.runQueue.currentConcurrencyOfQueue(
+      environment,
+      searchParams.queue
+    );
+
+    return json({
+      id: environment.id,
+      concurrencyLimit,
+      currentConcurrency,
+      queueConcurrencyLimit,
+      queueCurrentConcurrency,
+    });
+  }
+
+  return json({ id: environment.id, concurrencyLimit, currentConcurrency });
+}