diff options
| author | 2024-02-26 13:21:50 +0800 | |
|---|---|---|
| committer | 2024-02-26 13:21:50 +0800 | |
| commit | e6e9453a024dff943ea52b4fb588f2cf7fec509b (patch) | |
| tree | 966fb87d7f01d3c916ef3fe66ec14897ec3b0069 /examples/plugins/draftbottles/__init__.py | |
| parent | 3eb819d7c48020df35336e141c38ab4d9926c21e (diff) | |
| download | HydroRoll-e6e9453a024dff943ea52b4fb588f2cf7fec509b.tar.gz HydroRoll-e6e9453a024dff943ea52b4fb588f2cf7fec509b.zip | |
refactor(command): 命令词法解析器 (#82)
* refactor(command): 重构命令路由
* feat(lua): 包装异步方法`self.event.reply` => `msg:echo`
* feat(lua): 包装异步输入流方法`self.event.ask` => `msg:ask`
* 'Refactored by Sourcery' (#83)
Co-authored-by: Sourcery AI <>
* feat(Token|Lexer): 添加`Token`与`Lexer`类
* refactor(Lexer): 词法分析器添加`advance`方法
* chore: lint code
* refactor: sync gensokyo adapter
* feat: Cli parser (#85)
* feat(cli): 添加`Cli`类,解析命令行参数
* fix: 修复错误的`dest`与`action`
* feat(cli): 实现`install_package` 与 `build_template`
* feat(cli): 实现`-c|--config`指令配置镜像常量等
* feat(cli): 使用高效率的异步网络库`aiohttp`
* fix(cli): `TYPE_CHECKING` with partially module `typing`
* refactor!: examples, tests, src... rewrite in rust
---------
Co-authored-by: sourcery-ai[bot] <58596630+sourcery-ai[bot]@users.noreply.github.com>
Diffstat (limited to 'examples/plugins/draftbottles/__init__.py')
| -rw-r--r-- | examples/plugins/draftbottles/__init__.py | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/examples/plugins/draftbottles/__init__.py b/examples/plugins/draftbottles/__init__.py new file mode 100644 index 0000000..1a71f1d --- /dev/null +++ b/examples/plugins/draftbottles/__init__.py @@ -0,0 +1,48 @@ +from typing import Union +from iamai import Plugin, Event, Depends +from iamai.log import logger +from .config import Config +from iamai.event import MessageEvent +from .database import Database +from .permission import Permission +from .workroutes import WorkRoutes +from .inspector import Inspector + + +class Bottles(Plugin, config=Config): + database: Database = Depends() + permission: Permission = Depends() + workroutes: WorkRoutes = Depends() + inspector: Inspector = Depends() + + def __init__(self): + self.text = None + self.prefix = None + self.suffix = None + + async def handle(self) -> None: + self.namespace = next( + ( + key + for key, value in self.config.command_list.items() + if value == self.prefix + ), + "", + ) + if method := getattr(self.inspector, self.namespace, None): + result = await method(self.suffix, self.config) + if result: + await self.event.reply(result) + + async def rule(self) -> bool: + if not isinstance(self.event, MessageEvent): + return False + if not self.permission.is_admin(): + return False + self.text = self.event.get_plain_text() + for prefix in list(self.config.command_list.values()): + if self.text.startswith(prefix): + self.prefix = prefix + self.suffix = self.text[len(self.prefix) + 1 :] + return True + return False |
