Whitelist ready

src/app/api/idnot/UserController.ts

@@ -5,11 +5,14 @@ import { Service } from "typedi";
 import AuthService, { IUserJwtPayload } from "@Services/common/AuthService/AuthService";
 
 import IdNotService from "@Services/common/IdNotService/IdNotService";
+import WhitelistService from "@Services/common/WhitelistService/WhitelistService";
+import User from "le-coffre-resources/dist/SuperAdmin";
+import UsersService from "@Services/super-admin/UsersService/UsersService";
 
 @Controller()
 @Service()
 export default class UserController extends ApiController {
-	constructor(private authService: AuthService, private idNotService: IdNotService) {
+	constructor(private authService: AuthService, private idNotService: IdNotService, private whitelistService: WhitelistService, private userService: UsersService) {
 		super();
 	}
 
@@ -25,6 +28,7 @@ export default class UserController extends ApiController {
 			if (!code) throw new Error("code is required");						
 
 			const idNotToken = await this.idNotService.getIdNotToken(code);
+			
 			if(!idNotToken) {
 				this.httpValidationError(response, "IdNot token undefined");
 				return;
@@ -35,6 +39,27 @@ export default class UserController extends ApiController {
 				this.httpUnauthorized(response);
 				return;
 			}
+
+			//Whitelist feature
+			//Get user with contact
+			const prismaUser = await this.userService.getByUid(user.uid, {contact: true });
+			if (!prismaUser) {
+				this.httpNotFoundRequest(response, "user not found");
+				return;
+			}
+
+			//Hydrate user to be able to use his contact
+			const userHydrated = User.hydrate<User>(prismaUser, { strategy: "excludeAll" });
+
+			//Check if user is whitelisted
+			const isWhitelisted = await this.whitelistService.getByEmail(userHydrated.contact!.email);
+
+			//If not whitelisted, return 409 Not whitelisted
+			if (!isWhitelisted) {
+				this.httpNotWhitelisted(response);
+				return;
+			}
+
 			await this.idNotService.updateUser(user.uid);			
 			await this.idNotService.updateOffice(user.office_uid);
 

src/common/databases/migrations/20231220084359_whitelist/migration.sql

@@ -0,0 +1,16 @@
+-- CreateTable
+CREATE TABLE "whitelist" (
+    "uid" TEXT NOT NULL,
+    "email" VARCHAR(255) NOT NULL,
+    "active" BOOLEAN NOT NULL DEFAULT true,
+    "created_at" TIMESTAMP(3) DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" TIMESTAMP(3),
+
+    CONSTRAINT "whitelist_pkey" PRIMARY KEY ("uid")
+);
+
+-- CreateIndex
+CREATE UNIQUE INDEX "whitelist_uid_key" ON "whitelist"("uid");
+
+-- CreateIndex
+CREATE UNIQUE INDEX "whitelist_email_key" ON "whitelist"("email");

src/common/databases/schema.prisma

@@ -72,6 +72,15 @@ model Users {
   @@map("users")
 }
 
+model Whitelist {
+  uid        String    @id @unique @default(uuid())
+  email      String    @unique @db.VarChar(255)
+  active     Boolean   @default(true)
+  created_at DateTime? @default(now())
+  updated_at DateTime? @updatedAt
+  @@map("whitelist")
+}
+
 model Offices {
   uid            String          @id @unique @default(uuid())
   idNot          String          @unique @db.VarChar(255)

src/common/repositories/WhitelistRepository.ts

@@ -0,0 +1,39 @@
+import Database from "@Common/databases/database";
+import BaseRepository from "@Repositories/BaseRepository";
+import { Service } from "typedi";
+import { Prisma } from "prisma/prisma-client";
+
+@Service()
+export default class WhitelistRepository extends BaseRepository {
+	constructor(private database: Database) {
+		super();
+	}
+	protected get model() {
+		return this.database.getClient().whitelist;
+	}
+	protected get instanceDb() {
+		return this.database.getClient();
+	}
+
+	/**
+	 * @description : Find many whitelist
+	 */
+	public async findMany(query: Prisma.WhitelistFindManyArgs) {
+		query.take = Math.min(query.take || this.defaultFetchRows, this.maxFetchRows);
+		return this.model.findMany(query);
+	}
+
+    /**
+	 * @description : find unique by email
+	 */
+	public async findOneByEmail(email: string) {
+		return this.model.findUnique({
+			where: {
+				email: email,
+			},
+		});
+	}
+
+
+
+}

src/common/system/controller-pattern/BaseController.ts

@@ -48,6 +48,10 @@ export default abstract class BaseController {
 		return this.httpResponse(response, HttpCodes.FORBIDDEN, responseData);
 	}
 
+	protected httpNotWhitelisted(response: Response, responseData: IResponseData = "Not whitelisted") {
+		return this.httpResponse(response, HttpCodes.VALIDATION_ERROR, responseData);
+	}
+
 	protected httpResponse(response: Response, httpCode: HttpCodes, responseData: IResponseData = {}) {
 		if (responseData instanceof Error) {
 			throw responseData;

src/services/common/WhitelistService/WhitelistService.ts

@@ -0,0 +1,14 @@
+import WhitelistRepository from "@Repositories/WhitelistRepository";
+import BaseService from "@Services/BaseService";
+import { Service } from "typedi";
+
+@Service()
+export default class WhitelistService extends BaseService {
+	constructor(private whitelistRepository: WhitelistRepository) {
+		super();
+	}
+
+	public async getByEmail(email: string): Promise<any> { 
+		return this.whitelistRepository.findOneByEmail(email);
+	}
+}