Began chat encrpytion integration and ECC key generation on registration

3 files changed, 31 insertions, 6 deletions

diff --git a/main/app/sprinkles/admin/src/Controller/PostController.php b/main/app/sprinkles/admin/src/Controller/PostController.php
index 2441be4..98bee5a 100644
--- a/main/app/sprinkles/admin/src/Controller/PostController.php
+++ b/main/app/sprinkles/admin/src/Controller/PostController.php
@@ -79,9 +79,8 @@ class PostController extends SimpleController
             $uploadedFile->moveTo(__DIR__ . '/../../../../../uploads' . DIRECTORY_SEPARATOR . $filename);
 
             // Store in Database
-            DB::table('image_posts')->insert(
-                ['UserID' => $currentUser->id, 'File' => $filename]
-            );
+            DB::table('image_posts')
+                ->insert(['UserID' => $currentUser->id, 'File' => $filename]);
 
             $response->write('Uploaded successfully! <br/>');
         }
diff --git a/main/app/sprinkles/admin/src/Controller/UserController.php b/main/app/sprinkles/admin/src/Controller/UserController.php
index 46d0f0f..30a8d30 100644
--- a/main/app/sprinkles/admin/src/Controller/UserController.php
+++ b/main/app/sprinkles/admin/src/Controller/UserController.php
@@ -231,6 +231,32 @@ class UserController extends SimpleController
         return $response->withStatus(200);
     }
 
+
+    /**
+     * Sets the users public key
+     * Request type: POST
+     */
+    public function setPublicKey($request, $response, $args) {
+        $user = $this->getUserFromParams($args);
+
+        if (!$user) {
+            throw new NotFoundException($request, $response);
+        }
+
+        $classMapper = $this->ci->classMapper;
+        $requestedUser = $classMapper->staticMethod('user', 'where', 'user_name', $args['user_name'])
+            ->first();
+
+        if ($user->id === $requestedUser->id) {
+            $PublicKey = $request->getParsedBody()["PublicKey"];
+            Capsule::table('public_keys')
+                ->insert(['UserID' => $requestedUser->id, 'Key' => $PublicKey]);
+            return $response->withStatus(200);
+        } else {
+            throw new ForbiddenException();
+        }
+    }
+
     /**
      * Processes the request to delete an existing user.
      *
diff --git a/main/app/sprinkles/admin/src/Controller/WormholeController.php b/main/app/sprinkles/admin/src/Controller/WormholeController.php
index 2ed7e68..d70fbbc 100644
--- a/main/app/sprinkles/admin/src/Controller/WormholeController.php
+++ b/main/app/sprinkles/admin/src/Controller/WormholeController.php
@@ -83,9 +83,9 @@ class WormholeController extends SimpleController
     private function verifyAccessToken($args) {
         $currentUser = $this->ci->currentUser; // FOR DATABASE QUERY
         $access_token = $args['access_token'];
-        if (DB::table('public_keys')
-            ->where('UserID', 1)
-            ->where('Key', '=', $access_token)
+        if (DB::table('access_token')
+            ->where('id', 1)
+            ->where('token', '=', $access_token)
             ->exists()) {
             return true;
         } else {