first commit

2026-10-15 18:27:46 +08:00
commit 68eb19ea35
88 changed files with 10319 additions and 0 deletions
@@ -0,0 +1,77 @@
+"""微信机器人框架的统一入口：python main.py。"""
+from __future__ import annotations
+
+import argparse
+import json
+import logging
+import math
+from logging.handlers import RotatingFileHandler
+from pathlib import Path
+import sys
+
+from wxbot_framework.config import ConfigStore
+from wxbot_framework.locking import InstanceLock
+from wxbot_framework.runtime import BotApplication
+
+ROOT = Path(__file__).resolve().parent
+
+
+def configure_logging(config):
+    directory = ROOT / "logs"
+    directory.mkdir(exist_ok=True)
+    formatter = logging.Formatter("%(asctime)s %(levelname)s [%(name)s] %(message)s")
+    console = logging.StreamHandler()
+    console.setFormatter(formatter)
+    file = RotatingFileHandler(directory / "bot.log", maxBytes=2 * 1024 * 1024,
+                               backupCount=3, encoding="utf-8")
+    file.setFormatter(formatter)
+    logging.basicConfig(level=config.get("log_level", "INFO").upper(), handlers=[console, file], force=True)
+
+
+def main():
+    if hasattr(sys.stdout, "reconfigure"):
+        sys.stdout.reconfigure(encoding="utf-8")
+        sys.stderr.reconfigure(encoding="utf-8")
+    parser = argparse.ArgumentParser(description=__doc__)
+    parser.add_argument("--config", type=Path, default=ROOT / "bot_config.json")
+    parser.add_argument("--check", action="store_true", help="检查配置、清单及启用插件语法，不注入微信")
+    parser.add_argument("--list-plugins", action="store_true", help="列出可用插件，不注入微信")
+    parser.add_argument("--run-seconds", type=float, default=0, help="有限时长检查；默认 0 持续运行")
+    args = parser.parse_args()
+    if not math.isfinite(args.run_seconds) or args.run_seconds < 0:
+        parser.error("运行时长必须是非负有限数值")
+    try:
+        store = ConfigStore(args.config)
+        config = store.load()
+        if not (args.check or args.list_plugins) and config.get("wechat_account") == "wxid_example_robot":
+            parser.error("请先在本地 bot_config.json 中将 wechat_account 占位符改为自己的机器人账号 ID")
+        app = BotApplication(ROOT, store)
+        if args.check or args.list_plugins:
+            specs = app.plugins.discover()
+            if app.plugins.load_errors:
+                raise ValueError(str(app.plugins.load_errors))
+            for spec in specs.values():
+                enabled = config.get("plugins", {}).get(spec.name, {}).get("enabled", True)
+                if args.check and enabled:
+                    for source in spec.path.parent.rglob("*.py"):
+                        compile(source.read_bytes(), str(source), "exec")
+                    if spec.name in ("idiom_chain", "chat_summary", "mention_chat"):
+                        from wxbot_framework.llm import LLMClient
+                        LLMClient.from_file(store.path.parent / config.get("llm_config_file", "llm_config.json"))
+                print(json.dumps({"name": spec.name, "enabled": enabled, "api": spec.api,
+                                  "description": spec.description,
+                                  "missing_capabilities": sorted(spec.requires - app.plugins.capabilities)}, ensure_ascii=False))
+            if args.check:
+                print("配置与插件语法检查通过")
+            return 0
+        configure_logging(config)
+        with InstanceLock(ROOT / ".runtime/bot.lock"):
+            app.run(seconds=args.run_seconds)
+        return 0
+    except Exception:
+        logging.getLogger("Main").exception("启动或运行失败")
+        return 1
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())