AndroidRemoteController/nodejs/ef-compiler/actions/fun/ai/text2img.js

const path = require('path')
const fs = require('fs')
const aiModule = require(path.join(__dirname, '../../../../ai/ai.js'))

function resolveSavePath(savePath, folderPath) {
  if (!savePath || typeof savePath !== 'string') return null
  const trimmed = savePath.trim()
  if (path.isAbsolute(trimmed) || /^[A-Za-z]:/.test(trimmed)) return trimmed
  return folderPath ? path.join(folderPath, trimmed) : path.resolve(trimmed)
}

/** 入参：prompt, model, savePath（可选，图片保存路径；有则返回 b64 并写入文件，否则返回 url） */
async function executeText2img({ prompt, model, savePath, folderPath }) {
  const p = prompt != null ? String(prompt).trim() : ''
  const m = model != null ? String(model).trim().toLowerCase() : ''
  const action = m === 'doubao' ? 'doubao_text2img' : 'text2img'
  const outPath = savePath ? resolveSavePath(savePath, folderPath) : null
  try {
    const result = await aiModule.run(action, p, outPath)
    if (!result.success) return { success: false, error: result.error || 'text2img 失败' }
    const data = result.data
    const item = data?.data?.[0]
    if (!item) return { success: false, error: 'text2img 无返回数据' }
    if (item.b64_json && outPath) {
      const dir = path.dirname(outPath)
      if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true })
      fs.writeFileSync(outPath, Buffer.from(item.b64_json, 'base64'))
      return { success: true, value: outPath, path: outPath }
    }
    const url = item.url || ''
    return { success: true, value: url, path: url || undefined }
  } catch (e) {
    return { success: false, error: (e && (e.message || String(e))) || 'text2img 异常' }
  }
}

module.exports = { executeText2img }