org.jmeterplugins.repository.plugins.PluginSuggester Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jmeter-plugins-manager Show documentation
UI-based plugins management
There is a newer version: 1.10
Show newest version
package org.jmeterplugins.repository.plugins;

import org.apache.jmeter.gui.GuiPackage;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.log.Logger;
import org.jmeterplugins.repository.Plugin;
import org.jmeterplugins.repository.PluginManager;

import java.awt.*;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;

public class PluginSuggester {
    private static final Logger log = LoggingManager.getLoggerForClass();

    protected TestPlanAnalyzer analyzer;
    protected String testPlan;
    private final PluginManager pmgr;

    public PluginSuggester(PluginManager pmgr) {
        this.pmgr = pmgr;
        analyzer = new TestPlanAnalyzer();
    }

    public void checkAndSuggest(String msg) {
        Set pluginsToInstall = findPluginsToInstall(msg);
        if (pluginsToInstall.size() > 0) {

            pmgr.togglePlugins(pluginsToInstall, true);

            Frame parent = (GuiPackage.getInstance() != null) ? GuiPackage.getInstance().getMainFrame() : null;
            SuggestDialog dialog = new SuggestDialog(parent, pmgr, pluginsToInstall, testPlan);
            dialog.setVisible(true);
            dialog.setAlwaysOnTop(true);
        }
    }

    protected Set findPluginsToInstall(String msg) {
        if (msg != null && msg.contains("Loading file")) {
            testPlan = msg.substring(msg.indexOf(": ") + 2);
            if (!"null".equals(testPlan)) {
                return analyzeTestPlan(testPlan);
            }
        }
        return Collections.emptySet();
    }

    public Set analyzeTestPlan(String path) {
        Set nonExistentClasses = analyzer.analyze(path);
        if (nonExistentClasses.size() > 0) {
            return findPluginsFromClasses(nonExistentClasses);
        }
        return Collections.emptySet();
    }


    protected Set findPluginsFromClasses(Set nonExistentClasses) {
        try {
            pmgr.load();
        } catch (Throwable throwable) {
            log.warn("Cannot load plugins repo: ", throwable);
            return Collections.emptySet();
        }
        final Set availablePlugins = pmgr.getAvailablePlugins();
        final Set pluginsToInstall = new HashSet<>();
        for (Plugin plugin : availablePlugins) {
            if (plugin.containsComponentClasses(nonExistentClasses)) {
                pluginsToInstall.add(plugin);
            }
        }

        if (pluginsToInstall.isEmpty()) {
            log.warn("Plugins Manager were unable to find plugins to satisfy Test Plan requirements. " +
                    "To help improve, please report following list to https://jmeter-plugins.org/support/: " +
                    Arrays.toString(nonExistentClasses.toArray()));
        }

        return pluginsToInstall;
    }

    public TestPlanAnalyzer getAnalyzer() {
        return analyzer;
    }

    public void setAnalyzer(TestPlanAnalyzer analyzer) {
        this.analyzer = analyzer;
    }

}