src/infini/handler.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

from infini.input import Input
from infini.output import Output
from infini.typing import List, RouterType, Callable
from infini.queue import EventQueue


class Handler:
    handlers: List[RouterType]

    def input(self, input: Input):
        queue = self.match(input.get_plain_text())
        if queue.is_empty():
            yield Output.empty()
            return
        while not queue.is_empty():
            output = queue.pop()(input)
            # TODO 允许 Handler 函数为迭代器
            # if isinstance(output, GeneratorType):
            #     for o in output:
            #         yield o
            yield output
            if output.block:
                return

    def match(self, text: str) -> EventQueue[Callable[[Input], Output]]:
        queue = EventQueue()

        for handler in self.handlers:
            if handler["router"].match(text):
                queue.push(handler["priority"], handler["handler"])

        return queue