From 6286bcaf4619aee563e7d31fbe07d506422616b5 Mon Sep 17 00:00:00 2001
From: Damodar Lohani <dlohani48@gmail.com>
Date: Wed, 30 Mar 2022 06:51:59 +0000
Subject: [PATCH] fix for console

---
 app/realtime.php | 109 ++++++++++++++++++++++++-----------------------
 1 file changed, 56 insertions(+), 53 deletions(-)

diff --git a/app/realtime.php b/app/realtime.php
index bdb0d4c81b..a2393b13fd 100644
--- a/app/realtime.php
+++ b/app/realtime.php
@@ -510,61 +510,64 @@ $server->onMessage(function (int $connection, string $message) use ($server, $re
         $cache = new Cache(new RedisCache($redis));
         $database = new Database(new MariaDB($db), $cache);
         $database->setDefaultDatabase(App::getEnv('_APP_DB_SCHEMA', 'appwrite'));
-        $database->setNamespace('_console');
-        $project = $database->getDocument('projects', $realtime->connections[$connection]['projectId']);
-        if($project->isEmpty()) {
-            throw new Exception("Project not found: {$realtime->connections[$connection]['projectId']}");
-        }
-        $secrets = $project->getAttribute('databaseSecrets');
-        $displacement = $project->getAttribute('databaseSecretsDisplacement', 0);
-        $version = $displacement + \count($secrets);
-        
-        $filters['encrypt'] = [
-            'encode' => function($value) use($version, $secrets) {
-
-                $key = $secrets[\count($secrets)-1];
-                $iv = OpenSSL::randomPseudoBytes(OpenSSL::cipherIVLength(OpenSSL::CIPHER_AES_128_GCM));
-                $tag = null;
-                $value = json_encode([
-                    'data' => OpenSSL::encrypt($value, OpenSSL::CIPHER_AES_128_GCM, $key, 0, $iv, $tag),
-                    'method' => OpenSSL::CIPHER_AES_128_GCM,
-                    'iv' => \bin2hex($iv),
-                    'tag' => \bin2hex($tag ?? ''),
-                    'version' => $version,
-                ]);
-
-                $key = App::getEnv('_APP_OPENSSL_KEY_V1');
-                $iv = OpenSSL::randomPseudoBytes(OpenSSL::cipherIVLength(OpenSSL::CIPHER_AES_128_GCM));
-                $tag = null;
-                return json_encode([
-                    'data' => OpenSSL::encrypt($value, OpenSSL::CIPHER_AES_128_GCM, $key, 0, $iv, $tag),
-                    'method' => OpenSSL::CIPHER_AES_128_GCM,
-                    'iv' => \bin2hex($iv),
-                    'tag' => \bin2hex($tag ?? ''),
-                    'version' => '1',
-                ]);
-            },
-            'decode' => function($value) use($secrets, $displacement) {
-                if(is_null($value)) {
-                    return null;
-                }
-                
-                $value = json_decode($value, true);
-                $key = App::getEnv('_APP_OPENSSL_KEY_V'.$value['version']);
-        
-                $value = OpenSSL::decrypt($value['data'], $value['method'], $key, 0, hex2bin($value['iv']), hex2bin($value['tag']));
-
-                $value = json_decode($value, true);
-                $version = ($value['version'] ?? 1) - $displacement;
-                $key = $secrets[$version - 1];
-        
-                return OpenSSL::decrypt($value['data'], $value['method'], $key, 0, hex2bin($value['iv']), hex2bin($value['tag']));
+        $projectId = $realtime->connections[$connection]['projectId'];
+        if($projectId != 'console') {
+            $database->setNamespace('_console');
+            $project = $database->getDocument('projects', $projectId);
+            if($project->isEmpty()) {
+                throw new Exception("Project not found: {$projectId}");
             }
-        ];
-        
-        $database = new Database(new MariaDB($db), $cache, $filters);
+            $secrets = $project->getAttribute('databaseSecrets');
+            $displacement = $project->getAttribute('databaseSecretsDisplacement', 0);
+            $version = $displacement + \count($secrets);
+            
+            $filters['encrypt'] = [
+                'encode' => function($value) use($version, $secrets) {
+    
+                    $key = $secrets[\count($secrets)-1];
+                    $iv = OpenSSL::randomPseudoBytes(OpenSSL::cipherIVLength(OpenSSL::CIPHER_AES_128_GCM));
+                    $tag = null;
+                    $value = json_encode([
+                        'data' => OpenSSL::encrypt($value, OpenSSL::CIPHER_AES_128_GCM, $key, 0, $iv, $tag),
+                        'method' => OpenSSL::CIPHER_AES_128_GCM,
+                        'iv' => \bin2hex($iv),
+                        'tag' => \bin2hex($tag ?? ''),
+                        'version' => $version,
+                    ]);
+    
+                    $key = App::getEnv('_APP_OPENSSL_KEY_V1');
+                    $iv = OpenSSL::randomPseudoBytes(OpenSSL::cipherIVLength(OpenSSL::CIPHER_AES_128_GCM));
+                    $tag = null;
+                    return json_encode([
+                        'data' => OpenSSL::encrypt($value, OpenSSL::CIPHER_AES_128_GCM, $key, 0, $iv, $tag),
+                        'method' => OpenSSL::CIPHER_AES_128_GCM,
+                        'iv' => \bin2hex($iv),
+                        'tag' => \bin2hex($tag ?? ''),
+                        'version' => '1',
+                    ]);
+                },
+                'decode' => function($value) use($secrets, $displacement) {
+                    if(is_null($value)) {
+                        return null;
+                    }
+                    
+                    $value = json_decode($value, true);
+                    $key = App::getEnv('_APP_OPENSSL_KEY_V'.$value['version']);
+            
+                    $value = OpenSSL::decrypt($value['data'], $value['method'], $key, 0, hex2bin($value['iv']), hex2bin($value['tag']));
+    
+                    $value = json_decode($value, true);
+                    $version = ($value['version'] ?? 1) - $displacement;
+                    $key = $secrets[$version - 1];
+            
+                    return OpenSSL::decrypt($value['data'], $value['method'], $key, 0, hex2bin($value['iv']), hex2bin($value['tag']));
+                }
+            ];
+            
+            $database = new Database(new MariaDB($db), $cache, $filters);
+        }
         $database->setDefaultDatabase(App::getEnv('_APP_DB_SCHEMA', 'appwrite'));
-        $database->setNamespace("_{$realtime->connections[$connection]['projectId']}");
+        $database->setNamespace("_{$projectId}");
 
         /*
          * Abuse Check