feat(mvp): added support for token settings in py sdk

Author: h3yAlias

src/rushdb/client.py

@@ -11,6 +11,7 @@
 import urllib.request
 from typing import Any, Dict, Optional
 
+from .utils.token_prefix import extract_mixed_properties_from_token
 from .api.labels import LabelsAPI
 from .api.properties import PropertiesAPI
 from .api.records import RecordsAPI
@@ -102,6 +103,15 @@ def __init__(self, api_key: str, base_url: Optional[str] = None):
             ...     base_url="https://my-rushdb.company.com/api/v1"
             ... )
         """
+        settings, raw_key = extract_mixed_properties_from_token(api_key)
+
+        self.server_settings = {
+            'customDB':    settings['customDB'],
+            'managedDB':   settings['managedDB'],
+            'selfHosted':  settings['selfHosted'],
+            'canceled':    settings['canceled'],
+            'plan_type':   settings['planType'],
+        } if settings else None
         self.base_url = (base_url or self.DEFAULT_BASE_URL).rstrip("/")
         self.api_key = api_key
         self.records = RecordsAPI(self)

src/rushdb/utils/token_prefix.py

@@ -0,0 +1,30 @@
+import re
+from typing import Dict, Optional, Tuple
+
+PlanPrefix = {
+    'initial':      'in',
+    'extended':     'ex',
+    'fullFeatured': 'ff',
+}
+
+def extract_mixed_properties_from_token(
+        prefixed_token: str
+) -> Tuple[Optional[Dict[str, bool]], str]:
+    matched = re.match(r'^([a-z]{2})_([01]{4}\d*)_(.+)$', prefixed_token)
+    if not matched:
+        return None, prefixed_token
+
+    prefix, bits, raw = matched.groups()
+    plan = next((p for p, plan in PlanPrefix.items() if plan == prefix), None)
+    if plan is None:
+        return None, prefixed_token
+
+    b_custom, b_managed, b_self, b_canceled = bits[:4]
+    settings = {
+        'planType':    plan,
+        'customDB':    b_custom  == '1',
+        'managedDB':   b_managed == '1',
+        'selfHosted':  b_self    == '1',
+        'canceled':    b_canceled== '1',
+    }
+    return settings, raw