@doufunao123/asset-gateway 0.4.0 → 0.6.0

package/dist/index.js

@@ -1,7 +1,7 @@
 #!/usr/bin/env node
 
 // src/index.ts
-import { Command as Command8 } from "commander";
+import { Command as Command9 } from "commander";
 
 // src/commands/auth.ts
 import { existsSync as existsSync2, unlinkSync } from "fs";
@@ -70,7 +70,7 @@ function normalizeError(error2) {
 
 // src/meta.ts
 var CLI_NAME = "asset-gateway";
-var CLI_VERSION = "0.4.0";
+var CLI_VERSION = "0.6.0";
 var CLI_DESCRIPTION = "Universal asset generation gateway CLI";
 var DEFAULT_GATEWAY_URL = "https://assets.xiaomao.chat";
 
@@ -527,7 +527,7 @@ async function saveOutput(result, assetType, outputDir) {
 function createGenerateCommand() {
   const command = new Command3("generate").description("Generate assets via the gateway");
   command.addCommand(
-    new Command3("image").description("Generate an image from a text prompt").requiredOption("--prompt <text>", "Image description prompt").option("--provider <id>", "Provider to use").option("--transparent", "Request transparent background").option("--model <model>", "Model to use").option("--size <size>", "Image size (e.g. 1024x1024)").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
+    new Command3("image").description("Generate an image from a text prompt").requiredOption("--prompt <text>", "Image description prompt").option("--provider <id>", "Provider to use").option("--transparent", "Request transparent background").option("--model <model>", "Model to use").option("--size <size>", "Image size (e.g. 1024x1024)").option("--input <url>", "Input image URL for editing (Gemini/Grok)").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
       try {
         const ctx = createContext(this);
         const body = {
@@ -538,6 +538,7 @@ function createGenerateCommand() {
         if (options.transparent) body.transparent = true;
         if (options.model) body.model = options.model;
         if (options.size) body.size = options.size;
+        if (options.input) body.input_file = options.input;
         const data = await ctx.client.post("/api/generate", body);
         const localPath = await saveOutput(data, "image", options.outputDir);
         if (localPath) data.local_path = localPath;
@@ -610,14 +611,25 @@ function createGenerateCommand() {
     })
   );
   command.addCommand(
-    new Command3("model").description("Generate a 3D model").option("--image <url>", "Reference image URL").option("--prompt <text>", "Model description prompt").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
+    new Command3("model").description("Generate a 3D model").option("--image <url>", "Reference image URL").option("--prompt <text>", "Model description prompt").option("--model-version <v>", "Tripo model version (e.g. P1-20260311)").option("--face-limit <n>", "Max face count (48-20000)").option("--pbr", "Enable PBR textures").option("--texture-quality <q>", "Texture quality: standard or detailed").option("--auto-size", "Auto-scale to real-world dimensions").option("--negative-prompt <text>", "Negative prompt").option("--multiview <urls>", "4 image URLs comma-separated: front,left,back,right").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
       try {
         const ctx = createContext(this);
         const body = {
           asset_type: "model3d"
         };
+        const params = {};
         if (options.image) body.input_file = options.image;
         if (options.prompt) body.prompt = options.prompt;
+        if (options.modelVersion) params.model_version = options.modelVersion;
+        if (options.faceLimit) params.face_limit = Number(options.faceLimit);
+        if (options.pbr) params.pbr = true;
+        if (options.textureQuality) params.texture_quality = options.textureQuality;
+        if (options.autoSize) params.auto_size = true;
+        if (options.negativePrompt) params.negative_prompt = options.negativePrompt;
+        if (options.multiview) {
+          params.multiview = String(options.multiview).split(",").map((value) => value.trim()).filter(Boolean);
+        }
+        if (Object.keys(params).length > 0) body.params = params;
         const data = await ctx.client.post("/api/generate", body);
         const localPath = await saveOutput(data, "model3d", options.outputDir);
         if (localPath) data.local_path = localPath;
@@ -789,12 +801,205 @@ function createProcessCommand() {
   return command;
 }
 
-// src/commands/provider.ts
+// src/commands/process3d.ts
+import { mkdirSync as mkdirSync4, writeFileSync as writeFileSync4 } from "fs";
+import { join as join4 } from "path";
 import { Command as Command6 } from "commander";
+function infer3dExtension(format) {
+  const map = {
+    fbx: "fbx",
+    usdz: "usdz",
+    obj: "obj",
+    stl: "stl",
+    gltf: "gltf",
+    "3mf": "3mf",
+    glb: "glb"
+  };
+  return map[String(format ?? "glb").toLowerCase()] ?? "glb";
+}
+async function saveProcess3dOutput(data, operation, outputDir, format) {
+  if (!data.output_url) {
+    return null;
+  }
+  const ext = infer3dExtension(format);
+  const timestamp = Date.now();
+  const filePath = join4(outputDir, `${operation}_${timestamp}.${ext}`);
+  mkdirSync4(outputDir, { recursive: true });
+  const response = await fetch(String(data.output_url));
+  if (!response.ok) {
+    return null;
+  }
+  const buffer = Buffer.from(await response.arrayBuffer());
+  writeFileSync4(filePath, buffer);
+  return filePath;
+}
+function createProcess3dCommand() {
+  const command = new Command6("process3d").description("3D model post-processing via Tripo pipeline");
+  command.addCommand(
+    new Command6("convert").description("Convert 3D model format (FBX/USDZ/OBJ/STL/GLTF/3MF)").requiredOption("--task-id <id>", "Tripo task ID from generate model").requiredOption("--format <fmt>", "Target format: FBX, USDZ, OBJ, STL, GLTF, 3MF").option("--quad", "Enable quad remeshing").option("--face-limit <n>", "Max face count").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
+      try {
+        const ctx = createContext(this);
+        const params = {
+          format: options.format
+        };
+        if (options.quad) params.quad = true;
+        if (options.faceLimit) params.face_limit = Number(options.faceLimit);
+        const data = await ctx.client.post("/api/process3d", {
+          task_id: options.taskId,
+          operation: "convert",
+          params
+        });
+        const localPath = await saveProcess3dOutput(data, "convert", options.outputDir, options.format);
+        if (localPath) data.local_path = localPath;
+        printSuccess("process3d.convert", data, ctx);
+      } catch (error2) {
+        printError("process3d.convert", error2);
+      }
+    })
+  );
+  command.addCommand(
+    new Command6("texture").description("Re-texture a 3D model with new materials").requiredOption("--task-id <id>", "Tripo task ID").option("--prompt <text>", "Texture description prompt").option("--pbr", "Enable PBR materials").option("--quality <q>", "Texture quality: standard or detailed").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
+      try {
+        const ctx = createContext(this);
+        const params = {};
+        if (options.prompt) params.prompt = options.prompt;
+        if (options.pbr) params.pbr = true;
+        if (options.quality) params.quality = options.quality;
+        const data = await ctx.client.post("/api/process3d", {
+          task_id: options.taskId,
+          operation: "texture",
+          params
+        });
+        const localPath = await saveProcess3dOutput(data, "texture", options.outputDir);
+        if (localPath) data.local_path = localPath;
+        printSuccess("process3d.texture", data, ctx);
+      } catch (error2) {
+        printError("process3d.texture", error2);
+      }
+    })
+  );
+  command.addCommand(
+    new Command6("rig").description("Auto-rig a 3D model (add skeleton for animation)").requiredOption("--task-id <id>", "Tripo task ID").option("--format <fmt>", "Output format: glb or fbx", "glb").option("--spec <spec>", "Rig spec: mixamo or tripo", "mixamo").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
+      try {
+        const ctx = createContext(this);
+        const data = await ctx.client.post("/api/process3d", {
+          task_id: options.taskId,
+          operation: "rig",
+          params: {
+            format: options.format,
+            spec: options.spec
+          }
+        });
+        const localPath = await saveProcess3dOutput(data, "rig", options.outputDir, options.format);
+        if (localPath) data.local_path = localPath;
+        printSuccess("process3d.rig", data, ctx);
+      } catch (error2) {
+        printError("process3d.rig", error2);
+      }
+    })
+  );
+  command.addCommand(
+    new Command6("animate").description("Apply preset animation to a rigged model").requiredOption("--task-id <id>", "Tripo task ID (from rig step)").requiredOption("--animation <preset>", "Animation preset (e.g. preset:walk, preset:idle, preset:run)").option("--format <fmt>", "Output format: glb or fbx", "glb").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
+      try {
+        const ctx = createContext(this);
+        const data = await ctx.client.post("/api/process3d", {
+          task_id: options.taskId,
+          operation: "animate",
+          params: {
+            animation: options.animation,
+            format: options.format
+          }
+        });
+        const localPath = await saveProcess3dOutput(data, "animate", options.outputDir, options.format);
+        if (localPath) data.local_path = localPath;
+        printSuccess("process3d.animate", data, ctx);
+      } catch (error2) {
+        printError("process3d.animate", error2);
+      }
+    })
+  );
+  command.addCommand(
+    new Command6("reduce").description("Reduce polygon count (high-poly to low-poly)").requiredOption("--task-id <id>", "Tripo task ID").option("--face-limit <n>", "Target face count").option("--quad", "Use quad topology").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
+      try {
+        const ctx = createContext(this);
+        const params = {};
+        if (options.faceLimit) params.face_limit = Number(options.faceLimit);
+        if (options.quad) params.quad = true;
+        const data = await ctx.client.post("/api/process3d", {
+          task_id: options.taskId,
+          operation: "reduce",
+          params
+        });
+        const localPath = await saveProcess3dOutput(data, "reduce", options.outputDir);
+        if (localPath) data.local_path = localPath;
+        printSuccess("process3d.reduce", data, ctx);
+      } catch (error2) {
+        printError("process3d.reduce", error2);
+      }
+    })
+  );
+  command.addCommand(
+    new Command6("stylize").description("Apply artistic style to a model").requiredOption("--task-id <id>", "Tripo task ID").requiredOption("--style <style>", "Style: lego, voxel, voronoi, minecraft").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
+      try {
+        const ctx = createContext(this);
+        const data = await ctx.client.post("/api/process3d", {
+          task_id: options.taskId,
+          operation: "stylize",
+          params: {
+            style: options.style
+          }
+        });
+        const localPath = await saveProcess3dOutput(data, "stylize", options.outputDir);
+        if (localPath) data.local_path = localPath;
+        printSuccess("process3d.stylize", data, ctx);
+      } catch (error2) {
+        printError("process3d.stylize", error2);
+      }
+    })
+  );
+  command.addCommand(
+    new Command6("segment").description("Segment mesh into logical parts").requiredOption("--task-id <id>", "Tripo task ID").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
+      try {
+        const ctx = createContext(this);
+        const data = await ctx.client.post("/api/process3d", {
+          task_id: options.taskId,
+          operation: "segment",
+          params: {}
+        });
+        const localPath = await saveProcess3dOutput(data, "segment", options.outputDir);
+        if (localPath) data.local_path = localPath;
+        printSuccess("process3d.segment", data, ctx);
+      } catch (error2) {
+        printError("process3d.segment", error2);
+      }
+    })
+  );
+  command.addCommand(
+    new Command6("prerigcheck").description("Check if a model can be rigged").requiredOption("--task-id <id>", "Tripo task ID").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
+      try {
+        const ctx = createContext(this);
+        const data = await ctx.client.post("/api/process3d", {
+          task_id: options.taskId,
+          operation: "prerigcheck",
+          params: {}
+        });
+        const localPath = await saveProcess3dOutput(data, "prerigcheck", options.outputDir);
+        if (localPath) data.local_path = localPath;
+        printSuccess("process3d.prerigcheck", data, ctx);
+      } catch (error2) {
+        printError("process3d.prerigcheck", error2);
+      }
+    })
+  );
+  return command;
+}
+
+// src/commands/provider.ts
+import { Command as Command7 } from "commander";
 function createProviderCommand() {
-  const command = new Command6("provider").description("Provider management");
+  const command = new Command7("provider").description("Provider management");
   command.addCommand(
-    new Command6("list").description("List available providers").action(async function() {
+    new Command7("list").description("List available providers").action(async function() {
       try {
         const ctx = createContext(this);
         const data = await ctx.client.get("/api/providers");
@@ -805,7 +1010,7 @@ function createProviderCommand() {
     })
   );
   command.addCommand(
-    new Command6("health").description("Check provider health").argument("[name]", "Specific provider name").action(async function(name) {
+    new Command7("health").description("Check provider health").argument("[name]", "Specific provider name").action(async function(name) {
       try {
         const ctx = createContext(this);
         const path = name ? `/api/providers/${encodeURIComponent(name)}/health` : "/api/providers/health";
@@ -820,11 +1025,11 @@ function createProviderCommand() {
 }
 
 // src/commands/upload.ts
-import { Command as Command7 } from "commander";
+import { Command as Command8 } from "commander";
 function createUploadCommand() {
-  const command = new Command7("upload").description("Upload and manage assets");
+  const command = new Command8("upload").description("Upload and manage assets");
   command.addCommand(
-    new Command7("file").description("Upload a file and get a public URL").argument("<path>", "Path to file to upload").action(async function(filePath) {
+    new Command8("file").description("Upload a file and get a public URL").argument("<path>", "Path to file to upload").action(async function(filePath) {
       const ctx = createContext(this);
       try {
         const data = await ctx.client.uploadFile(filePath);
@@ -835,7 +1040,7 @@ function createUploadCommand() {
     })
   );
   command.addCommand(
-    new Command7("list").description("List uploaded assets").action(async function() {
+    new Command8("list").description("List uploaded assets").action(async function() {
       const ctx = createContext(this);
       try {
         const data = await ctx.client.get("/api/assets");
@@ -846,7 +1051,7 @@ function createUploadCommand() {
     })
   );
   command.addCommand(
-    new Command7("delete").description("Delete an uploaded asset (admin only)").argument("<filename>", "Filename to delete").action(async function(filename) {
+    new Command8("delete").description("Delete an uploaded asset (admin only)").argument("<filename>", "Filename to delete").action(async function(filename) {
       const ctx = createContext(this);
       try {
         const data = await ctx.client.delete(`/api/assets/${encodeURIComponent(filename)}`);
@@ -860,13 +1065,14 @@ function createUploadCommand() {
 }
 
 // src/index.ts
-var program = new Command8().name("asset-gateway").description("Universal asset generation gateway CLI").version(CLI_VERSION).option(
+var program = new Command9().name("asset-gateway").description("Universal asset generation gateway CLI").version(CLI_VERSION).option(
   "--gateway-url <url>",
   `Gateway URL (default: $ASSET_GATEWAY_URL, auth config, or ${DEFAULT_GATEWAY_URL})`
 ).option("--token <token>", "API token for authentication").option("--human", "Human-readable output instead of JSON").option("--fields <fields>", "Comma-separated list of output fields");
 program.addCommand(createAuthCommand());
 program.addCommand(createGenerateCommand());
 program.addCommand(createProcessCommand());
+program.addCommand(createProcess3dCommand());
 program.addCommand(createProviderCommand());
 program.addCommand(createUploadCommand());
 program.addCommand(createJobCommand());

package/package.json

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