115 lines
3.7 KiB
JavaScript
115 lines
3.7 KiB
JavaScript
/**
|
||
* 光湖视频AI系统 · Kling 可灵 API 适配器
|
||
* D136+ · 铸渊 ICE-GL-ZY001 · 2026-06-21
|
||
*
|
||
* 接入可灵AI视频生成API。价格: kling-v2.6-pro $0.049/s (约1.8元/5秒)
|
||
* 非Seedance——用于低成本验证提示词和流程
|
||
*/
|
||
|
||
const fs = require('fs');
|
||
const path = require('path');
|
||
const https = require('https');
|
||
const { loadVideoAiEnv } = require('./env-loader');
|
||
|
||
loadVideoAiEnv();
|
||
|
||
// API配置
|
||
// 仅从苍耳本机环境读取；密钥绝不写入仓库。
|
||
const API_KEY = process.env.KLING_API_KEY || '';
|
||
const BASE_URL = 'api.klingapi.com';
|
||
const POLL_INTERVAL = 3000; // 可灵更快，3秒轮询
|
||
const MAX_POLL = 60; // 最多3分钟
|
||
|
||
// 输出路径
|
||
const OUT_DIR = (() => {
|
||
const jzao = '/Volumes/JZAO/铸渊-ICE-GL-ZY001/OUT-输出/视频/zai-fu-fei-xiu-xian/ep01';
|
||
return fs.existsSync(jzao) ? jzao : path.resolve(__dirname, '../outputs/shots');
|
||
})();
|
||
|
||
function apiRequest(method, path_, body = null) {
|
||
if (!API_KEY) {
|
||
return Promise.reject(new Error('KLING_API_KEY 未设置。请仅在苍耳本机的仓库外密钥文件或环境变量中配置。'));
|
||
}
|
||
return new Promise((resolve, reject) => {
|
||
const options = {
|
||
hostname: BASE_URL,
|
||
path: path_,
|
||
method,
|
||
headers: {
|
||
'Authorization': `Bearer ${API_KEY}`,
|
||
'Content-Type': 'application/json',
|
||
},
|
||
};
|
||
const req = https.request(options, (res) => {
|
||
let data = '';
|
||
res.on('data', chunk => data += chunk);
|
||
res.on('end', () => {
|
||
try { resolve(JSON.parse(data)); }
|
||
catch (e) { resolve({ raw: data }); }
|
||
});
|
||
});
|
||
req.on('error', reject);
|
||
if (body) req.write(JSON.stringify(body));
|
||
req.end();
|
||
});
|
||
}
|
||
|
||
function downloadFile(url, dest) {
|
||
return new Promise((resolve, reject) => {
|
||
const file = fs.createWriteStream(dest);
|
||
const proto = url.startsWith('https') ? https : require('http');
|
||
proto.get(url, (res) => {
|
||
if (res.statusCode >= 300 && res.statusCode < 400) {
|
||
return downloadFile(res.headers.location, dest).then(resolve).catch(reject);
|
||
}
|
||
res.pipe(file);
|
||
file.on('finish', () => { file.close(); resolve(dest); });
|
||
}).on('error', reject);
|
||
});
|
||
}
|
||
|
||
async function generateKlingVideo({ prompt, duration = 5, outputPath, model = 'kling-v2.6-pro' }) {
|
||
console.log('[Kling] 提交文生视频...');
|
||
|
||
// 提交
|
||
const submit = await apiRequest('POST', '/v1/videos/text2video', {
|
||
model,
|
||
prompt,
|
||
duration: Math.min(duration, 10),
|
||
aspect_ratio: '16:9',
|
||
mode: 'standard',
|
||
});
|
||
|
||
if (!submit.task_id) {
|
||
throw new Error(`提交失败: ${JSON.stringify(submit)}`);
|
||
}
|
||
console.log(`[Kling] 任务: ${submit.task_id}`);
|
||
|
||
// 轮询
|
||
for (let i = 1; i <= MAX_POLL; i++) {
|
||
await new Promise(r => setTimeout(r, POLL_INTERVAL));
|
||
const status = await apiRequest('GET', `/v1/videos/${submit.task_id}`);
|
||
|
||
if (status.status === 'completed') {
|
||
const videoUrl = status.video_url || status.output?.video_url;
|
||
if (!videoUrl) throw new Error('任务完成但无视频URL');
|
||
|
||
const out = outputPath || path.join(OUT_DIR, `kling-${submit.task_id}.mp4`);
|
||
fs.mkdirSync(path.dirname(out), { recursive: true });
|
||
console.log(`[Kling] 下载: ${videoUrl.substring(0,60)}...`);
|
||
await downloadFile(videoUrl, out);
|
||
console.log(`[Kling] ✅ ${path.basename(out)}`);
|
||
return { videoPath: out, taskId: submit.task_id };
|
||
}
|
||
|
||
if (status.status === 'failed') {
|
||
throw new Error(`生成失败: ${status.error || JSON.stringify(status)}`);
|
||
}
|
||
|
||
if (i % 10 === 0) console.log(`[Kling] 生成中... (${i}/${MAX_POLL})`);
|
||
}
|
||
throw new Error('超时');
|
||
}
|
||
|
||
module.exports = { generateKlingVideo };
|