From f07a4133e5564ffd9200752eb6736f0ec1f2d7e0 Mon Sep 17 00:00:00 2001
From: Kevin <kevin.c.souza@gmail.com>
Date: Mon, 16 Jun 2025 16:44:47 -0300
Subject: [PATCH] resolve okr exclusion error

---
 .../modules/key-result/key-result.provider.ts | 56 ++++++++++++++++++-
 1 file changed, 55 insertions(+), 1 deletion(-)

diff --git a/src/core/modules/key-result/key-result.provider.ts b/src/core/modules/key-result/key-result.provider.ts
index 835d3dfa..8c5995eb 100644
--- a/src/core/modules/key-result/key-result.provider.ts
+++ b/src/core/modules/key-result/key-result.provider.ts
@@ -1,6 +1,6 @@
 import { Injectable } from '@nestjs/common'
 import { uniqBy, pickBy, omitBy, identity, isEmpty, maxBy, flatten, keyBy, uniq } from 'lodash'
-import { Any, Brackets, DeleteResult, FindConditions, In, Raw } from 'typeorm'
+import { Any, Brackets, DeleteResult, EntityManager, FindConditions, In, Raw } from 'typeorm'
 
 import { ConfidenceTagAdapter } from '@adapters/confidence-tag/confidence-tag.adapters'
 import { CONFIDENCE_TAG_THRESHOLDS } from '@adapters/confidence-tag/confidence-tag.constants'
@@ -65,6 +65,7 @@ export class KeyResultProvider extends CoreEntityProvider<KeyResult, KeyResultIn
     public readonly timeline: KeyResultTimelineProvider,
     protected readonly repository: KeyResultRepository,
     private readonly analyticsProvider: AnalyticsProvider,
+    private readonly entityManager: EntityManager,
   ) {
     super(KeyResultProvider.name, repository)
   }
@@ -694,6 +695,33 @@ export class KeyResultProvider extends CoreEntityProvider<KeyResult, KeyResultIn
       keyResultId: id,
     })
     const updates = await this.keyResultUpdateProvider.delete({ keyResultId: id })
+
+    // Task is a new table that will not be implemented in this backend
+    // Task exclusion will be manual for now
+    await this.entityManager.query(
+      `
+      DELETE FROM task_comment
+      WHERE task_id IN (SELECT id FROM task WHERE key_result_id = $1) 
+    `,
+      [id],
+    )
+
+    await this.entityManager.query(
+      `
+      DELETE FROM task_history
+      WHERE task_id IN (SELECT id FROM task WHERE key_result_id = $1)
+      `,
+      [id],
+    )
+
+    await this.entityManager.query(
+      `
+      DELETE FROM task
+      WHERE key_result_id = $1
+      `,
+      [id],
+    )
+
     const keyResult = await this.delete({ id })
 
     return {
@@ -713,6 +741,32 @@ export class KeyResultProvider extends CoreEntityProvider<KeyResult, KeyResultIn
     const checkMarks = await this.keyResultCheckMarkProvider.deleteFromObjective(objectiveId)
     const updates = await this.keyResultUpdateProvider.deleteFromObjective(objectiveId)
 
+    // Task is a new table that will not be implemented in this backend
+    // Task exclusion will be manual for now
+    await this.entityManager.query(
+      `
+      DELETE FROM task_comment
+      WHERE task_id IN (SELECT task.id FROM task JOIN key_result kr ON kr.id = task.key_result_id WHERE kr.objective_id = $1)
+    `,
+      [objectiveId],
+    )
+
+    await this.entityManager.query(
+      `
+      DELETE FROM task_history
+      WHERE task_id IN (SELECT task.id FROM task JOIN key_result kr ON kr.id = task.key_result_id WHERE kr.objective_id = $1)
+      `,
+      [objectiveId],
+    )
+
+    await this.entityManager.query(
+      `
+      DELETE FROM task
+      WHERE key_result_id IN (SELECT id FROM key_result WHERE objective_id = $1 )
+      `,
+      [objectiveId],
+    )
+
     const keyResults = await this.delete({
       objectiveId,
     })