Medios-Macina/Provider/zeroxzero.py

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."""

    def upload(self, file_path: str, **kwargs: Any) -> str:
        from API.HTTP import HTTPClient
        from 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