From 8db775e4a4e85b55218435a1a989461f62f31f36 Mon Sep 17 00:00:00 2001
From: Yuuta Liang <yuutaw@student.cs.ubc.ca>
Date: Wed, 29 Nov 2023 14:46:37 -0800
Subject: Fix UMLGen: exclude enums from ui.tui package

Signed-off-by: Yuuta Liang <yuutaw@student.cs.ubc.ca>
---
 umlgen/src/main/java/moe/yuuta/umlgen/Main.kt | 1 +
 1 file changed, 1 insertion(+)

(limited to 'umlgen/src/main')

diff --git a/umlgen/src/main/java/moe/yuuta/umlgen/Main.kt b/umlgen/src/main/java/moe/yuuta/umlgen/Main.kt
index 6d1a967..de62f67 100644
--- a/umlgen/src/main/java/moe/yuuta/umlgen/Main.kt
+++ b/umlgen/src/main/java/moe/yuuta/umlgen/Main.kt
@@ -88,6 +88,7 @@ object Main {
     private fun getEnums(): List<EnumDeclaration> =
             TargetRegistry.getInstance().getNodes()
                     .filter { (_, node) -> node.isEnumDeclaration }
+                    .filter { (pkg, _) -> pkg != null && !pkg.startsWith("ui.tui") }
                     .map { (_, node) -> node.asEnumDeclaration() }
 
     private fun parseAssocAnnotation(field: FieldDeclaration,
-- 
cgit v1.2.3