From 462edd4e13050d8ca31f290ed0acbcf1f03fb275 Mon Sep 17 00:00:00 2001
From: Pranav C <pranavxc@gmail.com>
Date: Thu, 20 Oct 2022 11:50:26 +0530
Subject: [PATCH] feat(gui): implement user delete method

Signed-off-by: Pranav C <pranavxc@gmail.com>
---
 packages/nc-gui/components/org-user/index.vue |   7 +-
 .../nocodb/src/lib/meta/api/orgUserApis.ts    | 139 +++++++++++-------
 packages/nocodb/src/lib/models/ProjectUser.ts |   6 +
 packages/nocodb/src/lib/models/SyncSource.ts  |   9 ++
 packages/nocodb/src/lib/models/User.ts        |   9 +-
 5 files changed, 106 insertions(+), 64 deletions(-)

diff --git a/packages/nc-gui/components/org-user/index.vue b/packages/nc-gui/components/org-user/index.vue
index ef24e8827f..38cfd8d6d8 100644
--- a/packages/nc-gui/components/org-user/index.vue
+++ b/packages/nc-gui/components/org-user/index.vue
@@ -8,7 +8,7 @@ const { api, isLoading } = useApi()
 
 let users = $ref<UserType[]>([])
 
-const currentPage = $ref(1)
+let currentPage = $ref(1)
 
 const currentLimit = $ref(10)
 
@@ -21,6 +21,7 @@ const pagination = reactive({
   pageSize: 10,
 })
 const loadUsers = async (page = currentPage, limit = currentLimit) => {
+  currentPage = page
   try {
     const response: any = await api.orgUsers.list({
       query: {
@@ -55,6 +56,8 @@ const updateRole = async (userId: string, roles: Role) => {
 const deleteUser = async (userId: string) => {
   Modal.confirm({
     title: 'Are you sure you want to delete this user?',
+    type:"warn",
+    content: 'On deleting, user will remove from from organization and any sync source(Airtable) created by user will get removed',
     onOk: async () => {
       try {
         await api.orgUsers.delete(userId)
@@ -94,7 +97,7 @@ const deleteUser = async (userId: string) => {
         :data-source="users"
         :pagination="pagination"
         :loading="isLoading"
-        @change="loadUsers"
+        @change="loadUsers($event.current)"
       >
         <template #emptyText>
           <a-empty :image="Empty.PRESENTED_IMAGE_SIMPLE" :description="$t('labels.noData')" />
diff --git a/packages/nocodb/src/lib/meta/api/orgUserApis.ts b/packages/nocodb/src/lib/meta/api/orgUserApis.ts
index dda8d7c95d..a373fbc338 100644
--- a/packages/nocodb/src/lib/meta/api/orgUserApis.ts
+++ b/packages/nocodb/src/lib/meta/api/orgUserApis.ts
@@ -1,47 +1,78 @@
-import { Router } from 'express';
-import { v4 as uuidv4 } from 'uuid';
-import validator from 'validator';
-import { OrgUserRoles } from '../../../enums/OrgUserRoles';
-import Audit from '../../models/Audit';
-import User from '../../models/User';
-import { metaApiMetrics } from '../helpers/apiMetrics';
-import { NcError } from '../helpers/catchError';
-import { extractProps } from '../helpers/extractProps';
-import ncMetaAclMw from '../helpers/ncMetaAclMw';
-import { PagedResponseImpl } from '../helpers/PagedResponse';
-import { randomTokenString } from '../helpers/stringHelpers';
-import { Tele } from 'nc-help';
-import { sendInviteEmail } from './projectUserApis';
+import { Router } from 'express'
+import { v4 as uuidv4 } from 'uuid'
+import validator from 'validator'
+import { OrgUserRoles } from '../../../enums/OrgUserRoles'
+import Audit from '../../models/Audit'
+import ProjectUser from '../../models/ProjectUser'
+import SyncSource from '../../models/SyncSource'
+import User from '../../models/User'
+import Noco from '../../Noco'
+import { metaApiMetrics } from '../helpers/apiMetrics'
+import { NcError } from '../helpers/catchError'
+import { extractProps } from '../helpers/extractProps'
+import ncMetaAclMw from '../helpers/ncMetaAclMw'
+import { PagedResponseImpl } from '../helpers/PagedResponse'
+import { randomTokenString } from '../helpers/stringHelpers'
+import { Tele } from 'nc-help'
+import { sendInviteEmail } from './projectUserApis'
 
 async function userList(req, res) {
   res.json(
     new PagedResponseImpl(await User.list(req.query), {
       ...req.query,
       count: await User.count(req.query),
-    })
-  );
+    }),
+  )
 }
 
 async function userUpdate(req, res) {
-  const updateBody = extractProps(req.body, ['roles']);
+  const updateBody = extractProps(req.body, ['roles'])
 
-  const user = await User.get(req.params.userId);
+  const user = await User.get(req.params.userId)
 
   if (user.roles.includes(OrgUserRoles.SUPER)) {
-    NcError.badRequest('Cannot update super admin roles');
+    NcError.badRequest('Cannot update super admin roles')
   }
 
-  res.json(await User.update(req.params.userId, updateBody));
+  res.json(await User.update(req.params.userId, updateBody))
 }
 
 async function userDelete(req, res) {
-  const user = await User.get(req.params.userId);
+  const ncMeta = await Noco.ncMeta.startTransaction()
+  try {
+    const user = await User.get(req.params.userId, ncMeta)
 
-  if (user.roles.includes(OrgUserRoles.SUPER)) {
-    NcError.badRequest('Cannot delete super admin');
+    if (user.roles.includes(OrgUserRoles.SUPER)) {
+      NcError.badRequest('Cannot delete super admin')
+    }
+
+    // delete project user entry and assign to super admin
+    const projectUsers = await ProjectUser.getProjectsList(
+      req.params.userId,
+      ncMeta,
+    )
+
+    // TODO: assign super admin as project owner
+    for (const projectUser of projectUsers) {
+      await ProjectUser.delete(
+        projectUser.project_id,
+        projectUser.fk_user_id,
+        ncMeta
+      );
+    }
+
+    // delete sync source entry
+    await SyncSource.deleteByUserId(req.params.userId, ncMeta)
+
+    // delete user
+    await User.delete(req.params.userId, ncMeta)
+
+    await ncMeta.commit()
+  } catch (e) {
+    await ncMeta.rollback(e)
   }
 
-  res.json(await User.delete(req.params.userId));
+  res.json(await User.delete(req.params.userId))
 }
 
 async function userAdd(req, res, next) {
@@ -50,34 +81,34 @@ async function userAdd(req, res, next) {
     req.body.roles &&
     ![OrgUserRoles.VIEWER, OrgUserRoles.CREATOR].includes(req.body.roles)
   ) {
-    NcError.badRequest('Invalid role');
+    NcError.badRequest('Invalid role')
   }
 
   // extract emails from request body
   const emails = (req.body.email || '')
     .toLowerCase()
     .split(/\s*,\s*/)
-    .map((v) => v.trim());
+    .map((v) => v.trim())
 
   // check for invalid emails
-  const invalidEmails = emails.filter((v) => !validator.isEmail(v));
+  const invalidEmails = emails.filter((v) => !validator.isEmail(v))
 
   if (!emails.length) {
-    return NcError.badRequest('Invalid email address');
+    return NcError.badRequest('Invalid email address')
   }
   if (invalidEmails.length) {
-    NcError.badRequest('Invalid email address : ' + invalidEmails.join(', '));
+    NcError.badRequest('Invalid email address : ' + invalidEmails.join(', '))
   }
 
-  const invite_token = uuidv4();
-  const error = [];
+  const invite_token = uuidv4()
+  const error = []
 
   for (const email of emails) {
     // add user to project if user already exist
-    const user = await User.getByEmail(email);
+    const user = await User.getByEmail(email)
 
     if (user) {
-      NcError.badRequest('User already exist');
+      NcError.badRequest('User already exist')
     } else {
       try {
         // create new user with invite token
@@ -87,10 +118,10 @@ async function userAdd(req, res, next) {
           email,
           roles: OrgUserRoles.VIEWER,
           token_version: randomTokenString(),
-        });
+        })
 
-        const count = await User.count();
-        Tele.emit('evt', { evt_type: 'org:user:invite', count });
+        const count = await User.count()
+        Tele.emit('evt', { evt_type: 'org:user:invite', count })
 
         await Audit.insert({
           project_id: req.params.projectId,
@@ -99,23 +130,23 @@ async function userAdd(req, res, next) {
           user: req.user.email,
           description: `invited ${email} to ${req.params.projectId} project `,
           ip: req.clientIp,
-        });
+        })
         // in case of single user check for smtp failure
         // and send back token if failed
         if (
           emails.length === 1 &&
           !(await sendInviteEmail(email, invite_token, req))
         ) {
-          return res.json({ invite_token, email });
+          return res.json({ invite_token, email })
         } else {
-          sendInviteEmail(email, invite_token, req);
+          sendInviteEmail(email, invite_token, req)
         }
       } catch (e) {
-        console.log(e);
+        console.log(e)
         if (emails.length === 1) {
-          return next(e);
+          return next(e)
         } else {
-          error.push({ email, error: e.message });
+          error.push({ email, error: e.message })
         }
       }
     }
@@ -124,31 +155,31 @@ async function userAdd(req, res, next) {
   if (emails.length === 1) {
     res.json({
       msg: 'success',
-    });
+    })
   } else {
-    return res.json({ invite_token, emails, error });
+    return res.json({ invite_token, emails, error })
   }
 }
 
-const router = Router({ mergeParams: true });
+const router = Router({ mergeParams: true })
 router.get(
   '/api/v1/users',
   metaApiMetrics,
-  ncMetaAclMw(userList, 'userList', [OrgUserRoles.SUPER])
-);
+  ncMetaAclMw(userList, 'userList', [OrgUserRoles.SUPER]),
+)
 router.patch(
   '/api/v1/users/:userId',
   metaApiMetrics,
-  ncMetaAclMw(userUpdate, 'userUpdate', [OrgUserRoles.SUPER])
-);
+  ncMetaAclMw(userUpdate, 'userUpdate', [OrgUserRoles.SUPER]),
+)
 router.delete(
   '/api/v1/users/:userId',
   metaApiMetrics,
-  ncMetaAclMw(userDelete, 'userAdd', [OrgUserRoles.SUPER])
-);
+  ncMetaAclMw(userDelete, 'userAdd', [OrgUserRoles.SUPER]),
+)
 router.post(
   '/api/v1/users',
   metaApiMetrics,
-  ncMetaAclMw(userAdd, 'userDelete', [OrgUserRoles.SUPER])
-);
-export default router;
+  ncMetaAclMw(userAdd, 'userDelete', [OrgUserRoles.SUPER]),
+)
+export default router
diff --git a/packages/nocodb/src/lib/models/ProjectUser.ts b/packages/nocodb/src/lib/models/ProjectUser.ts
index 219e9140ac..353951fdfd 100644
--- a/packages/nocodb/src/lib/models/ProjectUser.ts
+++ b/packages/nocodb/src/lib/models/ProjectUser.ts
@@ -183,4 +183,10 @@ export default class ProjectUser {
       project_id: projectId,
     });
   }
+
+  static async getProjectsList(userId:string, ncMeta = Noco.ncMeta): Promise<ProjectUser[]> {
+    return await ncMeta.metaList2(null, null, MetaTable.PROJECT_USERS, {
+      condition: { fk_user_id: userId },
+    });
+  }
 }
diff --git a/packages/nocodb/src/lib/models/SyncSource.ts b/packages/nocodb/src/lib/models/SyncSource.ts
index 3c14aa81d2..39f7bf7489 100644
--- a/packages/nocodb/src/lib/models/SyncSource.ts
+++ b/packages/nocodb/src/lib/models/SyncSource.ts
@@ -1,3 +1,4 @@
+import { NcError } from '../meta/helpers/catchError';
 import Noco from '../Noco';
 import { MetaTable } from '../utils/globals';
 import { extractProps } from '../meta/helpers/extractProps';
@@ -132,4 +133,12 @@ export default class SyncSource {
       syncSourceId
     );
   }
+
+  static async deleteByUserId(userId: string, ncMeta = Noco.ncMeta) {
+    if (!userId) NcError.badRequest('User Id is required');
+
+    return await ncMeta.metaDelete(null, null, MetaTable.SYNC_SOURCE, {
+      fk_user_id: userId,
+    });
+  }
 }
diff --git a/packages/nocodb/src/lib/models/User.ts b/packages/nocodb/src/lib/models/User.ts
index b839cb02bd..e408243f90 100644
--- a/packages/nocodb/src/lib/models/User.ts
+++ b/packages/nocodb/src/lib/models/User.ts
@@ -4,19 +4,12 @@ import { CacheGetType, CacheScope, MetaTable } from '../utils/globals';
 import Noco from '../Noco';
 import { extractProps } from '../meta/helpers/extractProps';
 import NocoCache from '../cache/NocoCache';
-import { NcError } from '../meta/helpers/catchError';
-import { UserType } from 'nocodb-sdk';
-import { NcError } from '../meta/helpers/catchError';
-import { CacheGetType, CacheScope, MetaTable } from '../utils/globals';
-import Noco from '../Noco';
-import { extractProps } from '../meta/helpers/extractProps';
-import NocoCache from '../cache/NocoCache';
 
 export default class User implements UserType {
   id: string;
 
   /** @format email */
-  email: string
+  email: string;
 
   password?: string;
   salt?: string;