Fixed up the DocBuffer a bit and added a basic XML writer.
[jsvc.git] / src / dolda / jsvc / next / DocBuffer.java
1 package dolda.jsvc.next;
2
3 import java.util.*;
4 import org.w3c.dom.*;
5
6 public class DocBuffer {
7     public final Document doc;
8     private final Map<String, Node> cursors = new HashMap<String, Node>();
9     public static final String ns = "jsvc:next:buffer";
10     
11     public DocBuffer(String ns, String root, String doctype, String pubid, String sysid) {
12         doc = DomUtil.document(ns, root, doctype, pubid, sysid);
13     }
14
15     public DocBuffer(String ns, String root) {
16         this(ns, root, null, null, null);
17     }
18     
19     private Node findcursor(Node c, String name) {
20         if(c instanceof Element) {
21             Element el = (Element)c;
22             String ns = el.getNamespaceURI();
23             if((ns != null) && ns.equals(DocBuffer.ns) && el.getTagName().equals("cursor") && el.getAttributeNS(null, "name").equals(name))
24                 return(c);
25         }
26         for(Node n = c.getFirstChild(); n != null; n = n.getNextSibling()) {
27             Node r = findcursor(n, name);
28             if(r != null)
29                 return(r);
30         }
31         return(null);
32     }
33
34     private Node cursor(String name) {
35         Node n;
36         if((n = cursors.get(name)) != null) {
37             return(n);
38         }
39         if((n = findcursor(doc, name)) == null)
40             return(null);
41         cursors.put(name, n);
42         return(n);
43     }
44     
45     public void insert(String cursor, Node n) {
46         Node c = cursor(cursor);
47         if(c == null)
48             throw(new RuntimeException("No such cursor: `" + cursor + "'"));
49         c.getParentNode().insertBefore(doc.importNode(n, true), c);
50     }
51     
52     public Element makecursor(String name) {
53         Element el = doc.createElementNS(ns, "cursor");
54         Attr a = doc.createAttributeNS(null, "name");
55         a.setValue(name);
56         el.setAttributeNodeNS(a);
57         return(el);
58     }
59
60     public Element el(String ns, String nm, Node contents, String... attrs) {
61         Element el = doc.createElementNS(ns, nm);
62         if(contents != null)
63             el.appendChild(contents);
64         for(String attr : attrs) {
65             int p = attr.indexOf('=');
66             el.setAttribute(attr.substring(0, p), attr.substring(p + 1));
67         }
68         return(el);
69     }
70     
71     public Text text(String text) {
72         return(doc.createTextNode(text));
73     }
74     
75     public void finalise() {
76         Node n = doc;
77         while(true) {
78             Node nx;
79             if(n.getFirstChild() != null) {
80                 nx = n.getFirstChild();
81             } else if(n.getNextSibling() != null) {
82                 nx = n.getNextSibling();
83             } else {
84                 for(nx = n.getParentNode(); nx != null; nx = nx.getParentNode()) {
85                     if(nx.getNextSibling() != null) {
86                         nx = nx.getNextSibling();
87                         break;
88                     }
89                 }
90             }
91             String ns = n.getNamespaceURI();
92             if((ns != null) && ns.equals(DocBuffer.ns))
93                 n.getParentNode().removeChild(n);
94             if(nx == null)
95                 break;
96             else
97                 n = nx;
98         }
99     }
100 }