From 71d2cb27844699710a3976ee49347d6fc28dfcaa Mon Sep 17 00:00:00 2001 From: 苏向夜 Date: Mon, 29 Jan 2024 18:13:43 +0800 Subject: fix(router): add Command router --- src/infini/router.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'src') diff --git a/src/infini/router.py b/src/infini/router.py index 127f83be..c84e11a4 100644 --- a/src/infini/router.py +++ b/src/infini/router.py @@ -38,3 +38,17 @@ class Endswith(Router): def match(self, input: str) -> bool: input = input.strip() return any([input.endswith(sign) for sign in self.signs]) + + +class Command(Router): + type: Literal["command"] = "command" + prefix: tuple = (".", "/") + + def match(self, input: str) -> bool: + input = input.strip() + if input: + if input.startswith(self.prefix): + input = input[1:] + return any([input.startswith(sign) for sign in self.signs]) + + return False -- cgit v1.2.3-70-g09d2