From 5a1ce860a8ed8ec09b5f801060ffbff8adef06c4 Mon Sep 17 00:00:00 2001
From: rem <remi.arnal@etu.uca.fr>
Date: Wed, 13 Dec 2023 13:58:37 +0100
Subject: [PATCH] :sparkles: add GET /user/ai/{category}

---
 app/gateway/user_gateway.php | 17 +++++++++++++++++
 app/routes.php               | 19 ++++++++++++++++++-
 2 files changed, 35 insertions(+), 1 deletion(-)

diff --git a/app/gateway/user_gateway.php b/app/gateway/user_gateway.php
index b7cc832..2417d17 100644
--- a/app/gateway/user_gateway.php
+++ b/app/gateway/user_gateway.php
@@ -92,6 +92,23 @@ class UserGateway
         return ["email" => $results[0]['email'], "username" => $results[0]['username']];
     }
 
+    public function getModelByCategory(string $uuid, string $category)
+    {
+        $query = "SELECT model FROM trained_model WHERE user_id = :user_uuid and category = LOWER(:category);";
+        try {
+            $this->con->executeQuery($query, array(
+                ':user_uuid' => array($uuid, PDO::PARAM_STR),
+                ':category' => array($category, PDO::PARAM_STR)
+            ));
+            $results = $this->con->getResults();
+        } catch (PDOException) {
+            return -1;
+        }
+        if (count($results) === 0) return  1;
+
+        return ["model" => $results[0]['model']];
+    }
+
     public function updateMail(string $uuid, string $new_email)
     {
         $query = "UPDATE user SET email=:new_email WHERE id=:uuid;";
diff --git a/app/routes.php b/app/routes.php
index 79c39b8..c28ccaa 100644
--- a/app/routes.php
+++ b/app/routes.php
@@ -40,7 +40,7 @@ return function (App $app) {
         return $res;
     });
 
-    #### ACCOUNT ####
+    // ===== ACCOUNT =====
     // Create User 
     $app->post('/user', function (Request $req, Response $res) {
         if (!Helpers::validJson((string) $req->getBody(), array("email", "hash", "username"))) {
@@ -114,6 +114,23 @@ return function (App $app) {
         return $res;
     });
 
+    $app->get('/user/ai/{category}', function(Request $req, Response $res, $args) {
+        if (!(new Token)->verifyToken($req->getHeader('Authorization'))) {
+            return $res->withStatus(401);
+        }       
+        $token = $req->getHeader('Authorization')[0];
+        $category = $args['category'];
+
+        $uuid = (new Token)->getUuidFromToken($token);
+        $code = (new UserGateway)->getModelByCategory($uuid, $category);
+        
+        if($code === -1) return $res->withStatus(500);
+        else if($code === 1) return $res->withStatus(404);
+
+        $res->getBody()->write(json_encode($code));
+        return $res->withStatus(200);
+    });
+
     // Update Mail
     $app->put('/user/email', function (Request $req, Response $res) {
         if (!(new Token)->verifyToken($req->getHeader('Authorization'))) {