chore: commit all changes

examples/aur_mcp_server.py

@@ -0,0 +1,196 @@
+import aiohttp
+import asyncio
+from datetime import datetime
+import json
+from mcp.server.fastmcp import FastMCP
+
+# 初始化 FastMCP 服务器
+mcp = FastMCP("ArchLinux-Pkg-Search")
+
+async def process_aur_info_response(task_coro):
+    try:
+        resp = await task_coro
+        resp.raise_for_status()
+        data = await resp.json()
+        if isinstance(data, dict) and "results" in data and isinstance(data["results"], list):
+             return data
+        else:
+             return None
+    except Exception:
+        return None
+
+@mcp.tool()
+async def search_pkg(pkg_name: str, repo: str = None) -> str:
+    """
+    搜索 Arch Linux 官方仓库和 AUR 的软件包。
+    
+    Args:
+        pkg_name: 要搜索的包名 (例如 linux, yay)
+        repo: 可选，指定官方仓库 (例如 core, extra)
+    """
+    if repo:
+        repo = repo[0].upper() + repo[1:]
+
+    timeout = aiohttp.ClientTimeout(total=10)
+
+    async with aiohttp.ClientSession(timeout=timeout) as session:
+        # 1. 首先尝试搜索官方仓库
+        search_url = f"https://archlinux.org/packages/search/json/?name={pkg_name}"
+        if repo:
+            search_url += f"&repo={repo}"
+
+        try:
+            async with session.get(search_url) as resp:
+                resp.raise_for_status()
+                data = await resp.json()
+                results = data.get("results", [])
+
+                if results:
+                    result = results[0]
+                    last_update_str = "N/A"
+                    if result.get("last_update"):
+                        try:
+                            dt_obj = datetime.fromisoformat(result["last_update"].replace("Z", "+00:00"))
+                            last_update_str = dt_obj.strftime('%Y-%m-%d %H:%M:%S')
+                        except ValueError:
+                            last_update_str = result.get('last_update', 'N/A').replace("T", " ").replace("Z", "")
+
+                    msg = (
+                        f"仓库：{result.get('repo', 'N/A')}
+"
+                        f"包名：{result.get('pkgname', 'N/A')}
+"
+                        f"版本：{result.get('pkgver', 'N/A')}
+"
+                        f"描述：{result.get('pkgdesc', 'N/A')}
+"
+                        f"打包：{result.get('packager', 'N/A')}
+"
+                        f"上游：{result.get('url', 'N/A')}
+"
+                        f"更新日期：{last_update_str}"
+                    )
+                    return msg
+
+        except Exception as e:
+            pass # 官方仓库失败则继续查 AUR
+
+        # 2. 尝试 AUR
+        aur_suggest_url = f"https://aur.archlinux.org/rpc/v5/suggest/{pkg_name}"
+        suggestions = []
+        try:
+            async with session.get(aur_suggest_url) as resp:
+                resp.raise_for_status()
+                suggestions = await resp.json()
+                if not isinstance(suggestions, list):
+                     suggestions = []
+        except Exception:
+            pass
+
+        if not suggestions:
+             suggestions = [pkg_name]
+
+        aur_info_base_url = "https://aur.archlinux.org/rpc/v5/info/"
+        target_pkg_info = None
+
+        if len(suggestions) == 1 or suggestions[0] == pkg_name:
+            target_name = suggestions[0]
+            aur_info_url = f"{aur_info_base_url}{target_name}"
+            try:
+                async with session.get(aur_info_url) as resp:
+                     resp.raise_for_status()
+                     search_map = await resp.json()
+                     if search_map.get("results") and isinstance(search_map["results"], list) and len(search_map["results"]) > 0:
+                         target_pkg_info = search_map["results"][0]
+            except Exception:
+                 pass
+        else:
+            fetch_tasks = []
+            for suggestion in suggestions:
+                 fetch_tasks.append(process_aur_info_response(session.get(f"{aur_info_base_url}{suggestion}")))
+
+            aur_responses = await asyncio.gather(*fetch_tasks)
+
+            best_result = None
+            max_votes = -1.0
+
+            for result_data in aur_responses:
+                 if isinstance(result_data, Exception) or result_data is None:
+                      continue
+
+                 if result_data.get("results"):
+                     pkg_info = result_data["results"][0]
+                     try:
+                         votes = float(pkg_info.get("NumVotes", 0.0) or 0.0)
+                     except (ValueError, TypeError):
+                         votes = 0.0
+                     if votes > max_votes:
+                         max_votes = votes
+                         best_result = pkg_info
+
+            if best_result:
+                target_pkg_info = best_result
+
+        # 3. 格式化并返回 AUR 结果
+        if target_pkg_info:
+            maintainer = target_pkg_info.get("Maintainer") or "孤儿包"
+            out_of_date_ts = target_pkg_info.get("OutOfDate")
+            out_of_date_str = ""
+            if out_of_date_ts:
+                try:
+                    out_of_date_dt = datetime.fromtimestamp(float(out_of_date_ts))
+                    out_of_date_str = f"过期时间：{out_of_date_dt.strftime('%Y-%m-%d %H:%M:%S')}
+"
+                except (ValueError, TypeError, OSError):
+                     pass
+
+            upstream_url = target_pkg_info.get("URL") or "无"
+            co_maintainers = target_pkg_info.get("CoMaintainers") or []
+            co_maintainers_str = ""
+            if co_maintainers and isinstance(co_maintainers, list):
+                 co_maintainers_str = f" ( {' '.join(map(str, co_maintainers))} )"
+
+            last_modified_ts = target_pkg_info.get("LastModified")
+            last_modified_str = "N/A"
+            if last_modified_ts:
+                try:
+                     last_modified_dt = datetime.fromtimestamp(float(last_modified_ts))
+                     last_modified_str = last_modified_dt.strftime('%Y-%m-%d %H:%M:%S')
+                except (ValueError, TypeError, OSError):
+                     pass
+
+            num_votes = 0.0
+            try:
+                num_votes = float(target_pkg_info.get("NumVotes", 0.0) or 0.0)
+            except (ValueError, TypeError):
+                pass
+
+            pkg_display_name = target_pkg_info.get('Name', 'N/A')
+
+            msg = (
+                 f"仓库：AUR
+"
+                 f"包名：{pkg_display_name}
+"
+                 f"版本：{target_pkg_info.get('Version', 'N/A')}
+"
+                 f"描述：{target_pkg_info.get('Description', 'N/A')}
+"
+                 f"维护者：{maintainer}{co_maintainers_str}
+"
+                 f"上游：{upstream_url}
+"
+                 f"{out_of_date_str}"
+                 f"更新时间：{last_modified_str}
+"
+                 f"投票：{num_votes:.0f}
+"
+                 f"AUR 链接：https://aur.archlinux.org/packages/{pkg_display_name}"
+             )
+            return msg
+
+        return f"没有在官方仓库或 AUR 中找到名为 '{pkg_name}' 的相关软件。"
+
+if __name__ == "__main__":
+    # 以 stdio 模式运行 MCP 服务器
+    mcp.run(transport='stdio')