| /* |
| * Copyright 2000-2010 JetBrains s.r.o. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| /* |
| * XSD/DTD Model generator tool |
| * |
| * By Gregory Shrago |
| * 2002 - 2006 |
| */ |
| package org.jetbrains.idea.devkit.dom.generator; |
| |
| import org.apache.xerces.xs.XSObject; |
| |
| import javax.xml.namespace.QName; |
| import java.io.PrintWriter; |
| import java.util.*; |
| |
| /** |
| * @author Gregory.Shrago |
| * @author Konstantin Bulenkov |
| */ |
| public class ModelDesc { |
| final Map<String, String> name2replaceMap = new HashMap<String, String>(); |
| final Map<QName, String> qname2FileMap = new HashMap<QName, String>(); |
| final Map<String, NamespaceDesc> nsdMap = new HashMap<String, NamespaceDesc>(); |
| final Map<String, TypeDesc> jtMap = new TreeMap<String, TypeDesc>(); |
| |
| |
| public String getNSDPrefix(TypeDesc td) { |
| return getNSDPrefix(td.xsNamespace, td.name, td.type == TypeDesc.TypeEnum.ENUM); |
| } |
| |
| public String getNSDPrefix(String namespace, String name, boolean isEnum) { |
| final int lastIdx = name.lastIndexOf("."); |
| if (lastIdx > -1) return name.substring(0, lastIdx + 1); |
| final NamespaceDesc nsd = getNSD(namespace); |
| if (isEnum && nsd.enumPkg != null) { |
| return nsd.enumPkg + "."; |
| } |
| if (nsd.pkgNames != null) { |
| final QName qname = new QName(namespace, name); |
| final String files = qname2FileMap.get(qname); |
| if (files != null) { |
| for (int i = 0; i < nsd.pkgNames.length; i += 2) { |
| String file = nsd.pkgNames[i]; |
| String pkg = nsd.pkgNames[i + 1]; |
| if (files.contains(":" + file + ":")) { |
| return pkg + "."; |
| } |
| } |
| } |
| } |
| return nsd.pkgName != null && nsd.pkgName.length() > 0 ? nsd.pkgName + "." : ""; |
| } |
| |
| public NamespaceDesc getNSD(String namespace) { |
| NamespaceDesc nsd = nsdMap.get(namespace); |
| if (nsd == null) { |
| nsd = nsdMap.get(""); |
| } |
| return nsd; |
| } |
| |
| |
| public String toJavaTypeName(String tname, String ns) { |
| final int lastIndex = tname.lastIndexOf('.'); |
| String xmlName = lastIndex>-1? tname.substring(lastIndex+1): tname; |
| NamespaceDesc nsd = getNSD(ns); |
| if (ns == null || !ns.endsWith(".dtd")) { |
| if (xmlName.endsWith(Util.ANONYMOUS_ELEM_TYPE_SUFFIX)) xmlName = xmlName.substring(0, xmlName.length() - Util.ANONYMOUS_ELEM_TYPE_SUFFIX.length()); |
| else if (xmlName.endsWith(Util.ANONYMOUS_ATTR_TYPE_SUFFIX)) xmlName = xmlName.substring(0, xmlName.length() - Util.ANONYMOUS_ATTR_TYPE_SUFFIX.length()); |
| else if (xmlName.endsWith(Util.TYPE_SUFFIX)) xmlName = xmlName.substring(0, xmlName.length() - Util.TYPE_SUFFIX.length()); |
| } |
| String rc = Util.capitalize(Util.toJavaName(xmlName)); |
| if (nsd.prefix != null && nsd.prefix.length()>0 && !rc.startsWith(nsd.prefix)) { |
| rc = nsd.prefix + rc; |
| } |
| if (Util.RESERVED_NAMES_MAP.containsKey(rc)) { |
| rc = Util.RESERVED_NAMES_MAP.get(rc); |
| } |
| if (name2replaceMap.containsKey(rc)) { |
| rc = Util.expandProperties(name2replaceMap.get(rc), nsd.props); |
| } |
| |
| return rc; |
| } |
| |
| public String toJavaQualifiedTypeName(XSObject xs, Map<String, NamespaceDesc> nsdMap, boolean isEnum) { |
| String typeName = toJavaTypeName(xs.getName(), xs.getNamespace()); |
| return getNSDPrefix(xs.getNamespace(), xs.getName(), isEnum) + typeName; |
| } |
| |
| public String toJavaQualifiedTypeName(String namespace, String xmlname, boolean isEnum) { |
| return getNSDPrefix(namespace, xmlname, isEnum) + toJavaTypeName(xmlname, namespace); |
| } |
| |
| void dump(final PrintWriter out) { |
| // out.println("-- qname2FileMap ---"); |
| // out.println(qname2FileMap); |
| out.println("-- nsdMap ---"); |
| for (Map.Entry<String, NamespaceDesc> entry : nsdMap.entrySet()) { |
| out.println("namespace key: "+entry.getKey()); |
| dumpNamespace(entry.getValue(), out); |
| } |
| out.println("-- jtMap ---"); |
| for (Map.Entry<String, TypeDesc> entry : jtMap.entrySet()) { |
| out.println("type key: "+entry.getKey()); |
| dumpTypeDesc(entry.getValue(), out); |
| } |
| } |
| |
| private void dumpTypeDesc(TypeDesc td, PrintWriter out) { |
| final ArrayList<String> superList; |
| if (td.supers !=null) { |
| superList = new ArrayList<String>(); |
| for (TypeDesc aSuper : td.supers) { |
| superList.add(getNSDPrefix(aSuper)+aSuper.name); |
| } |
| } else superList = null; |
| out.println(" name " + td.name); |
| out.println(" type " + td.type); |
| out.println(" xsName " + td.xsName); |
| out.println(" xsNS " + td.xsNamespace); |
| out.println(" dups " + td.duplicates); |
| out.println(" supers " + (superList!=null?superList:"null")); |
| out.println(" doc " + (td.documentation != null ? td.documentation.length() : "null")); |
| for (Map.Entry<String, FieldDesc> entry : td.fdMap.entrySet()) { |
| out.println(" field key: " + entry.getKey()); |
| dumpFieldDesc(td, entry.getValue(), out); |
| } |
| } |
| |
| private void dumpFieldDesc(TypeDesc td, FieldDesc fd, PrintWriter out) { |
| out.println(" name " + fd.name); |
| if (td.type == TypeDesc.TypeEnum.ENUM) return; |
| out.println(" clType " + fd.clType); |
| out.println(" required " + fd.required); |
| out.println(" index " + fd.idx+"/"+fd.realIndex); |
| out.println(" choiceOpt " + fd.choiceOpt); |
| out.println(" choice " + (fd.choice!=null?fd.choice.length:"null")); |
| out.println(" content " + fd.contentQualifiedName); |
| out.println(" dupIdx " + fd.duplicateIndex); |
| out.println(" elName " + fd.elementName); |
| out.println(" elType " + fd.elementType); |
| out.println(" def " + fd.def); |
| out.println(" doc " + (fd.documentation != null? fd.documentation.length(): "null")); |
| } |
| |
| private void dumpNamespace(NamespaceDesc value, PrintWriter out) { |
| if (value.skip) return; |
| out.println(" name "+value.name); |
| out.println(" prefix "+value.prefix); |
| out.println(" pkgName "+value.pkgName); |
| out.println(" pkgNames "+(value.pkgNames != null?Arrays.asList(value.pkgNames):"null")); |
| out.println(" enumPkg "+value.enumPkg); |
| out.println(" super "+value.superClass); |
| } |
| } |