1 files changed, 61 insertions, 0 deletions
diff --git a/src/Controllers/Client/UserController.php b/src/Controllers/Client/UserController.php
index 277ba78..63442ec 100755
--- a/src/Controllers/Client/UserController.php
+++ b/src/Controllers/Client/UserController.php
@@ -4,10 +4,12 @@ namespace App\Controllers\Client;
 
 use App\Database;
 use App\Errors\AppException;
+use App\Errors\NotFoundError;
 use App\Errors\UnauthorizedError;
 use App\Models\Device;
 use App\Models\User;
 use App\Support\RequestValidator;
+use Cassandra\Exception\UnauthorizedException;
 use Matrix\Enums\ErrorCode;
 use Matrix\Responses\ClientAccountWhoamiGetResponse;
 use Symfony\Component\HttpFoundation\Request;
@@ -120,4 +122,63 @@ class UserController
     
     return new JsonResponse();
   }
+  
+  #[Route(path: "/_matrix/client/v3/user/{userId}/account_data/{type}", methods: ["GET"])]
+  public function getAccountData(Request $request): Response
+  {
+    $user = User::authenticateWithRequest($request);
+    
+    $userId = $request->attributes->get("userId");
+    $type = $request->attributes->get("type");
+
+    if ($user->getId() !== $userId) {
+      throw new UnauthorizedError("Cannot get account data for other users.");
+    }
+    
+    $value = Database::getInstance()
+      ->query("select value from account_data where user_id=:user_id and key=:key", [
+        "user_id" => $userId,
+        "key" => $type,
+      ])
+      ->fetchColumn();
+    
+    if (empty($value)) {
+      throw new NotFoundError("Account data not found.");
+    }
+    
+    return new JsonResponse([
+      $type => $value,
+    ]);
+  }
+  
+  #[Route(path: "/_matrix/client/v3/user/{userId}/account_data/{type}", methods: ["PUT"])]
+  public function setAccountData(Request $request): Response
+  {
+    $user = User::authenticateWithRequest($request);
+    $body = json_decode($request->getContent(), true);
+    RequestValidator::validateJson();
+    
+    $userId = $request->attributes->get("userId");
+    $type = $request->attributes->get("type");
+
+    if ($user->getId() !== $userId) {
+      throw new UnauthorizedError("Cannot add account data for other users.");
+    }
+
+    Database::getInstance()
+      ->query(
+        <<<SQL
+          insert into account_data (key, value, user_id) values (:key, :value, :user_id)
+          on conflict (user_id, key) do update set
+          value = excluded.value
+        SQL,
+        [
+          "key" => $type,
+          "value" => $request->getContent(),
+          "user_id" => $userId,
+        ]
+      );
+        
+    return new JsonResponse();
+  }
 }