Process @PublicApi annotation and collect annotated packages.

1 files changed, 12 insertions, 4 deletions

diff --git a/bundle-plugin/src/main/java/com/yahoo/container/plugin/classanalysis/ClassFileMetaData.java b/bundle-plugin/src/main/java/com/yahoo/container/plugin/classanalysis/ClassFileMetaData.java
index 5601430a27f..7e2f59c1e4d 100644
--- a/bundle-plugin/src/main/java/com/yahoo/container/plugin/classanalysis/ClassFileMetaData.java
+++ b/bundle-plugin/src/main/java/com/yahoo/container/plugin/classanalysis/ClassFileMetaData.java
@@ -14,12 +14,12 @@ public class ClassFileMetaData {
 
     private final String name;
     private final Set<String> referencedClasses;
-    private final Optional<ExportPackageAnnotation> exportPackage;
+    private final PackageInfo packageInfo;
 
-    public ClassFileMetaData(String name, Set<String> referencedClasses, Optional<ExportPackageAnnotation> exportPackage) {
+    public ClassFileMetaData(String name, Set<String> referencedClasses, PackageInfo packageInfo) {
         this.name = name;
         this.referencedClasses = referencedClasses;
-        this.exportPackage = exportPackage;
+        this.packageInfo = packageInfo;
     }
 
     public String getName() {
@@ -30,8 +30,16 @@ public class ClassFileMetaData {
         return referencedClasses;
     }
 
+    public PackageInfo packageInfo() {
+        return packageInfo;
+    }
+
     public Optional<ExportPackageAnnotation> getExportPackage() {
-        return exportPackage;
+        return packageInfo.exportPackage();
+    }
+
+    public boolean isPublicApi() {
+        return packageInfo.isPublicApi();
     }
 
 }