new plugins

Author: blinkerbit

aird/cloud/__init__.py

@@ -23,9 +23,25 @@
     "OneDriveProvider",
     "encode_identifier",
     "decode_identifier",
+    "split_drive_path",
 ]
 
 
+def split_drive_path(path: str) -> list[str]:
+    """Split a OneDrive path into safe folder/file segments."""
+    out: list[str] = []
+    for raw in str(path or "").replace("\\", "/").split("/"):
+        name = raw.strip()
+        if not name or name in {".", ".."}:
+            continue
+        if "/" in name or "\\" in name or "\x00" in name:
+            continue
+        out.append(name[:255])
+        if len(out) >= 32:
+            break
+    return out
+
+
 class CloudProviderError(Exception):
     """Raised when a cloud provider encounters an operational error."""
 
@@ -119,6 +135,7 @@ def upload_file(
         parent_id: Optional[str] = None,
         size: Optional[int] = None,
         content_type: Optional[str] = None,
+        conflict: str = "rename",
     ) -> CloudFile:  # pragma: no cover - interface
         raise NotImplementedError
 
@@ -402,6 +419,7 @@ def upload_file(
         parent_id: Optional[str] = None,
         size: Optional[int] = None,
         content_type: Optional[str] = None,
+        conflict: str = "rename",
     ) -> CloudFile:
         if not name:
             raise CloudProviderError("A file name is required for Google Drive uploads")
@@ -490,6 +508,104 @@ def list_files(self, folder_id: str | None = None) -> List[CloudFile]:
             )
         return files
 
+    def _item_path_url(self, path: str, extra: str = "") -> str:
+        parts = split_drive_path(path)
+        if not parts:
+            return (
+                f"{self._base_url}/root/{extra.lstrip('/')}"
+                if extra
+                else f"{self._base_url}/root"
+            )
+        encoded = "/".join(quote(part, safe="") for part in parts)
+        if extra:
+            return f"{self._base_url}/root:/{encoded}:/{extra.lstrip('/')}"
+        return f"{self._base_url}/root:/{encoded}"
+
+    def _file_from_payload(self, payload: dict, *, default_name: str = "Unnamed") -> CloudFile:
+        return CloudFile(
+            id=payload.get("id", ""),
+            name=payload.get("name", default_name),
+            is_dir="folder" in payload,
+            size=_safe_int(payload.get("size")),
+            modified=payload.get("lastModifiedDateTime"),
+        )
+
+    def get_item_by_path(self, path: str) -> Optional[CloudFile]:
+        parts = split_drive_path(path)
+        if not parts:
+            return CloudFile(id="root", name="root", is_dir=True)
+        try:
+            response = requests.get(
+                self._item_path_url(path),
+                headers=self._headers(),
+                timeout=30,
+            )
+        except requests.RequestException as exc:
+            raise CloudProviderError(f"OneDrive request failed: {exc}") from exc
+        if response.status_code == 404:
+            return None
+        if response.status_code != 200:
+            raise CloudProviderError(
+                f"OneDrive item lookup failed ({response.status_code}): {response.text[:200]}"
+            )
+        return self._file_from_payload(response.json())
+
+    def ensure_folder_path(self, path: str) -> CloudFile:
+        parts = split_drive_path(path)
+        if not parts:
+            return CloudFile(id="root", name="root", is_dir=True)
+        existing = self.get_item_by_path("/".join(parts))
+        if existing is not None:
+            if not existing.is_dir:
+                raise CloudProviderError("OneDrive path exists and is not a folder")
+            return existing
+        parent_id: Optional[str] = None
+        current: list[str] = []
+        last = CloudFile(id="root", name="root", is_dir=True)
+        for name in parts:
+            current.append(name)
+            found = self.get_item_by_path("/".join(current))
+            if found is not None:
+                if not found.is_dir:
+                    raise CloudProviderError("OneDrive path exists and is not a folder")
+                last = found
+                parent_id = found.id
+                continue
+            last = self.ensure_folder(name, parent_id=parent_id)
+            parent_id = last.id
+        return last
+
+    def ensure_folder(self, name: str, parent_id: Optional[str] = None) -> CloudFile:
+        for item in self.list_files(parent_id):
+            if item.is_dir and item.name == name:
+                return item
+        if not parent_id or parent_id == "root":
+            url = f"{self._base_url}/root/children"
+        else:
+            url = f"{self._base_url}/items/{parent_id}/children"
+        try:
+            response = requests.post(
+                url,
+                headers=self._headers(),
+                json={
+                    "name": name,
+                    "folder": {},
+                    "@microsoft.graph.conflictBehavior": "fail",
+                },
+                timeout=30,
+            )
+        except requests.RequestException as exc:
+            raise CloudProviderError(f"OneDrive request failed: {exc}") from exc
+        if response.status_code in (200, 201):
+            return self._file_from_payload(response.json(), default_name=name)
+        if response.status_code == 409:
+            for item in self.list_files(parent_id):
+                if item.is_dir and item.name == name:
+                    return item
+        raise CloudProviderError(
+            f"OneDrive folder create failed ({response.status_code}): {response.text[:200]}"
+        )
+
     def download_file(self, file_id: str) -> CloudDownload:
         try:
             meta_resp = requests.get(
@@ -559,17 +675,10 @@ def _upload_simple_onedrive(
                 f"OneDrive upload failed ({response.status_code}): {response.text[:200]}"
             )
 
-        payload = response.json()
-        return CloudFile(
-            id=payload.get("id", ""),
-            name=payload.get("name", name),
-            is_dir="folder" in payload,
-            size=_safe_int(payload.get("size")),
-            modified=payload.get("lastModifiedDateTime"),
-        )
+        return self._file_from_payload(response.json(), default_name=name)
 
     def _upload_chunked_onedrive(
-        self, stream, name, parent_id, mime_type, size
+        self, stream, name, parent_id, mime_type, size, conflict: str = "rename"
     ) -> CloudFile:
         safe_name = quote(name, safe="")
         if not parent_id or parent_id == "root":
@@ -584,7 +693,11 @@ def _upload_chunked_onedrive(
                 session_url,
                 headers=self._headers(),
                 json={
-                    "item": {"@microsoft.graph.conflictBehavior": "rename"}
+                    "item": {
+                        "@microsoft.graph.conflictBehavior": conflict
+                        if conflict in {"rename", "replace", "fail"}
+                        else "rename"
+                    }
                 },
                 timeout=60,
             )
@@ -629,14 +742,7 @@ def _upload_chunked_onedrive(
                 ) from exc
 
             if upload_resp.status_code in (200, 201):
-                payload = upload_resp.json()
-                return CloudFile(
-                    id=payload.get("id", ""),
-                    name=payload.get("name", name),
-                    is_dir="folder" in payload,
-                    size=_safe_int(payload.get("size")),
-                    modified=payload.get("lastModifiedDateTime"),
-                )
+                return self._file_from_payload(upload_resp.json(), default_name=name)
 
             if upload_resp.status_code in (202, 204):
                 offset = end + 1
@@ -656,6 +762,7 @@ def upload_file(
         parent_id: Optional[str] = None,
         size: Optional[int] = None,
         content_type: Optional[str] = None,
+        conflict: str = "rename",
     ) -> CloudFile:
         if not name:
             raise CloudProviderError("A file name is required for OneDrive uploads")
@@ -677,7 +784,32 @@ def upload_file(
             return self._upload_simple_onedrive(
                 stream, name, parent_id, headers, mime_type
             )
-        return self._upload_chunked_onedrive(stream, name, parent_id, mime_type, size)
+        return self._upload_chunked_onedrive(
+            stream, name, parent_id, mime_type, size, conflict=conflict
+        )
+
+    def upload_file_at_path(
+        self,
+        stream: BinaryIO,
+        *,
+        drive_path: str,
+        size: Optional[int] = None,
+        content_type: Optional[str] = None,
+        conflict: str = "replace",
+    ) -> CloudFile:
+        parts = split_drive_path(drive_path)
+        if not parts:
+            raise CloudProviderError("A file path is required for OneDrive uploads")
+        parent = self.ensure_folder_path("/".join(parts[:-1])) if len(parts) > 1 else None
+        parent_id = None if parent is None or parent.id in ("", "root") else parent.id
+        return self.upload_file(
+            stream,
+            name=parts[-1],
+            parent_id=parent_id,
+            size=size,
+            content_type=content_type,
+            conflict=conflict,
+        )
 
 
 def encode_identifier(identifier: str) -> str:

aird/constants/__init__.py

@@ -49,11 +49,14 @@
     "webdav_server": False,
     "direct_messages": False,
     "gitlab_integration": False,
+    "onedrive_backup": False,
+    "onedrive_browser": False,
     "transfer_sendfile": True,
 }
 
 CHAT_SHARE_FOLDER = ".aird-shares"
 CHAT_STORE_FOLDER = ".aird-chats"
+AIRD_META_FOLDER = ".aird"
 
 # WebSocket connection configuration
 WEBSOCKET_CONFIG = {

aird/constants/admin.py

@@ -10,8 +10,11 @@
 URL_ADMIN = "/admin"
 URL_ADMIN_LOGIN = "/admin/login"
 URL_ADMIN_USERS = "/admin/users"
+URL_ADMIN_PLUGINS = "/admin/plugins"
+URL_ADMIN_FEATURE_FLAGS = "/admin/api/feature-flags"
 URL_ADMIN_LDAP = "/admin/ldap"
 URL_ADMIN_NETWORK_SHARES = "/admin/network-shares"
+URL_ADMIN_PATHS = "/admin/paths"
 # Network share error query params (URL-encoded)
 ERR_DB_UNAVAILABLE = "Database+unavailable"
 ERR_ALL_FIELDS_REQUIRED = "All+fields+are+required"

aird/constants/input_limits.py

@@ -77,6 +77,10 @@
 GITLAB_PROJECT_PATH_MAX_LEN = 512
 GITLAB_HOST_MAX_LEN = 256
 GITLAB_TOKEN_MAX_LEN = 16 * 1024
+ONEDRIVE_TOKEN_MAX_LEN = 16 * 1024
+ONEDRIVE_ROOT_PATH_MAX_LEN = 512
+PLUGIN_USERNAME_MAX_LEN = 128
+PLUGIN_MAX_USERNAMES = 500
 GITLAB_BOARD_ID_MAX = 10_000_000
 
 # Chat composer attachments in one message