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
    }
    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()
        if (credit < 10) {
          tried++
          logger.info(`账户${email}余额不足，尝试下一个账户`)
          continue
        }
        let description = e.msg.replace(/#((创作)?歌曲|suno|Suno)/, '')
        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('生成失败,请查看日志')
    }
  }
}