const {
  makeWASocket,
  Browsers,
  DisconnectReason,
  fetchLatestBaileysVersion,
  makeCacheableSignalKeyStore,
  makeInMemoryStore,
  useMultiFileAuthState,
  downloadMediaMessage,
  isJidUser, isJidGroup, isJidBroadcast, isJidStatusBroadcast
} = require('@whiskeysockets/baileys');
const fs = require('fs');
const path = require('path');
const { writeFile } = require('fs/promises');
const waEmitter = require('../emitter');
const serverConfig = require('../../config').server;

const { encodeJid, decodeJid, formatStatus, formatTimestamp, getFileExtension, uint8ArrayToBase64 } = require('./helper');
const generateId = require('../../utils/generateId.util');
const NodeCache = require('node-cache');
const P = require('pino');

// https://baileys.whiskeysockets.io/

const createWhatsApp = async phone => {
  let qrCode = null;
  const channelId = generateId();
  const whatsAppNo = phone;
  // 缓存 msgId-externalId，过期时间为 5 分钟
  const externalIdCache = new NodeCache({ stdTTL: 60*5 });
  // 缓存群信息，过期时间为 24 小时
  const groupSubjectCache = new NodeCache({ stdTTL: 60*60*24 });
  const logger = P({ timestamp: () => `,"time":"${new Date().toJSON()}"` }, P.destination('./logs/wa-logs-' + phone + '_' + channelId + '.txt'));
  logger.level = 'trace';
  const msgRetryCounterCache = new NodeCache();
  const storeFilename = './baileys_auth_info/baileys_store_'  + phone + '.json'
  const store = makeInMemoryStore({ logger });
  store?.readFromFile(storeFilename);
  // save every 10s
  setInterval(() => {
    store?.writeToFile(storeFilename);
  }, 10_000);

  const authStateFolder = './baileys_auth_info/' + phone;
  const { state, saveCreds } = await useMultiFileAuthState(authStateFolder);
  // fetch latest version of WA Web
  const { version, isLatest } = await fetchLatestBaileysVersion();
  const waVersion = version.join('.') + ', ' + (isLatest ? 'latest' : 'out');


  const start = () => {

    const waSocket = makeWASocket({
        version,
        logger,
        auth: {
          creds: state.creds,
          keys: makeCacheableSignalKeyStore(state.keys, logger),
        },
        // https://github.com/WhiskeySockets/Baileys/blob/31bc8ab/src/Utils/generics.ts#L21
        // https://github.com/WhiskeySockets/Baileys/blob/31bc8ab4e2c825c0d774875701ed07e20d05bdb6/WAProto/WAProto.proto
        browser: Browsers.macOS('SAFARI'),//Browsers.macOS('SAFARI'),//Browsers.ubuntu('IOS_PHONE'),//Browsers.baileys('WEAR_OS'),//
        msgRetryCounterCache,
        generateHighQualityLinkPreview: false,
        syncFullHistory: false,
      });

    store?.bind(waSocket.ev);

    const handleMessagesUpsert = async upsert => {
      console.info('messages.upsert: ', JSON.stringify(upsert, undefined, 2));

      if (upsert.type === 'notify') {
        for (const msg of upsert.messages) {

          // 没有类型的消息，先忽略
          if (!msg.message) {
            continue;
          }

          if (isJidStatusBroadcast(msg.key.remoteJid)) continue;

          const messageType = Object.keys(msg.message)[0];
          // 如果是群发(status@broadcast)，participant 是发送人，不然则是 remoteJid
          const remoteNo = isJidBroadcast(msg.key.remoteJid) ? decodeJid(msg.key.participant) : decodeJid(msg.key.remoteJid);
          const externalId = externalIdCache.get(msg.key.id);
          const isPersonal = isJidUser(msg.key.remoteJid) || isJidBroadcast(msg.key.remoteJid);
          const conversationType = isPersonal ? 'individual' : 'group';
          const isGroup = isJidGroup(msg.key.remoteJid);
          let groupSubject = groupSubjectCache.get(msg.key.remoteJid);
          if (isGroup && groupSubject === undefined) {
            const groupMetadata = await waSocket.groupMetadata(msg.key.remoteJid);
            groupSubject = groupMetadata.subject;
            groupSubjectCache.set(msg.key.remoteJid, groupMetadata.subject)
          }

          const emitEventName = msg.key.fromMe ? 'message:updated' : 'message:received';
          const msgDirection = msg.key.fromMe ? 'outbound' : 'inbound';
          const msgFrom = msg.key.fromMe ? whatsAppNo : remoteNo;
          const msgTo = msg.key.fromMe ? remoteNo : whatsAppNo;
          const msgStatus = msg.status === undefined ? '' : formatStatus(msg.status);

          if (msg.message?.conversation || msg.message?.extendedTextMessage?.text) {

            const text = msg.message?.conversation || msg.message?.extendedTextMessage?.text;

            waEmitter.emit(emitEventName, {
              id: msg.key.id,
              externalId,
              status: msgStatus,
              direction: msgDirection,
              from: msgFrom,
              to: msgTo,
              type: 'text',
              text: {
                body: text,
              },
              conversation: {
                type: conversationType,
                name: groupSubject,
              },
              customerProfile: {
                id: decodeJid(msg.key.participant),
                name: msg.pushName,
              },
              whatsAppNo,
              fromMe: msg.key.fromMe,
              eventSource: serverConfig.name + '.messages.upsert.notify',
              updateTime: formatTimestamp(msg.messageTimestamp),
            });
          } else if (messageType === 'imageMessage') {
            const imageMessage = msg.message.imageMessage;
            const fileExtension = getFileExtension(imageMessage.mimetype);
            const imageBuffer = await downloadMediaMessage(
              msg, 'buffer', {}, { logger, reuploadRequest: waSocket.updateMediaMessage, },
            );
            const imageFilename = './temp/image_'  + whatsAppNo + '_' + msg.key.id + fileExtension;
            await writeFile(imageFilename, imageBuffer);

            waEmitter.emit(emitEventName, {
              id: msg.key.id,
              externalId,
              status: msgStatus,
              direction: msgDirection,
              from: msgFrom,
              to: msgTo,
              type: 'image',
              image: {
                mimetype: imageMessage.mimetype,
                sha256: uint8ArrayToBase64(imageMessage.fileSha256),
                caption: imageMessage.caption,
                filePath: imageFilename,
                link_original: imageMessage.url,
              },
              conversation: {
                type: conversationType,
                name: groupSubject,
              },
              customerProfile: {
                id: decodeJid(msg.key.participant),
                name: msg.pushName,
              },
              whatsAppNo,
              fromMe: msg.key.fromMe,
              eventSource: serverConfig.name + '.messages.upsert.notify',
              updateTime: formatTimestamp(msg.messageTimestamp),
            });
          }
        }
      } else if (upsert.type === 'append') {
        for (const msg of upsert.messages) {
          if (msg.message?.conversation || msg.message?.extendedTextMessage?.text) {
            const text = msg.message?.conversation || msg.message?.extendedTextMessage?.text;
            // 如果是群发(status@broadcast)，participant 是发送人，不然则是 remoteJid
            const remoteNo = isJidBroadcast(msg.key.remoteJid) ? decodeJid(msg.key.participant) : decodeJid(msg.key.remoteJid);
            const externalId = externalIdCache.get(msg.key.id);
            const isPersonal = isJidUser(msg.key.remoteJid) || isJidBroadcast(msg.key.remoteJid);
            const conversationType = isPersonal ? 'individual' : 'group';

            const emitEventName = msg.key.fromMe ? 'message:updated' : 'message:received';
            const msgDirection = msg.key.fromMe ? 'outbound' : 'inbound';
            const msgFrom = msg.key.fromMe ? whatsAppNo : remoteNo;
            const msgTo = msg.key.fromMe ? remoteNo : whatsAppNo;
            const msgStatus = msg.status === undefined ? '' : formatStatus(msg.status);

            waEmitter.emit(emitEventName, {
              id: msg.key.id,
              externalId,
              status: msgStatus,
              direction: msgDirection,
              from: msgFrom,
              to: msgTo,
              type: 'text',
              text: {
                body: text,
              },
              conversation: {
                type: conversationType,
              },
              customerProfile: {
                id: decodeJid(msg.participant),
                name: msg.pushName,
              },
              whatsAppNo,
              fromMe: msg.key.fromMe,
              eventSource: serverConfig.name + '.messages.upsert.append',
              updateTime: formatTimestamp(msg.messageTimestamp),
            });
          }
        }
      }
    }

    const handleMessagesUpdate = async messageUpdate => {
      console.info('messages.update: ', JSON.stringify(messageUpdate, undefined, 2));

      for (const msg of messageUpdate) {

        // 没有明确标识状态的更新，忽略
        const ignore = msg.update === undefined || msg.update.status === undefined;

        if (ignore) continue;

        // 如果是群发(status@broadcast)，participant 是发送人，不然则是 remoteJid
        const remoteNo = isJidBroadcast(msg.key.remoteJid) ? decodeJid(msg.key.participant) : decodeJid(msg.key.remoteJid);
        const externalId = externalIdCache.get(msg.key.id);
        const isPersonal = isJidUser(msg.key.remoteJid) || isJidBroadcast(msg.key.remoteJid);
        const conversationType = isPersonal ? 'individual' : 'group';

        const msgDirection = msg.key.fromMe ? 'outbound' : 'inbound';
        const msgFrom = msg.key.fromMe ? whatsAppNo : remoteNo;
        const msgTo = msg.key.fromMe ? remoteNo : whatsAppNo;
        const msgStatus = formatStatus(msg.update.status);

        waEmitter.emit('message:updated', {
          id: msg.key.id,
          externalId,
          status: msgStatus,
          direction: msgDirection,
          from: msgFrom,
          to: msgTo,
          conversation: {
            type: conversationType,
          },
          customerProfile: {
            id: decodeJid(msg.key.participant),
            name: msg.pushName,
          },
          whatsAppNo,
          fromMe: msg.key.fromMe,
          eventSource: serverConfig.name + '.messages.updated',
          updateTime: formatTimestamp(new Date().getTime() / 1000),
        });
      }
    }

    const handleCredsUpdate = async () => {
      await saveCreds();
    }

    const sendMessageHandler = (event) => {
      const { to: number, externalId, ...content } = event;
      const jid = encodeJid(number);
      waSocket.sendMessage(
        jid, content
      ).then(msg => {
        externalIdCache.set(msg.key.id, externalId)
      }).catch(ex => {
        waEmitter.emit('message:updated', {
          id: generateId(),
          externalId,
          status: 'failed',
          direction: 'outbound',
          from: whatsAppNo,
          to: number,
          error: `发送消息出错 ` + ex,
          whatsAppNo,
          eventSource: serverConfig.name + '.sendMessage.catch',
          updateTime: formatTimestamp(new Date().getTime() / 1000),
        });
      });
    }

    waSocket.ev.on('connection.update', async update => {

      const { connection, lastDisconnect, qr } = update;

      if (connection === 'close') {
        waEmitter.off('request.' + whatsAppNo + '.send.message', sendMessageHandler);
        if((lastDisconnect?.error)?.output?.statusCode !== DisconnectReason.loggedOut) {
          start();
        } else {
          // logout 异步删除验证目录
          fs.rm(authStateFolder, { recursive: true, force: true }, (err) => {
            if (err) {
              return console.error(`Error deleting directory: ${err.message}`);
            }
            console.log('Directory deleted successfully: ', authStateFolder);
          });
          waEmitter.emit('connection:close', {
            whatsAppNo, channelId,
            eventSource: serverConfig.name + '.connection.update.close',
            status: 'offline',
          });
        }
      } else if (connection === 'open') {
        waEmitter.emit('connection:open', {
          status: 'open', whatsAppNo, channelId,
          eventSource: serverConfig.name + '.connection.update.open',
        });

        waEmitter.on('request.' + whatsAppNo + '.send.message', sendMessageHandler);
      } else if (qr !== undefined) {
        // WebSocket 创建成功等待扫码，如果没有扫码会更新 qr
        // 第一次一分钟，后面是 20 秒更新一次
        if (qrCode === null) {
          qrCode = qr;
          waEmitter.emit('creds:update', {
            id: generateId(),
            qr, whatsAppNo,
            server:serverConfig.name,
            eventSource: 'creds.update',
            createTime: formatTimestamp(new Date().getTime() / 1000),
          });
        } else {
          // 第一次二维码时效后退出，不需要等待更新二维码
          waSocket.logout(() => '二维码已过期');
        }

      }
    });

    waSocket.ev.on('creds.update', handleCredsUpdate);
    waSocket.ev.on('messages.upsert', handleMessagesUpsert);
    waSocket.ev.on('messages.update', handleMessagesUpdate);
  };

  return {
    createTimestamp: Date.now(),
    status: 'offline',
    version: waVersion,
    channelId: channelId,
    phone: phone,
    start,
  };
};

module.exports = {
  createWhatsApp,
};