Chris Arderne
46 lines
1.3 KiB
TypeScript
46 lines
1.3 KiB
TypeScript
import type { IOPacket } from "@trigger.dev/core/v3/utils/ioSerialization";
|
|
import { ServiceValidationError } from "~/v3/services/common.server";
|
|
|
|
export class MetadataTooLargeError extends ServiceValidationError {
|
|
constructor(message: string) {
|
|
super(message, 413);
|
|
this.name = "MetadataTooLargeError";
|
|
}
|
|
}
|
|
|
|
export function handleMetadataPacket(
|
|
metadata: any,
|
|
metadataType: string,
|
|
maximumSize: number
|
|
): IOPacket | undefined {
|
|
return handleMetadataPacketWithByteLength(metadata, metadataType, maximumSize)?.packet;
|
|
}
|
|
|
|
export function handleMetadataPacketWithByteLength(
|
|
metadata: any,
|
|
metadataType: string,
|
|
maximumSize: number
|
|
): { packet: IOPacket; byteLength: number } | undefined {
|
|
let metadataPacket: IOPacket | undefined = undefined;
|
|
|
|
if (typeof metadata === "string") {
|
|
metadataPacket = { data: metadata, dataType: metadataType };
|
|
}
|
|
|
|
if (metadataType === "application/json") {
|
|
metadataPacket = { data: JSON.stringify(metadata), dataType: "application/json" };
|
|
}
|
|
|
|
if (!metadataPacket || !metadataPacket.data) {
|
|
return;
|
|
}
|
|
|
|
const byteLength = Buffer.byteLength(metadataPacket.data, "utf8");
|
|
|
|
if (byteLength > maximumSize) {
|
|
throw new MetadataTooLargeError(`Metadata exceeds maximum size of ${maximumSize} bytes`);
|
|
}
|
|
|
|
return { packet: metadataPacket, byteLength };
|
|
}
|