org.simantics.xml.sax.base/src/org/simantics/xml/sax/base/ParserElement.java

   1 package org.simantics.xml.sax.base;
   2
   3 import java.io.Serializable;
   4 import java.util.ArrayList;
   5 import java.util.Collection;
   6 import java.util.HashMap;
   7 import java.util.List;
   8 import java.util.Map;
   9
  10 import org.simantics.db.Resource;
  11 import org.simantics.db.WriteGraph;
  12 import org.simantics.db.common.utils.ListUtils;
  13 import org.simantics.db.exception.DatabaseException;
  14 import org.simantics.layer0.Layer0;
  15 import org.simantics.utils.datastructures.MapList;
  16 import org.xml.sax.Attributes;
  17
  18 public class ParserElement implements Serializable {
  19         private static final long serialVersionUID = -5207502156942818875L;
  20         private String uri;
  21         private String localName;
  22         private String qName;
  23         Map<String, Attribute> attributes;
  24         MapList<Resource, ParserElement> lists;
  25
  26         XMLElementParser elementParser;
  27         XMLParser xmlParser;
  28
  29         Map<String,String> nsMap;
  30
  31         private ParserElement parent;
  32
  33         public ParserElement(String uri, String localName, String qName, Attributes attributes) {
  34                 this.uri = uri;
  35                 this.localName = localName;
  36                 this.qName = qName;
  37                 int attrCount = attributes.getLength();
  38                 this.attributes = new HashMap<>(attrCount);
  39                 for (int i = 0; i < attrCount; i++) {
  40                         this.attributes.put(attributes.getLocalName(i),new Attribute(attributes.getLocalName(i),attributes.getQName(i),attributes.getURI(i),attributes.getValue(i)));
  41                 }
  42                 processNS();
  43         }
  44
  45         public ParserElement(String uri, String localName, String qName, Collection<Attribute> attributes) {
  46                 this.uri = uri;
  47                 this.localName = localName;
  48                 this.qName = qName;
  49                 this.attributes = new HashMap<>(attributes.size());
  50                 for (Attribute a : attributes) {
  51                         this.attributes.put(a.localName,new Attribute(a.localName,a.qName,a.uri,a.value));
  52                 }
  53                 processNS();
  54         }
  55
  56         public ParserElement(ParserElement parent, String uri, String localName, String qName, Attributes attributes) {
  57                 this.parent = parent;
  58                 this.uri = uri;
  59                 this.localName = localName;
  60                 this.qName = qName;
  61                 int attrCount = attributes.getLength();
  62                 this.attributes = new HashMap<>(attrCount);
  63                 for (int i = 0; i < attrCount; i++) {
  64                         this.attributes.put(attributes.getLocalName(i),new Attribute(attributes.getLocalName(i),attributes.getQName(i),attributes.getURI(i),attributes.getValue(i)));
  65                 }
  66                 processNS();
  67         }
  68
  69         public ParserElement(ParserElement parent, String uri, String localName, String qName, Collection<Attribute> attributes) {
  70                 this.parent = parent;
  71                 this.uri = uri;
  72                 this.localName = localName;
  73                 this.qName = qName;
  74                 this.attributes = new HashMap<>(attributes.size());
  75                 for (Attribute a : attributes) {
  76                         this.attributes.put(a.localName,new Attribute(a.localName,a.qName,a.uri,a.value));
  77                 }
  78                 processNS();
  79         }
  80
  81         private void processNS() {
  82
  83                 nsMap = new HashMap<>();
  84                 for (Attribute a : attributes.values()) {
  85                         if (a.localName.startsWith("xmlns")) {
  86                                 String ns[] = a.localName.split(":");
  87                                 if (ns.length == 2) {
  88                                         nsMap.put(ns[1], a.value);
  89                                 }
  90                         }
  91                 }
  92                 Map<String, Attribute> newAttrs = new HashMap<>();
  93                 for (Attribute a : attributes.values()) {
  94                         String ns[] = a.localName.split(":");
  95                         if (ns.length == 2) {
  96                                 if ("xmlns".equals(ns[0]))
  97                                         continue;
  98                                 if ("xml".equals(ns[0]))
  99                                         continue;
 100                                 String namespace = getNS(ns[0]);
 101                                 if (namespace != null) {
 102                                         if (newAttrs.put(ns[1], new Attribute(ns[1], a.qName, namespace, a.value)) != null)
 103                                                 throw new RuntimeException("XML parser internal error, overlapping attribute names " + ns[1]);
 104                                 } else {
 105                                         throw new RuntimeException("XML parser error, namespace " + ns[0] + " for attribute " + a.localName + " not defined.");
 106                                 }
 107                         } else {
 108                                 newAttrs.put(a.localName, a);
 109                         }
 110                 }
 111                 attributes = newAttrs;
 112
 113                 String nameNs[] = qName.split(":");
 114                 if (nameNs.length == 2) {
 115                         localName = nameNs[1];
 116                         if (uri.length() == 0)
 117                                 uri = getNS(nameNs[0]) + "/"+localName;
 118                 } else if (localName.length() == 0) {
 119                         localName = qName;
 120                 }
 121         }
 122
 123         public String getNS(String key) {
 124                 String ns = nsMap.get(key);
 125                 if (ns == null && parent != null)
 126                         ns = parent.getNS(key);
 127                 return ns;
 128         }
 129
 130         public String getUri() {
 131                 return uri;
 132         }
 133
 134         public String getLocalName() {
 135                 return localName;
 136         }
 137
 138         public String getQName() {
 139                 return qName;
 140         }
 141
 142         public Collection<Attribute> getAttributes() {
 143                 return attributes.values();
 144         }
 145
 146         public Attribute getAttribute(String name) {
 147                 return attributes.get(name);
 148         }
 149
 150         private Resource data;
 151
 152         public void setData(Resource data) {
 153                 this.data = data;
 154         }
 155
 156         public Resource getData() {
 157                 return data;
 158         }
 159
 160         public void setElementParser(XMLElementParser parser) {
 161                 this.elementParser = parser;
 162         }
 163
 164         public XMLElementParser getElementParser() {
 165                 return elementParser;
 166         }
 167
 168         public void setXMLParser(XMLParser parser) {
 169                 this.xmlParser = parser;
 170         }
 171
 172         public XMLParser getXMLParser() {
 173                 return xmlParser;
 174         }
 175
 176         public void registerListChild(Resource predicate, ParserElement child) {
 177                 if (lists == null)
 178                         lists = new MapList<>();
 179                 lists.add(predicate, child);
 180         }
 181
 182         public void createLists(WriteGraph graph) throws DatabaseException {
 183                 if (lists == null)
 184                         return;
 185                 Layer0 L0 = Layer0.getInstance(graph);
 186                 for (Resource predicate : lists.getKeys()) {
 187                         List<Resource> children = new ArrayList<>();
 188                         for (ParserElement child : lists.getValues(predicate))
 189                                 children.add(child.getData());
 190                         Resource list = ListUtils.create(graph, L0.List, L0.List_Element, null, children);
 191                         graph.claim(getData(), predicate, list);
 192                 }
 193         }
 194
 195 }