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