|
| 1 | +from __future__ import annotations |
| 2 | + |
| 3 | +from typing import Optional |
| 4 | +from urllib.parse import quote |
| 5 | + |
| 6 | +import httpx |
| 7 | + |
| 8 | +from skillkit.models import ( |
| 9 | + CacheStats, |
| 10 | + CategoriesResponse, |
| 11 | + Category, |
| 12 | + HealthResponse, |
| 13 | + SearchResponse, |
| 14 | + Skill, |
| 15 | + TrendingResponse, |
| 16 | +) |
| 17 | + |
| 18 | + |
| 19 | +class SkillKitClient: |
| 20 | + def __init__(self, base_url: str = "http://localhost:3737", timeout: float = 30.0): |
| 21 | + self.base_url = base_url.rstrip("/") |
| 22 | + self.timeout = timeout |
| 23 | + self._client: Optional[httpx.AsyncClient] = None |
| 24 | + |
| 25 | + async def __aenter__(self) -> "SkillKitClient": |
| 26 | + self._client = httpx.AsyncClient(base_url=self.base_url, timeout=self.timeout) |
| 27 | + return self |
| 28 | + |
| 29 | + async def __aexit__(self, *args: object) -> None: |
| 30 | + if self._client: |
| 31 | + await self._client.aclose() |
| 32 | + self._client = None |
| 33 | + |
| 34 | + def _get_client(self) -> httpx.AsyncClient: |
| 35 | + if self._client is None: |
| 36 | + self._client = httpx.AsyncClient(base_url=self.base_url, timeout=self.timeout) |
| 37 | + return self._client |
| 38 | + |
| 39 | + async def health(self) -> HealthResponse: |
| 40 | + client = self._get_client() |
| 41 | + response = await client.get("/health") |
| 42 | + response.raise_for_status() |
| 43 | + data = response.json() |
| 44 | + return HealthResponse( |
| 45 | + status=data["status"], |
| 46 | + version=data["version"], |
| 47 | + skill_count=data.get("skillCount", 0), |
| 48 | + uptime=data.get("uptime", 0), |
| 49 | + ) |
| 50 | + |
| 51 | + async def search( |
| 52 | + self, |
| 53 | + query: str, |
| 54 | + limit: int = 20, |
| 55 | + include_content: bool = False, |
| 56 | + ) -> SearchResponse: |
| 57 | + client = self._get_client() |
| 58 | + params = {"q": query, "limit": str(limit)} |
| 59 | + if include_content: |
| 60 | + params["include_content"] = "true" |
| 61 | + response = await client.get("/search", params=params) |
| 62 | + response.raise_for_status() |
| 63 | + data = response.json() |
| 64 | + return SearchResponse( |
| 65 | + skills=[Skill(**s) for s in data["skills"]], |
| 66 | + total=data["total"], |
| 67 | + query=data["query"], |
| 68 | + limit=data["limit"], |
| 69 | + ) |
| 70 | + |
| 71 | + async def search_with_filters( |
| 72 | + self, |
| 73 | + query: str, |
| 74 | + limit: int = 20, |
| 75 | + include_content: bool = False, |
| 76 | + tags: Optional[list[str]] = None, |
| 77 | + category: Optional[str] = None, |
| 78 | + source: Optional[str] = None, |
| 79 | + ) -> SearchResponse: |
| 80 | + client = self._get_client() |
| 81 | + body: dict = {"query": query, "limit": limit, "include_content": include_content} |
| 82 | + filters: dict = {} |
| 83 | + if tags: |
| 84 | + filters["tags"] = tags |
| 85 | + if category: |
| 86 | + filters["category"] = category |
| 87 | + if source: |
| 88 | + filters["source"] = source |
| 89 | + if filters: |
| 90 | + body["filters"] = filters |
| 91 | + response = await client.post("/search", json=body) |
| 92 | + response.raise_for_status() |
| 93 | + data = response.json() |
| 94 | + return SearchResponse( |
| 95 | + skills=[Skill(**s) for s in data["skills"]], |
| 96 | + total=data["total"], |
| 97 | + query=data["query"], |
| 98 | + limit=data["limit"], |
| 99 | + ) |
| 100 | + |
| 101 | + async def get_skill(self, source: str, skill_id: str) -> Skill: |
| 102 | + client = self._get_client() |
| 103 | + parts = source.split("/", 1) |
| 104 | + owner = quote(parts[0], safe="") |
| 105 | + repo = quote(parts[1], safe="") if len(parts) > 1 else owner |
| 106 | + response = await client.get(f"/skills/{owner}/{repo}/{quote(skill_id, safe='')}") |
| 107 | + response.raise_for_status() |
| 108 | + return Skill(**response.json()) |
| 109 | + |
| 110 | + async def trending(self, limit: int = 20) -> list[Skill]: |
| 111 | + client = self._get_client() |
| 112 | + response = await client.get("/trending", params={"limit": str(limit)}) |
| 113 | + response.raise_for_status() |
| 114 | + data = response.json() |
| 115 | + return [Skill(**s) for s in data["skills"]] |
| 116 | + |
| 117 | + async def categories(self) -> list[Category]: |
| 118 | + client = self._get_client() |
| 119 | + response = await client.get("/categories") |
| 120 | + response.raise_for_status() |
| 121 | + data = response.json() |
| 122 | + return [Category(**c) for c in data["categories"]] |
| 123 | + |
| 124 | + async def cache_stats(self) -> CacheStats: |
| 125 | + client = self._get_client() |
| 126 | + response = await client.get("/cache/stats") |
| 127 | + response.raise_for_status() |
| 128 | + data = response.json() |
| 129 | + return CacheStats( |
| 130 | + hits=data["hits"], |
| 131 | + misses=data["misses"], |
| 132 | + size=data["size"], |
| 133 | + max_size=data.get("maxSize", 0), |
| 134 | + hit_rate=data.get("hitRate", 0.0), |
| 135 | + ) |
| 136 | + |
| 137 | + async def close(self) -> None: |
| 138 | + if self._client: |
| 139 | + await self._client.aclose() |
| 140 | + self._client = None |
0 commit comments