Add support for per-user quotas, by allowing a fake collab name "private-{username}"

Author: apdavison

api/simqueue/globals.py

@@ -26,3 +26,5 @@
     "Spikey": 1.0,
     "Demo": 1.0,
 }
+
+PRIVATE_SPACE = "private"

api/simqueue/oauth.py

@@ -7,6 +7,7 @@
 from fastapi import Security, HTTPException, status as status_codes
 
 from . import settings, db
+from .globals import PRIVATE_SPACE
 
 
 logger = logging.getLogger("simqueue")
@@ -75,6 +76,8 @@ def username(self):
         return self.preferred_username
 
     async def can_view(self, collab):
+        if collab == f"{PRIVATE_SPACE}-{self.username}":
+            return True
         # first of all, check team permissions
         target_team_names = {
             role: f"collab-{collab}-{role}" for role in ("viewer", "editor", "administrator")
@@ -91,6 +94,8 @@ async def can_view(self, collab):
             return collab_info.get("isPublic", False)
 
     def can_edit(self, collab):
+        if collab == f"{PRIVATE_SPACE}-{self.username}":
+            return True
         target_team_names = {
             role: f"collab-{collab}-{role}" for role in ("editor", "administrator")
         }
@@ -99,7 +104,7 @@ def can_edit(self, collab):
                 return True
 
     def get_collabs(self, access=["viewer", "editor", "administrator"]):
-        collabs = set()
+        collabs = set([f"{PRIVATE_SPACE}-{self.username}"])
         for team_access in self.roles.get("team", []):
             # note, if team information is missing from userinfo that means
             # the user is not a member of any collab