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,66 @@
+import type { MachinePreset } from "@trigger.dev/core/v3";
+import { MachineConfig, MachinePresetName } from "@trigger.dev/core/v3";
+import { defaultMachine, machines } from "~/services/platform.v3.server";
+import { logger } from "~/services/logger.server";
+
+export function machinePresetFromConfig(config: unknown): MachinePreset {
+  const parsedConfig = MachineConfig.safeParse(config);
+
+  if (!parsedConfig.success) {
+    logger.info("Failed to parse machine config", { config });
+
+    return machinePresetFromName("small-1x");
+  }
+
+  if (parsedConfig.data.preset) {
+    return machinePresetFromName(parsedConfig.data.preset);
+  }
+
+  if (parsedConfig.data.cpu && parsedConfig.data.memory) {
+    const name = derivePresetNameFromValues(parsedConfig.data.cpu, parsedConfig.data.memory);
+
+    return machinePresetFromName(name);
+  }
+
+  return machinePresetFromName("small-1x");
+}
+
+export function machinePresetFromName(name: MachinePresetName): MachinePreset {
+  return {
+    name,
+    ...machines[name],
+  };
+}
+
+export function machinePresetFromRun(run: { machinePreset: string | null }): MachinePreset | null {
+  const presetName = MachinePresetName.safeParse(run.machinePreset).data;
+
+  if (!presetName) {
+    return null;
+  }
+
+  return machinePresetFromName(presetName);
+}
+
+// Finds the smallest machine preset name that satisfies the given CPU and memory requirements
+function derivePresetNameFromValues(cpu: number, memory: number): MachinePresetName {
+  for (const [name, preset] of Object.entries(machines)) {
+    if (preset.cpu >= cpu && preset.memory >= memory) {
+      return name as MachinePresetName;
+    }
+  }
+
+  return defaultMachine;
+}
+
+export function allMachines(): Record<string, MachinePreset> {
+  return Object.fromEntries(
+    Object.entries(machines).map(([name, preset]) => [
+      name,
+      {
+        name: name as MachinePresetName,
+        ...preset,
+      },
+    ])
+  );
+}