/*******************************************************************************
* Copyright (c) 2008 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.util;
import org.eclipse.core.resources.IFile;
import org.eclipse.jface.viewers.StructuredSelection;
import org.eclipse.jface.viewers.ViewerFilter;
import org.eclipse.jface.window.Window;
import org.eclipse.jface.wizard.WizardDialog;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.IEditorInput;
import org.eclipse.ui.IFileEditorInput;
import org.eclipse.ui.PlatformUI;
import org.eclipse.wst.common.ui.internal.viewers.ResourceFilter;
import org.eclipse.wst.common.uriresolver.internal.util.URIHelper;
import org.eclipse.wst.xsd.ui.internal.editor.XSDEditorPlugin;
import org.eclipse.wst.xsd.ui.internal.util.TypesHelper;
import org.eclipse.wst.xsd.ui.internal.wizards.XSDSelectIncludeFileWizard;
import org.eclipse.xsd.XSDImport;
import org.eclipse.xsd.XSDInclude;
import org.eclipse.xsd.XSDRedefine;
import org.eclipse.xsd.XSDSchema;
import org.eclipse.xsd.impl.XSDImportImpl;
import org.w3c.dom.Element;
public class XSDDirectivesSchemaLocationUpdater
{
/**
* Modifies the schema location by opening the schema location dialog and
* processing the results. This method refactors the code in
* XSDImportSection$widgetSelected and SchemaLocationSection$widgetSelected
* and the processing in handleSchemaLocationChange()
*/
public static void updateSchemaLocation(XSDSchema xsdSchema, Object selection, boolean isInclude)
{
Shell shell = Display.getCurrent().getActiveShell();
IFile currentIFile = null;
IEditorInput editorInput = PlatformUI.getWorkbench().getActiveWorkbenchWindow().getActivePage().getActiveEditor().getEditorInput();
ViewerFilter filter;
if (editorInput instanceof IFileEditorInput)
{
currentIFile = ((IFileEditorInput) editorInput).getFile();
filter = new ResourceFilter(new String[] { ".xsd" }, //$NON-NLS-1$
new IFile[] { currentIFile }, null);
}
else
{
filter = new ResourceFilter(new String[] { ".xsd" }, //$NON-NLS-1$
null, null);
}
XSDSelectIncludeFileWizard fileSelectWizard = new XSDSelectIncludeFileWizard(xsdSchema, isInclude, XSDEditorPlugin.getXSDString("_UI_FILEDIALOG_SELECT_XML_SCHEMA"), //$NON-NLS-1$
XSDEditorPlugin.getXSDString("_UI_FILEDIALOG_SELECT_XML_DESC"), //$NON-NLS-1$
filter, new StructuredSelection(selection));
WizardDialog wizardDialog = new WizardDialog(shell, fileSelectWizard);
wizardDialog.create();
wizardDialog.setBlockOnOpen(true);
int result = wizardDialog.open();
if (result == Window.OK)
{
IFile selectedIFile = fileSelectWizard.getResultFile();
String schemaFileString;
if (selectedIFile != null && currentIFile != null)
{
schemaFileString = URIHelper.getRelativeURI(selectedIFile.getLocation(), currentIFile.getLocation());
}
else if (selectedIFile != null && currentIFile == null)
{
schemaFileString = selectedIFile.getLocationURI().toString();
}
else
{
schemaFileString = fileSelectWizard.getURL();
}
String attributeSchemaLocation = "schemaLocation"; //$NON-NLS-1$
if (selection instanceof XSDImport)
{
XSDImport xsdImport = (XSDImport) selection;
xsdImport.getElement().setAttribute(attributeSchemaLocation, schemaFileString); //$NON-NLS-1$
String namespace = fileSelectWizard.getNamespace();
if (namespace == null)
namespace = ""; //$NON-NLS-1$
XSDSchema externalSchema = fileSelectWizard.getExternalSchema();
java.util.Map map = xsdSchema.getQNamePrefixToNamespaceMap();
Element schemaElement = xsdSchema.getElement();
// update the xmlns in the schema element first, and then update the
// import element next so that the last change will be in the import element. This keeps the
// selection on the import element
TypesHelper helper = new TypesHelper(externalSchema);
String prefix = helper.getPrefix(namespace, false);
if (map.containsKey(prefix))
{
prefix = null;
}
if (prefix == null || (prefix != null && prefix.length() == 0))
{
StringBuffer newPrefix = new StringBuffer("pref"); //$NON-NLS-1$
int prefixExtension = 1;
while (map.containsKey(newPrefix.toString()) && prefixExtension < 100)
{
newPrefix = new StringBuffer("pref" + String.valueOf(prefixExtension)); //$NON-NLS-1$
prefixExtension++;
}
prefix = newPrefix.toString();
}
String attributeNamespace = "namespace"; //$NON-NLS-1$
if (namespace.length() > 0)
{
// if ns already in map, use its corresponding prefix
if (map.containsValue(namespace))
{
TypesHelper typesHelper = new TypesHelper(xsdSchema);
prefix = typesHelper.getPrefix(namespace, false);
}
else
// otherwise add to the map
{
schemaElement.setAttribute("xmlns:" + prefix, namespace); //$NON-NLS-1$
}
// prefixText.setText(prefix);
xsdImport.getElement().setAttribute(attributeNamespace, namespace);
}
// See https://bugs.eclipse.org/bugs/show_bug.cgi?id=155885
// Need to import otherwise the external schema is never
// resolved. One problem is that the schema location is still null,
// so the set types dialog will show types that belong to that schema
// with a null schema location. This should load resource
// into the resource set
if (selection instanceof XSDImportImpl) // redundant
{
XSDImportImpl xsdImportImpl = (XSDImportImpl) selection;
xsdImportImpl.importSchema();
}
}
else if (selection instanceof XSDInclude)
{
XSDInclude xsdInclude = (XSDInclude) selection;
xsdInclude.getElement().setAttribute(attributeSchemaLocation, schemaFileString);
}
else if (selection instanceof XSDRedefine)
{
XSDRedefine xsdRedefine = (XSDRedefine) selection;
xsdRedefine.getElement().setAttribute(attributeSchemaLocation, schemaFileString);
}
}
}
}