Can now just get Extension classes (#191)

2 files changed, 29 insertions, 1 deletions

diff --git a/pf4j/src/main/java/org/pf4j/AbstractPluginManager.java b/pf4j/src/main/java/org/pf4j/AbstractPluginManager.java
index 21df9c0..711d764 100644
--- a/pf4j/src/main/java/org/pf4j/AbstractPluginManager.java
+++ b/pf4j/src/main/java/org/pf4j/AbstractPluginManager.java
@@ -45,7 +45,7 @@ public abstract class AbstractPluginManager implements PluginManager {
 
     private Path pluginsRoot;
 
-    private ExtensionFinder extensionFinder;
+    protected ExtensionFinder extensionFinder;
 
     private PluginDescriptorFinder pluginDescriptorFinder;
 
@@ -526,6 +526,30 @@ public abstract class AbstractPluginManager implements PluginManager {
     }
 
     @Override
+    public <T> List<Class<T>> getExtensionClasses(Class<T> type) {
+        List<ExtensionWrapper<T>> extensionsWrapper = extensionFinder.find(type);
+        List<Class<T>> extensionClasses = new ArrayList<>(extensionsWrapper.size());
+        for (ExtensionWrapper<T> extensionWrapper : extensionsWrapper) {
+            @SuppressWarnings("unchecked")
+            Class<T> c = (Class<T>)extensionWrapper.getDescriptor().extensionClass;
+            extensionClasses.add(c);
+        }
+        return extensionClasses;
+    }
+
+    @Override
+    public <T> List<Class<T>> getExtensionClasses(Class<T> type, String pluginId) {
+        List<ExtensionWrapper<T>> extensionsWrapper = extensionFinder.find(type, pluginId);
+        List<Class<T>> extensionClasses = new ArrayList<>(extensionsWrapper.size());
+        for (ExtensionWrapper<T> extensionWrapper : extensionsWrapper) {
+            @SuppressWarnings("unchecked")
+            Class<T> c = (Class<T>)extensionWrapper.getDescriptor().extensionClass;
+            extensionClasses.add(c);
+        }
+        return extensionClasses;
+    }
+
+    @Override
     public <T> List<T> getExtensions(Class<T> type) {
         List<ExtensionWrapper<T>> extensionsWrapper = extensionFinder.find(type);
         List<T> extensions = new ArrayList<>(extensionsWrapper.size());
diff --git a/pf4j/src/main/java/org/pf4j/PluginManager.java b/pf4j/src/main/java/org/pf4j/PluginManager.java
index 06fc68e..2523870 100644
--- a/pf4j/src/main/java/org/pf4j/PluginManager.java
+++ b/pf4j/src/main/java/org/pf4j/PluginManager.java
@@ -131,6 +131,10 @@ public interface PluginManager {
 
 	ClassLoader getPluginClassLoader(String pluginId);
 
+    <T> List<Class<T>> getExtensionClasses(Class<T> type);
+
+    <T> List<Class<T>> getExtensionClasses(Class<T> type, String pluginId);
+
 	<T> List<T> getExtensions(Class<T> type);
 
     <T> List<T> getExtensions(Class<T> type, String pluginId);