import { LineClient } from 'sibdigital-messaging-apis/packages/messaging-api-line';
import { MessengerClient } from 'messaging-api-messenger';
import { SlackOAuthClient } from 'sibdigital-messaging-apis/packages/messaging-api-slack';
import { TelegramClient } from 'sibdigital-messaging-apis/packages/messaging-api-telegram';
import { ViberClient } from 'sibdigital-messaging-apis/packages/messaging-api-viber';

import LineBot from '../line/LineBot';
import MessengerBot from '../messenger/MessengerBot';
import SlackBot from '../slack/SlackBot';
import TelegramBot from '../telegram/TelegramBot';
import TwilioClient from '../whatsapp/TwilioClient';
import ViberBot from '../viber/ViberBot';
import WhatsappBot from '../whatsapp/WhatsappBot';
import { Channel } from '../types';

import getBottenderConfig from './getBottenderConfig';
import getSessionStore from './getSessionStore';

const BOT_MAP = {
  messenger: MessengerBot,
  line: LineBot,
  slack: SlackBot,
  telegram: TelegramBot,
  viber: ViberBot,
  whatsapp: WhatsappBot,
};

function getClient<C extends string>(
  channel: C
): C extends 'messenger'
  ? MessengerClient
  : C extends 'line'
  ? LineClient
  : C extends 'slack'
  ? SlackOAuthClient
  : C extends 'telegram'
  ? TelegramClient
  : C extends 'viber'
  ? ViberClient
  : C extends 'whatsapp'
  ? TwilioClient
  : any {
  const { channels = {} } = getBottenderConfig();
  const sessionStore = getSessionStore();

  const channelConfig = (channels as Record<string, any>)[channel];

  if (!channelConfig) {
    throw new Error(
      `getClient: ${channel} config is missing in \`bottender.config.js\`.`
    );
  }

  const ChannelBot = BOT_MAP[channel.split('_')[0] as Channel];

  const channelBot = new ChannelBot({
    ...channelConfig,
    sessionStore,
  } as any);

  return channelBot.connector.client as any;
}

export default getClient;