org.eclipse.m2e.binaryproject.ui/src/org/eclipse/m2e/binaryproject/ui/internal/BinaryProjectImportWizard.java

/*******************************************************************************
 * Copyright (c) 2012 Igor Fedorenko
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License 2.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *      Igor Fedorenko - initial API and implementation
 *******************************************************************************/
package org.eclipse.m2e.binaryproject.ui.internal;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;

import org.apache.maven.model.Dependency;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.jobs.Job;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.wizard.Wizard;
import org.eclipse.m2e.binaryproject.internal.AbstractBinaryProjectsImportJob;
import org.eclipse.m2e.core.embedder.ArtifactKey;
import org.eclipse.m2e.core.project.ProjectImportConfiguration;
import org.eclipse.m2e.core.ui.internal.actions.SelectionUtil;
import org.eclipse.m2e.core.ui.internal.wizards.MavenDependenciesWizardPage;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.ui.IImportWizard;
import org.eclipse.ui.IWorkbench;

@SuppressWarnings("restriction")
public class BinaryProjectImportWizard extends Wizard implements IImportWizard {

  private MavenDependenciesWizardPage artifactsPage;

  private List<Dependency> initialDependencies;

  public BinaryProjectImportWizard() {}

  @Override
  public void init(IWorkbench workbench, IStructuredSelection selection) {
    final List<Dependency> dependencies = new ArrayList<Dependency>();
    for (Iterator<?> it = selection.iterator(); it.hasNext();) {
      Object element = it.next();
      ArtifactKey artifactKey = SelectionUtil.getType(element, ArtifactKey.class);
      if (artifactKey != null) {
        Dependency d = new Dependency();
        d.setGroupId(artifactKey.getGroupId());
        d.setArtifactId(artifactKey.getArtifactId());
        d.setVersion(artifactKey.getVersion());
        d.setClassifier(artifactKey.getClassifier());
        dependencies.add(d);
      }
    }
    artifactsPage =
        new MavenDependenciesWizardPage(new ProjectImportConfiguration(), "Artifacts", "Select artifacts to import") {
          @Override
          protected void createAdvancedSettings(Composite composite, GridData gridData) {
            // TODO profile can theoretically be usedful
          }
        };
    artifactsPage.setDependencies(dependencies.toArray(new Dependency[dependencies.size()]));
    artifactsPage.addListener(event -> getContainer().updateButtons());
    this.initialDependencies = Collections.unmodifiableList(dependencies);
  }

  @Override
  public boolean performFinish() {
    final ArrayList<ArtifactKey> artifacts = new ArrayList<ArtifactKey>();

    for (Dependency dependency : artifactsPage.getDependencies()) {
      artifacts.add(new ArtifactKey(dependency.getGroupId(), dependency.getArtifactId(), dependency.getVersion(),
          dependency.getClassifier()));
    }

    if (artifacts.isEmpty()) {
      return false;
    }

    Job job = new AbstractBinaryProjectsImportJob() {

      @Override
      protected Collection<ArtifactKey> getArtifactKeys(IProgressMonitor monitor) throws CoreException {
        return artifacts;
      }
    };
    job.schedule();

    return true;
  }

  @Override
  public boolean canFinish() {
    return artifactsPage.getDependencies().length > 0;
  }

  @Override
  public void addPages() {
    addPage(artifactsPage);
  }

  public List<Dependency> getInitialDependencies() {
    return initialDependencies;
  }
}