/*******************************************************************************
 * Copyright (c) 2001, 2006 IBM Corporation 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:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.wst.xsd.ui.internal.common.commands;

import org.eclipse.wst.xsd.ui.internal.adapters.XSDBaseAdapter;
import org.eclipse.wst.xsd.ui.internal.common.util.Messages;
import org.eclipse.xsd.XSDComplexTypeDefinition;
import org.eclipse.xsd.XSDConcreteComponent;
import org.eclipse.xsd.XSDDerivationMethod;
import org.eclipse.xsd.XSDFactory;
import org.eclipse.xsd.XSDSimpleTypeDefinition;
import org.eclipse.xsd.XSDTypeDefinition;
import org.eclipse.xsd.XSDVariety;

public class SetBaseTypeCommand extends BaseCommand
{
  XSDConcreteComponent concreteComponent;
  XSDTypeDefinition baseType;
  XSDBaseAdapter adapter;
  

  public SetBaseTypeCommand(XSDConcreteComponent concreteComponent, XSDTypeDefinition baseType)
  {
    super(Messages._UI_ACTION_SET_BASE_TYPE);
    this.concreteComponent = concreteComponent;
    this.baseType = baseType;
  }

  public void execute()
  {
    try
    {
      beginRecording(concreteComponent.getElement());
      if (concreteComponent instanceof XSDComplexTypeDefinition)
      {
        XSDComplexTypeDefinition complexType = (XSDComplexTypeDefinition) concreteComponent;

        if (baseType instanceof XSDSimpleTypeDefinition)
        {
          if (!(complexType.getContent() instanceof XSDSimpleTypeDefinition))
          {
            XSDSimpleTypeDefinition simpleContent = XSDFactory.eINSTANCE.createXSDSimpleTypeDefinition();
            complexType.setContent(simpleContent);
          }
        }

        complexType.setBaseTypeDefinition(baseType);
        complexType.setDerivationMethod(XSDDerivationMethod.EXTENSION_LITERAL);
        formatChild(complexType.getElement());
      }
      else if (concreteComponent instanceof XSDSimpleTypeDefinition)
      {
        XSDSimpleTypeDefinition simpleType = (XSDSimpleTypeDefinition) concreteComponent;
        if (baseType instanceof XSDSimpleTypeDefinition)
        {
          XSDVariety variety = simpleType.getVariety();
          if (variety.getValue() == XSDVariety.ATOMIC)
          {
            simpleType.setBaseTypeDefinition((XSDSimpleTypeDefinition) baseType);
          }
          else if (variety.getValue() == XSDVariety.UNION)
          {
            simpleType.getMemberTypeDefinitions().add(baseType);
          }
          else if (variety.getValue() == XSDVariety.LIST)
          {
            simpleType.setItemTypeDefinition((XSDSimpleTypeDefinition) baseType);
          }
        }
      }
    }
    finally
    {
      endRecording();
    }
  }
}