import { ChatConfigurator, DataSource, DataSourceDecorator } from '../../shared/framework';
// @ts-ignore
import { CometChat } from '@cometchat/chat-sdk-react-native';
import { ExtensionConstants, ExtensionURLs } from '../ExtensionConstants';
import {
  CometChatLocalize,
  localize,
} from '../../shared/resources/CometChatLocalize';
import { CometChatMessageOption } from "../../shared/modals";
// @ts-ignore
import React from 'react';
import { CometChatTheme } from '../../shared/resources/CometChatTheme';
import {
  MessageOptionConstants,
} from '../../shared/constants/UIKitConstants';
import { MessageTranslationBubble } from './MessageTranslationBubble';
import { MessageTranslationConfigurationInterface } from './MessageTranslationExtension';
import { CometChatUIEventHandler } from '../../shared/events/CometChatUIEventHandler/CometChatUIEventHandler';
import { CometChatUIEvents, MessageEvents } from '../../shared/events';
import { messageStatus } from '../../shared/utils/CometChatMessageHelper';
import { ICONS } from './resources';
import { CometChatTextFormatter, CometChatUIKit, CometChatUrlsFormatter, MentionTextStyle } from '../../shared';
import { CommonUtils } from '../../shared/utils/CommonUtils';
import { AdditionalBubbleStylingParams, MessageBubbleAlignmentType } from '../../shared/base/Types';
import { TextStyle } from 'react-native';
import { anyObject } from '../../shared/utils';
export class MessageTranslationExtensionDecorator extends DataSourceDecorator {
  messageTranslationConfiguration?: MessageTranslationConfigurationInterface;

  translatedMessage: anyObject = {};
  constructor(
    dataSource: DataSource,
    messageTranslationConfiguration?: MessageTranslationConfigurationInterface
  ) {
    super(dataSource);
    if (messageTranslationConfiguration != undefined) {
      this.messageTranslationConfiguration = messageTranslationConfiguration;
    }
  }

  getId(): string {
    return 'MessageTranslation';
  }

  getTextMessageOptions(
    loggedInUser: CometChat.User,
    messageObject: CometChat.BaseMessage,
    group: CometChat.Group
  ): CometChatMessageOption[] {
    let optionsList: CometChatMessageOption[] = super.getTextMessageOptions(
      loggedInUser,
      messageObject,
      group
    );
    optionsList.push(this.getTranslateOption(messageObject));
    return optionsList;
  }

  getTranslateOption(messageObject: any): CometChatMessageOption {
    return {
      id: MessageOptionConstants.translateMessage,
      title: localize('TRANSLATE'),
      icon: ICONS.TRANSLATE,
      onPress: () => {
        this.translateMessage(messageObject);
      },
    };
  }

  getSetMetaData = (messageObj: any, messageTranslation: any) => {
    let metaData = messageObj.getMetadata();
    if (!metaData) {
      metaData = {};
    }
    if (metaData && !metaData['@injected']) {
      metaData = {
        ...metaData,
        '@injected': { extensions: { translate: {} } },
      };
    }
    if (
      metaData &&
      metaData['@injected'] &&
      metaData['@injected']['extensions']
    ) {
      let tempData = metaData['@injected']['extensions'];
      tempData = {
        ...metaData,
        '@injected': {
          ...metaData['@injected'],
          extensions: {
            ...metaData['@injected']['extensions'],
            translate: { [messageObj.id]: messageTranslation },
          },
        },
      };
      metaData = tempData;
    }

    if (
      metaData &&
      metaData['@injected'] &&
      metaData['@injected']['extensions']['translate']
    ) {
      let tempMetaData = {};
      let translateData = metaData['@injected']['extensions']['translate'];

      if (translateData) {
        translateData = {
          ...translateData,
          [messageObj.id]: messageTranslation,
        };
      } else {
        translateData[messageObj.id] = {
          [messageObj.id]: messageTranslation,
        };
      }
      tempMetaData = {
        ...metaData['@injected']['extensions']['translate'],
        ...translateData,
      };

      metaData['@injected']['extensions']['translate'] = tempMetaData;
    }

    messageObj.setMetadata(metaData);
    return {
      msg: messageObj,
      metaData: metaData['@injected']['extensions']['translate'],
    };
  };
  translateMessage = (message: any) => {
    const messageId = message.id;
    const messageText = message.text;
    let translateToLanguage = CometChatLocalize.getLocale();
    CometChatUIEventHandler.emitUIEvent(CometChatUIEvents.ccToggleBottomSheet, {
      isBottomSheetVisible: false,
    });
    CometChat.callExtension(
      ExtensionConstants.messageTranslation,
      'POST',
      ExtensionURLs.translate,
      {
        msgId: messageId,
        text: messageText,
        languages: [translateToLanguage],
      }
    )
      .then((result: any) => {
        if (
          result?.hasOwnProperty('translations') &&
          result['translations']['length']
        ) {
          const messageTranslation = result['translations'][0];
          let translatedMsg = this.getSetMetaData(
            message,
            messageTranslation['message_translated']
          );
          if (translatedMsg) {
            if (translatedMsg.metaData?.translate)
              this.translatedMessage = translatedMsg?.metaData?.translate;
          } else
            this.translatedMessage = {
              [message.id]: `${messageTranslation['message_translated']}`,
            };

          CometChatUIEventHandler.emitMessageEvent(
            MessageEvents.ccMessageEdited,
            {
              message: translatedMsg.msg,
              status: messageStatus.success,
            }
          );
        }
      })
      .catch((error: any) => {
        console.log(error);
      });
  };

  getTextMessageBubble(
    messageText: string,
    message: CometChat.TextMessage,
    alignment: MessageBubbleAlignmentType,
    theme: CometChatTheme,
    additionalParams?: AdditionalBubbleStylingParams
  ): JSX.Element | null{
    let tempTranslatedMsg: anyObject = {};
    let translatedMetaData: any = message.getMetadata();
    if (
      translatedMetaData &&
      translatedMetaData['@injected'] &&
      translatedMetaData['@injected']['extensions'] &&
      translatedMetaData['@injected']['extensions']['translate']
    ) {
      tempTranslatedMsg =
        translatedMetaData['@injected']['extensions']['translate'];
    }

    let loggedInUser = CometChatUIKit.loggedInUser;
    let mentionedUsers = message.getMentionedUsers();
    let textFormatters = [...(additionalParams?.textFormatters ?? [])];

    let linksTextFormatter = ChatConfigurator.getDataSource().getUrlsFormatter(loggedInUser as any);
    linksTextFormatter.setMessage(message);
    linksTextFormatter.setId("ccDefaultUrlsFormatterId")
    textFormatters.push(linksTextFormatter);

    if (!additionalParams?.disableMentions && mentionedUsers && mentionedUsers.length) {

      let mentionsTextFormatter =
        ChatConfigurator.getDataSource().getMentionsFormatter(loggedInUser as any);
      loggedInUser && mentionsTextFormatter.setLoggedInUser(loggedInUser);
      mentionsTextFormatter.setMessage(message);
      mentionsTextFormatter.setId("ccDefaultMentionFormatterId")
      let isUserSentMessage = alignment === "right";
      if (isUserSentMessage) {
        mentionsTextFormatter.setMentionsStyle(
          new MentionTextStyle({
            loggedInUserTextStyle: {
              color: theme.palette.getTertiary(),
              ...theme.typography.title2,
            } as TextStyle,
            textStyle: {
              color: theme.palette.getTertiary(),
              ...theme.typography.subtitle1,
            }  as TextStyle,
          })
        );
      } else {
        mentionsTextFormatter.setMentionsStyle(
          new MentionTextStyle({
            loggedInUserTextStyle: {
              color: theme.palette.getPrimary(),
              ...theme.typography.title2,
            } as TextStyle,
            textStyle: {
              color: theme.palette.getPrimary(),
              ...theme.typography.subtitle1,
            } as TextStyle,
          })
        );
      }

      textFormatters.push(mentionsTextFormatter);

    }
    let finalFormatters: any[] = [];

    let customerHasPassedUrlsFormatter;

    textFormatters.forEach(formatter => {
      if (formatter instanceof CometChatUrlsFormatter) {
        if (formatter.getId() !== "ccDefaultUrlsFormatterId") {
          customerHasPassedUrlsFormatter = true
        }
      }
      formatter.setMessage(message);
      let suggestionUsers = formatter.getSuggestionItems();
      suggestionUsers.length > 0 && formatter.setSuggestionItems(suggestionUsers);
      let _formatter = CommonUtils.clone(formatter);
      finalFormatters.push(_formatter);
    })

    if (customerHasPassedUrlsFormatter) {
      let customUrlsIndex = finalFormatters.findIndex((formatter: CometChatTextFormatter[]) => (formatter instanceof CometChatUrlsFormatter && formatter.getId() === "ccDefaultUrlsFormatterId"));
      if (customUrlsIndex > -1) {
        finalFormatters.splice(customUrlsIndex, 1);
      }
    }

    if (
      (tempTranslatedMsg && tempTranslatedMsg[message.getId()]) ||
      (this.translatedMessage && this.translatedMessage[message.getId()])
    ) {
      return (
        <MessageTranslationBubble
          translatedText={
            tempTranslatedMsg
              ? tempTranslatedMsg[message.getId()]
              : this.translatedMessage[message.getId()]
                ? this.translatedMessage[message.getId()]
                : ''
          }
          text={messageText}
          alignment={alignment}
          textFormatters={finalFormatters}
          {...this.messageTranslationConfiguration}
        />
      );
    }
    return super.getTextMessageBubble(messageText, message, alignment, theme, additionalParams);
  }
}