pre-migration commit

plugins/zeroxzero/__init__.py

@@ -0,0 +1,72 @@
+from __future__ import annotations
+
+import os
+import sys
+from typing import Any
+
+from ProviderCore.base import Provider
+from SYS.logger import log
+
+
+class ZeroXZero(Provider):
+    """File provider for 0x0.st."""
+
+    PLUGIN_NAME = "0x0"
+    PLUGIN_ALIASES = ("zeroxzero",)
+
+    def upload(self, file_path: str, **kwargs: Any) -> str:
+        from API.HTTP import HTTPClient
+        from SYS.models import ProgressFileReader
+
+        if not os.path.exists(file_path):
+            raise FileNotFoundError(f"File not found: {file_path}")
+
+        try:
+            headers = {
+                "User-Agent": "Medeia-Macina/1.0"
+            }
+            with HTTPClient(headers=headers) as client:
+                with open(file_path, "rb") as handle:
+                    try:
+                        total = os.path.getsize(file_path)
+                    except Exception:
+                        total = None
+                    wrapped = ProgressFileReader(
+                        handle,
+                        total_bytes=total,
+                        label="upload"
+                    )
+                    response = client.post(
+                        "https://0x0.st",
+                        files={
+                            "file": wrapped
+                        }
+                    )
+
+                if response.status_code == 200:
+                    uploaded_url = response.text.strip()
+                    try:
+                        pipe_obj = kwargs.get("pipe_obj")
+                        if pipe_obj is not None:
+                            from Store import Store
+
+                            Store(
+                                self.config,
+                                suppress_debug=True
+                            ).try_add_url_for_pipe_object(pipe_obj,
+                                                          uploaded_url)
+                    except Exception:
+                        pass
+
+                    return uploaded_url
+
+                raise Exception(
+                    f"Upload failed: {response.status_code} - {response.text}"
+                )
+
+        except Exception as exc:
+            log(f"[0x0] Upload error: {exc}", file=sys.stderr)
+            raise
+
+    def validate(self) -> bool:
+        return True