@mondaydotcomorg/monday-authorization 3.5.1 → 3.5.2-feat-shaime-support-entity-attributes-4-ddec1d3

package/src/authorization-attributes-service.ts

@@ -1,234 +1,46 @@
-import chunk from 'lodash/chunk.js';
-import { Api, FetcherConfig, HttpClient } from '@mondaydotcomorg/trident-backend-api';
-import { getTopicAttributes, sendToSns } from '@mondaydotcomorg/monday-sns';
-import { HttpFetcherError, RecursivePartial } from '@mondaydotcomorg/monday-fetch-api';
-import {
-  ResourceAttributeAssignment,
-  ResourceAttributeResponse,
-  ResourceAttributesOperation,
-} from './types/authorization-attributes-contracts';
-import { Resource } from './types/general';
-import { logger } from './authorization-internal-service';
-import { getAttributionsFromApi } from './attributions-service';
-import {
-  ASYNC_RESOURCE_ATTRIBUTES_MAX_OPERATIONS_PER_MESSAGE,
-  RESOURCE_ATTRIBUTES_SNS_UPDATE_OPERATION_MESSAGE_KIND,
-  SNS_ARN_ENV_VAR_NAME,
-  SNS_DEV_TEST_NAME,
-} from './constants/sns';
-import { APP_NAME, DEFAULT_FETCH_OPTIONS, ERROR_MESSAGES } from './constants';
-import type { TopicAttributesMap } from 'aws-sdk/clients/sns';
-
+import { AuthorizationAttributesMsService } from './authorization-attributes-ms-service';
+import { AuthorizationAttributesSnsService } from './authorization-attributes-sns-service';
+import { AuthorizationAttributesService as IAuthorizationAttributesService } from './types/authorization-attributes-service.interface';
+
+/**
+ * Main service class for managing resource and entity attributes.
+ * Provides access to both direct (MS) and SNS operations.
+ *
+ * @example
+ * const service = new AuthorizationAttributesService();
+ *
+ * // Use direct operations
+ * await service.direct().upsertResourceAttributes(accountId, assignments);
+ *
+ * // Use SNS operations
+ * await service.sns().upsertResourceAttributes(accountId, assignments, appName, actionId);
+ * ```
+ */
 export class AuthorizationAttributesService {
-  private static LOG_TAG = 'authorization_attributes';
-  private static API_PATHS = {
-    UPSERT_RESOURCE_ATTRIBUTES: '/attributes/{accountId}/resource',
-    DELETE_RESOURCE_ATTRIBUTES: '/attributes/{accountId}/resource/{resourceType}/{resourceId}',
-  } as const;
-  private httpClient: HttpClient;
-  private fetchOptions: RecursivePartial<FetcherConfig>;
-  private snsArn: string;
-
-  /**
-   * Public constructor to create the AuthorizationAttributesService instance.
-   * @param httpClient The HTTP client to use for API requests, if not provided, the default HTTP client from Api will be used.
-   * @param fetchOptions The fetch options to use for API requests, if not provided, the default fetch options will be used.
-   */
-  constructor(httpClient?: HttpClient, fetchOptions?: RecursivePartial<FetcherConfig>) {
-    if (!httpClient) {
-      httpClient = Api.getPart('httpClient');
-      if (!httpClient) {
-        throw new Error(ERROR_MESSAGES.HTTP_CLIENT_NOT_INITIALIZED);
-      }
-    }
-
-    if (!fetchOptions) {
-      fetchOptions = DEFAULT_FETCH_OPTIONS;
-    } else {
-      fetchOptions = {
-        ...DEFAULT_FETCH_OPTIONS,
-        ...fetchOptions,
-      };
-    }
-    this.httpClient = httpClient;
-    this.fetchOptions = fetchOptions;
-    this.snsArn = AuthorizationAttributesService.getSnsTopicArn();
-  }
-
-  /**
-   * Upsert resource attributes synchronously, performing http call to the authorization MS to assign the given attributes to the given resource.
-   * @param accountId
-   * @param resourceAttributeAssignments - Array of resource (resourceType, resourceId) and attribute (key, value) pairs to upsert in the authorization MS.
-   * e.g. [{ resourceType: 'board', resourceId: 123, key: 'board_kind', value: 'private' }]
-   * @returns ResourceAttributeResponse - The affected (created and updated_ resource attributes assignments in the `attributes` field.
-   */
-  async upsertResourceAttributes(
-    accountId: number,
-    resourceAttributeAssignments: ResourceAttributeAssignment[]
-  ): Promise<ResourceAttributeResponse> {
-    const attributionHeaders = getAttributionsFromApi();
-    try {
-      return await this.httpClient.fetch<ResourceAttributeResponse>(
-        {
-          url: {
-            appName: APP_NAME,
-            path: AuthorizationAttributesService.API_PATHS.UPSERT_RESOURCE_ATTRIBUTES.replace(
-              '{accountId}',
-              accountId.toString()
-            ),
-          },
-          method: 'POST',
-          headers: {
-            'Content-Type': 'application/json',
-            ...attributionHeaders,
-          },
-          body: JSON.stringify({ resourceAttributeAssignments }),
-        },
-        this.fetchOptions
-      );
-    } catch (err) {
-      if (err instanceof HttpFetcherError) {
-        throw new Error(ERROR_MESSAGES.REQUEST_FAILED('upsertResourceAttributes', err.status, err.message));
-      }
-      throw err;
-    }
-  }
+  private _directService: AuthorizationAttributesMsService | null = null;
+  private _snsService: AuthorizationAttributesSnsService | null = null;
 
   /**
-   * Delete resource attributes assignments synchronously, performing http call to the authorization MS to delete the given attributes from the given singular resource.
-   * @param accountId
-   * @param resource - The resource (resourceType, resourceId) to delete the attributes for.
-   * @param attributeKeys - Array of attribute keys to delete for the resource.
-   * @returns ResourceAttributeResponse - The affected (deleted) resource attributes assignments in the `attributes` field.
+   * Gets the direct (MS) service instance.
+   * Initializes the service on first access (lazy initialization).
+   * @returns IAuthorizationAttributesService instance
    */
-  async deleteResourceAttributes(
-    accountId: number,
-    resource: Resource,
-    attributeKeys: string[]
-  ): Promise<ResourceAttributeResponse> {
-    const attributionHeaders = getAttributionsFromApi();
-    if (!resource.id) {
-      throw new Error('Resource ID is required');
-    }
-    try {
-      return await this.httpClient.fetch<ResourceAttributeResponse>(
-        {
-          url: {
-            appName: APP_NAME,
-            path: AuthorizationAttributesService.API_PATHS.DELETE_RESOURCE_ATTRIBUTES.replace(
-              '{accountId}',
-              accountId.toString()
-            )
-              .replace('{resourceType}', resource.type)
-              .replace('{resourceId}', resource.id.toString()),
-          },
-          method: 'DELETE',
-          headers: {
-            'Content-Type': 'application/json',
-            ...attributionHeaders,
-          },
-          body: JSON.stringify({ keys: attributeKeys }),
-        },
-        this.fetchOptions
-      );
-    } catch (err) {
-      if (err instanceof HttpFetcherError) {
-        throw new Error(ERROR_MESSAGES.REQUEST_FAILED('deleteResourceAttributes', err.status, err.message));
-      }
-      throw err;
-    }
-  }
-
-  /**
-   *  Async function, this function only send the updates request to SNS and return before the change actually took place
-   * @param accountId
-   * @param appName - App name of the calling app
-   * @param callerActionIdentifier - action identifier
-   * @param resourceAttributeOperations - Array of operations to do on resource attributes.
-   * @return {Promise<ResourceAttributesOperation[]>} Array of sent operations
-   *  */
-  async updateResourceAttributesAsync(
-    accountId: number,
-    appName: string,
-    callerActionIdentifier: string,
-    resourceAttributeOperations: ResourceAttributesOperation[]
-  ): Promise<ResourceAttributesOperation[]> {
-    const topicArn: string = this.snsArn;
-    const sendToSnsPromises: Promise<ResourceAttributesOperation[]>[] = [];
-    const operationChucks = chunk(resourceAttributeOperations, ASYNC_RESOURCE_ATTRIBUTES_MAX_OPERATIONS_PER_MESSAGE);
-    for (const operationsChunk of operationChucks) {
-      sendToSnsPromises.push(
-        this.sendSingleSnsMessage(topicArn, accountId, appName, callerActionIdentifier, operationsChunk)
-      );
-    }
-    return (await Promise.all(sendToSnsPromises)).flat();
-  }
-
-  private async sendSingleSnsMessage(
-    topicArn: string,
-    accountId: number,
-    appName: string,
-    callerActionIdentifier: string,
-    operations: ResourceAttributesOperation[]
-  ): Promise<ResourceAttributesOperation[]> {
-    const payload = {
-      kind: RESOURCE_ATTRIBUTES_SNS_UPDATE_OPERATION_MESSAGE_KIND,
-      payload: {
-        accountId: accountId,
-        callerAppName: appName,
-        callerActionIdentifier: callerActionIdentifier,
-        operations: operations,
-      },
-    };
-    try {
-      await sendToSns(payload, topicArn);
-      return operations;
-    } catch (error) {
-      logger.error(
-        { error, tag: AuthorizationAttributesService.LOG_TAG },
-        'Authorization resource attributes async update: failed to send operations to SNS'
-      );
-      return [];
-    }
-  }
-
-  private static getSnsTopicArn(): string {
-    const arnFromEnv: string | undefined = process.env[SNS_ARN_ENV_VAR_NAME];
-    if (arnFromEnv) {
-      return arnFromEnv;
-    }
-    if (process.env.NODE_ENV === 'development' || process.env.NODE_ENV === 'test') {
-      return SNS_DEV_TEST_NAME;
+  direct(): IAuthorizationAttributesService {
+    if (this._directService === null) {
+      this._directService = new AuthorizationAttributesMsService();
     }
-    throw new Error('Unable to get sns topic arn from env variable');
+    return this._directService;
   }
 
   /**
-   * Checks we can contact the required SNS topic that used to send attribute updates to Authorization MS.
-   * This function can be used as health check for services that updating resource attributes in async is crucial.
-   * Note this function only verify the POD can contact AWS SDK and the topic exists, but the user still might get
-   * errors when pushing for the SNS (e.g: in case the AWS role of the POD don't have permissions to push messages).
-   * However, this is the best we can do without actually push dummy messages to the SNS.
-   * @return {Promise<boolean>} - true if succeeded
+   * Gets the SNS service instance.
+   * Initializes the service on first access (lazy initialization).
+   * @returns IAuthorizationAttributesService instance
    */
-  async asyncResourceAttributesHealthCheck(): Promise<boolean> {
-    try {
-      const requestedTopicArn: string = this.snsArn;
-      const attributes: TopicAttributesMap = await getTopicAttributes(requestedTopicArn);
-      const isHealthy = !(!attributes || !('TopicArn' in attributes) || attributes.TopicArn !== requestedTopicArn);
-      if (!isHealthy) {
-        logger.error(
-          { requestedTopicArn, snsReturnedAttributes: attributes, tag: AuthorizationAttributesService.LOG_TAG },
-          'authorization-attributes-service failed in health check'
-        );
-      }
-      return isHealthy;
-    } catch (error) {
-      logger.error(
-        { error, tag: AuthorizationAttributesService.LOG_TAG },
-        'authorization-attributes-service got error during health check'
-      );
-      return false;
+  sns(): IAuthorizationAttributesService {
+    if (this._snsService === null) {
+      this._snsService = new AuthorizationAttributesSnsService();
     }
+    return this._snsService;
   }
 }

package/src/authorization-attributes-sns-service.ts

@@ -0,0 +1,312 @@
+import chunk from 'lodash/chunk.js';
+import { getTopicAttributes, sendToSns } from '@mondaydotcomorg/monday-sns';
+import {
+  ResourceAttributeAssignment,
+  ResourceAttributeOperation,
+  EntityAttributeOperation,
+  AttributeOperation,
+} from './types/authorization-attributes-contracts';
+import { EntityAttributeAssignment } from './entity-attribute-assignment';
+import { Resource } from './types/general';
+import { logger } from './authorization-internal-service';
+import {
+  ASYNC_RESOURCE_ATTRIBUTES_MAX_OPERATIONS_PER_MESSAGE,
+  ASYNC_ENTITY_ATTRIBUTES_MAX_OPERATIONS_PER_MESSAGE,
+  RESOURCE_ATTRIBUTES_SNS_UPDATE_OPERATION_MESSAGE_KIND,
+  ENTITY_ATTRIBUTES_SNS_UPDATE_OPERATION_MESSAGE_KIND,
+  RESOURCE_SNS_ARN_ENV_VAR_NAME,
+  ENTITY_SNS_ARN_ENV_VAR_NAME,
+  RESOURCE_SNS_DEV_TEST_NAME,
+  ENTITY_SNS_DEV_TEST_NAME,
+  SnsTopicType,
+} from './constants/sns';
+import type { TopicAttributesMap } from 'aws-sdk/clients/sns';
+import { AuthorizationAttributesService } from './types/authorization-attributes-service.interface';
+import { EntityType } from './entity-attributes-constants';
+
+/**
+ * Service class for managing resource attributes asynchronously via SNS.
+ * Provides asynchronous operations to create/update and delete attributes on resources.
+ */
+export class AuthorizationAttributesSnsService implements AuthorizationAttributesService {
+  private static LOG_TAG = 'authorization_attributes';
+  private resourceSnsArn: string;
+  private entitySnsArn: string;
+
+  /**
+   * Public constructor to create the AuthorizationAttributesSnsService instance.
+   */
+  constructor() {
+    this.resourceSnsArn = AuthorizationAttributesSnsService.getSnsTopicArn(SnsTopicType.RESOURCE);
+    this.entitySnsArn = AuthorizationAttributesSnsService.getSnsTopicArn(SnsTopicType.ENTITY);
+  }
+
+  /**
+   * Async function to upsert resource attributes using SNS.
+   * Sends the updates request to SNS and returns before the change actually took place.
+   * @param accountId The account ID
+   * @param appName App name of the calling app
+   * @param callerActionIdentifier Action identifier
+   * @param resourceAttributeAssignments Array of resource attribute assignments to upsert
+   * @return Promise with array of sent operations
+   */
+  async upsertResourceAttributes(
+    accountId: number,
+    resourceAttributeAssignments: ResourceAttributeAssignment[],
+    appName?: string,
+    callerActionIdentifier?: string
+  ): Promise<ResourceAttributeOperation[]> {
+    if (!appName || !callerActionIdentifier) {
+      throw new Error('appName and callerActionIdentifier are required for SNS service');
+    }
+    const operations: ResourceAttributeOperation[] = resourceAttributeAssignments.map(assignment => ({
+      ...assignment,
+      operationType: AttributeOperation.UPSERT,
+    }));
+    return this.updateResourceAttributes(accountId, appName, callerActionIdentifier, operations);
+  }
+
+  /**
+   * Async function to delete resource attributes using SNS.
+   * Sends the updates request to SNS and returns before the change actually took place.
+   * @param accountId The account ID
+   * @param appName App name of the calling app
+   * @param callerActionIdentifier Action identifier
+   * @param resource The resource (resourceType, resourceId)
+   * @param attributeKeys Array of attribute keys to delete
+   * @return Promise with array of sent operations
+   */
+  async deleteResourceAttributes(
+    accountId: number,
+    resource: Resource,
+    attributeKeys: string[],
+    appName?: string,
+    callerActionIdentifier?: string
+  ): Promise<ResourceAttributeOperation[]> {
+    if (!appName || !callerActionIdentifier) {
+      throw new Error('appName and callerActionIdentifier are required for SNS service');
+    }
+    const operations: ResourceAttributeOperation[] = attributeKeys.map(
+      key =>
+        ({
+          resourceType: resource.type,
+          resourceId: resource.id!,
+          key,
+          operationType: AttributeOperation.DELETE,
+        } as ResourceAttributeOperation)
+    );
+    return this.updateResourceAttributes(accountId, appName, callerActionIdentifier, operations);
+  }
+
+  /**
+   * Async function to upsert entity attributes using SNS.
+   * Sends the updates request to SNS and returns before the change actually took place.
+   * @param accountId The account ID
+   * @param appName App name of the calling app
+   * @param callerActionIdentifier Action identifier
+   * @param entityAttributeAssignments Array of entity attribute assignments to upsert
+   * @return Promise with array of sent operations
+   */
+  async upsertEntityAttributes(
+    accountId: number,
+    entityAttributeAssignments: EntityAttributeAssignment[],
+    appName?: string,
+    callerActionIdentifier?: string
+  ): Promise<EntityAttributeOperation[]> {
+    if (!appName || !callerActionIdentifier) {
+      throw new Error('appName and callerActionIdentifier are required for SNS service');
+    }
+    const operations: EntityAttributeOperation[] = entityAttributeAssignments.map(assignment => {
+      return {
+        entityId: assignment.entityId,
+        entityType: assignment.entityType,
+        key: assignment.attributeKey,
+        value: assignment.attributeValue,
+        operationType: AttributeOperation.UPSERT,
+      };
+    });
+    return this.updateEntityAttributes(accountId, appName, callerActionIdentifier, operations);
+  }
+
+  /**
+   * Async function to delete entity attributes using SNS.
+   * Sends the updates request to SNS and returns before the change actually took place.
+   * @param accountId The account ID
+   * @param appName App name of the calling app
+   * @param callerActionIdentifier Action identifier
+   * @param entityType The entity type
+   * @param entityId The entity ID
+   * @param attributeKeys Array of attribute keys to delete
+   * @return Promise with array of sent operations
+   */
+  async deleteEntityAttributes(
+    accountId: number,
+    entityType: EntityType | string,
+    entityId: number,
+    attributeKeys: string[],
+    appName?: string,
+    callerActionIdentifier?: string
+  ): Promise<EntityAttributeOperation[]> {
+    if (!appName || !callerActionIdentifier) {
+      throw new Error('appName and callerActionIdentifier are required for SNS service');
+    }
+    const operations: EntityAttributeOperation[] = attributeKeys.map(
+      key =>
+        ({
+          entityType: entityType,
+          entityId,
+          key,
+          operationType: AttributeOperation.DELETE,
+        } as EntityAttributeOperation)
+    );
+    return this.updateEntityAttributes(accountId, appName, callerActionIdentifier, operations);
+  }
+
+  /**
+   *  Async function, this function only send the updates request to SNS and return before the change actually took place
+   * @param accountId
+   * @param appName - App name of the calling app
+   * @param callerActionIdentifier - action identifier
+   * @param resourceAttributeOperations - Array of operations to do on resource attributes.
+   * @return {Promise<ResourceAttributesOperation[]>} Array of sent operations
+   *  */
+  async updateResourceAttributes(
+    accountId: number,
+    appName: string,
+    callerActionIdentifier: string,
+    resourceAttributeOperations: ResourceAttributeOperation[]
+  ): Promise<ResourceAttributeOperation[]> {
+    const topicArn: string = this.resourceSnsArn;
+    const sendToSnsPromises: Promise<ResourceAttributeOperation[]>[] = [];
+    const operationChucks = chunk(resourceAttributeOperations, ASYNC_RESOURCE_ATTRIBUTES_MAX_OPERATIONS_PER_MESSAGE);
+    for (const operationsChunk of operationChucks) {
+      sendToSnsPromises.push(
+        this.sendSingleSnsMessage(
+          topicArn,
+          accountId,
+          appName,
+          callerActionIdentifier,
+          operationsChunk,
+          RESOURCE_ATTRIBUTES_SNS_UPDATE_OPERATION_MESSAGE_KIND,
+          'Authorization resource attributes async update: failed to send operations to SNS'
+        )
+      );
+    }
+    return (await Promise.all(sendToSnsPromises)).flat();
+  }
+
+  /**
+   *  Async function, this function only send the updates request to SNS and return before the change actually took place
+   * @param accountId
+   * @param appName - App name of the calling app
+   * @param callerActionIdentifier - action identifier
+   * @param entityAttributeOperations - Array of operations to do on entity attributes.
+   * @return {Promise<EntityAttributeOperation[]>} Array of sent operations
+   *  */
+  async updateEntityAttributes(
+    accountId: number,
+    appName: string,
+    callerActionIdentifier: string,
+    entityAttributeOperations: EntityAttributeOperation[]
+  ): Promise<EntityAttributeOperation[]> {
+    const topicArn: string = this.entitySnsArn;
+    const sendToSnsPromises: Promise<EntityAttributeOperation[]>[] = [];
+    const operationChucks = chunk(entityAttributeOperations, ASYNC_ENTITY_ATTRIBUTES_MAX_OPERATIONS_PER_MESSAGE);
+    for (const operationsChunk of operationChucks) {
+      sendToSnsPromises.push(
+        this.sendSingleSnsMessage(
+          topicArn,
+          accountId,
+          appName,
+          callerActionIdentifier,
+          operationsChunk,
+          ENTITY_ATTRIBUTES_SNS_UPDATE_OPERATION_MESSAGE_KIND,
+          'Authorization entity attributes async update: failed to send operations to SNS'
+        )
+      );
+    }
+    return (await Promise.all(sendToSnsPromises)).flat();
+  }
+
+  private async sendSingleSnsMessage<T extends ResourceAttributeOperation | EntityAttributeOperation>(
+    topicArn: string,
+    accountId: number,
+    appName: string,
+    callerActionIdentifier: string,
+    operations: T[],
+    kind: string,
+    errorLogMessage: string
+  ): Promise<T[]> {
+    const payload = {
+      kind,
+      payload: {
+        accountId: accountId,
+        callerAppName: appName,
+        callerActionIdentifier: callerActionIdentifier,
+        operations: operations,
+      },
+    };
+    try {
+      await sendToSns(payload, topicArn);
+      return operations;
+    } catch (error) {
+      logger.error({ error, tag: AuthorizationAttributesSnsService.LOG_TAG }, errorLogMessage);
+      return [];
+    }
+  }
+
+  private static getSnsTopicArn(type: SnsTopicType): string {
+    let envVarName: string;
+    let devTestName: string;
+
+    switch (type) {
+      case SnsTopicType.ENTITY:
+        envVarName = ENTITY_SNS_ARN_ENV_VAR_NAME;
+        devTestName = ENTITY_SNS_DEV_TEST_NAME;
+        break;
+      default:
+        // Default to resource SNS constants
+        envVarName = RESOURCE_SNS_ARN_ENV_VAR_NAME;
+        devTestName = RESOURCE_SNS_DEV_TEST_NAME;
+        break;
+    }
+
+    const arnFromEnv: string | undefined = process.env[envVarName];
+    if (arnFromEnv) {
+      return arnFromEnv;
+    }
+    if (process.env.NODE_ENV === 'development' || process.env.NODE_ENV === 'test') {
+      return devTestName;
+    }
+    throw new Error(`Unable to get ${type} sns topic arn from env variable`);
+  }
+
+  /**
+   * Checks we can contact the required SNS topic that used to send attribute updates to Authorization MS.
+   * This function can be used as health check for services that updating resource attributes in async is crucial.
+   * Note this function only verify the POD can contact AWS SDK and the topic exists, but the user still might get
+   * errors when pushing for the SNS (e.g: in case the AWS role of the POD don't have permissions to push messages).
+   * However, this is the best we can do without actually push dummy messages to the SNS.
+   * @return {Promise<boolean>} - true if succeeded
+   */
+  async asyncResourceAttributesHealthCheck(): Promise<boolean> {
+    try {
+      const requestedTopicArn: string = this.resourceSnsArn;
+      const attributes: TopicAttributesMap = await getTopicAttributes(requestedTopicArn);
+      const isHealthy = !(!attributes || !('TopicArn' in attributes) || attributes.TopicArn !== requestedTopicArn);
+      if (!isHealthy) {
+        logger.error(
+          { requestedTopicArn, snsReturnedAttributes: attributes, tag: AuthorizationAttributesSnsService.LOG_TAG },
+          'authorization-attributes-service failed in health check'
+        );
+      }
+      return isHealthy;
+    } catch (error) {
+      logger.error(
+        { error, tag: AuthorizationAttributesSnsService.LOG_TAG },
+        'authorization-attributes-service got error during health check'
+      );
+      return false;
+    }
+  }
+}

package/src/base-attribute-assignment.ts

@@ -0,0 +1,46 @@
+import { ValidationUtils } from './utils/validation';
+import isEqual from 'lodash/isEqual.js';
+import { EntityAttributeAssignment, ResourceAttributeDelete } from './types/authorization-attributes-contracts';
+
+/**
+ * Base class for attribute assignments (Resource or Entity)
+ * Provides common validation and functionality
+ */
+export abstract class BaseAttributeAssignment<TId extends number, TType extends string> {
+  public readonly id: TId;
+  public readonly type: TType;
+  public readonly attributeKey: string;
+  public readonly attributeValue: string;
+
+  constructor(
+    id: TId,
+    type: string,
+    attributeKey: string,
+    attributeValue: string,
+    validTypes: readonly string[],
+    idFieldName: string,
+    typeFieldName: string
+  ) {
+    const validated = ValidationUtils.validateAssignment<TType>(
+      { id, type, attributeKey, attributeValue },
+      validTypes as readonly TType[],
+      { id: idFieldName, type: typeFieldName }
+    );
+
+    this.id = validated.id as TId;
+    this.type = validated.type as TType;
+    this.attributeKey = validated.attributeKey;
+    this.attributeValue = validated.attributeValue;
+  }
+
+  /**
+   * Compares two assignments for equality
+   * @param other Another assignment instance
+   * @returns true if all properties are equal
+   */
+  equals(other: BaseAttributeAssignment<TId, TType>): boolean {
+    return isEqual(this, other);
+  }
+
+  abstract toDataTransferObject(): EntityAttributeAssignment | ResourceAttributeDelete;
+}

package/src/constants/sns.ts

@@ -1,5 +1,22 @@
-export const SNS_ARN_ENV_VAR_NAME = 'SHARED_AUTHORIZATION_SNS_ENDPOINT_RESOURCE_ATTRIBUTES';
-export const SNS_DEV_TEST_NAME =
+export enum SnsTopicType {
+  RESOURCE = 'resource',
+  ENTITY = 'entity',
+}
+
+// Resource SNS constants
+export const RESOURCE_SNS_ARN_ENV_VAR_NAME = 'SHARED_AUTHORIZATION_SNS_ENDPOINT_RESOURCE_ATTRIBUTES';
+export const RESOURCE_SNS_DEV_TEST_NAME =
   'arn:aws:sns:us-east-1:000000000000:monday-authorization-resource-attributes-sns-local';
+
+// Entity SNS constants
+export const ENTITY_SNS_ARN_ENV_VAR_NAME = 'SHARED_AUTHORIZATION_SNS_ENDPOINT_ENTITY_ATTRIBUTES';
+export const ENTITY_SNS_DEV_TEST_NAME =
+  'arn:aws:sns:us-east-1:000000000000:monday-authorization-entity-attributes-sns-local';
 export const RESOURCE_ATTRIBUTES_SNS_UPDATE_OPERATION_MESSAGE_KIND = 'resourceAttributeModification';
+export const ENTITY_ATTRIBUTES_SNS_UPDATE_OPERATION_MESSAGE_KIND = 'entityAttributeModification';
 export const ASYNC_RESOURCE_ATTRIBUTES_MAX_OPERATIONS_PER_MESSAGE = 100;
+export const ASYNC_ENTITY_ATTRIBUTES_MAX_OPERATIONS_PER_MESSAGE = 100;
+
+// Legacy exports for backward compatibility
+export const SNS_ARN_ENV_VAR_NAME = RESOURCE_SNS_ARN_ENV_VAR_NAME;
+export const SNS_DEV_TEST_NAME = RESOURCE_SNS_DEV_TEST_NAME;

package/src/entity-attribute-assignment.ts

@@ -0,0 +1,30 @@
+import { ENTITY_TYPES, EntityType } from './entity-attributes-constants';
+import { BaseAttributeAssignment } from './base-attribute-assignment';
+
+export class EntityAttributeAssignment extends BaseAttributeAssignment<number, EntityType> {
+  public readonly entityId: number;
+  public readonly entityType: EntityType;
+
+  constructor(entityId: number, entityType: string, attributeKey: string, attributeValue: string) {
+    super(entityId, entityType, attributeKey, attributeValue, ENTITY_TYPES, 'entityId', 'entityType');
+    this.entityId = entityId;
+    this.entityType = this.type;
+  }
+
+  toDataTransferObject() {
+    return {
+      entityId: this.entityId,
+      entityType: this.entityType,
+      key: this.attributeKey,
+      value: this.attributeValue,
+    };
+  }
+  /**
+   * Compares two assignments for equality
+   * @param other Another EntityAttributeAssignment instance
+   * @returns true if all properties are equal
+   */
+  equals(other: EntityAttributeAssignment): boolean {
+    return super.equals(other);
+  }
+}