AddConfigSetQuickFixProposal.java

/*******************************************************************************
 *  Copyright (c) 2012 VMware, Inc.
 *  All rights reserved. This program and the accompanying materials
 *  are made available under the terms of the Eclipse Public License v1.0
 *  which accompanies this distribution, and is available at
 *  http://www.eclipse.org/legal/epl-v10.html
 *
 *  Contributors:
 *      VMware, Inc. - initial API and implementation
 *******************************************************************************/
package org.springframework.ide.eclipse.quickfix.proposals;

import java.util.HashSet;
import java.util.Set;

import org.eclipse.core.resources.IFile;
import org.eclipse.jface.dialogs.Dialog;
import org.eclipse.jface.text.IDocument;
import org.eclipse.jface.text.contentassist.ICompletionProposal;
import org.eclipse.jface.wizard.Wizard;
import org.eclipse.jface.wizard.WizardDialog;
import org.eclipse.jface.wizard.WizardPage;
import org.eclipse.swt.SWT;
import org.eclipse.swt.events.ModifyEvent;
import org.eclipse.swt.events.ModifyListener;
import org.eclipse.swt.graphics.Image;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.layout.GridLayout;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.Label;
import org.eclipse.swt.widgets.Text;
import org.springframework.ide.eclipse.beans.core.BeansCorePlugin;
import org.springframework.ide.eclipse.beans.core.internal.model.BeansConfigSet;
import org.springframework.ide.eclipse.beans.core.internal.model.BeansProject;
import org.springframework.ide.eclipse.beans.core.internal.model.update.BeansModelUpdater;
import org.springframework.ide.eclipse.beans.core.model.IBean;
import org.springframework.ide.eclipse.beans.core.model.IBeansConfigSet;
import org.springframework.ide.eclipse.beans.core.model.IBeansModel;
import org.springframework.ide.eclipse.beans.ui.BeansUIImages;
import org.springframework.ide.eclipse.quickfix.QuickfixUtils;


/**
 * Quick fix proposal for creating a new config set
 * @author Terry Denney
 * @since 2.1
 */
public class AddConfigSetQuickFixProposal extends BeanAttributeQuickFixProposal implements ICompletionProposal {

	private class ConfigSetNameWizard extends Wizard {

		private String configSetName;

		private ConfigSetNameWizardPage page;

		public ConfigSetNameWizard() {
			setWindowTitle("Bean Reference Quick Fix");
		}

		@Override
		public void addPages() {
			page = new ConfigSetNameWizardPage();
			addPage(page);
		}

		public String getConfigSetName() {
			return configSetName;
		}

		@Override
		public boolean performFinish() {
			configSetName = page.getConfigSetName();
			return true;
		}
	}

	private class ConfigSetNameWizardPage extends WizardPage {

		private Text text;

		private String configSetName;

		private boolean canFinish;

		protected ConfigSetNameWizardPage() {
			super("configSetNameWizardPage");
		}

		@Override
		public boolean canFlipToNextPage() {
			return canFinish;
		}

		public void createControl(Composite parent) {
			setTitle("Create New Config Set");
			setMessage("Enter the name for the new config set.");

			parent.setLayout(new GridLayout());

			Composite control = new Composite(parent, SWT.NONE);
			control.setLayout(new GridLayout(2, false));
			control.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true));

			Label label = new Label(control, SWT.NONE);
			label.setText("Name: ");
			label.setLayoutData(new GridData(SWT.FILL, SWT.FILL, false, false));

			text = new Text(control, SWT.BORDER);
			text.setEditable(true);
			text.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, false));

			text.addModifyListener(new ModifyListener() {

				public void modifyText(ModifyEvent e) {
					configSetName = text.getText();
					update();
					getWizard().getContainer().updateButtons();
				}
			});

			setControl(control);
		}

		public String getConfigSetName() {
			return configSetName;
		}

		private void update() {
			String name = text.getText();
			if (name != null && name.length() > 0) {
				if (!project.hasConfigSet(name)) {
					canFinish = true;
					setErrorMessage(null);
				}
				else {
					canFinish = false;
					setErrorMessage("A config set with name " + name + " already exists.");
				}
			}
			else {
				canFinish = false;
				setErrorMessage(null);
			}
		}
	}

	private final String configName;

	private final String referencedConfigName;

	private final BeansProject project;

	private String configSetName;

	public AddConfigSetQuickFixProposal(int offset, int length, boolean missingEndQuote, IBean importBean, IFile file) {
		super(offset, length, missingEndQuote);

		this.configName = QuickfixUtils.getConfigName(file);
		this.referencedConfigName = QuickfixUtils.getConfigName(importBean.getElementResource());

		IBeansModel model = BeansCorePlugin.getModel();
		this.project = (BeansProject) model.getProject(file.getProject());
	}

	public AddConfigSetQuickFixProposal(int offset, int length, boolean missingEndQuote, IBean importBean, IFile file,
			String configSetName) {
		this(offset, length, missingEndQuote, importBean, file);
		this.configSetName = configSetName;
	}

	@Override
	public void applyQuickFix(IDocument document) {
		ConfigSetNameWizard wizard = new ConfigSetNameWizard();

		if (configSetName == null) {
			WizardDialog dialog = new WizardDialog(Display.getDefault().getActiveShell(), wizard);
			if (dialog.open() == Dialog.OK) {
				configSetName = wizard.getConfigSetName();
			}
		}

		if (configSetName != null) {
			Set<String> configNames = new HashSet<String>();
			configNames.add(configName);
			configNames.add(referencedConfigName);

			IBeansConfigSet configSet = new BeansConfigSet(project, configSetName, configNames,
					IBeansConfigSet.Type.MANUAL);
			project.addConfigSet(configSet);
			project.saveDescription();
			BeansModelUpdater.updateProject(project);
		}

	}

	public String getDisplayString() {
		return "Create new config set with " + configName + " and " + referencedConfigName;
	}

	public Image getImage() {
		return BeansUIImages.getImage(BeansUIImages.IMG_OBJS_CONFIG_SET);
	}

}