wehub-resource-sync
93 lines
2.5 KiB
TypeScript
93 lines
2.5 KiB
TypeScript
import { MAX_TAGS_PER_RUN } from "~/models/taskRunTag.server";
|
|
import { logger } from "~/services/logger.server";
|
|
import { getEntitlement } from "~/services/platform.v3.server";
|
|
import { MAX_ATTEMPTS } from "~/v3/services/triggerTask.server";
|
|
import { isFinalRunStatus } from "~/v3/taskStatus";
|
|
import type {
|
|
EntitlementValidationParams,
|
|
EntitlementValidationResult,
|
|
MaxAttemptsValidationParams,
|
|
ParentRunValidationParams,
|
|
TagValidationParams,
|
|
TriggerTaskValidator,
|
|
ValidationResult,
|
|
} from "../types";
|
|
import { validateProductionEntitlement } from "./validateProductionEntitlement.server";
|
|
import { ServiceValidationError } from "~/v3/services/common.server";
|
|
|
|
export class DefaultTriggerTaskValidator implements TriggerTaskValidator {
|
|
validateTags(params: TagValidationParams): ValidationResult {
|
|
const { tags } = params;
|
|
|
|
if (!tags) {
|
|
return { ok: true };
|
|
}
|
|
|
|
if (typeof tags === "string") {
|
|
return { ok: true };
|
|
}
|
|
|
|
if (tags.length > MAX_TAGS_PER_RUN) {
|
|
return {
|
|
ok: false,
|
|
error: new ServiceValidationError(
|
|
`Runs can only have ${MAX_TAGS_PER_RUN} tags, you're trying to set ${tags.length}.`
|
|
),
|
|
};
|
|
}
|
|
|
|
return { ok: true };
|
|
}
|
|
|
|
async validateEntitlement(
|
|
params: EntitlementValidationParams
|
|
): Promise<EntitlementValidationResult> {
|
|
return validateProductionEntitlement(params, getEntitlement);
|
|
}
|
|
|
|
validateMaxAttempts(params: MaxAttemptsValidationParams): ValidationResult {
|
|
const { taskId, attempt } = params;
|
|
|
|
if (attempt > MAX_ATTEMPTS) {
|
|
return {
|
|
ok: false,
|
|
error: new ServiceValidationError(
|
|
`Failed to trigger ${taskId} after ${MAX_ATTEMPTS} attempts.`
|
|
),
|
|
};
|
|
}
|
|
|
|
return { ok: true };
|
|
}
|
|
|
|
validateParentRun(params: ParentRunValidationParams): ValidationResult {
|
|
const { taskId, parentRun, resumeParentOnCompletion } = params;
|
|
|
|
// If there's no parent run specified, that's fine
|
|
if (!parentRun) {
|
|
return { ok: true };
|
|
}
|
|
|
|
// If we're not resuming the parent, we don't need to validate its status
|
|
if (!resumeParentOnCompletion) {
|
|
return { ok: true };
|
|
}
|
|
|
|
// Check if the parent run is in a final state
|
|
if (isFinalRunStatus(parentRun.status)) {
|
|
logger.debug("Parent run is in a terminal state", {
|
|
parentRun,
|
|
});
|
|
|
|
return {
|
|
ok: false,
|
|
error: new ServiceValidationError(
|
|
`Cannot trigger ${taskId} as the parent run has a status of ${parentRun.status}`
|
|
),
|
|
};
|
|
}
|
|
|
|
return { ok: true };
|
|
}
|
|
}
|