opencode-dashscope-imagegen 0.1.0

package/package.json

@@ -0,0 +1,34 @@
+{
+  "name": "opencode-dashscope-imagegen",
+  "version": "0.1.0",
+  "description": "OpenCode plugin: text-to-image generation via Alibaba DashScope (qwen-image / wan models). Adds an image_generate tool.",
+  "type": "module",
+  "main": "dist/index.js",
+  "exports": {
+    ".": "./dist/index.js"
+  },
+  "files": [
+    "dist",
+    "src"
+  ],
+  "scripts": {
+    "typecheck": "tsc --noEmit",
+    "build": "bun build src/index.ts --target node --format esm --outdir dist"
+  },
+  "keywords": [
+    "opencode",
+    "plugin",
+    "dashscope",
+    "qwen",
+    "image-generation",
+    "text-to-image"
+  ],
+  "license": "MIT",
+  "dependencies": {
+    "@opencode-ai/plugin": "latest"
+  },
+  "devDependencies": {
+    "typescript": "^5.5.0",
+    "@types/node": "^22.0.0"
+  }
+}

package/src/index.ts

@@ -0,0 +1,171 @@
+import type { Plugin } from "@opencode-ai/plugin";
+import { tool } from "@opencode-ai/plugin";
+import { writeFileSync, mkdirSync, existsSync, readFileSync } from "node:fs";
+import { join, isAbsolute, dirname } from "node:path";
+
+const LOG = "[dashscope-imagegen]";
+const ENDPOINT =
+  "https://dashscope.aliyuncs.com/api/v1/services/aigc/multimodal-generation/generation";
+const DEFAULT_MODEL = "qwen-image-2.0";
+
+console.log(`${LOG} plugin module loaded (pid=${process.pid}, platform=${process.platform})`);
+
+function configDir(): string {
+  return (
+    process.env.OPENCODE_CONFIG_DIR ??
+    join(process.env.USERPROFILE ?? process.env.HOME ?? ".", ".config", "opencode")
+  );
+}
+
+/** Resolve DashScope API key: env first, then opencode config (dashscope* providers). */
+function resolveApiKey(): string {
+  const fromEnv =
+    process.env.DASHSCOPE_IMAGEGEN_API_KEY ?? process.env.DASHSCOPE_API_KEY;
+  if (fromEnv) return fromEnv;
+
+  try {
+    const cfgPath = join(configDir(), "opencode.json");
+    const cfg = JSON.parse(readFileSync(cfgPath, "utf-8")) as {
+      provider?: Record<string, { options?: { apiKey?: string } }>;
+    };
+    for (const [name, prov] of Object.entries(cfg.provider ?? {})) {
+      if (name.startsWith("dashscope") && prov?.options?.apiKey) {
+        return prov.options.apiKey;
+      }
+    }
+  } catch {
+    // fall through
+  }
+  throw new Error(
+    "No DashScope API key found. Set DASHSCOPE_API_KEY env or configure a dashscope* provider in opencode.json.",
+  );
+}
+
+function outputDir(): string {
+  return (
+    process.env.DASHSCOPE_IMAGEGEN_DIR ?? join(configDir(), "gen-images")
+  );
+}
+
+async function generateImageUrl(
+  apiKey: string,
+  model: string,
+  prompt: string,
+  size: string,
+): Promise<string> {
+  const res = await fetch(ENDPOINT, {
+    method: "POST",
+    headers: {
+      Authorization: `Bearer ${apiKey}`,
+      "Content-Type": "application/json",
+    },
+    body: JSON.stringify({
+      model,
+      input: {
+        messages: [{ role: "user", content: [{ text: prompt }] }],
+      },
+      parameters: { size },
+    }),
+  });
+  const json = (await res.json()) as {
+    code?: string;
+    message?: string;
+    output?: {
+      choices?: { message?: { content?: { image?: string }[] } }[];
+    };
+  };
+  const url = json.output?.choices?.[0]?.message?.content?.[0]?.image;
+  if (!url) {
+    throw new Error(
+      `DashScope image generation failed: ${json.code ?? res.status} ${json.message ?? ""}`,
+    );
+  }
+  return url;
+}
+
+interface GenArgs {
+  prompt: string;
+  model?: string;
+  size?: string;
+  output_path?: string;
+}
+
+async function runGenerate(args: GenArgs) {
+  const apiKey = resolveApiKey();
+  const model = args.model ?? DEFAULT_MODEL;
+  const size = args.size ?? "1024*1024";
+
+  const url = await generateImageUrl(apiKey, model, args.prompt, size);
+  const imgRes = await fetch(url);
+  if (!imgRes.ok) {
+    throw new Error(`Image download failed: ${imgRes.status}`);
+  }
+  const buf = Buffer.from(await imgRes.arrayBuffer());
+
+  let outPath = args.output_path;
+  if (!outPath) {
+    const dir = outputDir();
+    if (!existsSync(dir)) mkdirSync(dir, { recursive: true });
+    const ts = new Date().toISOString().replace(/[:.]/g, "-");
+    outPath = join(dir, `gen-${ts}.png`);
+  } else if (!isAbsolute(outPath)) {
+    throw new Error("output_path must be an absolute path");
+  } else if (!existsSync(dirname(outPath))) {
+    mkdirSync(dirname(outPath), { recursive: true });
+  }
+
+  writeFileSync(outPath, buf);
+  const dataUrl = `data:image/png;base64,${buf.toString("base64")}`;
+  return {
+    title: "Generated image",
+    output: `Image generated and saved to: ${outPath}`,
+    attachments: [
+      {
+        type: "file" as const,
+        mime: "image/png",
+        url: dataUrl,
+        filename: outPath.split(/[\\/]/).pop(),
+      },
+    ],
+  };
+}
+
+const DashscopeImagegenPlugin: Plugin = async () => {
+  console.log(`${LOG} plugin() invoked, registering image_generate tool`);
+  return {
+    tool: {
+      image_generate: tool({
+        description:
+          "Generate an image from a text prompt via Alibaba DashScope (qwen-image-2.0 / qwen-image-3.0 / wan2.7-image). Saves PNG to disk and returns the absolute file path plus an image attachment.",
+        args: {
+          prompt: tool.schema.string().describe("Image description"),
+          model: tool.schema
+            .string()
+            .optional()
+            .describe("Model id (default: qwen-image-2.0)"),
+          size: tool.schema
+            .string()
+            .optional()
+            .describe("Size as WxH with star, e.g. 1024*1024 (default)"),
+          output_path: tool.schema
+            .string()
+            .optional()
+            .describe("Absolute output file path (default: auto-named in gen-images dir)"),
+        },
+        async execute(args) {
+          console.log(
+            `${LOG} execute: prompt="${args.prompt.slice(0, 80)}" model=${args.model ?? DEFAULT_MODEL} size=${args.size ?? "1024*1024"}`,
+          );
+          try {
+            return await runGenerate(args);
+          } catch (e) {
+            console.error(`${LOG} execute failed:`, e);
+            throw e;
+          }
+        },
+      }),
+    },
+  };
+};
+
+export default DashscopeImagegenPlugin;