fix: labels cascade delete

api/src/chat/repositories/label.repository.ts

@@ -82,9 +82,13 @@ export class LabelRepository extends BaseRepository<
     >,
     _criteria: TFilterQuery<Label>,
   ): Promise<void> {
-    const labels = await this.find(
-      typeof _criteria === 'string' ? { _id: _criteria } : _criteria,
-    );
+    const ids = Array.isArray(_criteria._id?.$in)
+      ? _criteria._id.$in
+      : Array.isArray(_criteria._id)
+        ? _criteria._id
+        : [_criteria._id];
+
+    const labels = await this.find({ _id: { $in: ids } });
     this.eventEmitter.emit('hook:label:delete', labels);
   }
 }

api/src/chat/services/block.service.ts

@@ -7,6 +7,7 @@
  */
 
 import { Injectable } from '@nestjs/common';
+import { OnEvent } from '@nestjs/event-emitter';
 
 import { AttachmentService } from '@/attachment/services/attachment.service';
 import EventWrapper from '@/channel/lib/EventWrapper';
@@ -25,6 +26,7 @@ import { getRandom } from '@/utils/helpers/safeRandom';
 import { BlockDto } from '../dto/block.dto';
 import { BlockRepository } from '../repositories/block.repository';
 import { Block, BlockFull, BlockPopulate } from '../schemas/block.schema';
+import { Label } from '../schemas/label.schema';
 import { Context } from '../schemas/types/context';
 import {
   BlockMessage,
@@ -604,4 +606,32 @@ export class BlockService extends BaseService<
     }
     throw new Error('Invalid message format.');
   }
+
+  /**
+   * Updates the `trigger_labels` and `assign_labels` fields of a block when a label is deleted.
+   *
+   *
+   * This method removes the deleted label from the `trigger_labels` and `assign_labels` fields of all blocks that have the label.
+   *
+   * @param label The label that is being deleted.
+   */
+  @OnEvent('hook:label:delete')
+  async handleLabelDelete(labels: Label[]) {
+    const blocks = await this.find({
+      $or: [
+        { trigger_labels: { $in: labels.map((l) => l.id) } },
+        { assign_labels: { $in: labels.map((l) => l.id) } },
+      ],
+    });
+
+    for (const block of blocks) {
+      const trigger_labels = block.trigger_labels.filter(
+        (labelId) => !labels.find((l) => l.id === labelId),
+      );
+      const assign_labels = block.assign_labels.filter(
+        (labelId) => !labels.find((l) => l.id === labelId),
+      );
+      await this.updateOne(block.id, { trigger_labels, assign_labels });
+    }
+  }
 }

api/src/chat/services/subscriber.service.ts

@@ -30,6 +30,7 @@ import { WebsocketGateway } from '@/websocket/websocket.gateway';
 
 import { SubscriberDto, SubscriberUpdateDto } from '../dto/subscriber.dto';
 import { SubscriberRepository } from '../repositories/subscriber.repository';
+import { Label } from '../schemas/label.schema';
 import {
   Subscriber,
   SubscriberFull,
@@ -208,4 +209,24 @@ export class SubscriberService extends BaseService<
       }
     }
   }
+
+  /**
+   * Updates the `labels` field of a subscriber when a label is deleted.
+   *
+   * This method removes the deleted label from the `labels` field of all subscribers that have the label.
+   *
+   * @param label The label that is being deleted.
+   */
+  @OnEvent('hook:label:delete')
+  async handleLabelDelete(labels: Label[]) {
+    const subscribers = await this.find({
+      labels: { $in: labels.map((l) => l.id) },
+    });
+    for (const subscriber of subscribers) {
+      const updatedLabels = subscriber.labels.filter(
+        (label) => !labels.find((l) => l.id === label),
+      );
+      await this.updateOne(subscriber.id, { labels: updatedLabels });
+    }
+  }
 }