pajlada
74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Optional
|
|
|
|
import logging
|
|
|
|
import clang.cindex
|
|
|
|
from .comment_commands import CommentCommands, json_transform, parse_comment_commands
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class EnumConstant:
|
|
def __init__(
|
|
self,
|
|
name: str,
|
|
) -> None:
|
|
self.name = name
|
|
self.json_name = name
|
|
self.tag: Optional[str] = None
|
|
|
|
self.dont_fail_on_deserialization: bool = False
|
|
|
|
def apply_comment_commands(self, comment_commands: CommentCommands) -> None:
|
|
for command, value in comment_commands:
|
|
match command:
|
|
case "json_rename":
|
|
# Rename the key that this field will use in json terms
|
|
log.debug(f"Rename json key from {self.json_name} to {value}")
|
|
self.json_name = value
|
|
case "json_dont_fail_on_deserialization":
|
|
# Don't fail when an optional object exists and its data is bad
|
|
log.debug(f"Don't fail on deserialization for {self.name}")
|
|
self.dont_fail_on_deserialization = bool(value.lower() == "true")
|
|
case "json_transform":
|
|
# Transform the key from whatever-case to case specified by `value`
|
|
self.json_name = json_transform(self.json_name, value)
|
|
case "json_inner":
|
|
# Do nothing on members
|
|
pass
|
|
case "json_tag":
|
|
# Rename the key that this field will use in json terms
|
|
log.debug(f"Applied json tag on {self.json_name}: {value}")
|
|
self.tag = value
|
|
case other:
|
|
log.warning(f"Unknown comment command found: {other} with value {value}")
|
|
|
|
@staticmethod
|
|
def from_node(node: clang.cindex.Cursor) -> EnumConstant:
|
|
assert node.type is not None
|
|
|
|
name = node.spelling
|
|
|
|
enum = EnumConstant(name)
|
|
|
|
if node.raw_comment is not None:
|
|
comment_commands = parse_comment_commands(node.raw_comment)
|
|
enum.apply_comment_commands(comment_commands)
|
|
|
|
return enum
|
|
|
|
def __eq__(self, other: object) -> bool:
|
|
if not isinstance(other, self.__class__):
|
|
return False
|
|
|
|
if self.name != other.name:
|
|
return False
|
|
|
|
return True
|
|
|
|
def __repr__(self) -> str:
|
|
return f"{self.name}"
|