@doufunao123/asset-gateway 0.3.0 → 0.5.0

package/dist/index.js

@@ -1,7 +1,7 @@
 #!/usr/bin/env node
 
 // src/index.ts
-import { Command as Command7 } from "commander";
+import { Command as Command8 } 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.2.0";
+var CLI_VERSION = "0.4.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;
@@ -694,12 +695,107 @@ function createJobCommand() {
   return command;
 }
 
-// src/commands/provider.ts
+// src/commands/process.ts
+import { mkdirSync as mkdirSync3, readFileSync as readFileSync2, writeFileSync as writeFileSync3 } from "fs";
+import { existsSync as existsSync3 } from "fs";
+import { join as join3 } from "path";
 import { Command as Command5 } from "commander";
+function readInputAsBase64(input) {
+  if (existsSync3(input)) {
+    const bytes = readFileSync2(input);
+    return `data:image/png;base64,${bytes.toString("base64")}`;
+  }
+  return input;
+}
+function saveProcessOutput(data, outputDir) {
+  const outputData = data.output_data;
+  if (typeof outputData !== "string" || !outputData) return null;
+  mkdirSync3(outputDir, { recursive: true });
+  const timestamp = Date.now();
+  const filePath = join3(outputDir, `processed_${timestamp}.png`);
+  writeFileSync3(filePath, Buffer.from(outputData, "base64"));
+  delete data.output_data;
+  return filePath;
+}
+function createProcessCommand() {
+  const command = new Command5("process").description("Post-process images (remove-bg, crop, resize, upscale)");
+  command.addCommand(
+    new Command5("remove-bg").description("Remove background from an image (AI-powered, BiRefNet)").requiredOption("--input <path>", "Input image (file path or URL)").option("--smart-crop", "Also crop to power-of-2 after removing background").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
+      try {
+        const ctx = createContext(this);
+        const ops = [{ op: "remove_bg" }];
+        if (options.smartCrop) {
+          ops.push({ op: "smart_crop", mode: "power_of2" });
+        }
+        const data = await ctx.client.post("/api/process", {
+          input: readInputAsBase64(options.input),
+          operations: ops
+        });
+        const localPath = saveProcessOutput(data, options.outputDir);
+        if (localPath) data.local_path = localPath;
+        printSuccess("process.remove-bg", data, ctx);
+      } catch (error2) {
+        printError("process.remove-bg", error2);
+      }
+    })
+  );
+  command.addCommand(
+    new Command5("crop").description("Smart crop an image (trim transparent borders)").requiredOption("--input <path>", "Input image (file path or URL)").option("--mode <mode>", "Crop mode: tightest or power_of2", "tightest").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
+      try {
+        const ctx = createContext(this);
+        const data = await ctx.client.post("/api/process", {
+          input: readInputAsBase64(options.input),
+          operations: [{ op: "smart_crop", mode: options.mode }]
+        });
+        const localPath = saveProcessOutput(data, options.outputDir);
+        if (localPath) data.local_path = localPath;
+        printSuccess("process.crop", data, ctx);
+      } catch (error2) {
+        printError("process.crop", error2);
+      }
+    })
+  );
+  command.addCommand(
+    new Command5("resize").description("Resize an image to exact dimensions").requiredOption("--input <path>", "Input image (file path or URL)").requiredOption("--width <n>", "Target width").requiredOption("--height <n>", "Target height").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
+      try {
+        const ctx = createContext(this);
+        const data = await ctx.client.post("/api/process", {
+          input: readInputAsBase64(options.input),
+          operations: [{ op: "resize", width: Number(options.width), height: Number(options.height) }]
+        });
+        const localPath = saveProcessOutput(data, options.outputDir);
+        if (localPath) data.local_path = localPath;
+        printSuccess("process.resize", data, ctx);
+      } catch (error2) {
+        printError("process.resize", error2);
+      }
+    })
+  );
+  command.addCommand(
+    new Command5("upscale").description("AI upscale an image (2x or 4x via Real-ESRGAN)").requiredOption("--input <path>", "Input image (file path or URL)").option("--scale <n>", "Scale factor (2 or 4)", "4").option("--output-dir <dir>", "Directory to save output", ".").action(async function(options) {
+      try {
+        const ctx = createContext(this);
+        const data = await ctx.client.post("/api/process", {
+          input: readInputAsBase64(options.input),
+          operations: [{ op: "upscale", scale: Number(options.scale) }]
+        });
+        const localPath = saveProcessOutput(data, options.outputDir);
+        if (localPath) data.local_path = localPath;
+        printSuccess("process.upscale", data, ctx);
+      } catch (error2) {
+        printError("process.upscale", error2);
+      }
+    })
+  );
+  return command;
+}
+
+// src/commands/provider.ts
+import { Command as Command6 } from "commander";
 function createProviderCommand() {
-  const command = new Command5("provider").description("Provider management");
+  const command = new Command6("provider").description("Provider management");
   command.addCommand(
-    new Command5("list").description("List available providers").action(async function() {
+    new Command6("list").description("List available providers").action(async function() {
       try {
         const ctx = createContext(this);
         const data = await ctx.client.get("/api/providers");
@@ -710,7 +806,7 @@ function createProviderCommand() {
     })
   );
   command.addCommand(
-    new Command5("health").description("Check provider health").argument("[name]", "Specific provider name").action(async function(name) {
+    new Command6("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";
@@ -725,11 +821,11 @@ function createProviderCommand() {
 }
 
 // src/commands/upload.ts
-import { Command as Command6 } from "commander";
+import { Command as Command7 } from "commander";
 function createUploadCommand() {
-  const command = new Command6("upload").description("Upload and manage assets");
+  const command = new Command7("upload").description("Upload and manage assets");
   command.addCommand(
-    new Command6("file").description("Upload a file and get a public URL").argument("<path>", "Path to file to upload").action(async function(filePath) {
+    new Command7("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);
@@ -740,7 +836,7 @@ function createUploadCommand() {
     })
   );
   command.addCommand(
-    new Command6("list").description("List uploaded assets").action(async function() {
+    new Command7("list").description("List uploaded assets").action(async function() {
       const ctx = createContext(this);
       try {
         const data = await ctx.client.get("/api/assets");
@@ -751,7 +847,7 @@ function createUploadCommand() {
     })
   );
   command.addCommand(
-    new Command6("delete").description("Delete an uploaded asset (admin only)").argument("<filename>", "Filename to delete").action(async function(filename) {
+    new Command7("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)}`);
@@ -765,12 +861,13 @@ function createUploadCommand() {
 }
 
 // src/index.ts
-var program = new Command7().name("asset-gateway").description("Universal asset generation gateway CLI").version(CLI_VERSION).option(
+var program = new Command8().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(createProviderCommand());
 program.addCommand(createUploadCommand());
 program.addCommand(createJobCommand());

package/package.json

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