From b14583ac9d946342e1f1bc451c824661ddee97b2 Mon Sep 17 00:00:00 2001
From: Mike Keen <mwk@mikekeen.com>
Date: Tue, 11 Jan 2022 15:48:49 -0500
Subject: [PATCH] feat: orgId filtering in units & projects

---
 src/controllers/project.controller.js | 8 ++++++--
 src/controllers/units.controller.js   | 6 ++++--
 src/routes/v1/resources/projects.js   | 1 +
 src/routes/v1/resources/units.js      | 1 +
 4 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/src/controllers/project.controller.js b/src/controllers/project.controller.js
index 96823853..7ed90acf 100644
--- a/src/controllers/project.controller.js
+++ b/src/controllers/project.controller.js
@@ -55,8 +55,9 @@ export const create = async (req, res) => {
 };
 
 export const findAll = async (req, res) => {
-  let { page, limit, search, orgUid, columns, useMock } = req.query;
-
+  let { page, limit, search, orgUid, columns } = req.query;
+  let where = orgUid ? { orgUid } : undefined;
+  
   const includes = [
     ProjectLocation,
     Qualification,
@@ -98,8 +99,11 @@ export const findAll = async (req, res) => {
       ...paginationParams(page, limit),
     };
 
+    console.log(query)
+    
     results = await Project.findAndCountAll({
       distinct: true,
+      where,
       ...query,
     });
   }
diff --git a/src/controllers/units.controller.js b/src/controllers/units.controller.js
index 1ebc7892..d4f303e4 100644
--- a/src/controllers/units.controller.js
+++ b/src/controllers/units.controller.js
@@ -56,8 +56,9 @@ export const create = async (req, res, next) => {
 };
 
 export const findAll = async (req, res) => {
-  let { page, limit, columns } = req.query;
-
+  let { page, limit, columns, orgUid } = req.query;
+  let where = orgUid ? { orgUid } : undefined;
+  
   const includes = [Qualification];
 
   if (columns) {
@@ -84,6 +85,7 @@ export const findAll = async (req, res) => {
         ...paginationParams(page, limit),
       }),
       page,
+      where,
       limit,
     ),
   );
diff --git a/src/routes/v1/resources/projects.js b/src/routes/v1/resources/projects.js
index 4bc1df57..a3d3018b 100644
--- a/src/routes/v1/resources/projects.js
+++ b/src/routes/v1/resources/projects.js
@@ -14,6 +14,7 @@ const querySchema = Joi.object()
     limit: Joi.number(),
     search: Joi.string(),
     columns: Joi.array().items(Joi.string()).single(),
+    orgUid: Joi.string(),
   })
   .with('page', 'limit');
 
diff --git a/src/routes/v1/resources/units.js b/src/routes/v1/resources/units.js
index 924e0e11..30391eb9 100644
--- a/src/routes/v1/resources/units.js
+++ b/src/routes/v1/resources/units.js
@@ -15,6 +15,7 @@ const querySchema = Joi.object()
     search: Joi.string(),
     warehouseUnitId: Joi.string(),
     columns: Joi.array().items(Joi.string()).single(),
+    orgUid: Joi.string(),
   })
   .with('page', 'limit');