import {
  AtType,
  ChatType,
  ElementType,
  Friend,
  Group,
  GroupMemberRole,
  PicSubType,
  RawMessage,
  SendMessageElement,
} from '../../../ntqqapi/types'
import { friends, getGroup, getGroupMember, getUidByUin, selfInfo } from '../../../common/data'
import {
  OB11MessageCustomMusic,
  OB11MessageData,
  OB11MessageDataType,
  OB11MessageFile,
  OB11MessageJson,
  OB11MessageMixType,
  OB11MessageMusic,
  OB11MessageNode,
  OB11PostSendMsg,
} from '../../types'
import { SendMsgElementConstructor } from '../../../ntqqapi/constructor'
import BaseAction from '../BaseAction'
import { ActionName, BaseCheckResult } from '../types'
import fs from 'node:fs'
import { decodeCQCode } from '../../cqcode'
import { dbUtil } from '../../../common/db'
import { ALLOW_SEND_TEMP_MSG, getConfigUtil } from '../../../common/config'
import { log } from '../../../common/utils/log'
import { sleep } from '../../../common/utils/helper'
import { uri2local } from '../../../common/utils'
import { NTQQGroupApi, NTQQMsgApi, NTQQUserApi, NTQQFriendApi } from '@/ntqqapi/api'
import { CustomMusicSignPostData, IdMusicSignPostData, MusicSign, MusicSignPostData } from '@/common/utils/sign'
import { Peer } from '@/ntqqapi/types/msg'

export interface ReturnDataType {
  message_id: number
}

export enum ContextMode {
  Normal = 0,
  Private = 1,
  Group = 2
}

export function convertMessage2List(message: OB11MessageMixType, autoEscape = false) {
  if (typeof message === 'string') {
    if (autoEscape === true) {
      message = [
        {
          type: OB11MessageDataType.text,
          data: {
            text: message,
          },
        },
      ]
    }
    else {
      message = decodeCQCode(message.toString())
    }
  }
  else if (!Array.isArray(message)) {
    message = [message]
  }
  return message
}

export async function createSendElements(
  messageData: OB11MessageData[],
  peer: Peer,
  ignoreTypes: OB11MessageDataType[] = [],
) {
  let sendElements: SendMessageElement[] = []
  let deleteAfterSentFiles: string[] = []
  for (let sendMsg of messageData) {
    if (ignoreTypes.includes(sendMsg.type)) {
      continue
    }
    switch (sendMsg.type) {
      case OB11MessageDataType.text: {
        const text = sendMsg.data?.text
        if (text) {
          sendElements.push(SendMsgElementConstructor.text(sendMsg.data!.text))
        }
      }
        break
      case OB11MessageDataType.at: {
        if (!peer) {
          continue
        }
        let atQQ = sendMsg.data?.qq
        if (atQQ) {
          atQQ = atQQ.toString()
          if (atQQ === 'all') {
            // todo：查询剩余的at全体次数
            const groupCode = peer.peerUid
            let remainAtAllCount = 1
            let isAdmin: boolean = true
            if (groupCode) {
              try {
                remainAtAllCount = (await NTQQGroupApi.getGroupAtAllRemainCount(groupCode)).atInfo
                  .RemainAtAllCountForUin
                log(`群${groupCode}剩余at全体次数`, remainAtAllCount)
                const self = await getGroupMember(groupCode, selfInfo.uin)
                isAdmin = self?.role === GroupMemberRole.admin || self?.role === GroupMemberRole.owner
              } catch (e) {
              }
            }
            if (isAdmin && remainAtAllCount > 0) {
              sendElements.push(SendMsgElementConstructor.at(atQQ, atQQ, AtType.atAll, '@全体成员'))
            }
          }
          else if (peer.chatType === ChatType.group) {
            const atMember = await getGroupMember(peer.peerUid, atQQ)
            if (atMember) {
              const display = `@${atMember.cardName || atMember.nick}`
              sendElements.push(
                SendMsgElementConstructor.at(atQQ, atMember.uid, AtType.atUser, display),
              )
            } else {
              const atNmae = sendMsg.data?.name
              const uid = await NTQQUserApi.getUidByUin(atQQ) || ''
              const display = atNmae ? `@${atNmae}` : ''
              sendElements.push(
                SendMsgElementConstructor.at(atQQ, uid, AtType.atUser, display),
              )
            }
          }
        }
      }
        break
      case OB11MessageDataType.reply: {
        let replyMsgId = sendMsg.data.id
        if (replyMsgId) {
          const replyMsg = await dbUtil.getMsgByShortId(parseInt(replyMsgId))
          if (replyMsg) {
            sendElements.push(
              SendMsgElementConstructor.reply(
                replyMsg.msgSeq,
                replyMsg.msgId,
                replyMsg.senderUin!,
                replyMsg.senderUin!,
              ),
            )
          }
        }
      }
        break
      case OB11MessageDataType.face: {
        const faceId = sendMsg.data?.id
        if (faceId) {
          sendElements.push(SendMsgElementConstructor.face(parseInt(faceId)))
        }
      }
        break
      case OB11MessageDataType.mface: {
        sendElements.push(
          SendMsgElementConstructor.mface(
            sendMsg.data.emoji_package_id,
            sendMsg.data.emoji_id,
            sendMsg.data.key,
            sendMsg.data.summary,
          ),
        )
      }
        break
      case OB11MessageDataType.image:
      case OB11MessageDataType.file:
      case OB11MessageDataType.video:
      case OB11MessageDataType.voice: {
        const data = (sendMsg as OB11MessageFile).data
        let file = data.file
        const payloadFileName = data?.name
        if (file) {
          const cache = await dbUtil.getFileCache(file)
          if (cache) {
            if (fs.existsSync(cache.filePath)) {
              file = 'file://' + cache.filePath
            }
            else if (cache.downloadFunc) {
              await cache.downloadFunc()
              file = cache.filePath
            }
            else if (cache.url) {
              file = cache.url
            }
            log('找到文件缓存', file)
          }
          const { path, isLocal, fileName, errMsg } = await uri2local(file)
          if (errMsg) {
            throw errMsg
          }
          if (path) {
            if (!isLocal) {
              // 只删除http和base64转过来的文件
              deleteAfterSentFiles.push(path)
            }
            if (sendMsg.type === OB11MessageDataType.file) {
              log('发送文件', path, payloadFileName || fileName)
              sendElements.push(await SendMsgElementConstructor.file(path, payloadFileName || fileName))
            }
            else if (sendMsg.type === OB11MessageDataType.video) {
              log('发送视频', path, payloadFileName || fileName)
              let thumb = sendMsg.data?.thumb
              if (thumb) {
                let uri2LocalRes = await uri2local(thumb)
                if (uri2LocalRes.success) {
                  thumb = uri2LocalRes.path
                }
              }
              sendElements.push(await SendMsgElementConstructor.video(path, payloadFileName || fileName, thumb))
            }
            else if (sendMsg.type === OB11MessageDataType.voice) {
              sendElements.push(await SendMsgElementConstructor.ptt(path))
            }
            else if (sendMsg.type === OB11MessageDataType.image) {
              sendElements.push(
                await SendMsgElementConstructor.pic(
                  path,
                  sendMsg.data.summary || '',
                  <PicSubType>parseInt(sendMsg.data?.subType?.toString()!) || 0,
                ),
              )
            }
          }
        }
      }
        break
      case OB11MessageDataType.json: {
        sendElements.push(SendMsgElementConstructor.ark(sendMsg.data.data))
      }
        break
      case OB11MessageDataType.poke: {
        let qq = sendMsg.data?.qq || sendMsg.data?.id
      }
        break
      case OB11MessageDataType.dice: {
        const resultId = sendMsg.data?.result
        sendElements.push(SendMsgElementConstructor.dice(resultId))
      }
        break
      case OB11MessageDataType.RPS: {
        const resultId = sendMsg.data?.result
        sendElements.push(SendMsgElementConstructor.rps(resultId))
      }
        break
    }
  }

  return {
    sendElements,
    deleteAfterSentFiles,
  }
}

export async function sendMsg(
  peer: Peer,
  sendElements: SendMessageElement[],
  deleteAfterSentFiles: string[],
  waitComplete = true,
) {
  if (!sendElements.length) {
    throw '消息体无法解析，请检查是否发送了不支持的消息类型'
  }
  // 计算发送的文件大小
  let totalSize = 0
  for (const fileElement of sendElements) {
    try {
      if (fileElement.elementType === ElementType.PTT) {
        totalSize += fs.statSync(fileElement.pttElement.filePath).size
      }
      if (fileElement.elementType === ElementType.FILE) {
        totalSize += fs.statSync(fileElement.fileElement.filePath).size
      }
      if (fileElement.elementType === ElementType.VIDEO) {
        totalSize += fs.statSync(fileElement.videoElement.filePath).size
      }
      if (fileElement.elementType === ElementType.PIC) {
        totalSize += fs.statSync(fileElement.picElement.sourcePath).size
      }
    } catch (e) {
      log('文件大小计算失败', e, fileElement)
    }
  }
  log('发送消息总大小', totalSize, 'bytes')
  let timeout = ((totalSize / 1024 / 100) * 1000) + 5000  // 100kb/s
  log('设置消息超时时间', timeout)
  const returnMsg = await NTQQMsgApi.sendMsg(peer, sendElements, waitComplete, timeout)
  log('消息发送结果', returnMsg)
  returnMsg.msgShortId = await dbUtil.addMsg(returnMsg)
  deleteAfterSentFiles.map((f) => fs.unlink(f, () => {
  }))
  return returnMsg
}

async function createContext(payload: OB11PostSendMsg, contextMode: ContextMode): Promise<Peer> {
  // This function determines the type of message by the existence of user_id / group_id,
  // not message_type.
  // This redundant design of Ob11 here should be blamed.

  if ((contextMode === ContextMode.Group || contextMode === ContextMode.Normal) && payload.group_id) {
    const group = (await getGroup(payload.group_id))! // checked before
    return {
      chatType: ChatType.group,
      peerUid: group.groupCode
    }
  }
  if ((contextMode === ContextMode.Private || contextMode === ContextMode.Normal) && payload.user_id) {
    const Uid = await NTQQUserApi.getUidByUin(payload.user_id.toString())
    const isBuddy = await NTQQFriendApi.isBuddy(Uid!)
    //console.log("[调试代码] UIN:", payload.user_id, " UID:", Uid, " IsBuddy:", isBuddy)
    return {
      chatType: isBuddy ? ChatType.friend : ChatType.temp,
      peerUid: Uid!,
      guildId: payload.group_id || ''//临时主动发起时需要传入群号
    }
  }
  throw '请指定 group_id 或 user_id'
}

export class SendMsg extends BaseAction<OB11PostSendMsg, ReturnDataType> {
  actionName = ActionName.SendMsg

  protected async check(payload: OB11PostSendMsg): Promise<BaseCheckResult> {
    const messages = convertMessage2List(payload.message)
    const fmNum = this.getSpecialMsgNum(messages, OB11MessageDataType.node)
    if (fmNum && fmNum != messages.length) {
      return {
        valid: false,
        message: '转发消息不能和普通消息混在一起发送,转发需要保证message只有type为node的元素',
      }
    }
    const musicNum = this.getSpecialMsgNum(messages, OB11MessageDataType.music)
    if (musicNum && messages.length > 1) {
      return {
        valid: false,
        message: '音乐消息不可以和其他消息混在一起发送',
      }
    }
    if (payload.message_type !== 'private' && payload.group_id && !(await getGroup(payload.group_id))) {
      return {
        valid: false,
        message: `群${payload.group_id}不存在`,
      }
    }
    if (payload.user_id && payload.message_type !== 'group') {
      const uid = await NTQQUserApi.getUidByUin(payload.user_id.toString())
      const isBuddy = await NTQQFriendApi.isBuddy(uid!)
      // 此处有问题
      if (!isBuddy) {
        //return { valid: false, message: '异常消息' }
      }
    }
    return {
      valid: true,
    }
  }

  protected async _handle(payload: OB11PostSendMsg) {
    const peer = await createContext(payload, ContextMode.Normal)
    const messages = convertMessage2List(
      payload.message,
      payload.auto_escape === true || payload.auto_escape === 'true',
    )
    if (this.getSpecialMsgNum(messages, OB11MessageDataType.node)) {
      try {
        const returnMsg = await this.handleForwardNode(peer, messages as OB11MessageNode[])
        return { message_id: returnMsg?.msgShortId! }
      } catch (e: any) {
        throw '发送转发消息失败 ' + e.toString()
      }
    }
    else if (this.getSpecialMsgNum(messages, OB11MessageDataType.music)) {
      const music = messages[0] as OB11MessageMusic
      if (music) {
        const { musicSignUrl } = getConfigUtil().getConfig()
        if (!musicSignUrl) {
          throw '音乐签名地址未配置'
        }
        const { type } = music.data
        if (!['qq', '163', 'custom'].includes(type)) {
          throw `不支持的音乐类型 ${type}`
        }
        const postData: MusicSignPostData = { ...music.data }
        if (type === 'custom' && music.data.content) {
          const data = postData as CustomMusicSignPostData
          data.singer = music.data.content
          delete (data as OB11MessageCustomMusic['data']).content
        }
        if (type === 'custom') {
          const customMusicData = music.data as CustomMusicSignPostData
          if (!customMusicData.url) {
            throw '自定义音卡缺少参数url'
          }
          if (!customMusicData.audio) {
            throw '自定义音卡缺少参数audio'
          }
          if (!customMusicData.title) {
            throw '自定义音卡缺少参数title'
          }
        }
        if (type === 'qq' || type === '163') {
          const idMusicData = music.data as IdMusicSignPostData
          if (!idMusicData.id) {
            throw '音乐卡片缺少id参数'
          }
        }
        let jsonContent: string
        try {
          jsonContent = await new MusicSign(musicSignUrl).sign(postData)
          if (!jsonContent) {
            throw '音乐消息生成失败，提交内容有误或者签名服务器签名失败'
          }
        } catch (e) {
          throw `签名音乐消息失败：${e}`
        }
        messages[0] = {
          type: OB11MessageDataType.json,
          data: { data: jsonContent },
        } as OB11MessageJson
      }
    }
    // log("send msg:", peer, sendElements)
    const { sendElements, deleteAfterSentFiles } = await createSendElements(messages, peer)
    if (sendElements.length === 1) {
      if (sendElements[0] === null) {
        return { message_id: 0 }
      }
    }
    const returnMsg = await sendMsg(peer, sendElements, deleteAfterSentFiles)
    deleteAfterSentFiles.map((f) => fs.unlink(f, () => {
    }))
    return { message_id: returnMsg.msgShortId! }
  }

  private getSpecialMsgNum(message: OB11MessageData[], msgType: OB11MessageDataType): number {
    if (Array.isArray(message)) {
      return message.filter((msg) => msg.type == msgType).length
    }
    return 0
  }

  private async cloneMsg(msg: RawMessage): Promise<RawMessage | undefined> {
    log('克隆的目标消息', msg)
    let sendElements: SendMessageElement[] = []
    for (const ele of msg.elements) {
      sendElements.push(ele as SendMessageElement)
      // Object.keys(ele).forEach((eleKey) => {
      //     if (eleKey.endsWith("Element")) {
      //     }
    }
    if (sendElements.length === 0) {
      log('需要clone的消息无法解析，将会忽略掉', msg)
    }
    log('克隆消息', sendElements)
    try {
      const nodeMsg = await NTQQMsgApi.sendMsg(
        {
          chatType: ChatType.friend,
          peerUid: selfInfo.uid,
        },
        sendElements,
        true,
      )
      await sleep(500)
      return nodeMsg
    } catch (e) {
      log(e, '克隆转发消息失败,将忽略本条消息', msg)
    }
  }

  // 返回一个合并转发的消息id
  private async handleForwardNode(destPeer: Peer, messageNodes: OB11MessageNode[]) {
    const selfPeer = {
      chatType: ChatType.friend,
      peerUid: selfInfo.uid,
    }
    let nodeMsgIds: string[] = []
    // 先判断一遍是不是id和自定义混用
    let needClone =
      messageNodes.filter((node) => node.data.id).length && messageNodes.filter((node) => !node.data.id).length
    for (const messageNode of messageNodes) {
      // 一个node表示一个人的消息
      let nodeId = messageNode.data.id
      // 有nodeId表示一个子转发消息卡片
      if (nodeId) {
        let nodeMsg = await dbUtil.getMsgByShortId(parseInt(nodeId))
        if (!needClone) {
          nodeMsgIds.push(nodeMsg?.msgId!)
        }
        else {
          if (nodeMsg?.peerUid !== selfInfo.uid) {
            const cloneMsg = await this.cloneMsg(nodeMsg!)
            if (cloneMsg) {
              nodeMsgIds.push(cloneMsg.msgId)
            }
          }
        }
      }
      else {
        // 自定义的消息
        // 提取消息段，发给自己生成消息id
        try {
          const { sendElements, deleteAfterSentFiles } = await createSendElements(
            convertMessage2List(messageNode.data.content),
            destPeer
          )
          log('开始生成转发节点', sendElements)
          let sendElementsSplit: SendMessageElement[][] = []
          let splitIndex = 0
          for (const ele of sendElements) {
            if (!sendElementsSplit[splitIndex]) {
              sendElementsSplit[splitIndex] = []
            }

            if (ele.elementType === ElementType.FILE || ele.elementType === ElementType.VIDEO) {
              if (sendElementsSplit[splitIndex].length > 0) {
                splitIndex++
              }
              sendElementsSplit[splitIndex] = [ele]
              splitIndex++
            }
            else {
              sendElementsSplit[splitIndex].push(ele)
            }
            log(sendElementsSplit)
          }
          // log("分割后的转发节点", sendElementsSplit)
          for (const eles of sendElementsSplit) {
            const nodeMsg = await sendMsg(selfPeer, eles, [], true)
            nodeMsgIds.push(nodeMsg.msgId)
            await sleep(500)
            log('转发节点生成成功', nodeMsg.msgId)
          }
          deleteAfterSentFiles.map((f) => fs.unlink(f, () => {
          }))
        } catch (e) {
          log('生成转发消息节点失败', e)
        }
      }
    }

    // 检查srcPeer是否一致，不一致则需要克隆成自己的消息, 让所有srcPeer都变成自己的，使其保持一致才能够转发
    let nodeMsgArray: Array<RawMessage> = []
    let srcPeer: Peer | null = null
    let needSendSelf = false
    for (const [index, msgId] of nodeMsgIds.entries()) {
      const nodeMsg = await dbUtil.getMsgByLongId(msgId)
      if (nodeMsg) {
        nodeMsgArray.push(nodeMsg)
        if (!srcPeer) {
          srcPeer = { chatType: nodeMsg.chatType, peerUid: nodeMsg.peerUid }
        }
        else if (srcPeer.peerUid !== nodeMsg.peerUid) {
          needSendSelf = true
          srcPeer = selfPeer
        }
      }
    }
    log('nodeMsgArray', nodeMsgArray)
    nodeMsgIds = nodeMsgArray.map((msg) => msg.msgId)
    if (needSendSelf) {
      log('需要克隆转发消息')
      for (const [index, msg] of nodeMsgArray.entries()) {
        if (msg.peerUid !== selfInfo.uid) {
          const cloneMsg = await this.cloneMsg(msg)
          if (cloneMsg) {
            nodeMsgIds[index] = cloneMsg.msgId
          }
        }
      }
    }
    // elements之间用换行符分隔
    // let _sendForwardElements: SendMessageElement[] = []
    // for(let i = 0; i < sendForwardElements.length; i++){
    //     _sendForwardElements.push(sendForwardElements[i])
    //     _sendForwardElements.push(SendMsgElementConstructor.text("\n\n"))
    // }
    // const nodeMsg = await NTQQApi.sendMsg(selfPeer, _sendForwardElements, true);
    // nodeIds.push(nodeMsg.msgId)
    // await sleep(500);
    // 开发转发
    if (nodeMsgIds.length === 0) {
      throw Error('转发消息失败，节点为空')
    }
    try {
      log('开发转发', nodeMsgIds)
      return await NTQQMsgApi.multiForwardMsg(srcPeer!, destPeer, nodeMsgIds)
    } catch (e) {
      log('forward failed', e)
      return null
    }
  }
}

export default SendMsg