Enum and other improvements

Author: Saturn745

src/main/kotlin/generator/LuaEmitter.kt

@@ -1,6 +1,7 @@
 package generator
 
 import parser.ParsedClass
+import parser.Visibility
 import util.extractJavaDocInfo
 import util.mapJavaTypeToLua
 
@@ -18,7 +19,7 @@ class LuaEmitter {
 
         // Process fields (variables)
         parsedClass.fields.forEach { field ->
-            sb.appendLine("---@field ${field.name} ${mapJavaTypeToLua(field.type)}")
+            sb.appendLine("---@field ${field.visibility.toString().lowercase()} ${field.name} ${mapJavaTypeToLua(field.type)}")
         }
 
         parsedClass.constructors.forEach { constructor ->
@@ -58,6 +59,12 @@ class LuaEmitter {
                 sb.appendLine("---@param $paramName${if (!param.required) "?" else ""} $luaType ${param.comment ?: ""}")
             }
 
+            when (method.visibility) {
+                Visibility.PUBLIC -> sb.appendLine("---@public")
+                Visibility.PROTECTED -> sb.appendLine("---@protected")
+                Visibility.PRIVATE -> sb.appendLine("---@private")
+            }
+
             sb.appendLine("---@return ${mapJavaTypeToLua(method.returnType)} ${doc.returnComment ?: ""}")
 
             if (doc.mainComment.isNotBlank()) {

src/main/kotlin/parser/JavaSourceParser.kt

@@ -44,7 +44,8 @@ class JavaSourceParser : ClassParser {
                 fields = enumDecl.entries.map {
                     ParsedField(
                         name = it.nameAsString,
-                        type = "string",
+                        type = enumDecl.nameAsString,
+                        visibility = Visibility.PUBLIC,
                         comment = it.comment.map { c -> extractJavaDocInfo(c.content).mainComment }.orElse(null)
                     )
                 },
@@ -91,6 +92,7 @@ class JavaSourceParser : ClassParser {
                     required = !isNullableComment,
                 )
             },
+            visibility = Visibility.fromString(constructor.accessSpecifier.toString().uppercase()),
             comment = constructor.comment.map { extractJavaDocInfo(it.content).mainComment }.orElse(null),
             isDeprecated = constructor.annotations.any { it.nameAsString == "Deprecated" },
             isAsync = false, // Constructors are not async
@@ -112,6 +114,7 @@ class JavaSourceParser : ClassParser {
                     required = !isNullableComment,
                 )
             },
+            visibility = Visibility.fromString(method.accessSpecifier.toString().uppercase()),
             comment = method.comment.map { extractJavaDocInfo(it.content).mainComment }.orElse(null),
             isDeprecated = method.annotations.any { it.nameAsString == "Deprecated" },
             isAsync = method.nameAsString.contains(
@@ -126,6 +129,7 @@ class JavaSourceParser : ClassParser {
             ParsedField(
                 name = variable.nameAsString,
                 type = field.elementType.asString(),
+                visibility = Visibility.fromString(field.accessSpecifier.toString().uppercase()),
                 comment = field.comment.map { extractJavaDocInfo(it.content).mainComment }.orElse(null)
             )
         }

src/main/kotlin/parser/ParsedField.kt

@@ -3,5 +3,6 @@ package parser
 data class ParsedField(
     val name: String,
     val type: String,
+    val visibility: Visibility,
     val comment: String?
 )

src/main/kotlin/parser/ParsedMethod.kt

@@ -3,6 +3,7 @@ package parser
 data class ParsedMethod(
     val name: String,
     val returnType: String,
+    val visibility: Visibility,
     val parameters: List<ParsedParameter>,
     val comment: String?,
     val isDeprecated: Boolean = false,

src/main/kotlin/parser/Visibility.kt

@@ -0,0 +1,16 @@
+package parser
+
+enum class Visibility {
+    PUBLIC, PROTECTED, PRIVATE;
+
+    companion object {
+        fun fromString(visibility: String): Visibility {
+            return when (visibility) {
+                "PUBLIC" -> PUBLIC
+                "PROTECTED" -> PROTECTED
+                "PRIVATE" -> PRIVATE
+                else -> PUBLIC
+            }
+        }
+    }
+}