TestRoots/watchdog

package nl.tudelft.watchdog.eclipse.ui.wizards;

import java.util.function.BiConsumer;
import java.util.function.Consumer;

import org.eclipse.jface.resource.ImageDescriptor;
import org.eclipse.jface.wizard.WizardDialog;
import org.eclipse.jface.wizard.WizardPage;
import org.eclipse.swt.SWT;
import org.eclipse.swt.events.MouseAdapter;
import org.eclipse.swt.events.MouseEvent;
import org.eclipse.swt.events.SelectionAdapter;
import org.eclipse.swt.events.SelectionEvent;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.layout.RowLayout;
import org.eclipse.swt.widgets.Button;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Label;
import org.eclipse.swt.widgets.Link;
import org.eclipse.swt.widgets.Text;

import nl.tudelft.watchdog.core.ui.wizards.YesNoDontKnowChoice;
import nl.tudelft.watchdog.eclipse.Activator;
import nl.tudelft.watchdog.eclipse.ui.util.BrowserOpenerSelection;

import static nl.tudelft.watchdog.core.ui.wizards.WizardStrings.*;

/**
 * A single registration step that has two options: 1. The user already has a
 * registration, for which they have to enter their ID. 2. The user has to
 * create a registration in the {@link #getRegistrationPanel()}.
 */
abstract class RegistrationStep extends WizardPage {

    private Composite dynamicContent;
    private Composite container;
    private boolean isPageComplete;
    private WizardDialog dialog;

    protected RegistrationStep(String pageName, WizardDialog dialog) {
        super(pageName);
        this.dialog = dialog;
    }

    @Override
    public boolean isPageComplete() {
        return isPageComplete;
    }

    @Override
    public void createControl(Composite parent) {
        container = new Composite(parent, SWT.NONE);
        container.setLayout(createRowLayout(SWT.VERTICAL));

        this.createUserRegistrationIntroduction(container);
        this.createUserIsRegisteredQuestion(container);

        this.setControl(container);
    }

    abstract void createUserRegistrationIntroduction(Composite container);

    abstract String getRegistrationType();

    private void createUserIsRegisteredQuestion(Composite parent) {
        Composite questionContainer = new Composite(parent, SWT.NONE);
        questionContainer.setLayout(createRowLayout(SWT.HORIZONTAL));

        Label question = new Label(questionContainer, SWT.NONE);
        question.setText("Do you want to create a new WatchDog " + this.getRegistrationType() + " registration?");

        Composite buttons = new Composite(questionContainer, SWT.NONE);
        buttons.setLayout(createRowLayout(SWT.HORIZONTAL));

        Button yes = new Button(buttons, SWT.RADIO);
        yes.setText("yes");
        whenSelectedCreatePanelAndUpdateUI(yes, getRegistrationPanel());

        Button no = new Button(buttons, SWT.RADIO);
        no.setText("no");
        whenSelectedCreatePanelAndUpdateUI(no, getIdInputPanel());

        dynamicContent = new Composite(container, SWT.NONE);
        dynamicContent.setLayout(createRowLayout(SWT.VERTICAL));
    }

    private void whenSelectedCreatePanelAndUpdateUI(Button button,
            BiConsumer<Composite, Consumer<Boolean>> compositeConstructor) {
        button.addSelectionListener(new SelectionAdapter() {
            @Override
            public void widgetSelected(SelectionEvent e) {
                if (button.getSelection()) {
                    dynamicContent.dispose();
                    dynamicContent = new Composite(container, SWT.NONE);
                    dynamicContent.setLayout(createRowLayout(SWT.VERTICAL));

                    compositeConstructor.accept(dynamicContent, (hasValidId) -> {
                        isPageComplete = hasValidId;
                        dialog.updateButtons();

                        forceLayoutRecalc();
                    });

                    forceLayoutRecalc();
                }
            }
        });
    }

    private void forceLayoutRecalc() {
        container.layout(true, true);
        container.redraw();
        container.update();

        // Make sure to update the shell height, as the new panels could potentially
        // overflow. In this case, force the minimum height to make sure all elements
        // are properly visible on the screen.
        dialog.getShell().pack();
        dialog.getShell().layout(true, true);
        dialog.getShell().redraw();
        dialog.getShell().update();
    }

    abstract BiConsumer<Composite, Consumer<Boolean>> getIdInputPanel();

    abstract BiConsumer<Composite, Consumer<Boolean>> getRegistrationPanel();

    /**
     * The default margin of a RowLayout is 3, per {@link RowLayout#marginLeft}.
     * Explicitly disable it for our use-cases to prevent a vertical alignment
     * mismatch.
     *
     * @return A rowLayout without any margin.
     */
    static RowLayout createRowLayout(int alignment) {
        RowLayout rowLayout = new RowLayout(alignment);

        rowLayout.fill = true;
        rowLayout.marginLeft = 0;

        return rowLayout;
    }

    static void createErrorMessageLabel(Composite container, Exception exception) {
        String exceptionMessage = exception.getMessage();
        if (exceptionMessage != null) {
            Label exceptionText = new Label(container, SWT.WRAP);
            int truncateThreshold = Integer.min(exceptionMessage.length(), 100);
            exceptionText.setText(exception.getMessage().substring(0, truncateThreshold));
        }

        new Label(container, SWT.NONE).setText(CONNECTED_TO_INTERNET);
        Link link = new Link(container, SWT.WRAP);
        link.setText(PLEASE_CONTACT_US + Links.OUR_WEBSITE.toHTMLURL() + ". " + HELP_TROUBLESHOOT);
        link.addSelectionListener(new BrowserOpenerSelection());
    }

    /**
     * Create a label that is linked to the text field. Clicking on the label will
     * select the text field.
     *
     * @param labelText
     *            The text of the label.
     * @param tooltip
     *            The tooltip for both elements.
     * @param container
     *            The parent container.
     * @return The created text field.
     */
    static Text createLinkedLabelTextField(String labelText, String tooltip, Composite container) {
        Label label = new Label(container, SWT.NONE);
        label.setLayoutData(new GridData(SWT.FILL, SWT.BEGINNING, true, false));
        label.setToolTipText(tooltip);
        label.setText(labelText);

        Text input = new Text(container, SWT.NONE);
        input.setLayoutData(new GridData(SWT.FILL, SWT.BEGINNING, true, false));
        input.setToolTipText(tooltip);
        input.setTextLimit(150);
        label.addMouseListener(new MouseAdapter() {
            @Override
            public void mouseDown(MouseEvent e) {
                input.forceFocus();
            }

            @Override
            public void mouseDoubleClick(MouseEvent e) {
                input.forceFocus();
            }
        });
        return input;
    }

    static Label createLogo(Composite logoContainer, String imageLocation) {
        Label watchdogLogo = new Label(logoContainer, SWT.NONE);

        ImageDescriptor watchdogLogoImageDescriptor = Activator.imageDescriptorFromPlugin(Activator.PLUGIN_ID,
                imageLocation);
        watchdogLogo.setImage(watchdogLogoImageDescriptor.createImage());

        return watchdogLogo;
    }

    static YesNoDontknowButtonGroup createYesNoDontKnowQuestionWithLabel(String labelText, Composite container) {
        new Label(container, SWT.NONE).setText(labelText);

        YesNoDontknowButtonGroup buttons = new YesNoDontknowButtonGroup(container);
        buttons.setLayout(createRowLayout(SWT.HORIZONTAL));

        buttons.addButton("Yes", YesNoDontKnowChoice.Yes);
        buttons.addButton("No", YesNoDontKnowChoice.No);
        buttons.addButton("Don't know", YesNoDontKnowChoice.DontKnow);

        return buttons;
    }

    static class YesNoDontknowButtonGroup extends Composite {
        YesNoDontKnowChoice selected = YesNoDontKnowChoice.DontKnow;

        YesNoDontknowButtonGroup(Composite parent) {
            super(parent, SWT.NONE);
        }

        void addButton(String text, YesNoDontKnowChoice choice) {
            Button button = new Button(this, SWT.RADIO);
            button.setText(text);
            button.addSelectionListener(new SelectionAdapter() {
                @Override
                public void widgetSelected(SelectionEvent e) {
                    selected = choice;
                }
            });
        }
    }

}