:recycle: refactor(generator): rename Generator to TextGenerator and add TODO

2 files changed, 6 insertions, 6 deletions

diff --git a/src/infini/generator.py b/src/infini/generator.py
index 3f386010..cc752d70 100644
--- a/src/infini/generator.py
+++ b/src/infini/generator.py
@@ -4,7 +4,7 @@ from infini.exceptions import UnknownEvent
 from jinja2 import Template
 
 
-class Generator:
+class TextGenerator:  # TODO 兼容多类型事件
     events: Dict[str, str]
     global_variables: Dict[str, str | Callable]
 
diff --git a/src/infini/loader.py b/src/infini/loader.py
index 1dbe7779..ba7eb546 100644
--- a/src/infini/loader.py
+++ b/src/infini/loader.py
@@ -1,6 +1,6 @@
 from importlib.util import spec_from_file_location
 from infini.core import Core
-from infini.generator import Generator
+from infini.generator import TextGenerator
 from infini.handler import Handler
 from infini.interceptor import Interceptor
 from infini.register import Register
@@ -147,7 +147,7 @@ class Loader:
     def inject_core(self, core: Core):
         pre_interceptor = Interceptor()
         handler = Handler()
-        generator = Generator()
+        generator = TextGenerator()
         interceptor = Interceptor()
 
         self.inject_pre_interceptor(pre_interceptor)
@@ -192,12 +192,12 @@ class Loader:
         self.inject_handler(handler)
         return handler
 
-    def inject_generator(self, generator: Generator):
+    def inject_generator(self, generator: TextGenerator):
         generator.events = self.events
         generator.global_variables = self.global_variables
 
-    def into_generator(self) -> Generator:
-        generator = Generator()
+    def into_generator(self) -> TextGenerator:
+        generator = TextGenerator()
         self.inject_generator(generator)
         return generator