import { Context, Dict, h, MessageEncoder } from '@satorijs/core'
import { DingtalkBot } from './bot'
import { SendMessageData } from './types'
// @ts-ignore
import { Entry } from '@cordisjs/plugin-server-temp'

export const escape = (val: string) =>
  val
    .replace(/(?<!\u200b)[\*_~`]/g, '\u200B$&')
    .replace(/([\\`*_{}[\]\-(#!>])/g, '\\$&')
    .replace(/([\-\*]|\d\.) /g, '\u200B$&')
    .replace(/^(\s{4})/gm, '\u200B&nbsp;&nbsp;&nbsp;&nbsp;')

export const unescape = (val: string) =>
  val
    .replace(/\u200b([\*_~`])/g, '$1')

export class DingtalkMessageEncoder<C extends Context = Context> extends MessageEncoder<C, DingtalkBot<C>> {
  buffer = ''

  /**
   * Markdown: https://open.dingtalk.com/document/isvapp/robot-message-types-and-data-format
   */
  hasRichContent = true

  async flush(): Promise<void> {
    if (this.buffer.length && !this.hasRichContent) {
      await this.sendMessage('sampleText', {
        content: this.buffer,
      })
    } else if (this.buffer.length && this.hasRichContent) {
      await this.sendMessage('sampleMarkdown', {
        text: this.buffer.replace(/\n/g, '\n\n'),
      })
    }
  }

  // https://open.dingtalk.com/document/orgapp/the-robot-sends-a-group-message
  async sendMessage<T extends keyof SendMessageData>(msgType: T, msgParam: SendMessageData[T]) {
    const { processQueryKey } = this.session.isDirect ? await this.bot.internal.batchSendOTO({
      msgKey: msgType,
      msgParam: JSON.stringify(msgParam),
      robotCode: this.bot.config.appkey,
      userIds: [this.session.channelId],
    }) : await this.bot.internal.orgGroupSend({
      // https://open.dingtalk.com/document/orgapp/types-of-messages-sent-by-robots
      msgKey: msgType,
      msgParam: JSON.stringify(msgParam),
      robotCode: this.bot.config.appkey,
      openConversationId: this.channelId,
    })
    const session = this.bot.session()
    session.messageId = processQueryKey
    session.channelId = this.session.channelId
    session.guildId = this.session.guildId
    session.app.emit(session, 'send', session)
    this.results.push({
      id: processQueryKey,
    })
  }

  // https://open.dingtalk.com/document/orgapp/upload-media-files?spm=ding_open_doc.document.0.0.3b166172ERBuHw
  async uploadMedia(attrs: Dict) {
    const { data, type } = await this.bot.ctx.http.file(attrs.src || attrs.url, attrs)
    const form = new FormData()
    const value = new Blob([data], { type })
    let mediaType: string
    if (type.startsWith('image/') || type.startsWith('video/')) {
      mediaType = type.split('/')[0]
    } else if (type.startsWith('audio/')) {
      mediaType = 'voice'
    } else {
      mediaType = 'file'
    }
    form.append('type', mediaType)
    form.append('media', value)
    const { media_id } = await this.bot.oldHttp.post('/media/upload', form)
    return media_id
  }

  private listType: 'ol' | 'ul' = null

  async visit(element: h) {
    const { type, attrs, children } = element
    if (type === 'text') {
      this.buffer += escape(attrs.content)
    } else if ((type === 'img' || type === 'image') && (attrs.src || attrs.url)) {
      const src = attrs.src || attrs.url
      // await this.flush()
      // await this.sendMessage('sampleImageMsg', {
      //   photoURL: src,
      // })
      if (await this.bot.http.isLocal(src)) {
        const temp = this.bot.ctx.get('server.temp')
        if (!temp) {
          return this.bot.logger.warn('missing temporary file service, cannot send assets with private url')
        }
        const entry: Entry | undefined = await temp.create(src)
        this.buffer += `![${attrs.alt ?? ''}](${entry.url})`
      } else {
        this.buffer += `![${attrs.alt ?? ''}](${src})`
      }
    } else if (type === 'message') {
      await this.flush()
      await this.render(children)
    } else if (type === 'at') {
      this.buffer += `@${attrs.id}`
    } else if (type === 'br') {
      this.buffer += '\n'
    } else if (type === 'p') {
      if (!this.buffer.endsWith('\n')) this.buffer += '\n'
      await this.render(children)
      if (!this.buffer.endsWith('\n')) this.buffer += '\n'
      this.buffer += '\n'
    } else if (type === 'b' || type === 'strong') {
      this.buffer += ` **`
      await this.render(children)
      this.buffer += `** `
    } else if (type === 'i' || type === 'em') {
      this.buffer += ` *`
      await this.render(children)
      this.buffer += `* `
    } else if (type === 'a' && attrs.href) {
      this.buffer += `[`
      await this.render(children)
      this.buffer += `](${encodeURI(attrs.href)})`
    } else if (type === 'ul' || type === 'ol') {
      this.listType = type
      await this.render(children)
      this.listType = null
    } else if (type === 'li') {
      if (!this.buffer.endsWith('\n')) this.buffer += '\n'
      if (this.listType === 'ol') {
        this.buffer += `1. `
      } else if (this.listType === 'ul') {
        this.buffer += '- '
      }
      this.render(children)
      this.buffer += '\n'
    } else if (type === 'blockquote') {
      if (!this.buffer.endsWith('\n')) this.buffer += '\n'
      this.buffer += '> '
      await this.render(children)
      this.buffer += '\n\n'
    }
  }
}