@doufunao123/asset-gateway 0.12.0 → 0.12.2

package/dist/index.js

@@ -623,11 +623,11 @@ function createDescribeCommand() {
 }
 
 // src/commands/generate.ts
-import { mkdirSync as mkdirSync2, writeFileSync as writeFileSync2 } from "fs";
-import { dirname as dirname2, join as join2 } from "path";
+import { existsSync as existsSync3, mkdirSync as mkdirSync2, readFileSync as readFileSync2, writeFileSync as writeFileSync2 } from "fs";
+import { dirname as dirname2, extname, join as join2 } from "path";
 import { Command as Command3 } from "commander";
 function inferExtension(assetType) {
-  const map = { image: "png", audio: "mp3", music: "mp3", tts: "mp3", video: "mp4", model3d: "glb", text: "txt", sprite: "png" };
+  const map = { image: "png", audio: "mp3", music: "mp3", tts: "mp3", video: "mp4", model3d: "glb", text: "txt", sprite: "png", world: "spz" };
   return map[assetType] ?? "bin";
 }
 function stripDataUri(data) {
@@ -925,11 +925,18 @@ function createGenerateCommand() {
         if (options.negativePrompt) params.negative_prompt = options.negativePrompt;
         if (options.seed) params.seed = Number(options.seed);
         if (options.matteColor) params.matte_color = options.matteColor;
+        let inputFile = options.input;
+        if (existsSync3(inputFile)) {
+          const ext = extname(inputFile).toLowerCase();
+          const mime = ext === ".jpg" || ext === ".jpeg" ? "image/jpeg" : "image/png";
+          const b64 = readFileSync2(inputFile).toString("base64");
+          inputFile = `data:${mime};base64,${b64}`;
+        }
         const body = {
           asset_type: "sprite",
           prompt: options.prompt,
           model: options.model,
-          input_file: options.input,
+          input_file: inputFile,
           params
         };
         const data = await ctx.client.post("/api/generate", body);
@@ -941,6 +948,39 @@ function createGenerateCommand() {
       }
     })
   );
+  command.addCommand(
+    new Command3("world").description("Generate a 3D world/environment using WorldLabs Marble").requiredOption("--prompt <text>", "Text prompt describing the environment").option("--input <path>", "Input image (local path or URL) for image-to-world").option("--model <model>", "Model: marble-1.0-draft, marble-1.0, marble-1.1, marble-1.1-plus", "marble-1.1").option("--display-name <name>", "Display name for the generated world").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
+      try {
+        const ctx = createContext(this);
+        const params = {};
+        if (options.displayName) params.display_name = options.displayName;
+        let inputFile;
+        if (options.input) {
+          if (existsSync3(options.input)) {
+            const ext = extname(options.input).toLowerCase();
+            const mime = ext === ".jpg" || ext === ".jpeg" ? "image/jpeg" : "image/png";
+            const b64 = readFileSync2(options.input).toString("base64");
+            inputFile = `data:${mime};base64,${b64}`;
+          } else {
+            inputFile = options.input;
+          }
+        }
+        const body = {
+          asset_type: "world",
+          prompt: options.prompt,
+          model: options.model,
+          params
+        };
+        if (inputFile) body.input_file = inputFile;
+        const data = await ctx.client.post("/api/generate", body);
+        const localPath = await saveOutput(data, "world", options.outputDir);
+        if (localPath) data.local_path = localPath;
+        printSuccess("generate.world", data, ctx);
+      } catch (error2) {
+        printError("generate.world", error2);
+      }
+    })
+  );
   return command;
 }
 
@@ -990,13 +1030,13 @@ function createJobCommand() {
 }
 
 // src/commands/process.ts
-import { mkdirSync as mkdirSync3, readFileSync as readFileSync2, writeFileSync as writeFileSync3 } from "fs";
-import { existsSync as existsSync3 } from "fs";
+import { mkdirSync as mkdirSync3, readFileSync as readFileSync3, writeFileSync as writeFileSync3 } from "fs";
+import { existsSync as existsSync4 } from "fs";
 import { join as join3 } from "path";
 import { Command as Command5 } from "commander";
 function readInputAsBase64(input) {
-  if (existsSync3(input)) {
-    const bytes = readFileSync2(input);
+  if (existsSync4(input)) {
+    const bytes = readFileSync3(input);
     return `data:image/png;base64,${bytes.toString("base64")}`;
   }
   return input;
@@ -1495,11 +1535,11 @@ function createUploadCommand() {
 }
 
 // src/commands/voice.ts
-import { existsSync as existsSync4, readFileSync as readFileSync3 } from "fs";
-import { extname } from "path";
+import { existsSync as existsSync5, readFileSync as readFileSync4 } from "fs";
+import { extname as extname2 } from "path";
 import { Command as Command9 } from "commander";
 function inferAudioMime(filePath) {
-  const extension = extname(filePath).toLowerCase();
+  const extension = extname2(filePath).toLowerCase();
   const map = {
     ".mp3": "audio/mpeg",
     ".wav": "audio/wav",
@@ -1513,10 +1553,10 @@ function inferAudioMime(filePath) {
   return map[extension] ?? "application/octet-stream";
 }
 function readAudioAsBase64(filePath) {
-  if (!existsSync4(filePath)) {
+  if (!existsSync5(filePath)) {
     throw configError(`Audio file not found: ${filePath}`);
   }
-  const bytes = readFileSync3(filePath);
+  const bytes = readFileSync4(filePath);
   return {
     audio_base64: bytes.toString("base64"),
     audio_mime: inferAudioMime(filePath)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@doufunao123/asset-gateway",
-  "version": "0.12.0",
+  "version": "0.12.2",
   "description": "Universal asset generation gateway CLI",
   "type": "module",
   "bin": {