From ce95ad845cdb36c305f645c41f94485f44767c3e Mon Sep 17 00:00:00 2001
From: Chenjianyu <chenjianyu@@easycorp.ltd>
Date: Mon, 24 Jul 2023 16:25:50 +0800
Subject: [PATCH] * add get product and project object data method.

---
 module/ai/model.php | 26 +++++++++++++++++---------
 1 file changed, 17 insertions(+), 9 deletions(-)

diff --git a/module/ai/model.php b/module/ai/model.php
index 7a6378481c..96bc2bdb9b 100644
--- a/module/ai/model.php
+++ b/module/ai/model.php
@@ -671,15 +671,23 @@ class aiModel extends model
         $object     = new stdclass();
         $objectData = new stdclass();
 
-        /* Query necessary object data from zentao. TODO: add more later. */
-        if($module == 'story')
-        {
-            if(isset($sourceGroups['story'])) $object->story = $this->loadModel('story')->getById($objectId);
-        }
-        if($module == 'execution')
-        {
-            if(isset($sourceGroups['execution'])) $object->execution = $this->loadModel('execution')->getByID($objectId);
-            if(isset($sourceGroups['tasks']))     $object->tasks     = $this->loadModel('task')->getExecutionTasks($objectId);
+        /* Query necessary object data from zentao. */
+        switch($module){
+            case 'story':
+                if(isset($sourceGroups['story'])) $object->story = $this->loadModel('story')->getById($objectId);
+                break;
+            case 'execution':
+                if(isset($sourceGroups['execution'])) $object->execution = $this->loadModel('execution')->getByID($objectId);
+                if(isset($sourceGroups['tasks']))     $object->tasks     = $this->loadModel('task')->getExecutionTasks($objectId);
+                break;
+            case 'product':
+                if(isset($sourceGroups['product'])) $object->product = $this->loadModel('product')->getById($objectId);
+                break;
+            case 'project':
+                if(isset($sourceGroups['project'])) $object->project = $this->loadModel('project')->getById($objectId);
+                break;
+            case 'my':
+                /* TODO: add more later. */
         }
         if($module == 'doc')
         {