wehub-resource-sync
67 lines
1.9 KiB
TypeScript
67 lines
1.9 KiB
TypeScript
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,
|
|
},
|
|
])
|
|
);
|
|
}
|