plugins/org.eclipse.oomph.projectcopy/src/org/eclipse/oomph/projectcopy/ProjectCopyAction.java

/*
 * Copyright (c) 2015 Eike Stepper (Berlin, Germany) and others.
 * 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:
 *    Eike Stepper - initial API and implementation
 */
package org.eclipse.oomph.projectcopy;

import org.eclipse.oomph.util.IOUtil;
import org.eclipse.oomph.util.XMLUtil;
import org.eclipse.oomph.util.XMLUtil.ElementUpdater;

import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IProjectDescription;
import org.eclipse.core.resources.IWorkspace;
import org.eclipse.core.resources.IWorkspaceRoot;
import org.eclipse.core.resources.IWorkspaceRunnable;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.core.runtime.Path;
import org.eclipse.core.runtime.Platform;
import org.eclipse.core.runtime.Status;
import org.eclipse.core.runtime.jobs.Job;
import org.eclipse.jface.action.IAction;
import org.eclipse.jface.dialogs.Dialog;
import org.eclipse.jface.dialogs.IInputValidator;
import org.eclipse.jface.dialogs.InputDialog;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.IObjectActionDelegate;
import org.eclipse.ui.IWorkbenchPart;

import org.w3c.dom.Element;

import javax.xml.parsers.DocumentBuilder;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;

/**
 * @author Eike Stepper
 */
public class ProjectCopyAction implements IObjectActionDelegate
{
  private Shell shell;

  private ISelection selection;

  public ProjectCopyAction()
  {
  }

  public void setActivePart(IAction action, IWorkbenchPart targetPart)
  {
    shell = targetPart.getSite().getShell();
  }

  public void selectionChanged(IAction action, ISelection selection)
  {
    this.selection = selection;
  }

  public void run(IAction action)
  {
    final IProject source = (IProject)((IStructuredSelection)selection).getFirstElement();
    final File folder = source.getLocation().toFile();

    try
    {
      final IWorkspaceRoot ROOT = ResourcesPlugin.getWorkspace().getRoot();
      final File parentFolder = folder.getParentFile();

      InputDialog dialog = new InputDialog(shell, "Copy Project", "Name of the new project:", source.getName(), new IInputValidator()
      {
        public String isValid(String newName)
        {
          IProject newProject = ROOT.getProject(newName);
          if (newProject.exists())
          {
            return "Project " + newName + " already exists.";
          }

          File newFolder = new File(parentFolder, newName);
          if (newFolder.exists())
          {
            return "Location " + newFolder.getAbsolutePath() + " already exists.";
          }

          return null;
        }
      });

      if (dialog.open() == Dialog.OK)
      {
        final String newName = dialog.getValue();

        new Job("Copy project")
        {
          @Override
          protected IStatus run(IProgressMonitor monitor)
          {
            try
            {
              ResourcesPlugin.getWorkspace().run(new IWorkspaceRunnable()
              {
                public void run(IProgressMonitor monitor) throws CoreException
                {
                  try
                  {
                    File newFolder = new File(parentFolder, newName);
                    copyTree(folder, source.getName(), newName, folder, newFolder);
                    importProject(newFolder, newName);
                  }
                  catch (Exception ex)
                  {
                    ProjectCopyPlugin.INSTANCE.coreException(ex);
                  }
                }
              }, monitor);

              return Status.OK_STATUS;
            }
            catch (CoreException ex)
            {
              return ex.getStatus();
            }
          }
        }.schedule();
      }
    }
    catch (Exception ex)
    {
      ProjectCopyPlugin.INSTANCE.log(ex);
    }
  }

  private static void copyTree(File folder, String oldName, String newName, File source, File target) throws Exception
  {
    if (source.isDirectory())
    {
      IOUtil.mkdirs(target);
      File[] files = source.listFiles();
      for (File file : files)
      {
        String name = file.getName();
        copyTree(folder, oldName, newName, new File(source, name), new File(target, name));
      }
    }
    else
    {
      if (source.equals(new File(folder, ".project")))
      {
        Replacer replacer = new Replacer("<name>" + oldName + "</name>", "<name>" + newName + "</name>");
        replacer.copy(source, target);
      }
      else if (source.equals(new File(new File(folder, "META-INF"), "MANIFEST.MF")))
      {
        Replacer replacer = new Replacer("Bundle-SymbolicName: " + oldName, "Bundle-SymbolicName: " + newName);
        replacer.copy(source, target);
      }
      else if (source.equals(new File(folder, "feature.xml")))
      {
        oldName = replaceFeatureSuffix(oldName, "");
        newName = replaceFeatureSuffix(newName, "");
        Replacer replacer = new Replacer("id=\"" + oldName + "\"", "id=\"" + newName + "\"");
        replacer.copy(source, target);
      }
      else if (source.equals(new File(folder, "component.def")))
      {
        oldName = replaceFeatureSuffix(oldName, ".feature.group");
        newName = replaceFeatureSuffix(newName, ".feature.group");
        Replacer replacer = new Replacer("id=\"" + oldName + "\"", "id=\"" + newName + "\"");
        replacer.copy(source, target);
      }
      else if (source.equals(new File(folder, "pom.xml")))
      {
        oldName = replaceFeatureSuffix(oldName, "");
        newName = replaceFeatureSuffix(newName, "");

        DocumentBuilder documentBuilder = XMLUtil.createDocumentBuilder();
        Element rootElement = XMLUtil.loadRootElement(documentBuilder, source);

        ElementUpdater artifactIDUpdater = new ElementUpdater(rootElement, "artifactId");
        String oldContents = IOUtil.readUTF8(source);
        String newContents = artifactIDUpdater.update(oldContents, newName);
        IOUtil.writeUTF8(target, newContents);
      }
      else
      {
        IOUtil.copyFile(source, target);
      }
    }
  }

  private static String replaceFeatureSuffix(String name, String newSuffix)
  {
    if (name.endsWith("-feature"))
    {
      name = name.substring(0, name.length() - "-feature".length());
      name += newSuffix;
    }

    return name;
  }

  private static void importProject(final File folder, final String name) throws CoreException
  {
    final IWorkspace workspace = ResourcesPlugin.getWorkspace();
    workspace.run(new IWorkspaceRunnable()
    {
      public void run(IProgressMonitor monitor) throws CoreException
      {
        Path locationPath = new Path(folder.getAbsolutePath());
        if (Platform.getLocation().isPrefixOf(locationPath))
        {
          locationPath = null;
        }

        IProjectDescription description = workspace.newProjectDescription(name);
        description.setLocation(locationPath);

        IProject project = workspace.getRoot().getProject(name);
        project.create(description, monitor);

        if (!project.isOpen())
        {
          project.open(monitor);
        }
      }
    }, new NullProgressMonitor());
  }

  /**
   * @author Eike Stepper
   */
  public static class Replacer
  {
    private final String search;

    private final String replace;

    private boolean done;

    public Replacer(String search, String replace)
    {
      this.search = search;
      this.replace = replace;
    }

    public final void copy(File source, File target) throws IOException
    {
      BufferedReader reader = null;
      BufferedWriter writer = null;

      try
      {
        reader = new BufferedReader(new FileReader(source));
        writer = new BufferedWriter(new FileWriter(target));
        copyText(reader, writer);
      }
      finally
      {
        IOUtil.close(reader);
        IOUtil.close(writer);
      }
    }

    private void copyText(BufferedReader reader, BufferedWriter writer) throws IOException
    {
      String line;
      while ((line = reader.readLine()) != null)
      {
        if (!done)
        {
          int start = line.indexOf(search);
          if (start != -1)
          {
            line = line.substring(0, start) + replace + line.substring(start + search.length());
            done = true;
          }
        }

        writer.write(line);
        writer.newLine();
      }
    }
  }
}