upstream-twenty/packages/twenty-server/test/integration/utils/query-messaging.util.ts

import gql from 'graphql-tag';

import {
  type CalendarChannelContactAutoCreationPolicy,
  type MessageFolderImportPolicy,
} from 'twenty-shared/types';

import { type CalendarChannelDTO } from 'src/engine/metadata-modules/calendar-channel/dtos/calendar-channel.dto';
import { type ConnectedAccountDTO } from 'src/engine/metadata-modules/connected-account/dtos/connected-account.dto';
import { type MessageChannelDTO } from 'src/engine/metadata-modules/message-channel/dtos/message-channel.dto';
import { type MessageFolderDTO } from 'src/engine/metadata-modules/message-folder/dtos/message-folder.dto';

import { makeMetadataAPIRequest } from 'test/integration/metadata/suites/utils/make-metadata-api-request.util';
import { waitForAllJobsToFinish } from 'test/integration/utils/wait-for-all-jobs-to-finish.util';

type MetadataAPIResponse = {
  body: { data: Record<string, unknown>; errors?: { message: string }[] };
};

export const getDataOrThrow = (response: MetadataAPIResponse) => {
  if (response.body.errors?.length) {
    throw new Error(
      `Metadata API request failed: ${response.body.errors
        .map((error) => error.message)
        .join('; ')}`,
    );
  }

  return response.body.data;
};

export type MessageFolderDto = Pick<
  MessageFolderDTO,
  'id' | 'name' | 'isSynced' | 'isSentFolder' | 'pendingSyncAction'
>;

export type MessageChannelDto = Pick<
  MessageChannelDTO,
  | 'id'
  | 'handle'
  | 'connectedAccountId'
  | 'syncStatus'
  | 'syncStage'
  | 'syncStageStartedAt'
  | 'throttleFailureCount'
  | 'throttleRetryAfter'
  | 'visibility'
>;

export type CalendarChannelDto = Pick<
  CalendarChannelDTO,
  | 'id'
  | 'handle'
  | 'connectedAccountId'
  | 'syncStatus'
  | 'syncStage'
  | 'syncStageStartedAt'
  | 'throttleFailureCount'
  | 'isContactAutoCreationEnabled'
  | 'contactAutoCreationPolicy'
  | 'visibility'
>;

export type ConnectedAccountDto = Pick<
  ConnectedAccountDTO,
  'id' | 'handle' | 'provider' | 'lastCredentialsRefreshedAt' | 'authFailedAt'
>;

type MessageChannelUpdate = {
  messageFolderImportPolicy?: MessageFolderImportPolicy;
  isSyncEnabled?: boolean;
  isContactAutoCreationEnabled?: boolean;
};

type CalendarChannelUpdate = {
  isSyncEnabled?: boolean;
  isContactAutoCreationEnabled?: boolean;
  contactAutoCreationPolicy?: CalendarChannelContactAutoCreationPolicy;
};

const MESSAGE_CHANNEL_FIELDS = gql`
  fragment TestMessageChannelFields on MessageChannel {
    id
    handle
    connectedAccountId
    syncStatus
    syncStage
    syncStageStartedAt
    throttleFailureCount
    throttleRetryAfter
    visibility
  }
`;

export const queryMessageChannels = async (): Promise<MessageChannelDto[]> => {
  const response = await makeMetadataAPIRequest({
    query: gql`
      query MessageChannelsForTest {
        myMessageChannels {
          ...TestMessageChannelFields
        }
      }
      ${MESSAGE_CHANNEL_FIELDS}
    `,
  });

  return getDataOrThrow(response).myMessageChannels as MessageChannelDto[];
};

export const queryMessageChannel = async ({
  connectedAccountId,
  channelId,
}: {
  connectedAccountId: string;
  channelId: string;
}): Promise<MessageChannelDto> => {
  const response = await makeMetadataAPIRequest({
    query: gql`
      query MessageChannelForTest($connectedAccountId: UUID) {
        myMessageChannels(connectedAccountId: $connectedAccountId) {
          ...TestMessageChannelFields
        }
      }
      ${MESSAGE_CHANNEL_FIELDS}
    `,
    variables: { connectedAccountId },
  });

  const channel = (
    getDataOrThrow(response).myMessageChannels as MessageChannelDto[]
  ).find((candidate) => candidate.id === channelId);

  if (!channel) {
    throw new Error(`Message channel ${channelId} not found`);
  }

  return channel;
};

export const queryMessageFolders = async (
  messageChannelId: string,
): Promise<MessageFolderDto[]> => {
  const response = await makeMetadataAPIRequest({
    query: gql`
      query MessageFoldersForTest($messageChannelId: UUID) {
        myMessageFolders(messageChannelId: $messageChannelId) {
          id
          name
          isSynced
          isSentFolder
          pendingSyncAction
        }
      }
    `,
    variables: { messageChannelId },
  });

  return getDataOrThrow(response).myMessageFolders as MessageFolderDto[];
};

export const queryCalendarChannels = async (
  connectedAccountId: string,
): Promise<CalendarChannelDto[]> => {
  const response = await makeMetadataAPIRequest({
    query: gql`
      query CalendarChannelsForTest($connectedAccountId: UUID) {
        myCalendarChannels(connectedAccountId: $connectedAccountId) {
          id
          handle
          connectedAccountId
          syncStatus
          syncStage
          syncStageStartedAt
          throttleFailureCount
          isContactAutoCreationEnabled
          contactAutoCreationPolicy
          visibility
        }
      }
    `,
    variables: { connectedAccountId },
  });

  return getDataOrThrow(response).myCalendarChannels as CalendarChannelDto[];
};

export const queryCalendarChannel = async ({
  connectedAccountId,
  calendarChannelId,
}: {
  connectedAccountId: string;
  calendarChannelId: string;
}): Promise<CalendarChannelDto> => {
  const channel = (await queryCalendarChannels(connectedAccountId)).find(
    (candidate) => candidate.id === calendarChannelId,
  );

  if (!channel) {
    throw new Error(`Calendar channel ${calendarChannelId} not found`);
  }

  return channel;
};

export const queryConnectedAccount = async (
  connectedAccountId: string,
): Promise<ConnectedAccountDto> => {
  const response = await makeMetadataAPIRequest({
    query: gql`
      query ConnectedAccountsForTest {
        myConnectedAccounts {
          id
          handle
          provider
          lastCredentialsRefreshedAt
          authFailedAt
        }
      }
    `,
  });

  const account = (
    getDataOrThrow(response).myConnectedAccounts as ConnectedAccountDto[]
  ).find((candidate) => candidate.id === connectedAccountId);

  if (!account) {
    throw new Error(`Connected account ${connectedAccountId} not found`);
  }

  return account;
};

export const updateMessageChannel = async (
  messageChannelId: string,
  update: MessageChannelUpdate,
): Promise<void> => {
  const response = await makeMetadataAPIRequest({
    query: gql`
      mutation UpdateMessageChannelForTest($input: UpdateMessageChannelInput!) {
        updateMessageChannel(input: $input) {
          id
        }
      }
    `,
    variables: {
      input: {
        id: messageChannelId,
        update,
      },
    },
  });

  getDataOrThrow(response);
};

export const updateCalendarChannel = async (
  calendarChannelId: string,
  update: CalendarChannelUpdate,
): Promise<void> => {
  const response = await makeMetadataAPIRequest({
    query: gql`
      mutation UpdateCalendarChannelForTest(
        $input: UpdateCalendarChannelInput!
      ) {
        updateCalendarChannel(input: $input) {
          id
        }
      }
    `,
    variables: {
      input: {
        id: calendarChannelId,
        update,
      },
    },
  });

  getDataOrThrow(response);
};

export const startChannelSync = async (
  connectedAccountId: string,
): Promise<void> => {
  const response = await makeMetadataAPIRequest({
    query: gql`
      mutation StartChannelSyncForTest($connectedAccountId: UUID!) {
        startChannelSync(connectedAccountId: $connectedAccountId) {
          success
        }
      }
    `,
    variables: { connectedAccountId },
  });

  getDataOrThrow(response);
};

export const deleteConnectedAccount = async (
  connectedAccountId: string,
): Promise<void> => {
  const response = await makeMetadataAPIRequest({
    query: gql`
      mutation DeleteConnectedAccountForTest($id: UUID!) {
        deleteConnectedAccount(id: $id) {
          id
        }
      }
    `,
    variables: { id: connectedAccountId },
  });

  getDataOrThrow(response);

  await waitForAllJobsToFinish();
};