"""真寻与 NoneBot 共用的运行编排；不导入宿主框架。""" import hashlib from dataclasses import replace import json import logging from pathlib import Path import shlex from ._vendor.farm_core import Catalogue, FarmService, Identity, SQLiteStore from ._vendor.farm_core.domain import FarmError from ._vendor.farm_core.storage import SCHEMA_VERSION from ._vendor.farm_core.presentation import ( plot_ids_text, receipt_text, snapshot_text, ) from ._vendor.farm_render import FarmRenderer, RenderOptions, RenderService from ._vendor.farm_render.seasons import season_for from ._vendor.farm_render.shop import shop_page from ._vendor.farm_render.details import detail_rows, details_text from ._vendor.farm_render.warehouse import ( warehouse_page, warehouse_rows, warehouse_text, ) logger = logging.getLogger(__name__) COMMANDS = { "开通农场":"register", "我的农场":"farm", "农场详述":"details", "我的农场币":"inventory", "我的点券":"inventory", "我的种子":"seeds", "我的作物":"crops", "种子商店":"shop", "农场帮助":"help", "购买种子":"buy", "播种":"sow", "收获":"harvest", "铲除":"eradicate", "作物加锁":"lock", "作物解锁":"unlock", "出售作物":"sell", "农场除虫":"clear_pests", "除虫":"clear_pests", "地块编号":"plot_ids", "偷菜":"steal", } HELP = ( "开通农场 [名称]｜我的农场 [高清]｜农场详述 [页码]\n" "地块编号｜查看当前土地编号及开垦位置\n" "偷菜 @目标｜每日最多5次成功偷取\n" "我的种子/我的作物 [页码] [文字]，默认图片。\n" "全部图片使用农场图片质量配置；仅我的农场支持高清。\n" "种子商店 [页码]｜购买种子 作物 数量｜播种 作物 [数量]\n" "收获｜铲除｜农场除虫｜我的种子 [页码]｜我的作物 [页码]\n" "作物加锁 作物｜作物解锁 作物｜出售作物 [作物] [数量]\n" "作物可填写名称或 #编号；同名作物请使用商店或仓库中的编号。\n" "每成功播种1块地获得2经验；收获经验按作物结算。\n" "每铲除1块枯萎地块获得2经验；每清除1只虫获得1经验。\n" "新播种及下一季每半个基础生长周期生虫一次，间隔至少2小时。\n" "每只虫使生长速度降低5%；成熟时每只虫使本季减产5%。\n" "成熟后除虫不能恢复本季减产；生长期间可发送：农场除虫。\n" "我的农场按北京时间自动切换春夏秋冬背景；高清使用相同季节。\n" "旧版农场数据在首次使用时迁移，绑定当前机器人和平台。" ) ERRORS = { 'cannot_steal_self': '不能偷自己的农场。', 'different_farm_scope': '只能偷取当前机器人、平台内的农场。', 'target_not_registered': '目标尚未开通当前农场。', 'steal_daily_limit': '今日成功偷菜已达5次，北京时间零点恢复。', "legacy_identity_bound":"旧农场已绑定其他机器人或平台，请联系管理员。", "legacy_account_conflict":"新旧农场账号冲突，未覆盖，请联系管理员。", "legacy_schema_unsupported":"旧农场数据库结构不兼容，请联系管理员。", "legacy_data_invalid":"旧农场数据异常，迁移已回滚，请联系管理员。", "not_registered":"请先发送：开通农场", "unknown_plant":"没有这种作物，请查看种子商店。", "ambiguous_plant":"存在同名作物，请使用商店或仓库中的 #编号。", "plant_not_purchasable":"该种子当前不出售，已有种子仍可播种。", "plant_disabled":"该作物已停用，暂不能购买或播种。", "unsupported_crop_type":"该条目不支持作为普通作物种植。", "invalid_integer":"数量必须为有效的正整数。", "insufficient_funds":"余额不足。", "insufficient_stock":"库存不足。", "insufficient_empty_plots":"空地不足。", "level_required":"等级不足。", "plant_locked":"该作物已锁定。", "nothing_to_harvest":"当前没有成熟作物。", "nothing_to_eradicate":"当前没有枯萎作物。", "nothing_to_sell":"当前没有可出售作物。", "nothing_to_clear":"当前没有虫子需要清除。", "operation_conflict":"该事件已处理但参数不一致，请发送新消息。", } def operation_id(bot_id, platform, user_id, message_id, action): if message_id is None or str(message_id) == "": return None return hashlib.sha256(json.dumps( [str(bot_id), platform, user_id, str(message_id), action], ensure_ascii=False, ).encode()).hexdigest() class Runtime: def __init__(self, data_dir, font_path=None, deployment="zhenxun-preview", image_format="AUTO", *, font_dir=None, font_setting="FARM_PREVIEW_FONT", image_quality=75, image_quality_provider=None): RenderOptions(quality=image_quality) self.image_quality = image_quality self.image_quality_provider = image_quality_provider self.data_dir = Path(data_dir).resolve() self.font_path = font_path self.font_setting = font_setting self.font_dir = Path(font_dir or "resources/font") self.deployment = deployment self.image_format = image_format self.store = SQLiteStore(self.data_dir / "farm.sqlite3") self.service = FarmService(self.store, Catalogue.bundled()) self.rendering = None self.started = False self.log_handler = None self.legacy_source = self.data_dir / 'farm.db' async def start(self): if self.started: return self.data_dir.mkdir(parents=True, exist_ok=True) self.log_handler = logging.FileHandler( self.data_dir / "farm-preview.log", encoding="utf-8", ) self.log_handler.setFormatter(logging.Formatter( "%(asctime)s %(levelname)s %(name)s %(message)s", )) logging.getLogger(__package__).setLevel(logging.INFO) logging.getLogger(__package__).addHandler(self.log_handler) try: await self.store.initialize(catalogue=self.service.catalogue) except Exception: logger.error("farm preview database initialization failed") await self.close() raise if self.font_path: candidates = [Path(self.font_path)] else: candidates = [self.font_dir / name for name in ( "HYWenHei-85W.ttf", "msyh.ttf", "syht.otf", "syst.otf", "AlibabaHealthFont2.0CN-85B.ttf", )] candidates.extend(Path(name) for name in ( "C:/Windows/Fonts/msyh.ttc", "C:/Windows/Fonts/simsun.ttc", "/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc", "/usr/share/fonts/truetype/wqy/wqy-microhei.ttc", )) for candidate in candidates: if not candidate.is_file(): continue try: self.rendering = RenderService(FarmRenderer(candidate)) logger.info("farm preview font selected file=%s", candidate.name) break except (ValueError, OSError): logger.warning("farm preview font candidate is not loadable") if self.rendering is None: logger.warning("farm preview font unavailable; text mode only") self.started = True logger.info("farm preview initialized schema=%s", SCHEMA_VERSION) async def close(self): try: if self.rendering: await self.rendering.close() finally: await self.store.close() self.started = False if self.log_handler: logging.getLogger(__package__).removeHandler(self.log_handler) self.log_handler.close() self.log_handler = None async def execute( self, action, arguments, *, bot_id, platform, user_id, message_id, display_name="农场玩家", avatar=None, target_user_id=None, ): if not self.started: return "农场尚未初始化，请检查启动日志。" try: who = Identity(self.deployment, str(bot_id), platform, str(user_id)) try: await self.store.migrate_legacy( self.legacy_source, who, self.service.catalogue, self.service.clock(), ) except (FarmError, KeyError, TypeError) as error: logger.exception('farm legacy migration failed') code = getattr(error, 'code', '') return ERRORS.get( code if code.startswith('legacy_') else '', '旧农场数据无法迁移，请联系管理员检查日志。', ) args = shlex.split(arguments) if action == "farm" and args: aliases = { "detail":"details", "my-point":"inventory", "my-vipPoint":"inventory", "my-seed":"seeds", "my-plant":"crops", "seed-shop":"shop", "buy-seed":"buy", "sowing":"sow", "harvest":"harvest", "eradicate":"eradicate", "lock-plant":"lock", "unlock-plant":"unlock", "sell-plant":"sell", "farm-help":"help", "clear-pests":"clear_pests", } if args[0] in aliases: action = aliases[args[0]] args = args[1:] op = operation_id( bot_id, platform, str(user_id), message_id, action, ) if action == "help": return HELP image_actions = ('farm', 'plot_ids', 'details', 'shop', 'seeds', 'crops') if action in image_actions and action != 'farm' and '高清' in args: return '仅我的农场支持高清，请移除高清参数后重试。' quality = self.image_quality text_mode = (action in ('seeds', 'crops') and args and args[-1] == '文字') if action in image_actions and self.rendering and not text_mode: try: if self.image_quality_provider is not None: quality = self.image_quality_provider() RenderOptions(quality=quality) except (TypeError, ValueError): logger.warning('farm image quality config invalid') return '农场图片质量配置无效，请设置为1～95的整数。' if action == "shop": page = int(args[0]) if args else 1 if len(args) > 1: raise ValueError() try: chunk, pages = shop_page(self.service.catalogue, page) except ValueError: return '页码无效，请发送：种子商店 1' if self.rendering: account = None try: account = (await self.service.snapshot(who)).account except FarmError as error: if error.code != 'not_registered': raise return await self.rendering.seed_shop( chunk, account, page, pages, image_format=self.image_format, quality=quality, ) return f'种子商店 第{page}/{pages}页\n' + '\n'.join( f'{p.name}（#{p.plant_id}）：' f'{p.cost}{"点券" if p.is_vip else "农场币"}' f' / 等级{p.level}' for p in chunk ) + ( '\n购买种子 #编号 数量 或 购买种子 种子名 数量' + (f'\n下一页：种子商店 {page+1}' if page < pages else '') ) if action == 'details': if len(args) > 1: raise ValueError() page = int(args[0]) if args else 1 snapshot = await self.service.snapshot(who) now = self.service.clock() try: rows, pages = detail_rows( self.rendering.renderer if self.rendering else None, who, snapshot, now, page, ) except ValueError: return '页码无效，请发送：农场详述 1' if not self.rendering: return details_text(rows, page, pages) return await self.rendering.farm_details( who, snapshot, now, page, season_for(now), image_format=self.image_format, avatar=avatar, quality=quality, ) if action in ('crops', 'seeds'): title = '我的作物' if action == 'crops' else '我的种子' mode = args.pop() if args and args[-1] == '文字' else '' if len(args) > 1: raise ValueError() try: page = int(args[0]) if args else 1 except ValueError: return f'页码无效，请发送：{title} 1' snapshot = await self.service.snapshot(who) rows = warehouse_rows(snapshot, self.service.catalogue, action) try: _, pages = warehouse_page(rows, page) except ValueError: return f'页码无效，请发送：{title} 1' if not self.rendering or mode == '文字': return warehouse_text(rows, snapshot, action, page, pages) return await self.rendering.warehouse( rows, snapshot, action, page, pages, self.image_format, quality=quality, ) if action == "inventory": if args: raise ValueError() return snapshot_text(await self.service.snapshot(who)) if action in ("farm", "plot_ids"): if args not in ([], ["高清"]): raise ValueError() if not self.rendering: if action == "plot_ids": return plot_ids_text(await self.service.snapshot(who)) return f"请配置 {self.font_setting} 为可用中文字体路径。" now = self.service.clock() theme = season_for(now) logger.info("farm preview seasonal theme=%s", theme) options = ( RenderOptions(width=1536, quality=max(quality, 85), format=self.image_format, theme=theme) if args else RenderOptions(format=self.image_format, theme=theme, quality=quality) ) if action == "plot_ids": options = replace(options, show_plot_ids=True) picture = await self.rendering.my_farm( self.service, who, options=options, avatar=avatar, supported=("JPEG",), now=now, ) return picture.content if op is None: return "无法取得稳定消息标识，本次写操作未执行。" if action == 'steal': if target_user_id is None: return '请发送：偷菜 @目标，不支持直接输入QQ号。' if args: return '请只指定一个偷菜目标，不要附加QQ号或其他文字。' target = Identity(who.deployment, who.realm, who.platform, str(target_user_id)) receipt = await self.service.steal(who, target, op) elif action == "register": name = " ".join(args) if args else display_name name = " ".join(str(name).split())[:64] or "农场玩家" receipt = await self.service.register(who, name, op) elif action in ("buy", "sow", "sell"): if action == "sell" and not args: receipt = await self.service.sell(who, None, None, op) else: if len(args) not in (1, 2) or (action == "buy" and len(args) != 2): raise ValueError() count = int(args[1]) if len(args) == 2 else None receipt = await getattr(self.service, action)( who, args[0], count, op, ) elif action in ("lock", "unlock"): if len(args) != 1: raise ValueError() receipt = await self.service.lock( who, args[0], action == "lock", op, ) elif action in ("harvest", "eradicate", "clear_pests"): if args: raise ValueError() receipt = await getattr(self.service, action)(who, op) else: return HELP text = receipt_text(receipt) if action == 'register': if receipt.data['created']: count = receipt.data.get('plot_count') granted = (f'已获得{count}块土地' if count is not None else '已获得初始土地') text += f'\n{granted}，发送「地块编号」查看位置。' else: text += '\n发送「地块编号」查看土地位置。' return text except FarmError as error: return ERRORS.get(error.code, "操作未完成：" + error.code) except ValueError: return "参数有误，请发送：农场帮助"