From 09c972a92e085fe977e4335d31a8f7e98971e4be Mon Sep 17 00:00:00 2001
From: Naofal <naofal@naofal.xyz>
Date: Sun, 30 Mar 2025 23:56:03 +0300
Subject: [PATCH] Collect package tags

---
 src/main/java/xyz/naofal/jtags/TreeVisitor.java | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/main/java/xyz/naofal/jtags/TreeVisitor.java b/src/main/java/xyz/naofal/jtags/TreeVisitor.java
index 8dc1ba4..d7128d0 100644
--- a/src/main/java/xyz/naofal/jtags/TreeVisitor.java
+++ b/src/main/java/xyz/naofal/jtags/TreeVisitor.java
@@ -31,6 +31,10 @@ public class TreeVisitor extends TreePathScanner<Void, TreeVisitorContext> {
 
     logger.fine(() -> "Collecting tags in file: " + p.getLocation());
 
+    if (node.getPackage() != null) {
+      scan(node.getPackage(), p);
+    }
+
     return scan(node.getTypeDecls(), p);
   }
 
@@ -38,7 +42,11 @@ public class TreeVisitor extends TreePathScanner<Void, TreeVisitorContext> {
   public Void visitPackage(PackageTree node, TreeVisitorContext p) {
     Tag tag =
         new Tag(
-            TagKind.PACKAGE, node.getPackageName().toString(), p.getLocation(), p.getLine(node));
+            TagKind.PACKAGE, node.getPackageName().toString(), p.getLocation().getParent(), "");
+
+    if (tags.contains(tag)) { 
+      return null;
+    }
 
     logger.finer(() -> "Package: " + tag);