calendar/packages/features/ee/integration-attribute-sync/services/IntegrationAttributeSyncService.ts

import type { CredentialRepository } from "@calcom/features/credentials/repositories/CredentialRepository";
import type { TeamRepository } from "@calcom/features/ee/teams/repositories/TeamRepository";

import { enabledAppSlugs } from "../constants";
import {
  type IIntegrationAttributeSyncRepository,
  type ISyncFormData,
  type ITeamCondition,
  type IAttributeSyncRule,
  AttributeSyncIntegrations,
  ConditionIdentifierEnum,
} from "../repositories/IIntegrationAttributeSyncRepository";
import { attributeSyncRuleSchema, type ZCreateAttributeSyncSchema } from "../schemas/zod";

export class DuplicateAttributeWithinSyncError extends Error {
  constructor(public attributeId: string) {
    super("Attribute is already mapped in this sync");
    this.name = "DuplicateAttributeWithinSyncError";
  }
}

export class DuplicateAttributeAcrossSyncsError extends Error {
  constructor(public attributeId: string) {
    super("Attribute is already mapped in another sync");
    this.name = "DuplicateAttributeAcrossSyncsError";
  }
}

export class UnauthorizedAttributeError extends Error {
  constructor(public attributeIds: string[]) {
    super("One or more attributes do not belong to this organization");
    this.name = "UnauthorizedAttributeError";
  }
}

export class CredentialNotFoundError extends Error {
  constructor() {
    super("Credential not found");
    this.name = "CredentialNotFoundError";
  }
}

interface IIntegrationAttributeSyncServiceDeps {
  credentialRepository: CredentialRepository;
  integrationAttributeSyncRepository: IIntegrationAttributeSyncRepository;
  teamRepository: TeamRepository;
}

export class IntegrationAttributeSyncService {
  constructor(private readonly deps: IIntegrationAttributeSyncServiceDeps) {}

  private extractTeamIdsFromRule(rule: IAttributeSyncRule): number[] {
    return rule.conditions
      .filter((c): c is ITeamCondition => c.identifier === ConditionIdentifierEnum.TEAM_ID)
      .flatMap((c) => c.value);
  }

  private async validateTeamsBelongToOrg(teamIds: number[], organizationId: number): Promise<void> {
    if (teamIds.length === 0) return;

    const invalidTeams = await this.deps.teamRepository.findTeamsNotBelongingToOrgByIds({
      teamIds,
      orgId: organizationId,
    });

    if (invalidTeams.length > 0) {
      throw new Error(
        `Teams do not belong to this organization: ${invalidTeams.map((t) => t.id).join(", ")}`
      );
    }
  }

  private validateWithinSyncUniqueness(mappings: { attributeId: string }[]): void {
    const seenAttributes = new Set<string>();
    for (const mapping of mappings) {
      if (seenAttributes.has(mapping.attributeId)) {
        throw new DuplicateAttributeWithinSyncError(mapping.attributeId);
      }
      seenAttributes.add(mapping.attributeId);
    }
  }

  private async validateCrossSyncUniqueness(
    organizationId: number,
    mappings: { attributeId: string }[],
    excludeSyncId?: string
  ): Promise<void> {
    const existingMappedIds =
      await this.deps.integrationAttributeSyncRepository.getMappedAttributeIdsByOrganization(
        organizationId,
        excludeSyncId
      );
    const existingSet = new Set(existingMappedIds);

    for (const mapping of mappings) {
      if (existingSet.has(mapping.attributeId)) {
        throw new DuplicateAttributeAcrossSyncsError(mapping.attributeId);
      }
    }
  }

  private async validateAttributeOwnership(organizationId: number, attributeIds: string[]): Promise<void> {
    const validAttributeIds =
      await this.deps.integrationAttributeSyncRepository.getAttributeIdsByOrganization(
        organizationId,
        attributeIds
      );

    if (validAttributeIds.length !== attributeIds.length) {
      const validSet = new Set(validAttributeIds);
      const invalidIds = attributeIds.filter((id) => !validSet.has(id));
      throw new UnauthorizedAttributeError(invalidIds);
    }
  }

  async getEnabledAppCredentials(organizationId: number) {
    return this.deps.credentialRepository.findByTeamIdAndSlugs({
      teamId: organizationId,
      slugs: enabledAppSlugs,
    });
  }

  async getAllIntegrationAttributeSyncs(organizationId: number) {
    return this.deps.integrationAttributeSyncRepository.getByOrganizationId(organizationId);
  }

  async getById(id: string) {
    return this.deps.integrationAttributeSyncRepository.getById(id);
  }

  async createAttributeSync(input: ZCreateAttributeSyncSchema, organizationId: number) {
    const credential = await this.deps.credentialRepository.findByIdAndTeamId({
      id: input.credentialId,
      teamId: organizationId,
    });

    if (!credential) {
      throw new CredentialNotFoundError();
    }

    const parsedRule = attributeSyncRuleSchema.parse(input.rule);

    const teamIds = this.extractTeamIdsFromRule(parsedRule);
    await this.validateTeamsBelongToOrg(teamIds, organizationId);

    const integrationValue = credential.app?.slug || credential.type;
    if (!Object.values(AttributeSyncIntegrations).includes(integrationValue as AttributeSyncIntegrations)) {
      throw new Error(`Unsupported integration type: ${integrationValue}`);
    }

    this.validateWithinSyncUniqueness(input.syncFieldMappings);

    await this.validateCrossSyncUniqueness(organizationId, input.syncFieldMappings);

    const attributeIds = input.syncFieldMappings.map((m) => m.attributeId);
    await this.validateAttributeOwnership(organizationId, attributeIds);

    return this.deps.integrationAttributeSyncRepository.create({
      name: input.name,
      organizationId,
      integration: integrationValue as AttributeSyncIntegrations,
      credentialId: input.credentialId,
      enabled: input.enabled,
      rule: parsedRule,
      syncFieldMappings: input.syncFieldMappings,
    });
  }

  async updateIncludeRulesAndMappings(data: ISyncFormData) {
    const { syncFieldMappings, rule, ruleId, ...integrationAttributeSync } = data;

    this.validateWithinSyncUniqueness(syncFieldMappings);

    await this.validateCrossSyncUniqueness(data.organizationId, syncFieldMappings, data.id);

    const attributeIds = syncFieldMappings.map((m) => m.attributeId);
    await this.validateAttributeOwnership(data.organizationId, attributeIds);

    const existingFieldMappings = await this.deps.integrationAttributeSyncRepository.getSyncFieldMappings(
      data.id
    );

    const parsedRule = attributeSyncRuleSchema.parse(rule);

    const teamIds = this.extractTeamIdsFromRule(parsedRule);
    await this.validateTeamsBelongToOrg(teamIds, data.organizationId);

    const incomingMappingIds = new Set(
      syncFieldMappings.reduce((ids, mapping) => {
        if ("id" in mapping) ids.push(mapping.id);
        return ids;
      }, [] as string[])
    );

    const fieldMappingsToDelete = existingFieldMappings
      .filter((mapping) => !incomingMappingIds.has(mapping.id))
      .map((mapping) => mapping.id);

    const fieldMappingsToCreate = syncFieldMappings.filter((m) => !("id" in m));
    const fieldMappingsToUpdate = syncFieldMappings.filter((m): m is typeof m & { id: string } => "id" in m);

    await this.deps.integrationAttributeSyncRepository.updateTransactionWithRuleAndMappings({
      integrationAttributeSync,
      attributeSyncRule: {
        id: ruleId,
        rule: parsedRule,
      },
      fieldMappingsToCreate,
      fieldMappingsToUpdate,
      fieldMappingsToDelete,
    });
  }
  async deleteById(id: string) {
    return this.deps.integrationAttributeSyncRepository.deleteById(id);
  }

  async getAllByCredentialId(credentialId: number) {
    return this.deps.integrationAttributeSyncRepository.getAllByCredentialId(credentialId);
  }
}