session-desktop/ts/opengroup/opengroupV2/OpenGroupAPIV2CompactPoll.ts

import { getV2OpenGroupRoomByRoomId } from '../../data/opengroups';
import {
  OpenGroupRequestCommonType,
  OpenGroupV2CompactPollRequest,
  parseMessages,
} from './ApiUtil';
import { parseStatusCodeFromOnionRequest } from './OpenGroupAPIV2Parser';
import _ from 'lodash';
import { sendViaOnion } from '../../session/onions/onionSend';
import { OpenGroupManagerV2 } from './OpenGroupManagerV2';
import { OpenGroupMessageV2 } from './OpenGroupMessageV2';

const COMPACT_POLL_ENDPOINT = 'compact_poll';

export const compactFetchEverything = async (
  rooms: Array<OpenGroupRequestCommonType>
): Promise<null | any> => {
  // fetch all we need
  const compactPollRequest = await getCompactPollRequest(rooms);
  if (!compactPollRequest) {
    window.log.info('Nothing found to be fetched. returning');
    return null;
  }

  const result = await sendOpenGroupV2RequestCompactPoll(compactPollRequest);
  const statusCode = parseStatusCodeFromOnionRequest(result);
  if (statusCode !== 200) {
    return null;
  }
  return result;
};

/**
 * This return body to be used to do the compactPoll
 */
const getCompactPollRequest = async (
  rooms: Array<OpenGroupRequestCommonType>
): Promise<null | OpenGroupV2CompactPollRequest> => {
  // first verify the rooms we got are all from on the same server
  let firstUrl: string;
  if (rooms) {
    firstUrl = rooms[0].serverUrl;
    const anotherUrl = rooms.some(r => r.serverUrl !== firstUrl);
    if (anotherUrl) {
      throw new Error('CompactPoll is designed for a single server');
    }
  } else {
    window.log.warn('CompactPoll: No room given. nothing to do');
    return null;
  }

  const allServerPubKeys: Array<string> = [];

  const roomsRequestInfos = _.compact(
    await Promise.all(
      rooms.map(async ({ roomId, serverUrl }) => {
        try {
          const fetchedInfo = await getV2OpenGroupRoomByRoomId({
            serverUrl,
            roomId,
          });
          if (!fetchedInfo) {
            window.log.warn('Could not find this room getMessages');
            return null;
          }

          const {
            lastMessageFetchedServerID,
            lastMessageDeletedServerID,
            token,
            serverPublicKey,
          } = fetchedInfo;
          allServerPubKeys.push(serverPublicKey);
          const roomRequestContent: Record<string, any> = {
            room_id: roomId,
            auth_token: token || '',
          };
          if (lastMessageDeletedServerID) {
            roomRequestContent.from_deletion_server_id = lastMessageDeletedServerID;
          }

          if (lastMessageFetchedServerID) {
            roomRequestContent.from_message_server_id = lastMessageFetchedServerID;
          }

          return roomRequestContent;
        } catch (e) {
          window.log.warn('failed to fetch roominfos for room', roomId);
          return null;
        }
      })
    )
  );
  if (!roomsRequestInfos?.length) {
    return null;
  }
  // double check that all those server pubkeys are the same
  let firstPubkey: string;
  if (allServerPubKeys?.length) {
    firstPubkey = allServerPubKeys[0];
    const allMatch = allServerPubKeys.every(p => p === firstPubkey);
    if (!allMatch) {
      window.log.warn('All pubkeys do not match:', allServerPubKeys);
      return null;
    }
  } else {
    window.log.warn('No pubkeys found:', allServerPubKeys);
    return null;
  }
  const body = JSON.stringify({
    requests: roomsRequestInfos,
  });
  return {
    body,
    server: firstUrl,
    serverPubKey: firstPubkey,
    endpoint: COMPACT_POLL_ENDPOINT,
  };
};

/**
 * This call is separate as a lot of the logic is custom (statusCode handled separately, etc)
 */
async function sendOpenGroupV2RequestCompactPoll(
  request: OpenGroupV2CompactPollRequest
): Promise<Object | null> {
  const { server, endpoint, body, serverPubKey } = request;
  // this will throw if the url is not valid
  const builtUrl = new URL(`${server}/${endpoint}`);

  console.warn(`sending compactPoll request: ${request.body}`);

  const res = await sendViaOnion(serverPubKey, builtUrl, {
    method: 'POST',
    body,
  });

  const statusCode = parseStatusCodeFromOnionRequest(res);
  if (!statusCode) {
    window.log.warn(
      'sendOpenGroupV2Request Got unknown status code; res:',
      res
    );
    return res as object;
  }

  const results = await parseCompactPollResults(res);

  throw new Error(
    'See how we handle needs of new tokens, and save stuff to db (last deleted, ... conversation commit, etc'
  );

  return res as object;
}

type ParsedRoomCompactPollResults = {
  roomId: string;
  deletions: Array<number>;
  messages: Array<OpenGroupMessageV2>;
  moderators: Array<string>;
};

const parseCompactPollResult = async (
  singleRoomResult: any
): Promise<ParsedRoomCompactPollResults | null> => {
  const {
    room_id,
    deletions: rawDeletions,
    messages: rawMessages,
    moderators: rawMods,
  } = singleRoomResult;

  if (
    !room_id ||
    rawDeletions === undefined ||
    rawMessages === undefined ||
    rawMods === undefined
  ) {
    window.log.warn('Invalid compactPoll result', singleRoomResult);
    return null;
  }

  const validMessages = await parseMessages(rawMessages);
  const moderators = rawMods as Array<string>;
  const deletions = rawDeletions as Array<number>;

  return { roomId: room_id, deletions, messages: validMessages, moderators };
};

const parseCompactPollResults = async (
  res: any
): Promise<Array<ParsedRoomCompactPollResults> | null> => {
  if (
    !res ||
    !res.result ||
    !res.result.results ||
    !res.result.results.length
  ) {
    return null;
  }
  const arrayOfResults = res.result.results as Array<any>;

  const parsedResults: Array<ParsedRoomCompactPollResults> = _.compact(
    await Promise.all(arrayOfResults.map(parseCompactPollResult))
  );

  if (!parsedResults || !parsedResults.length) {
    return null;
  }
  return parsedResults;
};