add route to fetch the current pod count, locked behind a key that lives in the config.php.

Author: ylebre

config.php.example

@@ -43,6 +43,10 @@
 
 	const PUBSUB_SERVER = "wss://pubsub:8080";
 
+	// Token / key that needs to be in the request to fetch the pod
+	// count from the storage api
+	const POD_COUNT_KEY = "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx";
+
 	// Array of IPs that can be trusted. If they are in this list, they
 	// won't be locked out after failed login attempts;
 	const TRUSTED_IPS = [];

lib/Routes/SolidStorageProvider.php

@@ -31,5 +31,31 @@ public static function respondToStorageNew() {
 			header("Content-type: application/json");
 			echo json_encode($responseData, JSON_PRETTY_PRINT | JSON_THROW_ON_ERROR);
 		}
+
+		public static function respondToPodCount() {
+			$requestFactory = new ServerRequestFactory();
+			$rawRequest = $requestFactory->fromGlobals($_SERVER, $_GET, $_POST, $_COOKIE, $_FILES);
+
+			if (!defined('POD_COUNT_KEY')) {
+				header("HTTP/1.1 404 Not found");
+				exit();
+			}
+			if (!isset($_SERVER['HTTP_POD_COUNT_KEY'])) {
+				header("HTTP/1.1 404 Not found");
+				exit();
+			}
+			if ($_SERVER['HTTP_POD_COUNT_KEY'] !== POD_COUNT_KEY) {
+				header("HTTP/1.1 400 Bad Request");
+				exit();
+			}
+
+			$podCount = StorageServer::getPodCount();
+			$responseData = array(
+				"count" => $podCount
+			);
+			header("HTTP/1.1 200 OK");
+			header("Content-type: application/json");
+			echo json_encode($responseData, JSON_PRETTY_PRINT | JSON_THROW_ON_ERROR);
+		}
 	}
 

lib/StorageServer.php

@@ -48,6 +48,20 @@ public static function createStorage($ownerWebId) {
 			];
 		}
 
+		public static function getPodCount() {
+			Db::connect();
+			$query = Db::$pdo->prepare(
+				'SELECT count(storage_id) as podCount FROM storage'
+			);
+			$query->execute();
+			$result = $query->fetchAll();
+
+			if (sizeof($result) === 1) {
+				return $result[0]['podCount'];
+			}
+			return false;
+		}
+
                 public static function storageIdExists($storageId) {
                         Db::connect();
                         $query = Db::$pdo->prepare(

www/storage/index.php

@@ -17,6 +17,10 @@
 	switch($method) {
 		case "GET":
 			switch ($request) {
+				case "/api/podcount":
+				case "/api/podcount/":
+					SolidStorageProvider::respondToPodCount();
+				break;
 				default:
 					header($_SERVER['SERVER_PROTOCOL'] . " 404 Not found");
 				break;