65 lines
2.2 KiB
JavaScript
65 lines
2.2 KiB
JavaScript
|
#!/usr/bin/env node
|
|||
|
|
/**
|
|||
|
|
* image-api-bridge.js · Python ↔ Node 图片API桥接
|
|||
|
|
* D144 · 铸渊 ICE-GL-ZY001
|
|||
|
|
*
|
|||
|
|
* Python 通过 subprocess 调用本脚本，stdin 输入 JSON 命令，stdout 输出 JSON 结果。
|
|||
|
|
* 解决 char-hero-design-packer.py 依赖 JS image-api-adapter 的问题。
|
|||
|
|
*
|
|||
|
|
* 用法:
|
|||
|
|
* echo '{"action":"generate_character_ref","charId":"CHAR-003","l0":"...","l1":"..."}' | node image-api-bridge.js
|
|||
|
|
* echo '{"action":"generate_image","prompt":"...","size":"2048x2048","filename":"test.png"}' | node image-api-bridge.js
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
// 桥接模式：所有日志输出到 stderr，stdout 只输出最终 JSON 结果
|
|||
|
|
console.log = (...args) => process.stderr.write(args.join(' ') + '\n');
|
|||
|
|
console.warn = (...args) => process.stderr.write(args.join(' ') + '\n');
|
|||
|
|
console.error = (...args) => process.stderr.write(args.join(' ') + '\n');
|
|||
|
|
|
|||
|
|
const { generateImage, generateCharacterRef } = require('./image-api-adapter');
|
|||
|
|
|
|||
|
|
let input = '';
|
|||
|
|
process.stdin.setEncoding('utf8');
|
|||
|
|
process.stdin.on('data', chunk => input += chunk);
|
|||
|
|
process.stdin.on('end', async () => {
|
|||
|
|
try {
|
|||
|
|
const req = JSON.parse(input || '{}');
|
|||
|
|
let result;
|
|||
|
|
|
|||
|
|
switch (req.action) {
|
|||
|
|
case 'generate_character_ref': {
|
|||
|
|
const { charId, l0, l1 } = req;
|
|||
|
|
if (!charId) throw new Error('缺少 charId');
|
|||
|
|
result = await generateCharacterRef({
|
|||
|
|
charId,
|
|||
|
|
l0Descriptor: l0 || '中国古代修仙少年，16岁，亚洲面孔',
|
|||
|
|
l1Config: l1 ? { clothing: l1 } : undefined,
|
|||
|
|
});
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
case 'generate_image': {
|
|||
|
|
const { prompt, size, filename, outputPath } = req;
|
|||
|
|
if (!prompt) throw new Error('缺少 prompt');
|
|||
|
|
result = await generateImage({ prompt, size, filename, outputPath });
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
default:
|
|||
|
|
throw new Error(`未知操作: ${req.action}`);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
process.stdout.write(JSON.stringify({ ok: true, data: result }) + '\n');
|
|||
|
|
process.exit(0);
|
|||
|
|
} catch (err) {
|
|||
|
|
process.stdout.write(JSON.stringify({ ok: false, error: err.message }) + '\n');
|
|||
|
|
process.exit(1);
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// stdin 已关闭（pipe 模式），直接触发 end
|
|||
|
|
if (process.stdin.isTTY) {
|
|||
|
|
process.stderr.write('用法: echo \'{"action":"..."}\' | node image-api-bridge.js\n');
|
|||
|
|
process.exit(1);
|
|||
|
|
}
|