rez_core 5.0.67 → 5.0.68

package/src/module/workflow/service/task.service.ts

@@ -1,540 +1,547 @@
-import { Inject, Injectable } from '@nestjs/common';
-import { EntityServiceImpl } from 'src/module/meta/service/entity-service-impl.service';
-import { UserData } from 'src/module/user/entity/user.entity';
-import { TaskRepository } from '../repository/task.repository';
-import { ActionDataService } from './action-data.service';
-import { DataSource, EntityManager } from 'typeorm';
-import { ActivityLogService } from './activity-log.service';
-import { ACTIVITY_CATEGORIES } from '../repository/activity-log.repository';
-import { MediaDataService } from 'src/module/meta/service/media-data.service';
-import { ActionHandler } from 'src/module/workflow-automation/interface/action.decorator';
-import * as moment from 'moment';
-import { ConfigService } from '@nestjs/config';
-import axios from 'axios';
-import { ChecksumAlgorithm } from '@aws-sdk/client-s3';
-import { ReflectionHelper } from '../../../utils/service/reflection-helper.service';
-
-@Injectable()
-@ActionHandler('add_task')
-export class TaskService extends EntityServiceImpl {
-  constructor(
-    private readonly taskRepository: TaskRepository,
-    private readonly actionDataService: ActionDataService,
-    private readonly dataSource: DataSource,
-    @Inject('ActivityLogService')
-    private readonly activityLogService: ActivityLogService,
-    private readonly mediaDataService: MediaDataService,
-    private readonly configService: ConfigService
-  ) {
-    super();
-  }
-  name: string = 'add_task';
-
-  async execute(payload: any): Promise<any> {
-    let { entity, user, config } = payload;
-
-    const { organization_id, level_id, level_type } = entity;
-
-    if (entity.entity_type !== 'LEAD') {
-      entity = await this.getEntityData('LEAD', entity.parent_id, user);
-    }
-
-    // 3. Build entityData
-    const entityData = {
-      mapped_entity_id: entity.id,
-      mapped_entity_type: entity.entity_type,
-      task_owner: entity.lead_owner,
-      stage_id: entity.stageId,
-      status: config.status,
-      due_date: Date.now() + 2,
-      due_time: Date.now(),
-      description: config.description,
-      is_mandatory: config.is_mandatory,
-      name: config.name,
-      entity_type: 'TASK',
-    };
-
-    return await this.createEntity(entityData, user);
-  }
-
-  async createEntity(
-    entityData: any,
-    loggedInUser: any,
-    manager?: EntityManager | null,
-    appcode?: string,
-  ): Promise<any> {
-    if (entityData && typeof entityData['is_mandatory'] === 'string') {
-      entityData['is_mandatory'] = entityData['is_mandatory'] === '1';
-    }
-
-    // handling the date string for due_date and reminder_date
-    if (entityData?.due_date) {
-      // Interpret input 'YYYY-MM-DD' as local midnight (IST)
-      entityData.due_date = moment(entityData.due_date, 'YYYY-MM-DD')
-        .startOf('day') // 00:00 local time
-        .toDate(); // Convert to JS Date object for TypeORM
-    }
-
-    if (entityData?.reminder_date) {
-      entityData.reminder_date = moment(entityData.reminder_date, 'YYYY-MM-DD')
-        .startOf('day')
-        .toDate();
-    }
-
-    const createdEntity = await super.createEntity(
-      entityData,
-      loggedInUser,
-      manager,
-      appcode,
-    );
-
-    try {
-      const logData = {
-        mapped_entity_id: createdEntity.mapped_entity_id,
-        mapped_entity_type: createdEntity.mapped_entity_type,
-        title: `Task added`,
-        description: `A new task ${createdEntity.name} was added`,
-        category: ACTIVITY_CATEGORIES.TASK,
-        action: 'add',
-        appcode: loggedInUser.appcode,
-      };
-
-      await this.activityLogService.logActivity(logData, loggedInUser);
-    } catch (error) {
-      console.error(
-        'Failed to log activity for meeting:',
-        error?.message || error,
-      );
-      // Logging should not block main flow
-    }
-
-    let relationData = await this.dataSource.query(
-      `SELECT * FROM frm_entity_relation WHERE source_entity_type = $1  AND target_entity_type = $2`,
-      [createdEntity.mapped_entity_type, createdEntity.entity_type],
-    );
-
-    if (relationData) {
-      await this.dataSource.query(
-        `INSERT INTO frm_entity_relation_data (organization_id, source_entity_id, source_entity_type, target_entity_id, target_entity_type, relation_type) VALUES ($1, $2, $3, $4, $5 , $6)`,
-        [
-          loggedInUser.organization_id,
-          createdEntity.mapped_entity_id,
-          createdEntity.mapped_entity_type,
-          createdEntity.id,
-          createdEntity.entity_type,
-          relationData[0]?.relation_type,
-        ],
-      );
-    }
-
-    return createdEntity;
-  }
-
-  async updateEntity(
-    entityData,
-    loggedInUser: UserData,
-    appcode?: string,
-  ): Promise<any> {
-    const taskStatusQuery = `
-    SELECT name 
-    FROM frm_list_master_items 
-    WHERE id = $1 
-      AND organization_id = $2 
-      AND listtype = 'TKST'
-    LIMIT 1;
-  `;
-
-    const statusRows = await this.dataSource.query(taskStatusQuery, [
-      entityData.status,
-      loggedInUser.organization_id,
-    ]);
-
-    const oldStatusQuery = `
-    SELECT * 
-    FROM frm_wf_task_data
-    WHERE id = $1
-      AND organization_id = $2
-    LIMIT 1;
-  `;
-
-    const [oldRow] = await this.dataSource.query(oldStatusQuery, [
-      entityData.id,
-      loggedInUser.organization_id,
-    ]);
-
-    const oldStatusRows = await this.dataSource.query(taskStatusQuery, [
-      oldRow.status,
-      loggedInUser.organization_id,
-    ]);
-
-    const isStatusChanged =
-      statusRows.length > 0 &&
-      oldStatusRows.length > 0 &&
-      statusRows[0].name !== oldStatusRows[0].name;
-
-    const isCompletedStatus =
-      statusRows.length > 0 &&
-      statusRows[0].name?.toLowerCase() === 'completed';
-
-    // Sync flags based on status
-    entityData.is_completed = isCompletedStatus;
-    entityData.is_done = isCompletedStatus;
-
-    const isMandatory =
-      String(entityData.is_mandatory).toLowerCase() === 'true' ||
-      entityData.is_mandatory === true ||
-      entityData.is_mandatory === 1 ||
-      entityData.is_mandatory === '1';
-
-    const updatedEntity = await super.updateEntity(
-      {
-        ...entityData,
-        is_mandatory: isMandatory,
-      },
-      loggedInUser,
-      appcode,
-    );
-
-    // Convert boolean back to "1" / "0" for response
-    if (updatedEntity && typeof updatedEntity['is_mandatory'] === 'boolean') {
-      updatedEntity['is_mandatory'] = updatedEntity['is_mandatory'] ? '1' : '0';
-    }
-
-    try {
-      if (isCompletedStatus && isStatusChanged) {
-        const completedLogData = {
-          mapped_entity_id: updatedEntity.mapped_entity_id,
-          mapped_entity_type: updatedEntity.mapped_entity_type,
-          title: `Task completed`,
-          description: `${updatedEntity.code} was marked as completed`,
-          category: ACTIVITY_CATEGORIES.TASK,
-          action: 'completed',
-          appcode: loggedInUser.appcode,
-        };
-
-        await this.activityLogService.logActivity(
-          completedLogData,
-          loggedInUser,
-        );
-      } else {
-        const editLogData = {
-          mapped_entity_id: updatedEntity.mapped_entity_id,
-          mapped_entity_type: updatedEntity.mapped_entity_type,
-          title: `Task edited`,
-          description: `${updatedEntity.code} was edited`,
-          category: ACTIVITY_CATEGORIES.TASK,
-          action: 'edit',
-          appcode: loggedInUser.appcode,
-        };
-
-        await this.activityLogService.logActivity(editLogData, loggedInUser);
-      }
-    } catch (error) {
-      console.error(
-        'Failed to log activity for task:',
-        error?.message || error,
-      );
-      // Don't block main flow
-    }
-
-    return updatedEntity;
-  }
-
-  async getAllTaskByUserIdandStageId(
-    loggedInUser: UserData,
-    data,
-  ): Promise<any> {
-    const taskData = await this.taskRepository.getAllTaskByUserIdAndStageId({
-      user_id: data.user_id,
-      stage_id: data.stage_id,
-      mapped_entity_type: data.mapped_entity_type,
-      mapped_entity_id: data.mapped_entity_id,
-    });
-
-    const grouped: { mandatory: any[]; non_mandatory: any[] } = {
-      mandatory: [],
-      non_mandatory: [],
-    };
-
-    for (const task of taskData) {
-      if (task && task.is_mandatory) {
-        grouped.mandatory.push(task);
-      } else {
-        grouped.non_mandatory.push(task);
-      }
-    }
-
-    return grouped;
-  }
-
-  async getAllTask(
-    loggedInUser: UserData,
-    data: {
-      mapped_entity_type: string;
-      mapped_entity_id: number;
-      status?: string;
-      mandatory?: boolean;
-      overdue?: boolean;
-    },
-  ): Promise<any[]> {
-    const { mapped_entity_type, mapped_entity_id, status, mandatory, overdue } =
-      data;
-    const taskRepo = this.reflectionHelper.getRepoService('TaskDataEntity');
-    const qb = taskRepo
-      .createQueryBuilder("t")
-      .select("t.*")
-      .leftJoinAndSelect("frm_wf_stage", "s", "t.stage_id::text = s.id::text")
-      .leftJoinAndSelect("frm_wf_stage_group", "sg", "s.stage_group_id = sg.id")
-      .leftJoinAndSelect("frm_wf_action", "a", "t.action_id::text = a.id::text")
-      .where("t.mapped_entity_type = :mapped_entity_type", {
-        mapped_entity_type,
-      })
-      .andWhere("t.mapped_entity_id::text = :mapped_entity_id", {
-        mapped_entity_id: String(mapped_entity_id),
-      });
-
-    // ---------------------------
-    // OPTIONAL FILTERS
-    // ---------------------------
-
-    if (status) {
-      qb.andWhere("t.status::text = :status", { status: String(status) });
-    }
-
-    if (mandatory !== undefined) {
-      qb.andWhere("t.is_mandatory = :mandatory", {
-        mandatory: mandatory ? "1" : "0",
-      });
-    }
-
-    if (overdue) {
-      qb.andWhere(
-        `
-        (
-          t.due_date < CURRENT_DATE
-          OR (t.due_date = CURRENT_DATE AND t.due_time < CURRENT_TIME)
-        )
-      `,
-      ).andWhere("t.is_done::text = :is_done", { is_done: "0" });
-    }
-
-    qb.orderBy("t.created_date", "DESC");
-
-    const taskData = await qb.getRawMany();
-
-    // ------------------------------------------------
-    // PROFILE IMAGE LOGIC (same as your code)
-    // ------------------------------------------------
-
-    if (taskData?.length) {
-      for (const task of taskData) {
-        try {
-          const baseUrl = this.configService.get<string>("REDIRECT_BE_URL");
-          const queryParams = new URLSearchParams({
-            loggedInUser: JSON.stringify(loggedInUser),
-          }).toString();
-
-          const response = await axios.get(
-            `${baseUrl}/users/profile-image-url/${task.task_owner}?entity_type=USR&${queryParams}`,
-            { headers: { "Content-Type": "application/json" } },
-          );
-
-          task.created_by_name = response.data.name;
-          task.task_owner_name = response.data.name;
-          task.task_owner_profile = response.data.profile_image;
-        } catch (err) {
-          console.error("⚠ Error fetching profile:", err.message);
-        }
-      }
-    }
-
-    if (!taskData.length) return [];
-
-    // ------------------------------------------------
-    // STATUS LOOKUP
-    // ------------------------------------------------
-    const listMasterRepo = this.reflectionHelper.getRepoService('ListMasterItems');
-
-    const statuses = await listMasterRepo.find({
-      where: {
-        organization_id: loggedInUser.organization_id,
-        listtype: "TKST",
-      },
-      select: ["id", "name"],
-    });
-
-    const statusMap = new Map(statuses.map((row) => [String(row.id), row.name]));
-
-    return taskData.map((task) => ({
-      ...task,
-      task_status: statusMap.get(String(task.status)) || null,
-      action_name:
-        task.action_id === "0" || task.action_id === 0
-          ? "Generic"
-          : task.action_name,
-      is_mandatory: task.is_mandatory === "1" ? "1" : "0",
-    }));
-  }
-
-  async getEntityData(entityType: string, id: number, loggedInUser: any) {
-    const taskData = await super.getEntityData(entityType, id, loggedInUser);
-
-    if (taskData?.due_date) {
-      taskData.due_date = moment(taskData.due_date)
-        .local() // convert from UTC to local
-        .format('YYYY-MM-DD'); // return just the date string
-    }
-
-    if (taskData?.reminder_date) {
-      taskData.reminder_date = moment(taskData.reminder_date)
-        .local()
-        .format('YYYY-MM-DD');
-    }
-
-    return taskData;
-  }
-
-  async saveActionData(
-    action: any,
-    loggedInUser: UserData,
-    mapped_entity_id: number,
-    mapped_entity_type: string,
-  ): Promise<any> {
-    await this.taskRepository.saveActionDataInTask(
-      action,
-      loggedInUser,
-      mapped_entity_id,
-      mapped_entity_type,
-    );
-  }
-
-  async moveTask(
-    loggedInUser: UserData,
-    body: {
-      mapped_entity_type: string;
-      mapped_entity_id: number;
-      stage_id: number;
-      action_id: number;
-      reason_code?: string | number;
-      remark?: string;
-      stage_group_id?: number;
-    },
-  ): Promise<any> {
-    // Logic to move task based on the provided body parameters
-    // This could involve updating the task's stage, action, etc.
-
-    // update task status
-    this.taskRepository.updateTaskStatus(
-      loggedInUser,
-      body.mapped_entity_type,
-      body.mapped_entity_id,
-      body.stage_id,
-      body.action_id,
-    );
-
-    // update action status
-    await this.actionDataService.updateActionStatus(
-      loggedInUser,
-      body.mapped_entity_type,
-      body.mapped_entity_id,
-      body.stage_id,
-      body.action_id,
-    );
-
-    if (body.reason_code || body.remark) {
-      await this.createSystemNote(
-        {
-          reason_code: body.reason_code!,
-          remark: body.remark || '',
-          mapped_entity_id: body.mapped_entity_id,
-          stage_id: body.stage_id,
-          action_id: body.action_id,
-          stage_group_id: body.stage_group_id,
-        },
-        loggedInUser,
-      );
-    }
-
-    return 'Task moved successfully';
-  }
-
-  async deleteEntity(
-    entity_type: string,
-    taskId: number,
-    loggedInUser: UserData,
-  ): Promise<any> {
-    // Fetch the task before deleting
-    const task: any = await super.getEntityData(
-      entity_type,
-      taskId,
-      loggedInUser,
-    );
-
-    if (!task) {
-      throw new Error('Task not found');
-    }
-
-    // Check if the task is system-generated
-    if (task.is_system) {
-      throw new Error('Cannot delete system-generated tasks');
-    }
-
-    // Perform the deletion
-    await super.deleteEntity(entity_type, taskId, loggedInUser);
-
-    // Try to log the delete activity
-    try {
-      const logData = {
-        mapped_entity_id: task.mapped_entity_id,
-        mapped_entity_type: task.mapped_entity_type,
-        title: `Task deleted`,
-        description: `${task.name} was deleted`,
-        category: ACTIVITY_CATEGORIES.TASK,
-        action: 'delete',
-        appcode: loggedInUser.appcode,
-      };
-
-      await this.activityLogService.logActivity(logData, loggedInUser);
-    } catch (error) {
-      console.error(
-        'Failed to log activity for task deletion:',
-        error?.message || error,
-      );
-    }
-
-    return { message: 'Task deleted successfully' };
-  }
-
-  async createSystemNote(
-    entity: {
-      reason_code: string | number;
-      remark: string;
-      mapped_entity_id: number;
-      stage_id: number;
-      action_id: number;
-      stage_group_id?: number;
-    },
-    loggedInUser: any,
-  ) {
-    if (!entity || !entity.mapped_entity_id || !entity.reason_code) return null;
-
-    const [reason] = await this.dataSource.query(
-      `SELECT name FROM frm_list_master_items WHERE id = $1 AND organization_id = $2`,
-      [entity.reason_code, loggedInUser.organization_id],
-    );
-
-    const notePayload = {
-      note_title: reason ? reason.name : entity.reason_code,
-      note: entity.remark,
-      is_system: true,
-      mapped_entity_type: 'LEAD',
-      entity_type: 'NOTE',
-      mapped_entity_id: entity.mapped_entity_id,
-      stage_id: entity.stage_id,
-      action_id: entity.action_id,
-      stage_group_id: entity.stage_group_id,
-    } as any;
-
-    return await super.createEntity(notePayload, loggedInUser);
-  }
-}
+import { Inject, Injectable } from '@nestjs/common';
+import { EntityServiceImpl } from 'src/module/meta/service/entity-service-impl.service';
+import { UserData } from 'src/module/user/entity/user.entity';
+import { TaskRepository } from '../repository/task.repository';
+import { ActionDataService } from './action-data.service';
+import { DataSource, EntityManager } from 'typeorm';
+import { ActivityLogService } from './activity-log.service';
+import { ACTIVITY_CATEGORIES } from '../repository/activity-log.repository';
+import { MediaDataService } from 'src/module/meta/service/media-data.service';
+import { ActionHandler } from 'src/module/workflow-automation/interface/action.decorator';
+import * as moment from 'moment';
+import { ConfigService } from '@nestjs/config';
+import axios from 'axios';
+import { ChecksumAlgorithm } from '@aws-sdk/client-s3';
+import { ReflectionHelper } from '../../../utils/service/reflection-helper.service';
+
+@Injectable()
+@ActionHandler('add_task')
+export class TaskService extends EntityServiceImpl {
+  constructor(
+    private readonly taskRepository: TaskRepository,
+    private readonly actionDataService: ActionDataService,
+    private readonly dataSource: DataSource,
+    @Inject('ActivityLogService')
+    private readonly activityLogService: ActivityLogService,
+    private readonly mediaDataService: MediaDataService,
+    private readonly configService: ConfigService,
+  ) {
+    super();
+  }
+  name: string = 'add_task';
+
+  async execute(payload: any): Promise<any> {
+    let { entity, user, config } = payload;
+
+    const { organization_id, level_id, level_type } = entity;
+
+    if (entity.entity_type !== 'LEAD') {
+      entity = await this.getEntityData('LEAD', entity.parent_id, user);
+    }
+
+    // 3. Build entityData
+    const entityData = {
+      mapped_entity_id: entity.id,
+      mapped_entity_type: entity.entity_type,
+      task_owner: entity.lead_owner,
+      stage_id: entity.stageId,
+      status: config.status,
+      due_date: Date.now() + 2,
+      due_time: Date.now(),
+      description: config.description,
+      is_mandatory: config.is_mandatory,
+      name: config.name,
+      entity_type: 'TASK',
+    };
+
+    return await this.createEntity(entityData, user);
+  }
+
+  async createEntity(
+    entityData: any,
+    loggedInUser: any,
+    manager?: EntityManager | null,
+    appcode?: string,
+  ): Promise<any> {
+    if (entityData && typeof entityData['is_mandatory'] === 'string') {
+      entityData['is_mandatory'] = entityData['is_mandatory'] === '1';
+    }
+
+    // handling the date string for due_date and reminder_date
+    if (entityData?.due_date) {
+      // Interpret input 'YYYY-MM-DD' as local midnight (IST)
+      entityData.due_date = moment(entityData.due_date, 'YYYY-MM-DD')
+        .startOf('day') // 00:00 local time
+        .toDate(); // Convert to JS Date object for TypeORM
+    }
+
+    if (entityData?.reminder_date) {
+      entityData.reminder_date = moment(entityData.reminder_date, 'YYYY-MM-DD')
+        .startOf('day')
+        .toDate();
+    }
+
+    const createdEntity = await super.createEntity(
+      entityData,
+      loggedInUser,
+      manager,
+      appcode,
+    );
+
+    try {
+      const logData = {
+        mapped_entity_id: createdEntity.mapped_entity_id,
+        mapped_entity_type: createdEntity.mapped_entity_type,
+        title: `Task added`,
+        description: `A new task ${createdEntity.name} was added`,
+        category: ACTIVITY_CATEGORIES.TASK,
+        action: 'add',
+        appcode: loggedInUser.appcode,
+      };
+
+      await this.activityLogService.logActivity(logData, loggedInUser);
+    } catch (error) {
+      console.error(
+        'Failed to log activity for meeting:',
+        error?.message || error,
+      );
+      // Logging should not block main flow
+    }
+
+    let relationData = await this.dataSource.query(
+      `SELECT * FROM frm_entity_relation WHERE source_entity_type = $1  AND target_entity_type = $2`,
+      [createdEntity.mapped_entity_type, createdEntity.entity_type],
+    );
+
+    if (relationData) {
+      await this.dataSource.query(
+        `INSERT INTO frm_entity_relation_data (organization_id, source_entity_id, source_entity_type, target_entity_id, target_entity_type, relation_type) VALUES ($1, $2, $3, $4, $5 , $6)`,
+        [
+          loggedInUser.organization_id,
+          createdEntity.mapped_entity_id,
+          createdEntity.mapped_entity_type,
+          createdEntity.id,
+          createdEntity.entity_type,
+          relationData[0]?.relation_type,
+        ],
+      );
+    }
+
+    return createdEntity;
+  }
+
+  async updateEntity(
+    entityData,
+    loggedInUser: UserData,
+    appcode?: string,
+  ): Promise<any> {
+    const taskStatusQuery = `
+    SELECT name 
+    FROM frm_list_master_items 
+    WHERE id = $1 
+      AND organization_id = $2 
+      AND listtype = 'TKST'
+    LIMIT 1;
+  `;
+
+    const statusRows = await this.dataSource.query(taskStatusQuery, [
+      entityData.status,
+      loggedInUser.organization_id,
+    ]);
+
+    const oldStatusQuery = `
+    SELECT * 
+    FROM frm_wf_task_data
+    WHERE id = $1
+      AND organization_id = $2
+    LIMIT 1;
+  `;
+
+    const [oldRow] = await this.dataSource.query(oldStatusQuery, [
+      entityData.id,
+      loggedInUser.organization_id,
+    ]);
+
+    const oldStatusRows = await this.dataSource.query(taskStatusQuery, [
+      oldRow.status,
+      loggedInUser.organization_id,
+    ]);
+
+    const isStatusChanged =
+      statusRows.length > 0 &&
+      oldStatusRows.length > 0 &&
+      statusRows[0].name !== oldStatusRows[0].name;
+
+    const isCompletedStatus =
+      statusRows.length > 0 &&
+      statusRows[0].name?.toLowerCase() === 'completed';
+
+    // Sync flags based on status
+    entityData.is_completed = isCompletedStatus;
+    entityData.is_done = isCompletedStatus;
+
+    const isMandatory =
+      String(entityData.is_mandatory).toLowerCase() === 'true' ||
+      entityData.is_mandatory === true ||
+      entityData.is_mandatory === 1 ||
+      entityData.is_mandatory === '1';
+
+    const updatedEntity = await super.updateEntity(
+      {
+        ...entityData,
+        is_mandatory: isMandatory,
+      },
+      loggedInUser,
+      appcode,
+    );
+
+    // Convert boolean back to "1" / "0" for response
+    if (updatedEntity && typeof updatedEntity['is_mandatory'] === 'boolean') {
+      updatedEntity['is_mandatory'] = updatedEntity['is_mandatory'] ? '1' : '0';
+    }
+
+    try {
+      if (isCompletedStatus && isStatusChanged) {
+        const completedLogData = {
+          mapped_entity_id: updatedEntity.mapped_entity_id,
+          mapped_entity_type: updatedEntity.mapped_entity_type,
+          title: `Task completed`,
+          description: `${updatedEntity.code} was marked as completed`,
+          category: ACTIVITY_CATEGORIES.TASK,
+          action: 'completed',
+          appcode: loggedInUser.appcode,
+        };
+
+        await this.activityLogService.logActivity(
+          completedLogData,
+          loggedInUser,
+        );
+      } else {
+        const editLogData = {
+          mapped_entity_id: updatedEntity.mapped_entity_id,
+          mapped_entity_type: updatedEntity.mapped_entity_type,
+          title: `Task edited`,
+          description: `${updatedEntity.code} was edited`,
+          category: ACTIVITY_CATEGORIES.TASK,
+          action: 'edit',
+          appcode: loggedInUser.appcode,
+        };
+
+        await this.activityLogService.logActivity(editLogData, loggedInUser);
+      }
+    } catch (error) {
+      console.error(
+        'Failed to log activity for task:',
+        error?.message || error,
+      );
+      // Don't block main flow
+    }
+
+    return updatedEntity;
+  }
+
+  async getAllTaskByUserIdandStageId(
+    loggedInUser: UserData,
+    data,
+  ): Promise<any> {
+    const taskData = await this.taskRepository.getAllTaskByUserIdAndStageId({
+      user_id: data.user_id,
+      stage_id: data.stage_id,
+      mapped_entity_type: data.mapped_entity_type,
+      mapped_entity_id: data.mapped_entity_id,
+    });
+
+    const grouped: { mandatory: any[]; non_mandatory: any[] } = {
+      mandatory: [],
+      non_mandatory: [],
+    };
+
+    for (const task of taskData) {
+      if (task && task.is_mandatory) {
+        grouped.mandatory.push(task);
+      } else {
+        grouped.non_mandatory.push(task);
+      }
+    }
+
+    return grouped;
+  }
+
+  async getAllTask(
+    loggedInUser: UserData,
+    data: {
+      mapped_entity_type: string;
+      mapped_entity_id: number;
+      status?: string;
+      mandatory?: boolean;
+      overdue?: boolean;
+    },
+  ): Promise<any[]> {
+    const { mapped_entity_type, mapped_entity_id, status, mandatory, overdue } =
+      data;
+    const taskRepo = this.reflectionHelper.getRepoService('TaskDataEntity');
+
+    const qb = taskRepo
+      .createQueryBuilder('t')
+      .select('t.*')
+      .addSelect('sg.name', 'stage_group_name')
+      .addSelect('s.name', 'stage_name')
+      .addSelect('a.name', 'action_name')
+      .leftJoinAndSelect('frm_wf_stage', 's', 't.stage_id::text = s.id::text')
+      .leftJoinAndSelect('frm_wf_stage_group', 'sg', 's.stage_group_id = sg.id')
+      .leftJoinAndSelect('frm_wf_action', 'a', 't.action_id::text = a.id::text')
+      .where('t.mapped_entity_type = :mapped_entity_type', {
+        mapped_entity_type,
+      })
+      .andWhere('t.mapped_entity_id::text = :mapped_entity_id', {
+        mapped_entity_id: String(mapped_entity_id),
+      });
+
+    // ---------------------------
+    // OPTIONAL FILTERS
+    // ---------------------------
+
+    if (status) {
+      qb.andWhere('t.status::text = :status', { status: String(status) });
+    }
+
+    if (mandatory !== undefined) {
+      qb.andWhere('t.is_mandatory = :mandatory', {
+        mandatory: mandatory ? '1' : '0',
+      });
+    }
+
+    if (overdue) {
+      qb.andWhere(
+        `
+        (
+          t.due_date < CURRENT_DATE
+          OR (t.due_date = CURRENT_DATE AND t.due_time < CURRENT_TIME)
+        )
+      `,
+      ).andWhere('t.is_done::text = :is_done', { is_done: '0' });
+    }
+
+    qb.orderBy('t.created_date', 'DESC');
+
+    const taskData = await qb.getRawMany();
+
+    // ------------------------------------------------
+    // PROFILE IMAGE LOGIC (same as your code)
+    // ------------------------------------------------
+
+    if (taskData?.length) {
+      for (const task of taskData) {
+        try {
+          const baseUrl = this.configService.get<string>('REDIRECT_BE_URL');
+          const queryParams = new URLSearchParams({
+            loggedInUser: JSON.stringify(loggedInUser),
+          }).toString();
+
+          const response = await axios.get(
+            `${baseUrl}/users/profile-image-url/${task.task_owner}?entity_type=USR&${queryParams}`,
+            { headers: { 'Content-Type': 'application/json' } },
+          );
+
+          task.created_by_name = response.data.name;
+          task.task_owner_name = response.data.name;
+          task.task_owner_profile = response.data.profile_image;
+        } catch (err) {
+          console.error('⚠ Error fetching profile:', err.message);
+        }
+      }
+    }
+
+    if (!taskData.length) return [];
+
+    // ------------------------------------------------
+    // STATUS LOOKUP
+    // ------------------------------------------------
+    const listMasterRepo =
+      this.reflectionHelper.getRepoService('ListMasterItems');
+
+    const statuses = await listMasterRepo.find({
+      where: {
+        organization_id: loggedInUser.organization_id,
+        listtype: 'TKST',
+      },
+      select: ['id', 'name'],
+    });
+
+    const statusMap = new Map(
+      statuses.map((row) => [String(row.id), row.name]),
+    );
+
+    return taskData.map((task) => ({
+      ...task,
+      task_status: statusMap.get(String(task.status)) || null,
+      action_name:
+        task.action_id === '0' || task.action_id === 0
+          ? 'Generic'
+          : task.action_name,
+      is_mandatory: task.is_mandatory === '1' ? '1' : '0',
+    }));
+  }
+
+  async getEntityData(entityType: string, id: number, loggedInUser: any) {
+    const taskData = await super.getEntityData(entityType, id, loggedInUser);
+
+    if (taskData?.due_date) {
+      taskData.due_date = moment(taskData.due_date)
+        .local() // convert from UTC to local
+        .format('YYYY-MM-DD'); // return just the date string
+    }
+
+    if (taskData?.reminder_date) {
+      taskData.reminder_date = moment(taskData.reminder_date)
+        .local()
+        .format('YYYY-MM-DD');
+    }
+
+    return taskData;
+  }
+
+  async saveActionData(
+    action: any,
+    loggedInUser: UserData,
+    mapped_entity_id: number,
+    mapped_entity_type: string,
+  ): Promise<any> {
+    await this.taskRepository.saveActionDataInTask(
+      action,
+      loggedInUser,
+      mapped_entity_id,
+      mapped_entity_type,
+    );
+  }
+
+  async moveTask(
+    loggedInUser: UserData,
+    body: {
+      mapped_entity_type: string;
+      mapped_entity_id: number;
+      stage_id: number;
+      action_id: number;
+      reason_code?: string | number;
+      remark?: string;
+      stage_group_id?: number;
+    },
+  ): Promise<any> {
+    // Logic to move task based on the provided body parameters
+    // This could involve updating the task's stage, action, etc.
+
+    // update task status
+    this.taskRepository.updateTaskStatus(
+      loggedInUser,
+      body.mapped_entity_type,
+      body.mapped_entity_id,
+      body.stage_id,
+      body.action_id,
+    );
+
+    // update action status
+    await this.actionDataService.updateActionStatus(
+      loggedInUser,
+      body.mapped_entity_type,
+      body.mapped_entity_id,
+      body.stage_id,
+      body.action_id,
+    );
+
+    if (body.reason_code || body.remark) {
+      await this.createSystemNote(
+        {
+          reason_code: body.reason_code!,
+          remark: body.remark || '',
+          mapped_entity_id: body.mapped_entity_id,
+          stage_id: body.stage_id,
+          action_id: body.action_id,
+          stage_group_id: body.stage_group_id,
+        },
+        loggedInUser,
+      );
+    }
+
+    return 'Task moved successfully';
+  }
+
+  async deleteEntity(
+    entity_type: string,
+    taskId: number,
+    loggedInUser: UserData,
+  ): Promise<any> {
+    // Fetch the task before deleting
+    const task: any = await super.getEntityData(
+      entity_type,
+      taskId,
+      loggedInUser,
+    );
+
+    if (!task) {
+      throw new Error('Task not found');
+    }
+
+    // Check if the task is system-generated
+    if (task.is_system) {
+      throw new Error('Cannot delete system-generated tasks');
+    }
+
+    // Perform the deletion
+    await super.deleteEntity(entity_type, taskId, loggedInUser);
+
+    // Try to log the delete activity
+    try {
+      const logData = {
+        mapped_entity_id: task.mapped_entity_id,
+        mapped_entity_type: task.mapped_entity_type,
+        title: `Task deleted`,
+        description: `${task.name} was deleted`,
+        category: ACTIVITY_CATEGORIES.TASK,
+        action: 'delete',
+        appcode: loggedInUser.appcode,
+      };
+
+      await this.activityLogService.logActivity(logData, loggedInUser);
+    } catch (error) {
+      console.error(
+        'Failed to log activity for task deletion:',
+        error?.message || error,
+      );
+    }
+
+    return { message: 'Task deleted successfully' };
+  }
+
+  async createSystemNote(
+    entity: {
+      reason_code: string | number;
+      remark: string;
+      mapped_entity_id: number;
+      stage_id: number;
+      action_id: number;
+      stage_group_id?: number;
+    },
+    loggedInUser: any,
+  ) {
+    if (!entity || !entity.mapped_entity_id || !entity.reason_code) return null;
+
+    const [reason] = await this.dataSource.query(
+      `SELECT name FROM frm_list_master_items WHERE id = $1 AND organization_id = $2`,
+      [entity.reason_code, loggedInUser.organization_id],
+    );
+
+    const notePayload = {
+      note_title: reason ? reason.name : entity.reason_code,
+      note: entity.remark,
+      is_system: true,
+      mapped_entity_type: 'LEAD',
+      entity_type: 'NOTE',
+      mapped_entity_id: entity.mapped_entity_id,
+      stage_id: entity.stage_id,
+      action_id: entity.action_id,
+      stage_group_id: entity.stage_group_id,
+    } as any;
+
+    return await super.createEntity(notePayload, loggedInUser);
+  }
+}