@mevdragon/vidfarm-devcli 0.2.2 → 0.2.4

package/dist/src/app.js

@@ -1,3 +1,4 @@
+import { randomUUID } from "node:crypto";
 import { readFileSync, statSync } from "node:fs";
 import path from "node:path";
 import { Hono } from "hono";
@@ -60,6 +61,11 @@ const settingsProfileFormSchema = z.object({
     groupchat_url: z.union([z.literal(""), z.string().url()]).optional(),
     flockposter_api_key: z.string().max(1000).optional()
 });
+const developerPreviewPresignSchema = z.object({
+    file_name: z.string().min(1).max(255),
+    content_type: z.string().trim().min(1).max(255).optional(),
+    directory: z.string().trim().max(500).optional()
+});
 const listJobsQuerySchema = z.object({
     tracer: z.string().min(1).optional(),
     start_time: z.string().min(1).optional(),
@@ -195,6 +201,56 @@ function isAllowedAttachment(fileName, contentType) {
         || contentType === "text/plain"
         || contentType === "text/markdown";
 }
+function sanitizeStorageSubpath(value) {
+    if (!value) {
+        return "";
+    }
+    return value
+        .split("/")
+        .map((segment) => segment.trim().replace(/[^\w.-]+/g, "_"))
+        .filter((segment) => segment && segment !== "." && segment !== "..")
+        .join("/");
+}
+function inferAttachmentContentType(fileName) {
+    switch (path.extname(fileName).toLowerCase()) {
+        case ".png":
+            return "image/png";
+        case ".gif":
+            return "image/gif";
+        case ".jpg":
+        case ".jpeg":
+            return "image/jpeg";
+        case ".webp":
+            return "image/webp";
+        case ".svg":
+            return "image/svg+xml";
+        case ".mp4":
+        case ".m4v":
+            return "video/mp4";
+        case ".mov":
+            return "video/quicktime";
+        case ".webm":
+            return "video/webm";
+        case ".mp3":
+            return "audio/mpeg";
+        case ".wav":
+            return "audio/wav";
+        case ".m4a":
+            return "audio/mp4";
+        case ".aac":
+            return "audio/aac";
+        case ".ogg":
+            return "audio/ogg";
+        case ".pdf":
+            return "application/pdf";
+        case ".md":
+            return "text/markdown; charset=utf-8";
+        case ".txt":
+            return "text/plain; charset=utf-8";
+        default:
+            return "application/octet-stream";
+    }
+}
 function isFormFile(value) {
     return Boolean(value
         && typeof value === "object"
@@ -563,7 +619,7 @@ function requireCustomer(c) {
 }
 const requireAuth = async (c, next) => {
     try {
-        const customer = auth.authenticate(c.req.header("vidfarm-user-id"), c.req.header("vidfarm-api-key"));
+        const customer = auth.authenticate(c.req.header("vidfarm-api-key"));
         c.set("customer", customer);
         await next();
     }
@@ -877,7 +933,7 @@ app.get(`${TEMPLATES_PREFIX}/:templateId/jobs/:jobId`, (c) => {
     }
     return c.json(serializeJob(job));
 });
-app.get(`${TEMPLATES_PREFIX}/:templateId/jobs/:jobId/logs`, (c) => {
+app.get(`${TEMPLATES_PREFIX}/:templateId/jobs/:jobId/logs`, async (c) => {
     const customer = requireCustomer(c);
     const template = templateRegistry.get(c.req.param("templateId"));
     if (!template) {
@@ -892,7 +948,7 @@ app.get(`${TEMPLATES_PREFIX}/:templateId/jobs/:jobId/logs`, (c) => {
         end_time: c.req.query("end_time"),
         limit: c.req.query("limit")
     });
-    const logs = jobs.listLogs({
+    const logs = await jobs.listLogs({
         jobId: job.id,
         startTime: query.start_time,
         endTime: query.end_time,
@@ -914,6 +970,39 @@ app.post(`${TEMPLATES_PREFIX}/:templateId/jobs/:jobId/cancel`, (c) => {
     return c.json({ ok: true, job_id: job.id, status: "cancelled" });
 });
 app.get(`${USER_PREFIX}/me`, (c) => c.json({ customer: requireCustomer(c) }));
+app.post(`${USER_PREFIX}/me/developer/preview-media/presign`, async (c) => {
+    try {
+        const customer = requireCustomer(c);
+        const body = developerPreviewPresignSchema.parse(await c.req.json());
+        const fileName = sanitizeFileName(body.file_name);
+        const contentType = body.content_type?.trim() || inferAttachmentContentType(fileName);
+        if (!isAllowedAttachment(fileName, contentType)) {
+            return c.json({ error: `Unsupported preview media type: ${fileName}` }, 400);
+        }
+        const directory = sanitizeStorageSubpath(body.directory);
+        const storageKey = storage.developerScopedKey(customer.id, directory, randomUUID(), fileName);
+        const upload = await storage.createWriteUrl(storageKey, {
+            contentType,
+            publicRead: false,
+            expiresIn: 3600
+        });
+        return c.json({
+            file_name: fileName,
+            content_type: contentType,
+            storage_key: storageKey,
+            preview_media_url: buildAbsoluteUrl(c, `/template-media?key=${encodeURIComponent(storageKey)}`),
+            upload: {
+                method: upload.method,
+                url: upload.url,
+                headers: upload.headers,
+                expires_in_seconds: upload.expiresIn
+            }
+        });
+    }
+    catch (error) {
+        return c.json({ error: error instanceof Error ? error.message : "Unable to create preview upload URL." }, 400);
+    }
+});
 app.get(`${USER_PREFIX}/me/jobs`, (c) => {
     const query = listJobsQuerySchema.parse({
         template_id: c.req.query("template_id"),
@@ -969,8 +1058,8 @@ app.post(`${TEMPLATES_PREFIX}/sources`, async (c) => {
             slug_id: z.string().min(3).regex(/^[a-z0-9_]+$/i, "slug_id must contain only letters, numbers, and underscores."),
             repo_url: z.string().url(),
             branch: z.string().min(1).default("production"),
-            template_module_path: z.string().min(1),
-            skill_path: z.string().min(1).default("SKILL.md"),
+            template_module_path: z.string().regex(/(^|\/)vidfarm_template_[^/]+\/src\/template\.ts$/i, "template_module_path must point at a template entrypoint inside a folder that starts with vidfarm_template_, for example templates/vidfarm_template_example/src/template.ts."),
+            skill_path: z.string().min(1).optional(),
             install_command: z.string().min(1).default("npm install"),
             build_command: z.string().min(1).default("npm run build")
         }).parse(await c.req.json());