@mevdragon/vidfarm-devcli 0.1.0

package/dist/src/context.js

@@ -0,0 +1,198 @@
+import { config } from "./config.js";
+import { database } from "./db.js";
+import { createId } from "./lib/ids.js";
+export function createTemplateJobContext(input) {
+    const templateConfig = database.getTemplateConfig(input.customer.id, input.template.id);
+    const prefix = `templates/${input.template.id}/users/${input.customer.id}/jobs/${input.job.id}`;
+    return {
+        env: config.isProduction ? "production" : "development",
+        customer: input.customer,
+        templateConfig,
+        logger: {
+            debug(message, metadata = {}) {
+                database.addJobEvent({
+                    id: createId("evt"),
+                    jobId: input.job.id,
+                    level: "debug",
+                    message,
+                    metadata,
+                    progress: null,
+                    artifactKey: null
+                });
+            },
+            info(message, metadata = {}) {
+                database.addJobEvent({
+                    id: createId("evt"),
+                    jobId: input.job.id,
+                    level: "info",
+                    message,
+                    metadata,
+                    progress: null,
+                    artifactKey: null
+                });
+            },
+            warn(message, metadata = {}) {
+                database.addJobEvent({
+                    id: createId("evt"),
+                    jobId: input.job.id,
+                    level: "warn",
+                    message,
+                    metadata,
+                    progress: null,
+                    artifactKey: null
+                });
+            },
+            error(message, metadata = {}) {
+                database.addJobEvent({
+                    id: createId("evt"),
+                    jobId: input.job.id,
+                    level: "error",
+                    message,
+                    metadata,
+                    progress: null,
+                    artifactKey: null
+                });
+            },
+            progress(progress, message, metadata = {}) {
+                database.addJobEvent({
+                    id: createId("evt"),
+                    jobId: input.job.id,
+                    level: "info",
+                    message,
+                    metadata,
+                    progress,
+                    artifactKey: typeof metadata.artifactKey === "string" ? metadata.artifactKey : null
+                });
+                database.updateJobStatus({
+                    id: input.job.id,
+                    status: "running",
+                    progress
+                });
+            }
+        },
+        jobs: {
+            async enqueueChild({ operationName, workflowName, payload, providerHint }) {
+                const child = await input.jobs.enqueueChildJob({
+                    templateId: input.template.id,
+                    operationName,
+                    workflowName,
+                    tracer: input.job.tracer,
+                    payload,
+                    customerId: input.customer.id,
+                    parentJobId: input.job.id,
+                    providerHint
+                });
+                return { jobId: child.id };
+            }
+        },
+        storage: {
+            async putJson(key, value) {
+                const stored = await input.storage.putJson(`${prefix}/${key}`, value);
+                insertArtifact("json", stored.key, stored.url, {});
+                await input.billing.record({
+                    customerId: input.customer.id,
+                    jobId: input.job.id,
+                    templateId: input.template.id,
+                    type: "storage_write",
+                    costUsd: 0.005,
+                    chargeUsd: 0.01,
+                    metadata: { key: stored.key }
+                });
+                return stored;
+            },
+            async putText(key, value, contentType = "text/plain; charset=utf-8") {
+                const stored = await input.storage.putText(`${prefix}/${key}`, value, contentType);
+                insertArtifact("text", stored.key, stored.url, { contentType });
+                await input.billing.record({
+                    customerId: input.customer.id,
+                    jobId: input.job.id,
+                    templateId: input.template.id,
+                    type: "storage_write",
+                    costUsd: 0.005,
+                    chargeUsd: 0.01,
+                    metadata: { key: stored.key }
+                });
+                return stored;
+            },
+            async putBuffer(key, value, options = {}) {
+                const stored = await input.storage.putBuffer(`${prefix}/${key}`, value, options.contentType);
+                insertArtifact(options.kind ?? "binary", stored.key, stored.url, {
+                    ...(options.metadata ?? {}),
+                    contentType: options.contentType ?? null
+                });
+                await input.billing.record({
+                    customerId: input.customer.id,
+                    jobId: input.job.id,
+                    templateId: input.template.id,
+                    type: "storage_write",
+                    costUsd: 0.005,
+                    chargeUsd: 0.01,
+                    metadata: { key: stored.key }
+                });
+                return stored;
+            },
+            getPublicUrl(key) {
+                return input.storage.getPublicUrl(`${prefix}/${key}`);
+            }
+        },
+        billing: {
+            async record(record) {
+                await input.billing.record({
+                    customerId: input.customer.id,
+                    jobId: input.job.id,
+                    templateId: input.template.id,
+                    type: record.type,
+                    costUsd: record.costUsd,
+                    chargeUsd: record.chargeUsd ?? Number((record.costUsd * 2).toFixed(4)),
+                    metadata: record.metadata
+                });
+            }
+        },
+        providers: {
+            async generateText(request) {
+                return input.providers.generateText({
+                    customerId: input.customer.id,
+                    jobId: input.job.id,
+                    workerId: input.workerId,
+                    ...request
+                });
+            },
+            async generateImage(request) {
+                return input.providers.generateImage({
+                    customerId: input.customer.id,
+                    jobId: input.job.id,
+                    workerId: input.workerId,
+                    ...request
+                });
+            },
+            async analyzeImageLayout(request) {
+                return input.providers.analyzeImageLayout({
+                    customerId: input.customer.id,
+                    jobId: input.job.id,
+                    workerId: input.workerId,
+                    ...request
+                });
+            }
+        },
+        remotion: {
+            async render(request) {
+                return input.remotion.render({
+                    ...request,
+                    outputKey: request.outputKey ? `${prefix}/${request.outputKey}` : undefined
+                });
+            }
+        }
+    };
+    function insertArtifact(kind, storageKey, publicUrl, metadata) {
+        database.insertArtifact({
+            id: createId("art"),
+            jobId: input.job.id,
+            customerId: input.customer.id,
+            templateId: input.template.id,
+            kind,
+            storageKey,
+            publicUrl,
+            metadata
+        });
+    }
+}