✨ codegen events

Author: yanyongyu

codegen/__init__.py

@@ -1,6 +1,11 @@
-from typing import List
+import inspect
+from pathlib import Path
+from itertools import chain
 from dataclasses import dataclass
+from typing import List, Union, get_args, get_origin
 
+from jinja2 import Environment, PackageLoader
+from githubkit.webhooks.models import GitHubWebhookModel
 from githubkit.webhooks.types import webhook_action_types
 
 
@@ -10,11 +15,35 @@ class Data:
     payload_type: str
 
 
+def pascal_case(*names: str) -> str:
+    words = chain.from_iterable(name.split("_") for name in names)
+    return "".join(word if word.isupper() else word.capitalize() for word in words)
+
+
 def build_event():
+    imports: List[str] = []
     events: List[Data] = []
     for event, types in webhook_action_types.items():
         if not isinstance(types, dict):
-            events.append(Data(types.__name__, types.__name__))
+            imports.append(f"{types.__name__} as {types.__name__}Payload")
+            events.append(Data(types.__name__, f"{types.__name__}Payload"))
+            continue
 
         for action, type in types.items():
-            ...
+            if inspect.isclass(type) and issubclass(type, GitHubWebhookModel):
+                imports.append(f"{type.__name__} as {type.__name__}Payload")
+                events.append(Data(type.__name__, f"{type.__name__}Payload"))
+            else:
+                assert get_origin(type) is Union
+                imports.extend(model.__name__ for model in get_args(type))
+                events.append(
+                    Data(
+                        pascal_case(event, action),
+                        f"Union[{', '.join(model.__name__ for model in get_args(type))}]",
+                    )
+                )
+
+    env = Environment(loader=PackageLoader("codegen"))
+    template = env.get_template("event.py.jinja")
+    event_text = template.render(imports=imports, events=events)
+    Path("./nonebot/adapters/github/event.py").write_text(event_text)

codegen/templates/event.py.jinja

@@ -1,7 +1,11 @@
-from typing import Any
+from typing import Any, Union
 
+{% for import_ in imports %}
+from githubkit.webhooks.models import {{ import_ }}
+{% endfor %}
 from githubkit.utils import UNSET
 from nonebot.typing import overrides
+from nonebot.utils import escape_tag
 
 from nonebot.adapters import Event as BaseEvent
 
@@ -19,14 +23,24 @@ class Event(BaseEvent):
 
     @overrides(BaseEvent)
     def get_event_name(self) -> str:
-        return self.event_name + f".{action}" if (action := getattr(self, "action", UNSET)) else ""
+        return self.name + (
+            f".{action}" if (action := getattr(self.payload, "action", None)) else ""
+        )
 
     @overrides(BaseEvent)
     def get_event_description(self) -> str:
         return escape_tag(
-            f"{self.__class__.__name__} " +
-            (f"from sender {sender.login}" if (sender := getattr(self, "sender", UNSET)) else "") +
-            (f"in repository {repo.full_name}" if (repo := getattr(self, "repository", UNSET)) else "") +
+            f"{self.__class__.__name__} "
+            + (
+                f"from sender {sender.login}"
+                if (sender := getattr(self.payload, "sender", None))
+                else ""
+            )
+            + (
+                f"in repository {repo.full_name}"
+                if (repo := getattr(self.payload, "repository", None))
+                else ""
+            )
         )
 
     @overrides(BaseEvent)
@@ -35,7 +49,7 @@ class Event(BaseEvent):
 
     @overrides(BaseEvent)
     def get_user_id(self) -> str:
-        if sender := getattr(self, "sender", UNSET):
+        if sender := getattr(self.payload, "sender", None):
             return sender.login
         raise ValueError("Event has no context!")
 
@@ -46,3 +60,11 @@ class Event(BaseEvent):
     @overrides(BaseEvent)
     def is_tome(self) -> bool:
         return self.to_me
+
+
+{% for event in events %}
+
+class {{ event.class_name }}(Event):
+    payload: {{ event.payload_type }}
+
+{% endfor %}