TelegramPWA/src/api/gramjs/methods/reactions.ts

import BigInt from 'big-integer';
import { Api as GramJs } from '../../../lib/gramjs';

import type {
  ApiChat, ApiReaction, ApiSticker,
} from '../../types';

import {
  API_GENERAL_ID_LIMIT,
  REACTION_LIST_LIMIT,
  RECENT_REACTIONS_LIMIT,
  TOP_REACTIONS_LIMIT,
} from '../../../config';
import { split } from '../../../util/iteratees';
import {
  buildApiAvailableEffect,
  buildApiAvailableReaction,
  buildApiReaction,
  buildApiSavedReactionTag,
  buildMessagePeerReaction,
} from '../apiBuilders/reactions';
import { buildStickerFromDocument } from '../apiBuilders/symbols';
import {
  buildInputPaidReactionPrivacy,
  buildInputPeer,
  buildInputReaction,
  DEFAULT_PRIMITIVES,
  generateRandomTimestampedBigInt,
} from '../gramjsBuilders';
import localDb from '../localDb';
import { invokeRequest } from './client';

export function sendWatchingEmojiInteraction({
  chat,
  emoticon,
}: {
  chat: ApiChat; emoticon: string;
}) {
  return invokeRequest(new GramJs.messages.SetTyping({
    peer: buildInputPeer(chat.id, chat.accessHash),
    action: new GramJs.SendMessageEmojiInteractionSeen({
      emoticon,
    }),
  }), {
    abortControllerChatId: chat.id,
  });
}

export function sendEmojiInteraction({
  chat,
  emoticon,
  messageId,
  timestamps,
}: {
  chat: ApiChat; messageId: number; emoticon: string; timestamps: number[];
}) {
  return invokeRequest(new GramJs.messages.SetTyping({
    peer: buildInputPeer(chat.id, chat.accessHash),
    action: new GramJs.SendMessageEmojiInteraction({
      emoticon,
      msgId: messageId,
      interaction: new GramJs.DataJSON({
        data: JSON.stringify({
          v: 1,
          a: timestamps.map((t: number) => ({
            t,
            i: 1,
          })),
        }),
      }),
    }),
  }), {
    abortControllerChatId: chat.id,
  });
}

export async function fetchAvailableReactions() {
  const result = await invokeRequest(new GramJs.messages.GetAvailableReactions({
    hash: DEFAULT_PRIMITIVES.INT,
  }));

  if (!result || result instanceof GramJs.messages.AvailableReactionsNotModified) {
    return undefined;
  }

  result.reactions.forEach((reaction) => {
    if (reaction.staticIcon instanceof GramJs.Document) {
      localDb.documents[String(reaction.staticIcon.id)] = reaction.staticIcon;
    }
    if (reaction.selectAnimation instanceof GramJs.Document) {
      localDb.documents[String(reaction.selectAnimation.id)] = reaction.selectAnimation;
    }
    if (reaction.aroundAnimation instanceof GramJs.Document) {
      localDb.documents[String(reaction.aroundAnimation.id)] = reaction.aroundAnimation;
    }
    if (reaction.appearAnimation instanceof GramJs.Document) {
      localDb.documents[String(reaction.appearAnimation.id)] = reaction.appearAnimation;
    }
    if (reaction.centerIcon instanceof GramJs.Document) {
      localDb.documents[String(reaction.centerIcon.id)] = reaction.centerIcon;
    }
  });

  return result.reactions.map(buildApiAvailableReaction);
}

export async function fetchAvailableEffects() {
  const result = await invokeRequest(new GramJs.messages.GetAvailableEffects({
    hash: DEFAULT_PRIMITIVES.INT,
  }));

  if (!result || result instanceof GramJs.messages.AvailableEffectsNotModified) {
    return undefined;
  }

  const documentsMap = new Map(result.documents.map((doc) => [String(doc.id), doc]));

  result.documents.forEach((document) => {
    if (document instanceof GramJs.Document) {
      localDb.documents[String(document.id)] = document;
    }
  });

  const effects = result.effects.map(buildApiAvailableEffect);

  const stickers: ApiSticker[] = [];
  const emojis: ApiSticker[] = [];

  for (const effect of effects) {
    if (effect.effectAnimationId) {
      const document = documentsMap.get(effect.effectStickerId);
      const emoji = document && buildStickerFromDocument(document, false, effect.isPremium);
      if (emoji) emojis.push(emoji);
    } else {
      const document = localDb.documents[effect.effectStickerId];
      const sticker = buildStickerFromDocument(document);
      if (sticker) {
        stickers.push(sticker);
      }
    }
  }

  return { effects, emojis, stickers };
}

export function sendReaction({
  chat, messageId, reactions, shouldAddToRecent,
}: {
  chat: ApiChat;
  messageId: number;
  reactions?: ApiReaction[];
  shouldAddToRecent?: boolean;
}) {
  return invokeRequest(new GramJs.messages.SendReaction({
    reaction: reactions?.map((r) => buildInputReaction(r)),
    peer: buildInputPeer(chat.id, chat.accessHash),
    msgId: messageId,
    ...(shouldAddToRecent && { addToRecent: true }),
  }), {
    shouldReturnTrue: true,
    shouldThrow: true,
  });
}

export function sendPaidReaction({
  chat,
  messageId,
  count,
  isPrivate,
  peerId,
}: {
  chat: ApiChat;
  messageId: number;
  count: number;
  isPrivate?: boolean;
  peerId?: string;
}) {
  return invokeRequest(new GramJs.messages.SendPaidReaction({
    peer: buildInputPeer(chat.id, chat.accessHash),
    msgId: messageId,
    randomId: generateRandomTimestampedBigInt(),
    count,
    private: buildInputPaidReactionPrivacy(isPrivate, peerId),
  }), {
    shouldReturnTrue: true,
    shouldThrow: true,
  });
}

export function fetchMessageReactions({
  ids, chat,
}: {
  ids: number[]; chat: ApiChat;
}) {
  const chunks = split(ids, API_GENERAL_ID_LIMIT);
  chunks.forEach((chunkIds) => {
    invokeRequest(new GramJs.messages.GetMessagesReactions({
      id: chunkIds,
      peer: buildInputPeer(chat.id, chat.accessHash),
    }), {
      shouldReturnTrue: true,
      abortControllerChatId: chat.id,
    });
  });
}

export async function fetchMessageReactionsList({
  chat, messageId, reaction, offset,
}: {
  chat: ApiChat; messageId: number; reaction?: ApiReaction; offset?: string;
}) {
  const result = await invokeRequest(new GramJs.messages.GetMessageReactionsList({
    peer: buildInputPeer(chat.id, chat.accessHash),
    id: messageId,
    ...(reaction && { reaction: buildInputReaction(reaction) }),
    limit: REACTION_LIST_LIMIT,
    ...(offset && { offset }),
  }));

  if (!result) {
    return undefined;
  }

  const { nextOffset, reactions, count } = result;

  return {
    nextOffset,
    reactions: reactions.map(buildMessagePeerReaction).filter(Boolean),
    count,
  };
}

export function setDefaultReaction({
  reaction,
}: {
  reaction: ApiReaction;
}) {
  return invokeRequest(new GramJs.messages.SetDefaultReaction({
    reaction: buildInputReaction(reaction),
  }));
}

export async function fetchTopReactions({ hash }: { hash?: string }) {
  const result = await invokeRequest(new GramJs.messages.GetTopReactions({
    limit: TOP_REACTIONS_LIMIT,
    hash: hash ? BigInt(hash) : DEFAULT_PRIMITIVES.BIGINT,
  }));

  if (!result || result instanceof GramJs.messages.ReactionsNotModified) {
    return undefined;
  }

  return {
    hash: String(result.hash),
    reactions: result.reactions.map((r) => buildApiReaction(r)).filter(Boolean),
  };
}

export async function fetchRecentReactions({ hash }: { hash?: string }) {
  const result = await invokeRequest(new GramJs.messages.GetRecentReactions({
    limit: RECENT_REACTIONS_LIMIT,
    hash: hash ? BigInt(hash) : DEFAULT_PRIMITIVES.BIGINT,
  }));

  if (!result || result instanceof GramJs.messages.ReactionsNotModified) {
    return undefined;
  }

  return {
    hash: String(result.hash),
    reactions: result.reactions.map((r) => buildApiReaction(r)).filter(Boolean),
  };
}

export function clearRecentReactions() {
  return invokeRequest(new GramJs.messages.ClearRecentReactions());
}

export async function fetchDefaultTagReactions({ hash }: { hash?: string }) {
  const result = await invokeRequest(new GramJs.messages.GetDefaultTagReactions({
    hash: hash ? BigInt(hash) : DEFAULT_PRIMITIVES.BIGINT,
  }));

  if (!result || result instanceof GramJs.messages.ReactionsNotModified) {
    return undefined;
  }

  return {
    hash: String(result.hash),
    reactions: result.reactions.map((r) => buildApiReaction(r)).filter(Boolean),
  };
}

export async function fetchSavedReactionTags({ hash }: { hash?: string }) {
  const result = await invokeRequest(new GramJs.messages.GetSavedReactionTags({
    hash: hash ? BigInt(hash) : DEFAULT_PRIMITIVES.BIGINT,
  }));

  if (!result || result instanceof GramJs.messages.SavedReactionTagsNotModified) {
    return undefined;
  }

  return {
    hash: String(result.hash),
    tags: result.tags.map(buildApiSavedReactionTag).filter(Boolean),
  };
}

export function updateSavedReactionTag({
  reaction,
  title,
}: {
  reaction: ApiReaction;
  title?: string;
}) {
  return invokeRequest(new GramJs.messages.UpdateSavedReactionTag({
    reaction: buildInputReaction(reaction),
    title,
  }), {
    shouldReturnTrue: true,
  });
}