BROKENXAPI 2.0.1__tar.gz → 2.0.2__tar.gz

{brokenxapi-2.0.1 → brokenxapi-2.0.2}/BROKENXAPI.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: BROKENXAPI
-Version: 2.0.1
+Version: 2.0.2
 Summary: Official async Python SDK and CLI for BrokenX YouTube API
 Author-email: Mr Broken <brokenxnetwork@gmail.com>
 License: MIT License

{brokenxapi-2.0.1 → brokenxapi-2.0.2}/BROKENXAPI.egg-info/SOURCES.txt

@@ -11,6 +11,8 @@ BROKENXAPI.egg-info/requires.txt
 BROKENXAPI.egg-info/top_level.txt
 brokenxapi/__init__.py
 brokenxapi/__version__.py
+brokenxapi/auth.py
 brokenxapi/cli.py
+brokenxapi/client.py
 brokenxapi/exceptions.py
 brokenxapi/models.py

{brokenxapi-2.0.1 → brokenxapi-2.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: BROKENXAPI
-Version: 2.0.1
+Version: 2.0.2
 Summary: Official async Python SDK and CLI for BrokenX YouTube API
 Author-email: Mr Broken <brokenxnetwork@gmail.com>
 License: MIT License

brokenxapi-2.0.2/brokenxapi/__version__.py

@@ -0,0 +1,2 @@
+__version__ = "2.0.2"
+

brokenxapi-2.0.2/brokenxapi/auth.py

@@ -0,0 +1,11 @@
+from pathlib import Path
+
+AUTH_FILE = Path.home() / ".brokenxapi"
+
+def save_key(key: str):
+    AUTH_FILE.write_text(key.strip())
+
+def get_key() -> str | None:
+    if AUTH_FILE.exists():
+        return AUTH_FILE.read_text().strip()
+    return None

brokenxapi-2.0.2/brokenxapi/cli.py

@@ -0,0 +1,51 @@
+import argparse
+import asyncio
+from .client import BrokenXAPI
+from .auth import get_key, save_key
+from .__version__ import __version__
+
+def main():
+    parser = argparse.ArgumentParser("brokenx")
+    parser.add_argument("-v", "--version", action="store_true")
+
+    sub = parser.add_subparsers(dest="cmd")
+
+    auth = sub.add_parser("auth")
+    auth.add_argument("key")
+
+    search = sub.add_parser("search")
+    search.add_argument("query")
+
+    dl = sub.add_parser("download")
+    dl.add_argument("video_id")
+    dl.add_argument("-v", "--video", action="store_true")
+
+    args = parser.parse_args()
+
+    if args.version:
+        print(__version__)
+        return
+
+    if args.cmd == "auth":
+        save_key(args.key)
+        print("✅ API key saved")
+        return
+
+    api_key = get_key()
+    if not api_key:
+        print("❌ Please run: brokenx auth <API_KEY>")
+        return
+
+    asyncio.run(run(args, api_key))
+
+
+async def run(args, api_key):
+    async with BrokenXAPI(api_key) as api:
+        if args.cmd == "search":
+            res = await api.search(args.query)
+            print(res)
+
+        elif args.cmd == "download":
+            media = "video" if args.video else "audio"
+            res = await api.download(args.video_id, media)
+            print(res["telegram_url"])

brokenxapi-2.0.2/brokenxapi/client.py

@@ -0,0 +1,46 @@
+import aiohttp
+
+class BrokenXAPI:
+    def __init__(self, api_key: str, base_url: str = "https://mrbroken-brokenxbots.hf.space"):
+        self.api_key = api_key
+        self.base_url = base_url.rstrip("/")
+        self._session = None
+
+    async def __aenter__(self):
+        self._session = aiohttp.ClientSession(
+            headers={
+                "Authorization": f"Bearer {self.api_key}"
+            }
+        )
+        return self
+
+    async def __aexit__(self, exc_type, exc, tb):
+        if self._session:
+            await self._session.close()
+
+    async def _get(self, path: str, params: dict | None = None):
+        url = f"{self.base_url}{path}"
+        async with self._session.get(url, params=params) as r:
+            r.raise_for_status()
+            return await r.json()
+
+    # -------- PUBLIC METHODS --------
+
+    async def search(self, query: str, video: bool = False):
+        return await self._get(
+            "/search",
+            {
+                "q": query,
+                "video": video
+            }
+        )
+
+    async def download(self, video_id: str, media: str = "audio"):
+        if media not in ("audio", "video"):
+            raise ValueError("media must be 'audio' or 'video'")
+
+        return await self._get(
+            f"/download/{media}",
+            {"video_id": video_id}
+        )
+

brokenxapi-2.0.1/brokenxapi/__version__.py

@@ -1,2 +0,0 @@
-__version__ = "2.0.1"
-

brokenxapi-2.0.1/brokenxapi/cli.py

@@ -1,129 +0,0 @@
-import argparse
-import asyncio
-import json
-import os
-import sys
-import stat
-import importlib.metadata
-from pathlib import Path
-
-from brokenxapi import BrokenXAPI
-from brokenxapi.exceptions import BrokenXAPIError
-
-CONFIG_DIR = Path.home() / ".brokenx"
-CONFIG_FILE = CONFIG_DIR / "config.json"
-
-
-# ---------------- CONFIG HELPERS ----------------
-
-def save_api_key(api_key: str):
-    CONFIG_DIR.mkdir(exist_ok=True)
-    CONFIG_FILE.write_text(json.dumps({"api_key": api_key}))
-    CONFIG_FILE.chmod(stat.S_IRUSR | stat.S_IWUSR)  # 600
-
-
-def load_api_key():
-    if not CONFIG_FILE.exists():
-        return None
-    try:
-        data = json.loads(CONFIG_FILE.read_text())
-        return data.get("api_key")
-    except Exception:
-        return None
-
-
-def require_api_key():
-    key = load_api_key()
-    if not key:
-        print(
-            "❌ Not authenticated.\n"
-            "Run: brokenx auth <YOUR_API_KEY>",
-            file=sys.stderr,
-        )
-        sys.exit(1)
-    return key
-
-
-# ---------------- CLI CORE ----------------
-
-async def run_cli():
-    parser = argparse.ArgumentParser(
-        prog="brokenx",
-        description="BROKENXAPI Command Line Interface",
-    )
-
-    parser.add_argument(
-        "-v", "--version",
-        action="store_true",
-        help="Show BROKENXAPI version",
-    )
-
-    sub = parser.add_subparsers(dest="command")
-
-    # ---------- AUTH ----------
-    auth_cmd = sub.add_parser("auth", help="Authenticate with API key")
-    auth_cmd.add_argument("api_key", help="Your BROKENXAPI key")
-
-    # ---------- SEARCH ----------
-    search_cmd = sub.add_parser("search", help="Search YouTube")
-    search_cmd.add_argument("query", help="Search query")
-
-    # ---------- DOWNLOAD ----------
-    download_cmd = sub.add_parser("download", help="Download media")
-    download_cmd.add_argument("video_id", help="YouTube video ID")
-    download_cmd.add_argument(
-        "-v", "--video",
-        action="store_true",
-        help="Download video instead of audio",
-    )
-
-    args = parser.parse_args()
-
-    # ---------------- VERSION ----------------
-    if args.version:
-        print(importlib.metadata.version("BROKENXAPI"))
-        return
-
-    # ---------------- AUTH ----------------
-    if args.command == "auth":
-        try:
-            # 🔍 Verify key by doing a lightweight call
-            async with BrokenXAPI(api_key=args.api_key) as api:
-                await api.search("test")  # harmless validation
-            save_api_key(args.api_key)
-            print("✅ Authentication successful. API key saved.")
-        except BrokenXAPIError as e:
-            print(f"❌ Authentication failed: {e}", file=sys.stderr)
-            sys.exit(1)
-        return
-
-    # ---------------- OTHER COMMANDS ----------------
-    api_key = require_api_key()
-
-    async with BrokenXAPI(api_key=api_key) as api:
-        if args.command == "search":
-            result = await api.search(args.query)
-            print(json.dumps(result, indent=2))
-            return
-
-        if args.command == "download":
-            media_type = "video" if args.video else "audio"
-            result = await api.download(args.video_id, media_type)
-            print(json.dumps(result, indent=2))
-            return
-
-    parser.print_help()
-
-
-def main():
-    try:
-        asyncio.run(run_cli())
-    except KeyboardInterrupt:
-        pass
-    except BrokenXAPIError as e:
-        print(f"❌ {e}", file=sys.stderr)
-        sys.exit(1)
-
-
-if __name__ == "__main__":
-    main()