"""各平台适配器共用的运行编排；不导入宿主框架。""" import hashlib import asyncio 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.commands import ( command_map, command_specs, help_text, ) 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.bundle import ensure_assets, validate_options 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 = command_map('zhenxun') HELP = help_text(command_specs('zhenxun')) ERRORS = { 'land_origin_unknown': '农场初始土地来源无法确认，未补地或扣币，' '请联系管理员核查注册或迁移记录。', 'land_compensation_capacity': '旧用户补3块后将超出当前30个绘图位置，' '已保留原状，请联系管理员单独处理补偿。', 'land_layout_invalid': '地块编号或数量异常，未补地或扣币，请联系管理员核查。', 'land_rules_invalid': '开垦配置异常，未扣币，请联系管理员检查。', 'land_rule_unavailable': '下一块地的开垦规则尚未配置，未扣币。', '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, framework='zhenxun', resource_source='Github', service_url='', resource_options_provider=None): RenderOptions(quality=image_quality) validate_options(resource_source, service_url) self.resource_source = resource_source self.service_url = service_url self.resource_options_provider = resource_options_provider self.framework = framework self.command_specs = command_specs(framework) self.commands = command_map(framework) 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", )) try: if self.resource_options_provider is not None: self.resource_source, self.service_url = ( self.resource_options_provider() ) bundle_path = await asyncio.to_thread( ensure_assets, self.data_dir, self.resource_source, self.service_url, ) except Exception: # 不将下载URL、服务地址或可能带凭据的网络异常写入日志。 logger.error('farm assets unavailable; using text; ' 'check resource source and service address') bundle_path = None for candidate in candidates if bundle_path is not None else (): if not candidate.is_file(): continue try: self.rendering = RenderService(FarmRenderer( candidate, bundle_path=bundle_path, )) 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: args = shlex.split(arguments) # 帮助无需注册或迁移玩家数据，绘图失败可直接阅读文字。 if action == 'help': return await self.help_menu(args) 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 await self.help_menu(args) if action == 'reclaim' and args == ['条件']: conditions = await self.service.reclaim_conditions(who) return ( f"下一块：{conditions['slot']}号普通空地\n" f"所需等级：{conditions['required_level']}" f"（当前{conditions['level']}）\n" f"所需农场币：{conditions['coin_cost']}" f"（当前{conditions['point']}）\n" '满足条件后发送：开垦' ) 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", "reclaim"): if args: raise ValueError() receipt = await getattr(self.service, action)(who, op) else: return help_text(self.command_specs) 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 "参数有误，请发送：农场帮助" async def help_menu(self, args): if '高清' in args: return '仅我的农场支持高清，请移除高清参数后重试。' if args not in ([], ['文字']): return '参数有误，请发送：农场帮助 或 农场帮助 文字' fallback = help_text(self.command_specs) if args or self.rendering is None: return fallback quality = self.image_quality if self.image_quality_provider is not None: quality = self.image_quality_provider() try: RenderOptions(quality=quality) except (ValueError, TypeError): logger.warning('farm help image quality config invalid') return '农场图片质量配置无效，请设置为1～95的整数。' try: return await self.rendering.help_menu( self.command_specs, self.framework, self.image_format, quality, ) except (OSError, ValueError, RuntimeError): logger.exception('farm help rendering failed; using text') return fallback