Examples with SchemaPrefixChangeHandler org.eclipse.wst.xsd.ui.internal.nsedit.SchemaPrefixChangeHandler used on opensource projects

Example 1 with SchemaPrefixChangeHandler

use of org.eclipse.wst.xsd.ui.internal.nsedit.SchemaPrefixChangeHandler in project webtools.sourceediting by eclipse.

the class XSDEditNamespacesAction method run.

public void run() {
    if (element != null) {
        Shell shell = XMLCommonResources.getInstance().getWorkbench().getActiveWorkbenchWindow().getShell();
        String targetNamespace = null;
        if (xsdSchema != null) {
            targetNamespace = xsdSchema.getTargetNamespace();
        }
        XSDEditSchemaInfoDialog dialog = new XSDEditSchemaInfoDialog(shell, new Path(resourceLocation), targetNamespace);
        List namespaceInfoList = namespaceInfoManager.getNamespaceInfoList(element);
        List oldNamespaceInfoList = NamespaceInfo.cloneNamespaceInfoList(namespaceInfoList);
        // update the document in response to these changes
        for (Iterator i = namespaceInfoList.iterator(); i.hasNext(); ) {
            NamespaceInfo info = (NamespaceInfo) i.next();
            NamespaceInfo oldCopy = new NamespaceInfo(info);
            // $NON-NLS-1$
            info.setProperty("oldCopy", oldCopy);
        }
        dialog.setNamespaceInfoList(namespaceInfoList);
        dialog.create();
        dialog.getShell().setSize(500, 400);
        // $NON-NLS-1$
        dialog.getShell().setText(XMLCommonResources.getInstance().getString("_UI_MENU_EDIT_SCHEMA_INFORMATION_TITLE"));
        dialog.setBlockOnOpen(true);
        dialog.open();
        // $NON-NLS-1$
        String xsdPrefix = null;
        String origXSDPrefix = xsdSchema.getSchemaForSchemaQNamePrefix();
        String newTNSPrefix = "";
        if (dialog.getReturnCode() == Window.OK) {
            Element xsdSchemaElement = xsdSchema.getElement();
            DocumentImpl doc = (DocumentImpl) xsdSchemaElement.getOwnerDocument();
            List newInfoList = dialog.getNamespaceInfoList();
            // see if we need to rename any prefixes
            Map prefixMapping = createPrefixMapping(oldNamespaceInfoList, namespaceInfoList);
            String origTNSPrefix = null;
            Map origPrefixMap = xsdSchema.getQNamePrefixToNamespaceMap();
            for (Iterator iter = origPrefixMap.keySet().iterator(); iter.hasNext(); ) {
                String key = (String) iter.next();
                String ns = (String) origPrefixMap.get(key);
                if ((targetNamespace == null && ns == null) || targetNamespace != null && targetNamespace.equals(ns)) {
                    origTNSPrefix = key;
                    break;
                }
            }
            Map map2 = new Hashtable();
            for (Iterator iter = newInfoList.iterator(); iter.hasNext(); ) {
                NamespaceInfo ni = (NamespaceInfo) iter.next();
                String pref = ni.prefix;
                String uri = ni.uri;
                if (pref == null)
                    // $NON-NLS-1$
                    pref = "";
                if (uri == null)
                    // $NON-NLS-1$
                    uri = "";
                if (XSDConstants.isSchemaForSchemaNamespace(uri)) {
                    xsdPrefix = pref;
                }
                if (uri.equals(dialog.getTargetNamespace())) {
                    newTNSPrefix = pref;
                }
                map2.put(pref, uri);
            }
            if (map2.size() > 0) {
                try {
                    doc.getModel().beginRecording(this, XSDEditorPlugin.getXSDString("_UI_NAMESPACE_CHANGE"));
                    boolean targetNamespaceChanged = (targetNamespace != null && !targetNamespace.equals(dialog.getTargetNamespace()) || targetNamespace == null && dialog.getTargetNamespace() != null);
                    boolean tnsPrefixChanged = !newTNSPrefix.equals(origTNSPrefix);
                    boolean xsdPrefixChanged = (!(origXSDPrefix == null && xsdPrefix.equals("")) || (origXSDPrefix != null && !origXSDPrefix.equals(xsdPrefix)));
                    xsdSchema.setIncrementalUpdate(false);
                    // First handle the prefix change for the target namespace
                    if (tnsPrefixChanged) {
                        prefixMapping.remove(origTNSPrefix);
                        UpdateNamespaceInformationCommand command = new UpdateNamespaceInformationCommand(Messages._UI_ACTION_NAMESPACE_INFORMATION_CHANGE, xsdSchema, newTNSPrefix, targetNamespace);
                        command.execute();
                        xsdSchema.update();
                    }
                    // Second, handle the target namespace change
                    if (targetNamespaceChanged) {
                        // set the targetNamespace attribute
                        xsdSchema.setTargetNamespace(dialog.getTargetNamespace());
                        TargetNamespaceChangeHandler targetNamespaceChangeHandler = new TargetNamespaceChangeHandler(xsdSchema, targetNamespace, dialog.getTargetNamespace());
                        targetNamespaceChangeHandler.resolve();
                    }
                    if (prefixMapping.size() > 0) {
                        for (Iterator iter = prefixMapping.keySet().iterator(); iter.hasNext(); ) {
                            String oldPrefix = (String) iter.next();
                            String newPrefix = (String) prefixMapping.get(oldPrefix);
                            // Now update any references to this old prefix in the schema
                            // with the value of the new prefix
                            String ns = (String) origPrefixMap.get(oldPrefix);
                            // $NON-NLS-1$
                            SchemaPrefixChangeHandler spch = new SchemaPrefixChangeHandler(xsdSchema, newPrefix, ns != null ? ns : "");
                            spch.resolve();
                        }
                    }
                    namespaceInfoManager.removeNamespaceInfo(element);
                    namespaceInfoManager.addNamespaceInfo(element, newInfoList, false);
                    // Third, handle the schema for schema prefix change
                    if (xsdPrefixChanged) {
                        if (xsdPrefix != null && xsdPrefix.length() == 0) {
                            xsdSchema.setSchemaForSchemaQNamePrefix(null);
                        } else {
                            xsdSchema.setSchemaForSchemaQNamePrefix(xsdPrefix);
                        }
                        namespaceInfoManager.removeNamespaceInfo(element);
                        namespaceInfoManager.addNamespaceInfo(element, newInfoList, false);
                        xsdSchema.setIncrementalUpdate(true);
                        // Now change the references to any schema types/components ie. string --> xs:string
                        SchemaPrefixChangeHandler spch = new SchemaPrefixChangeHandler(xsdSchema, xsdPrefix);
                        spch.resolve();
                        // Change the prefix for all schema components
                        updateAllNodes(element, xsdPrefix);
                        // $NON-NLS-1$
                        prefixMapping.remove(origXSDPrefix != null ? origXSDPrefix : "");
                    }
                    xsdSchema.setIncrementalUpdate(true);
                } catch (Exception e) {
                } finally {
                    xsdSchema.update();
                    doc.getModel().endRecording(this);
                }
            }
        }
    }
}