plugins/org.eclipse.emf.cdo.migrator/src/org/eclipse/emf/cdo/internal/migrator/CDOMigrator.java

/***************************************************************************
 * Copyright (c) 2004 - 2008 Eike Stepper, Germany.
 * 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.emf.cdo.internal.migrator;

import org.eclipse.emf.codegen.ecore.genmodel.GenDelegationKind;
import org.eclipse.emf.codegen.ecore.genmodel.GenModel;
import org.eclipse.emf.common.util.EList;
import org.eclipse.emf.common.util.WrappedException;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IFolder;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.NullProgressMonitor;

import java.io.ByteArrayInputStream;
import java.io.InputStream;

/**
 * @author Eike Stepper
 */
public abstract class CDOMigrator
{
  public static final String ROOT_EXTENDS_CLASS = "org.eclipse.emf.internal.cdo.CDOObjectImpl";

  public static final String ROOT_EXTENDS_INTERFACE = "org.eclipse.emf.cdo.CDOObject";

  public static final String PLUGIN_VARIABLE = "CDO=org.eclipse.emf.cdo";

  public static final String CDO_MF_CONTENTS = "This is a marker file for bundles with CDO native models.\n";

  private CDOMigrator()
  {
  }

  public static String adjustGenModel(GenModel genModel, IProject project)
  {
    StringBuilder builder = new StringBuilder();

    if (genModel.getFeatureDelegation() != GenDelegationKind.REFLECTIVE_LITERAL)
    {
      genModel.setFeatureDelegation(GenDelegationKind.REFLECTIVE_LITERAL);
      builder.append("Set Feature Delegation = ");
      builder.append(GenDelegationKind.REFLECTIVE_LITERAL);
      builder.append("\n");
    }

    if (!ROOT_EXTENDS_CLASS.equals(genModel.getRootExtendsClass()))
    {
      genModel.setRootExtendsClass(ROOT_EXTENDS_CLASS);
      builder.append("Set Root Extends Class = ");
      builder.append(ROOT_EXTENDS_CLASS);
      builder.append("\n");
    }

    if (!ROOT_EXTENDS_INTERFACE.equals(genModel.getRootExtendsInterface()))
    {
      genModel.setRootExtendsInterface(ROOT_EXTENDS_INTERFACE);
      builder.append("Set Root Extends Interface = ");
      builder.append(ROOT_EXTENDS_INTERFACE);
      builder.append("\n");
    }

    EList<String> pluginVariables = genModel.getModelPluginVariables();
    if (!pluginVariables.contains(PLUGIN_VARIABLE))
    {
      pluginVariables.add(PLUGIN_VARIABLE);
      builder.append("Added Model Plugin Variables = ");
      builder.append(PLUGIN_VARIABLE);
      builder.append("\n");
    }

    IFolder folder = project.getFolder("META-INF");
    if (!folder.exists())
    {
      try
      {
        folder.create(true, true, new NullProgressMonitor());
        builder.append("Created META-INF folder\n");
      }
      catch (CoreException ex)
      {
        throw new WrappedException(ex);
      }
    }

    IFile file = folder.getFile("CDO.MF");
    if (!file.exists())
    {
      try
      {
        InputStream contents = new ByteArrayInputStream(CDO_MF_CONTENTS.getBytes());
        file.create(contents, true, new NullProgressMonitor());
        builder.append("Created CDO.MF marker file\n");
      }
      catch (CoreException ex)
      {
        throw new WrappedException(ex);
      }
    }

    return builder.length() == 0 ? null : builder.toString();
  }
}