mirror of
https://github.com/ikechan8370/chatgpt-plugin.git
synced 2025-12-16 21:37:11 +00:00
102 lines
3.4 KiB
JavaScript
102 lines
3.4 KiB
JavaScript
import plugin from '../../../lib/plugins/plugin.js'
|
||
import { SunoClient } from '../client/SunoClient.js'
|
||
import { Config } from '../utils/config.js'
|
||
import { downloadFile } from '../utils/common.js'
|
||
import common from '../../../lib/common/common.js'
|
||
|
||
export class Vocal extends plugin {
|
||
constructor (e) {
|
||
super({
|
||
name: 'ChatGPT-Plugin 音乐合成',
|
||
dsc: '基于Suno等AI的饮月生成！',
|
||
event: 'message',
|
||
priority: 500,
|
||
rule: [
|
||
{
|
||
reg: '^#((创作)?歌曲|suno|Suno)',
|
||
fnc: 'createSong',
|
||
permission: 'master'
|
||
}
|
||
]
|
||
})
|
||
this.task = [
|
||
{
|
||
// 设置十分钟左右的浮动
|
||
cron: '0/1 * * * ?',
|
||
// cron: '*/2 * * * *',
|
||
name: '保持suno心跳',
|
||
fnc: this.heartbeat.bind(this)
|
||
}
|
||
]
|
||
}
|
||
|
||
async heartbeat (e) {
|
||
let sessTokens = Config.sunoSessToken.split(',')
|
||
let clientTokens = Config.sunoClientToken.split(',')
|
||
for (let i = 0; i < sessTokens.length; i++) {
|
||
let sessToken = sessTokens[i]
|
||
let clientToken = clientTokens[i]
|
||
if (sessToken && clientToken) {
|
||
let client = new SunoClient({ sessToken, clientToken })
|
||
await client.heartbeat()
|
||
}
|
||
}
|
||
}
|
||
|
||
async createSong (e) {
|
||
if (!Config.sunoClientToken || !Config.sunoSessToken) {
|
||
await e.reply('未配置Suno Token')
|
||
return true
|
||
}
|
||
let description = e.msg.replace(/#((创作)?歌曲|suno|Suno)/, '')
|
||
if (description === '额度' || description === 'credit' || description === '余额') {
|
||
let sessTokens = Config.sunoSessToken.split(',')
|
||
let clientTokens = Config.sunoClientToken.split(',')
|
||
let tried = 0
|
||
let msg = ''
|
||
while (tried < sessTokens.length) {
|
||
let index = tried
|
||
let sess = sessTokens[index]
|
||
let clientToken = clientTokens[index]
|
||
let client = new SunoClient({ sessToken: sess, clientToken })
|
||
let { credit, email } = await client.queryCredit()
|
||
logger.info({ credit, email })
|
||
msg += `用户${email}余额：${credit}\n`
|
||
}
|
||
await e.reply(msg)
|
||
return true
|
||
}
|
||
await e.reply('正在生成，请稍后')
|
||
try {
|
||
let sessTokens = Config.sunoSessToken.split(',')
|
||
let clientTokens = Config.sunoClientToken.split(',')
|
||
let tried = 0
|
||
while (tried < sessTokens.length) {
|
||
let index = tried
|
||
let sess = sessTokens[index]
|
||
let clientToken = clientTokens[index]
|
||
let client = new SunoClient({ sessToken: sess, clientToken })
|
||
let { credit, email } = await client.queryCredit()
|
||
logger.info({ credit, email })
|
||
if (credit < 10) {
|
||
tried++
|
||
logger.info(`账户${email}余额不足，尝试下一个账户`)
|
||
continue
|
||
}
|
||
|
||
let songs = await client.createSong(description)
|
||
let messages = ['提示词：' + description]
|
||
for (let song of songs) {
|
||
messages.push(`歌名：${song.title}, 风格: ${song.metadata.tags}, 长度: ${song.metadata.duration}秒\n歌词：\n${song.metadata.prompt}`)
|
||
messages.push(segment.image(song.image_url))
|
||
let videoPath = await downloadFile(song.video_url, `suno/${song.title}.mp4`)
|
||
messages.push(segment.video(videoPath))
|
||
}
|
||
await e.reply(common.makeForwardMsg(e, messages, '音乐合成结果'))
|
||
}
|
||
} catch (err) {
|
||
console.error(err)
|
||
await e.reply('生成失败,请查看日志')
|
||
}
|
||
}
|
||
}
|