org.jmeterplugins.repository.plugins.SuggestDialog 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.action.ActionNames;
import org.apache.jmeter.gui.action.ActionRouter;
import org.apache.jorphan.gui.ComponentUtil;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.log.Logger;
import org.jmeterplugins.repository.GenericCallback;
import org.jmeterplugins.repository.Plugin;
import org.jmeterplugins.repository.PluginManager;
import org.jmeterplugins.repository.PluginManagerMenuItem;

import javax.swing.*;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.LinkedList;
import java.util.Set;

import static org.jmeterplugins.repository.PluginManagerDialog.SPACING;

public class SuggestDialog extends JDialog implements GenericCallback {
    private static final Logger log = LoggingManager.getLoggerForClass();
    private final PluginManager manager;
    private JLabel titleLabel = new JLabel("");
    private JLabel statusLabel = new JLabel("");

    public SuggestDialog(Frame parent, PluginManager manager, Set plugins, final String testPlan) {
        super(parent, "JMeter Plugins Manager", true);
        setLocationRelativeTo(parent);
        this.manager = manager;
        init(plugins, testPlan);
    }

    private void init(Set plugins, final String testPlan) {
        setLayout(new BorderLayout());
        setIconImage(PluginManagerMenuItem.getPluginsIcon(manager.hasAnyUpdates()).getImage());
        ComponentUtil.centerComponentInWindow(this);

        JPanel mainPanel = new JPanel(new BorderLayout(0, 0));
        mainPanel.setBorder(SPACING);

        final StringBuilder message = new StringBuilder("Your test plan requires following plugins:
");
        for (Plugin plugin : plugins) {
            message.append("").append(plugin.getName()).append("");
        }
        message.append("");
        message.append("");
        message.append("Plugins Manager can install it automatically. Following changes will be applied:");
        message.append("");
        titleLabel.setText(message.toString());
        mainPanel.add(titleLabel, BorderLayout.NORTH);

        mainPanel.add(getDetailsPanel(), BorderLayout.CENTER);

        mainPanel.add(getButtonsPanel(plugins, testPlan), BorderLayout.SOUTH);

        add(mainPanel, BorderLayout.CENTER);
        pack();
    }

    private JPanel getButtonsPanel(final Set plugins, final String testPlan) {
        final JButton btnYes = new JButton("Yes, install it");
        final JButton btnNo = new JButton("Cancel");
        final SuggestDialog dialog = this;
        btnYes.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent e) {
                btnYes.setEnabled(false);
                btnNo.setEnabled(false);

                new Thread() {
                    @Override
                    public void run() {
                        LinkedList options = new LinkedList<>();
                        options.add("-t");
                        options.add(testPlan);
                        manager.applyChanges(dialog, true, options);
                        dispose();
                        ActionRouter.getInstance().actionPerformed(new ActionEvent(this, 0, ActionNames.EXIT));
                    }
                }.start();
            }
        });

        btnNo.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent e) {
                manager.togglePlugins(plugins, false);
                dispose();
            }
        });

        JPanel buttons = new JPanel(new FlowLayout());
        buttons.add(btnYes);
        buttons.add(btnNo);
        JPanel btnPanel = new JPanel(new BorderLayout());
        btnPanel.setBorder(SPACING);
        btnPanel.add(buttons, BorderLayout.EAST);
        btnPanel.add(statusLabel, BorderLayout.WEST);
        return btnPanel;
    }

    private JPanel getDetailsPanel() {
        JTextArea messageLabel = new JTextArea(manager.getChangesAsText());
        messageLabel.setEditable(false);

        JPanel messagePanel = new JPanel(new BorderLayout());
        messagePanel.setBorder(SPACING);
        messagePanel.add(new JScrollPane(messageLabel));

        return messagePanel;
    }

    @Override
    public void notify(final String s) {
        SwingUtilities.invokeLater(
                new Runnable() {

                    @Override
                    public void run() {
                        statusLabel.setText(s);
                        repaint();
                    }
                });

    }
}