1 files changed, 35 insertions, 0 deletions
diff --git a/bundle-plugin/src/main/java/com/yahoo/container/plugin/classanalysis/ImportCollector.java b/bundle-plugin/src/main/java/com/yahoo/container/plugin/classanalysis/ImportCollector.java
new file mode 100644
index 00000000000..3946fe297f9
--- /dev/null
+++ b/bundle-plugin/src/main/java/com/yahoo/container/plugin/classanalysis/ImportCollector.java
@@ -0,0 +1,35 @@
+// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.container.plugin.classanalysis;
+
+import org.objectweb.asm.Type;
+
+import java.util.Collection;
+import java.util.Optional;
+import java.util.Set;
+
+/**
+ * @author ollivir
+ */
+public interface ImportCollector {
+    Set<String> imports();
+
+    default void addImportWithTypeDesc(String typeDescriptor) {
+        addImport(Type.getType(typeDescriptor));
+    }
+
+    default void addImport(Type type) {
+        addImport(Analyze.getClassName(type));
+    }
+
+    default void addImportWithInternalName(String name) {
+        addImport(Analyze.internalNameToClassName(name));
+    }
+
+    default void addImports(Collection<String> imports) {
+        imports().addAll(imports);
+    }
+
+    default void addImport(Optional<String> anImport) {
+        anImport.ifPresent(pkg -> imports().add(pkg));
+    }
+}