From 2de09b038dad056f0262593be4292529b2c59d62 Mon Sep 17 00:00:00 2001
From: Ryan Smith <ryan.smith1@bp.com>
Date: Mon, 26 Feb 2024 15:16:09 +0000
Subject: [PATCH] feat: remove auth to test iam connection

---
 src/handlers/privateResolver.ts | 15 ++++++++++-----
 src/handlers/publicResolver.ts  | 32 +++++++++++---------------------
 2 files changed, 21 insertions(+), 26 deletions(-)

diff --git a/src/handlers/privateResolver.ts b/src/handlers/privateResolver.ts
index d5becd7..356f26b 100644
--- a/src/handlers/privateResolver.ts
+++ b/src/handlers/privateResolver.ts
@@ -1,9 +1,14 @@
 import { AppSyncResolverHandler } from 'aws-lambda';
-import { MutationInsertSignUpDataArgs } from '../../appsync';
+import { MutationInsertSignUpDataArgs, SignUpData } from '../../appsync';
 import { logger } from '../shared/logger';
-import { SignUpDataRepository } from '../repository/signUpDataRepository';
+import { MongoClient } from 'mongodb';
 
-const signUpDataRepository = SignUpDataRepository.getInstance();
+const client = new MongoClient(
+  process?.env?.MONGODB_CONNECTION_STRING ?? 'mongodb://localhost:27017/'
+);
+
+const db = client.db('D2E');
+const collection = db.collection<SignUpData>('SignUps');
 
 export const handler: AppSyncResolverHandler<MutationInsertSignUpDataArgs, boolean> = async (
   event,
@@ -17,9 +22,9 @@ export const handler: AppSyncResolverHandler<MutationInsertSignUpDataArgs, boole
   logger.info(`${JSON.stringify(params)}`);
 
   switch (info.fieldName) {
-    case 'insertSignUpDataPrivate': {
+    case 'testPrivate': {
       const { id, email, type } = params;
-      const res = await signUpDataRepository.insert({ id, email, type });
+      const res = (await collection.insertOne({ id, email, type })).acknowledged;
       callback(null, res);
       break;
     }
diff --git a/src/handlers/publicResolver.ts b/src/handlers/publicResolver.ts
index 1a06e74..ac8f689 100644
--- a/src/handlers/publicResolver.ts
+++ b/src/handlers/publicResolver.ts
@@ -1,11 +1,15 @@
 import { AppSyncResolverHandler } from 'aws-lambda';
-import { School, QueryGetSchoolByNameArgs } from '../../appsync';
+import { QueryGetSchoolByNameArgs, SignUpData } from '../../appsync';
 import { logger } from '../shared/logger';
-import { SchoolDataRepository } from '../repository/schoolDataRepository';
+import { MongoClient } from 'mongodb';
 
-const schoolDataRepository = SchoolDataRepository.getInstance();
+const client = new MongoClient(
+  process?.env?.MONGODB_CONNECTION_STRING ?? 'mongodb://localhost:27017/'
+);
 
-export const handler: AppSyncResolverHandler<QueryGetSchoolByNameArgs, School> = async (
+const db = client.db('D2E');
+const collection = db.collection<SignUpData>('SignUps');
+export const handler: AppSyncResolverHandler<QueryGetSchoolByNameArgs, SignUpData[]> = async (
   event,
   context,
   callback
@@ -17,14 +21,9 @@ export const handler: AppSyncResolverHandler<QueryGetSchoolByNameArgs, School> =
   logger.info(`${JSON.stringify(params)}`);
 
   switch (info.fieldName) {
-    case 'getSchoolByNamePublic': {
-      const { name } = params;
-      const school = await schoolDataRepository.getByName(name);
-      if (!school) {
-        callback(null);
-        break;
-      }
-      callback(null, removeFields<School>(info.selectionSetList, school));
+    case 'testPublic': {
+      const res = await collection.find({}).toArray();
+      callback(null, res);
       break;
     }
     default: {
@@ -35,12 +34,3 @@ export const handler: AppSyncResolverHandler<QueryGetSchoolByNameArgs, School> =
 
   throw new Error('An unknown error occurred');
 };
-
-const removeFields = <T extends object>(selectionSetList: string[], obj: T): T => {
-  return Object.entries(obj).reduce((acc, [key, value]) => {
-    if (selectionSetList.includes(key)) {
-      acc = { ...acc, [key]: value as string };
-    }
-    return acc;
-  }, {} as T);
-};