Added newtype support for primitive type enum

projects/jdwp/codegen/new_type_generator.py

@@ -0,0 +1,27 @@
+"""New type Generator."""
+from projects.jdwp.defs.schema import PrimitiveType
+import typing
+
+
+def get_python_type(jdwp_type: PrimitiveType) -> str:
+    """Map JDWP type to Python type."""
+    mapping = {
+        PrimitiveType.STRING: "str",
+        PrimitiveType.INT: "int",
+        PrimitiveType.BYTE: "int",
+        PrimitiveType.BOOLEAN: "bool",
+    }
+    return mapping.get(jdwp_type, "int")
+
+
+def get_type_alias_definition(jdwp_type: PrimitiveType) -> str:
+    """Return the type alias definition for a given JDWP type."""
+    python_type = get_python_type(jdwp_type)
+    new_type_name = f"{jdwp_type.name.capitalize()}Type"
+    return f"{new_type_name} = typing.NewType('{new_type_name}', {python_type})"
+
+
+def generate_new_types():
+    for jdwp_type in PrimitiveType:
+        type_alias_definition = get_type_alias_definition(jdwp_type)
+        print(type_alias_definition)

projects/jdwp/defs/schema.py

@@ -16,6 +16,8 @@ class PrimitiveType(Enum):
 
     STRING = "string"
     BOOLEAN = "boolean"
+    INT = "int"
+    BYTE = "int"
     DICT = "dict"
     REFERENCE_TYPE_ID = "referenceTypeID"
     CLASS_LOADER = "classLoader"

projects/jdwp/tests/BUCK

@@ -3,4 +3,7 @@
 python_test(
   name = "tests",
   srcs = glob(["*.py"]),
+  deps = [
+       "//projects/jdwp:lib",
+    ],
 )

projects/jdwp/tests/test_primitve_type.py

@@ -0,0 +1,17 @@
+import unittest
+from projects.jdwp.codegen.new_type_generator import get_python_type
+from projects.jdwp.defs.schema import PrimitiveType
+
+
+class TestEnumMemberMapping(unittest.TestCase):
+    def test_enum_member_mappings(self):
+        for jdwp_type in PrimitiveType:
+            with self.subTest(jdwp_type=jdwp_type):
+                result = get_python_type(jdwp_type)
+                self.assertIsInstance(
+                    result, str, f"Mapping for {jdwp_type} is missing or not a string"
+                )
+
+
+if __name__ == "__main__":
+    unittest.main()

projects/jdwp/tests/test_type_alias_definition.py

@@ -0,0 +1,19 @@
+import unittest
+from projects.jdwp.defs.schema import PrimitiveType
+from projects.jdwp.codegen.new_type_generator import get_type_alias_definition
+
+
+class TestTypeAliasDefinition(unittest.TestCase):
+    def test_type_alias_definitions(self):
+        for jdwp_type in PrimitiveType:
+            with self.subTest(jdwp_type=jdwp_type):
+                expected_start = f"{jdwp_type.name.capitalize()}Type = typing.NewType"
+                definition = get_type_alias_definition(jdwp_type)
+                self.assertTrue(
+                    definition.startswith(expected_start),
+                    f"Definition for {jdwp_type} is incorrect",
+                )
+
+
+if __name__ == "__main__":
+    unittest.main()