1 files changed, 87 insertions, 0 deletions
diff --git a/solo-tool-project/src/solo_tool/storage.py b/solo-tool-project/src/solo_tool/storage.py
new file mode 100644
index 0000000..0c5577f
--- /dev/null
+++ b/solo-tool-project/src/solo_tool/storage.py
@@ -0,0 +1,87 @@
+from typing import Protocol
+from abc import abstractmethod
+
+from pathlib import Path
+from glob import glob
+import json
+import requests
+from os import getenv
+
+class StorageBackend(Protocol):
+    @abstractmethod
+    def listSessions(self) -> list[str]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def readSession(self, id: str) -> dict:
+        raise NotImplementedError
+
+    @abstractmethod
+    def writeSession(self, session: dict, id: str) -> None:
+        raise NotImplementedError
+
+    @abstractmethod
+    def writeRecording(self, recording: Path, destination: str) -> None:
+        raise NotImplementedError
+
+class FileSystemStorageBackend(StorageBackend):
+    def __init__(self, storagePath: str):
+        self._storagePath = Path(storagePath)
+
+    def listSessions(self) -> list[str]:
+        #return [Path(f).stem for f in glob(f"{self._storagePath / "sessions"}/*.json")]
+        return [Path(f).stem for f in glob(str(self._storagePath / "sessions" / "*.json"))]
+
+    def readSession(self, id: str) -> dict:
+        with open(self._storagePath / "sessions" / f"{id}.json", "r") as f:
+            session = json.load(f)
+        return session
+
+    def writeSession(self, session: dict, id: str) -> None:
+        with open(self._storagePath / "sessions" / f"{id}.json", "w") as f:
+            json.dump(session, f)
+
+    def writeRecording(self, recording: Path, destination: str) -> None:
+        pass
+
+class FileBrowserStorageBackend(StorageBackend):
+    def __init__(self, serverUrl: str):
+        self._baseUrl = serverUrl
+        self._username = getenv("ST_USER")
+        self._password = getenv("ST_PASS")
+        self._apiKey = self._getApiKey()
+
+    def listSessions(self) -> list[str]:
+        url = f"{self._baseUrl}/api/resources/sessions"
+        response = self._request("GET", url)
+        return [item["name"][0:-5] for item in response.json()["items"] if item["extension"] == ".json"]
+
+    def readSession(self, id: str) -> dict:
+        url = f"{self._baseUrl}/api/raw/sessions/{id}.json"
+        response = self._request("GET", url)
+        return json.loads(response.content)
+
+    def writeSession(self, session: dict, id: str) -> None:
+        url = f"{self._baseUrl}/api/resources/sessions/{id}.json"
+        self._request("PUT", url, json=session)
+
+    def writeRecording(self, recording: Path, destination: str) -> None:
+        url = f"{self._baseUrl}/api/resources/recordings/{destination}"
+        with open(recording, "rb") as file:
+            self._request("POST", url, {"Content-Type" : "audio/mpeg"}, data=file)
+
+    def _getApiKey(self) -> str:
+        response = requests.post(f"{self._baseUrl}/api/login", json={"username":self._username, "password":self._password})
+        return response.content
+
+    def _request(self, verb: str, url: str, moreHeaders: dict={}, **kwargs):
+        headers = moreHeaders | {"X-Auth" : self._apiKey}
+        response = requests.request(verb, url, headers=headers, **kwargs)
+        if response.status_code == requests.codes.UNAUTHORIZED:
+            # if unauthorized, the key might have expired
+            self._apiKey = self._getApiKey()
+            headers["X-Auth"] = self._apiKey
+            response = requests.request(verb, url, headers=headers, **kwargs)
+        response.raise_for_status()
+        return response
+