From 167e89b464e697a03ab963e5c0a93d2298cbdc5a Mon Sep 17 00:00:00 2001
From: Akhil Meka <akhil.reddy@qualyval.com>
Date: Fri, 14 Feb 2025 13:13:11 +0530
Subject: [PATCH] fix task model

---
 src/task/task.service.ts | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/task/task.service.ts b/src/task/task.service.ts
index 0e9a03d..c93045c 100644
--- a/src/task/task.service.ts
+++ b/src/task/task.service.ts
@@ -1,6 +1,7 @@
 import { AuthenticatedUser } from "../auth";
 import { getFilterObject, getSortObject, PageQueryParams } from "../pagination";
 import { generateId } from "../utils/id";
+import { taskPipeline } from "../utils/pipeline";
 import {
   CreateTaskInput,
   taskFields,
@@ -13,6 +14,13 @@ export async function createTask(
   input: CreateTaskInput,
   user: AuthenticatedUser
 ) {
+  if (!input.stage) {
+    input.stage = {
+      pipeline: taskPipeline,
+      currentStage: 0,
+    };
+  }
+
   if (!input.files) {
     return await taskModel.create({
       ...input,
@@ -90,6 +98,8 @@ export async function listTasks(params: PageQueryParams, tenantId: string) {
         pid: 1,
         title: 1,
         dueDate: 1,
+        documents: 1,
+        stage: 1,
         createdAt: 1,
         createdBy: {
           $let: {
@@ -181,6 +191,8 @@ export async function searchTasks(params: PageQueryParams, tenantId: string) {
         pid: 1,
         title: 1,
         dueDate: 1,
+        documents: 1,
+        stage: 1,
         createdAt: 1,
         createdBy: {
           $let: {
@@ -246,4 +258,4 @@ export async function newUpload(
       },
     }
   );
-}
\ No newline at end of file
+}