package bndtools.nature;

import java.io.ByteArrayInputStream;
import java.util.Collections;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Set;

import org.bndtools.api.BndtoolsConstants;
import org.bndtools.headless.build.manager.api.HeadlessBuildManager;
import org.bndtools.versioncontrol.ignores.manager.api.VersionControlIgnoresManager;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IProjectDescription;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.MultiStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jface.action.IAction;
import org.eclipse.jface.dialogs.ErrorDialog;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.ui.IObjectActionDelegate;
import org.eclipse.ui.IWorkbenchPart;

import aQute.bnd.build.Project;
import bndtools.Plugin;
import bndtools.preferences.BndPreferences;

public class ToggleNatureAction implements IObjectActionDelegate {

	private ISelection		selection;
	private IWorkbenchPart	targetPart;

	/*
	 * (non-Javadoc)
	 * @see org.eclipse.ui.IActionDelegate#run(org.eclipse.jface.action.IAction)
	 */
	@Override
	public void run(IAction action) {
		if (selection instanceof IStructuredSelection) {
			for (Iterator<?> it = ((IStructuredSelection) selection).iterator(); it.hasNext();) {
				Object element = it.next();
				IProject project = null;
				if (element instanceof IProject) {
					project = (IProject) element;
				} else if (element instanceof IAdaptable) {
					project = ((IAdaptable) element).getAdapter(IProject.class);
				}
				if (project != null) {
					boolean isJavaProject = false;
					try {
						isJavaProject = project.hasNature(JavaCore.NATURE_ID);
					} catch (CoreException e) {
						/* swallow */
					}
					if (isJavaProject) {
						IStatus status = toggleNature(JavaCore.create(project));
						if (!status.isOK())
							ErrorDialog.openError(targetPart.getSite()
								.getShell(), "Toggle Bnd Nature", null, status);
					}
				}
			}
		}
	}

	/*
	 * (non-Javadoc)
	 * @see
	 * org.eclipse.ui.IActionDelegate#selectionChanged(org.eclipse.jface.action
	 * .IAction, org.eclipse.jface.viewers.ISelection)
	 */
	@Override
	public void selectionChanged(IAction action, ISelection selection) {
		this.selection = selection;
	}

	/*
	 * (non-Javadoc)
	 * @see
	 * org.eclipse.ui.IObjectActionDelegate#setActivePart(org.eclipse.jface.
	 * action.IAction, org.eclipse.ui.IWorkbenchPart)
	 */
	@Override
	public void setActivePart(IAction action, IWorkbenchPart targetPart) {
		this.targetPart = targetPart;
	}

	private static void ensureBndBndExists(IProject project) throws CoreException {
		IFile bndfile = project.getFile(Project.BNDFILE);
		if (!bndfile.exists())
			bndfile.create(new ByteArrayInputStream(new byte[0]), false, null);
	}

	/**
	 * Toggles sample nature on a project
	 *
	 * @param project to have sample nature added or removed
	 */
	private static IStatus toggleNature(IJavaProject project) {
		try {
			/* Version control ignores */
			VersionControlIgnoresManager versionControlIgnoresManager = Plugin.getDefault()
				.getVersionControlIgnoresManager();
			Set<String> enabledIgnorePlugins = new BndPreferences()
				.getVersionControlIgnoresPluginsEnabled(versionControlIgnoresManager, project, null);

			/* Headless build files */
			HeadlessBuildManager headlessBuildManager = Plugin.getDefault()
				.getHeadlessBuildManager();
			Set<String> enabledPlugins = new BndPreferences().getHeadlessBuildPluginsEnabled(headlessBuildManager,
				null);

			IProject iProject = project.getProject();
			IProjectDescription description = iProject.getDescription();
			String[] natures = description.getNatureIds();

			List<String> headlessBuildWarnings = new LinkedList<>();

			for (int i = 0; i < natures.length; ++i) {
				if (BndtoolsConstants.NATURE_ID.equals(natures[i])) {
					// Remove the nature
					String[] newNatures = new String[natures.length - 1];
					System.arraycopy(natures, 0, newNatures, 0, i);
					System.arraycopy(natures, i + 1, newNatures, i, natures.length - i - 1);
					description.setNatureIds(newNatures);
					iProject.setDescription(description, null);

					/* Remove the headless build files */
					headlessBuildManager.setup(enabledPlugins, false, iProject.getLocation()
						.toFile(), false, enabledIgnorePlugins, headlessBuildWarnings);

					/*
					 * refresh the project; files were created outside of
					 * Eclipse API
					 */
					iProject.refreshLocal(IResource.DEPTH_INFINITE, null);

					return createStatus(
						"Obsolete build files may remain in the project. Please review the messages below.",
						Collections.<String> emptyList(), headlessBuildWarnings);
				}
			}

			/* Add the headless build files */
			headlessBuildManager.setup(enabledPlugins, false, iProject.getLocation()
				.toFile(), true, enabledIgnorePlugins, headlessBuildWarnings);

			// Add the nature
			ensureBndBndExists(iProject);
			String[] newNatures = new String[natures.length + 1];
			System.arraycopy(natures, 0, newNatures, 0, natures.length);
			newNatures[natures.length] = BndtoolsConstants.NATURE_ID;
			description.setNatureIds(newNatures);
			iProject.setDescription(description, null);

			/* refresh the project; files were created outside of Eclipse API */
			iProject.refreshLocal(IResource.DEPTH_INFINITE, null);

			return createStatus("Some build files could not be generated. Please review the messages below.",
				Collections.<String> emptyList(), headlessBuildWarnings);
		} catch (CoreException e) {
			return new Status(IStatus.ERROR, Plugin.PLUGIN_ID, 0, "Error occurred while toggling Bnd project nature",
				e);
		}
	}

	private static IStatus createStatus(String message, List<String> errors, List<String> warnings) {
		MultiStatus status = new MultiStatus(Plugin.PLUGIN_ID, 0, message, null);

		for (String error : errors)
			status.add(new Status(IStatus.ERROR, Plugin.PLUGIN_ID, 0, error, null));
		for (String warning : warnings)
			status.add(new Status(IStatus.WARNING, Plugin.PLUGIN_ID, 0, warning, null));

		return status;
	}
}
