1 files changed, 75 insertions, 10 deletions

diff --git a/src/Controllers/Client/KeyController.php b/src/Controllers/Client/KeyController.php
index b9ae61f..5e3245b 100644
--- a/src/Controllers/Client/KeyController.php
+++ b/src/Controllers/Client/KeyController.php
@@ -2,6 +2,7 @@
 
 namespace App\Controllers\Client;
 
+use App\Database;
 use App\Models\User;
 use App\Support\RequestValidator;
 use Matrix\Responses\ClientKeysUploadPostResponse;
@@ -12,6 +13,76 @@ use Symfony\Component\Routing\Attribute\Route;
 
 class KeyController
 {
+  #[Route(path: "/_matrix/client/r0/keys/upload", methods: ["POST"])]
+  #[Route(path: "/_matrix/client/v3/keys/upload", methods: ["POST"])]
+  public function upload(Request $request): Response
+  {
+    $user = User::authenticateWithRequest($request);
+    $body = json_decode($request->getContent(), true);
+    RequestValidator::validateJson();
+
+    if (! empty($body["device_keys"])) {
+      if ($body["device_keys"]["user_id"] !== $user->getId()) {}
+      if ($body["device_keys"]["user_id"] !== $user->getDeviceId()) {}
+      
+      Database::getInstance()->query(<<<SQL
+        insert into device_keys (supported_algorithms, keys, signatures, user_id, device_id)
+        values (:supported_algorithms, :keys, :signatures, :user_id, :device_id)
+      SQL, [
+        "supported_algorithms" => json_encode($body["device_keys"]["algorithms"]),
+        "keys" => json_encode($body["device_keys"]["keys"]),
+        "signatures" => json_encode($body["device_keys"]["signatures"]),
+        "user_id" => $user->getId(),
+        "device_id" => $user->getDeviceId(),
+      ]);
+    }
+
+    $oneTimeKeys = $body["one_time_keys"];
+    if (! empty($body["fallback_keys"])) {
+      $oneTimeKeys += $body["fallback_keys"];
+    }
+    
+    foreach ($oneTimeKeys as $identifier => $object) {
+      $identifierParts = explode(":", $identifier);
+
+      $algorithm = $identifierParts[0];
+      $id = $identifierParts[1];
+
+      $signatures = array_values($object["signatures"])[0];
+      $signatureIdentifier = array_keys($signatures)[0];
+      $signatureAlgorithm = explode(":", $signatureIdentifier)[0];
+      $signatureKey = array_values($signatures)[0];
+
+      $deviceId = explode(":", $signatureIdentifier)[1];
+      
+      Database::getInstance()->query(<<<SQL
+        insert into one_time_keys (id, key, algorithm, signature_key, signature_algorithm, is_fallback, user_id, device_id)
+        values (:id, :key, :algorithm, :signature_key, :signature_algorithm, :is_fallback, :user_id, :device_id)
+      SQL, [
+        "id" => $id,
+        "key" => $object["key"],
+        "algorithm" => $algorithm,
+        "signature_key" => $signatureKey,
+        "signature_algorithm" => $signatureAlgorithm,
+        "is_fallback" => ($object["fallback"] ?? false) ?: 0,
+        "user_id" => $user->getId(),
+        "device_id" => $deviceId,
+      ]);
+    }
+
+    # TODO: do that per algorithm
+    $currentCountOneTimeKeys = Database::getInstance()
+      ->query("select count(id) from one_time_keys where user_id=:userId and is_fallback=false", [
+        "userId" => $user->getId(),
+      ])
+      ->fetchColumn();
+
+    return new JsonResponse(new ClientKeysUploadPostResponse([
+      #"curve25519" => 0,
+      "signed_curve25519" => $currentCountOneTimeKeys,
+    ]));
+  }
+  
   #[Route(path: "/_matrix/client/v3/keys/query", methods: ["POST"])]
   public function query(Request $request): Response
   {
@@ -29,19 +100,13 @@ class KeyController
     ]);
   }
 
-  #[Route(path: "/_matrix/client/r0/keys/upload", methods: ["POST"])]
-  #[Route(path: "/_matrix/client/v3/keys/upload", methods: ["POST"])]
-  public function upload(Request $request): Response
+  #[Route(path: "/_matrix/client/v3/keys/claim", methods: ["POST"])]
+  public function claim(Request $request): Response
   {
     $user = User::authenticateWithRequest($request);
     $body = json_decode($request->getContent(), true);
     RequestValidator::validateJson();
-
-    foreach ($body["one_time_keys"] as $identifier => $object) {}
-
-    return new JsonResponse(new ClientKeysUploadPostResponse([
-      #"curve25519" => 0,
-      "signed_curve25519" => count($body["one_time_keys"]),
-    ]));
+    
+    return new JsonResponse();
   }
 }