@mevdragon/vidfarm-devcli 0.2.8 → 0.2.9

package/dist/src/context.js

@@ -5,12 +5,7 @@ import { jobLogStore } from "./services/job-logs.js";
 export function createTemplateJobContext(input) {
     const templateConfig = database.getTemplateConfig(input.customer.id, input.template.id);
     const prefix = input.storage.templateJobPrefix(input.template.id, input.customer.id, input.job.id);
-    const buildArtifactUrl = (storageKey, fallbackUrl) => {
-        if (config.STORAGE_DRIVER !== "s3") {
-            return fallbackUrl;
-        }
-        return `${config.PUBLIC_BASE_URL}/template-media?key=${encodeURIComponent(storageKey)}`;
-    };
+    const buildArtifactUrl = (storageKey, fallbackUrl) => fallbackUrl ?? input.storage.getPublicUrl(storageKey);
     return {
         env: config.isProduction ? "production" : "development",
         customer: input.customer,
@@ -97,15 +92,6 @@ export function createTemplateJobContext(input) {
                 const stored = await input.storage.putJson(`${prefix}/${key}`, value);
                 const artifactUrl = buildArtifactUrl(stored.key, stored.url);
                 insertArtifact("json", stored.key, artifactUrl, {});
-                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,
                     url: artifactUrl
@@ -115,15 +101,6 @@ export function createTemplateJobContext(input) {
                 const stored = await input.storage.putText(`${prefix}/${key}`, value, contentType);
                 const artifactUrl = buildArtifactUrl(stored.key, stored.url);
                 insertArtifact("text", stored.key, artifactUrl, { 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,
                     url: artifactUrl
@@ -136,15 +113,6 @@ export function createTemplateJobContext(input) {
                     ...(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,
                     url: artifactUrl
@@ -159,15 +127,22 @@ export function createTemplateJobContext(input) {
                 await input.billing.record({
                     customerId: input.customer.id,
                     jobId: input.job.id,
+                    tracer: input.job.tracer,
                     templateId: input.template.id,
                     type: record.type,
                     costUsd: record.costUsd,
-                    chargeUsd: record.chargeUsd ?? Number((record.costUsd * 2).toFixed(4)),
+                    chargeUsd: record.chargeUsd ?? Number((record.costUsd * 1.2).toFixed(4)),
+                    costCenterSlug: record.costCenterSlug,
+                    idempotencyKey: record.idempotencyKey,
+                    occurredAtMs: record.occurredAtMs,
                     metadata: record.metadata
                 });
             }
         },
         providers: {
+            listAvailableProviders() {
+                return input.providers.listAvailableProviders(input.customer.id);
+            },
             async generateText(request) {
                 return input.providers.generateText({
                     customerId: input.customer.id,
@@ -211,10 +186,35 @@ export function createTemplateJobContext(input) {
         },
         remotion: {
             async render(request) {
-                return input.remotion.render({
+                const render = await input.remotion.render({
                     ...request,
                     outputKey: request.outputKey ? `${prefix}/${request.outputKey}` : undefined
                 });
+                const estimatedCostUsd = "estimatedCostUsd" in render.metadata ? render.metadata.estimatedCostUsd : null;
+                const chargeUsd = "chargeUsd" in render.metadata ? render.metadata.chargeUsd : null;
+                if (render.metadata.mode === "lambda" && typeof estimatedCostUsd === "number" && estimatedCostUsd > 0) {
+                    await input.billing.record({
+                        customerId: input.customer.id,
+                        jobId: input.job.id,
+                        tracer: input.job.tracer,
+                        templateId: input.template.id,
+                        type: "render",
+                        costUsd: estimatedCostUsd,
+                        chargeUsd: typeof chargeUsd === "number" ? chargeUsd : undefined,
+                        costCenterSlug: "remotion_lambda",
+                        idempotencyKey: `remotion_lambda:${render.renderId}`,
+                        occurredAtMs: Date.now(),
+                        metadata: {
+                            ...render.metadata,
+                            render_id: render.renderId,
+                            output_url: render.outputUrl,
+                            composition_id: typeof render.metadata.compositionId === "string"
+                                ? render.metadata.compositionId
+                                : request.compositionId,
+                        }
+                    });
+                }
+                return render;
             }
         }
     };