add generate_getter() function

Author: MrMaydo

src/__init__.py

@@ -0,0 +1,57 @@
+import re
+from dataclasses import dataclass
+
+
+JAVA_KEYWORDS = {
+    "abstract", "assert", "boolean", "break", "byte", "case", "catch",
+    "char", "class", "const", "continue", "default", "do", "double",
+    "else", "enum", "extends", "final", "finally", "float", "for",
+    "goto", "if", "implements", "import", "instanceof", "int",
+    "interface", "long", "native", "new", "package", "private",
+    "protected", "public", "return", "short", "static", "strictfp",
+    "super", "switch", "synchronized", "this", "throw", "throws",
+    "transient", "try", "void", "volatile", "while"
+}
+
+JAVA_BUILTIN_TYPES = {
+    "Boolean", "Byte", "Character", "Double", "Float", "Integer", "List", "Long", "Short",
+    "Class", "Object", "String", "Void"
+}
+
+JAVA_LITERALS = {
+    "null", "true", "false"
+}
+
+
+@dataclass
+class Field:
+    name: str
+    type: str
+
+
+def generate_getter(attr: Field) -> str:
+    attr_name = attr.name
+    attr_type = attr.type
+
+    _validate_java_identifier(attr_name)
+
+    getter_name = "get" + attr_name[0].upper() + attr_name[1:]
+
+    getter = f"""
+    public {attr_type} {getter_name}() {{
+        return {attr_name};
+    }}
+    """
+    return getter
+
+
+def _validate_java_identifier(name: str) -> None:
+    if not name:
+        raise ValueError("Field name cannot be empty")
+
+    if name in JAVA_KEYWORDS | JAVA_BUILTIN_TYPES | JAVA_LITERALS:
+        raise ValueError(f"'{name}' is a Java reserved keyword")
+
+    if not re.match(r"^[A-Za-z_][A-Za-z0-9_]*$", name):
+        raise ValueError(f"Invalid Java identifier: '{name}'")
+

src/method_generator.py

@@ -0,0 +1,56 @@
+import pytest
+
+from src.method_generator import generate_getter, Field
+
+JAVA_KEYWORDS = {
+    "abstract", "assert", "boolean", "break", "byte", "case", "catch",
+    "char", "class", "const", "continue", "default", "do", "double",
+    "else", "enum", "extends", "final", "finally", "float", "for",
+    "goto", "if", "implements", "import", "instanceof", "int",
+    "interface", "long", "native", "new", "package", "private",
+    "protected", "public", "return", "short", "static", "strictfp",
+    "super", "switch", "synchronized", "this", "throw", "throws",
+    "transient", "try", "void", "volatile", "while"
+}
+
+JAVA_BUILTIN_TYPES = {
+    "Boolean", "Byte", "Character", "Double", "Float", "Integer", "List", "Long", "Short",
+    "Class", "Object", "String", "Void"
+}
+
+JAVA_LITERALS = {
+    "null", "true", "false"
+}
+
+
+def test_generate_getter_integer():
+    attr = Field(
+        name="exampleAttribute",
+        type="Integer"
+    )
+    expected = """
+    public Integer getExampleAttribute() {
+        return exampleAttribute;
+    }
+    """
+    assert (generate_getter(attr) == expected)
+
+
+def test_generate_getter_string():
+    attr = Field(name="someName", type="String")
+    expected = """
+    public String getSomeName() {
+        return someName;
+    }
+    """
+    assert (generate_getter(attr) == expected)
+
+
+def test_generate_getter_invalid_name():
+    EMPTY_NAMES = {None, ""}
+    invalid_names = JAVA_KEYWORDS | JAVA_BUILTIN_TYPES | JAVA_LITERALS | EMPTY_NAMES
+    for name in invalid_names:
+        attr = Field(name=name, type="String")
+        with pytest.raises(ValueError):
+            generate_getter(attr)
+