001    package org.codehaus.groovy.sandbox.markup;
002    /*
003    
004    Copyright 2004 (C) John Wilson. All Rights Reserved.
005    
006    Redistribution and use of this software and associated documentation
007    ("Software"), with or without modification, are permitted provided
008    that the following conditions are met:
009    
010    1. Redistributions of source code must retain copyright
011       statements and notices.  Redistributions must also contain a
012       copy of this document.
013    
014    2. Redistributions in binary form must reproduce the
015       above copyright notice, this list of conditions and the
016       following disclaimer in the documentation and/or other
017       materials provided with the distribution.
018    
019    3. The name "groovy" must not be used to endorse or promote
020       products derived from this Software without prior written
021       permission of The Codehaus.  For written permission,
022       please contact info@codehaus.org.
023    
024    4. Products derived from this Software may not be called "groovy"
025       nor may "groovy" appear in their names without prior written
026       permission of The Codehaus. "groovy" is a registered
027       trademark of The Codehaus.
028    
029    5. Due credit should be given to The Codehaus -
030       http://groovy.codehaus.org/
031    
032    THIS SOFTWARE IS PROVIDED BY THE CODEHAUS AND CONTRIBUTORS
033    ``AS IS'' AND ANY EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT
034    NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
035    FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
036    THE CODEHAUS OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
037    INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
038    (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
039    SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
040    HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
041    STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
042    ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
043    OF THE POSSIBILITY OF SUCH DAMAGE.
044    
045    */
046    
047    import groovy.lang.Closure;
048    import groovy.lang.ParameterArray;
049    
050    import java.util.Collections;
051    import java.util.HashMap;
052    import java.util.Map;
053    
054    public class BaseMarkupBuilder extends Builder {
055            public BaseMarkupBuilder(final Map namespaceMethodMap) {
056                    super(namespaceMethodMap);
057            }
058            
059            public Object bind(final Closure root) {
060                    return new Document(root, this.namespaceMethodMap);
061            }
062            
063            private static class Document extends Built {
064                    private Object out;
065                    private final Map pendingNamespaces = new HashMap();
066                    private final Map namespaces = new HashMap();
067                    private String prefix = "";
068                    
069                    public Document(final Closure root, final Map namespaceMethodMap) {
070                            super(root, namespaceMethodMap);
071                            
072                            this.namespaces.put("xml", "http://www.w3.org/XML/1998/namespace");                             // built in namespace
073                            this.namespaces.put("mkp", "http://www.codehaus.org/Groovy/markup/keywords");   // pseudo namespace for markup keywords
074                    }
075                    
076                    /* (non-Javadoc)
077                     * @see groovy.lang.GroovyObject#invokeMethod(java.lang.String, java.lang.Object)
078                     */
079                    public Object invokeMethod(final String name, final Object args) {
080                            final Object[] arguments = (Object[]) args;
081                            Map attrs = Collections.EMPTY_MAP;
082                            Object body = null;
083                            
084                            
085                            //
086                            // Sort the parameters out
087                            //
088                            for (int i = 0; i != arguments.length; i++) {
089                                    final Object arg = arguments[i];
090                                    
091                                    if (arg instanceof Map) {
092                                            attrs = (Map)arg;
093                                    } else if (arg instanceof Closure) {
094                                            final Closure c = ((Closure) arg);
095                                            
096                                            c.setDelegate(this);
097                                            body = c.asWritable();
098                                    } else {
099                                            body = arg;
100                                    }
101                            }
102                            
103                            //
104                            // call the closure corresponding to the tag
105                            //
106                            final Object uri;
107                            
108                            if (this.pendingNamespaces.containsKey(this.prefix)) {
109                                    uri = this.pendingNamespaces.get(this.prefix);
110                            } else if (this.namespaces.containsKey(this.prefix)) {
111                                    uri = this.namespaces.get(this.prefix);
112                            } else {
113                                    uri = ":";
114                            }
115                            
116                            final Object[] info  = (Object[])this.namespaceSpecificTags.get(uri);
117                            final Map tagMap = (Map)info[2];
118                            final Closure defaultTagClosure = (Closure)info[0];
119                            
120                            final String prefix = this.prefix;
121                            this.prefix = "";
122                            
123                            if (tagMap.containsKey(name)) {
124                                    return ((Closure)tagMap.get(name)).call(new ParameterArray(new Object[]{this, this.pendingNamespaces, this.namespaces, this.namespaceSpecificTags, prefix, attrs, body, this.out}));
125                            } else {
126                                    return defaultTagClosure.call(new ParameterArray(new Object[]{name, this, this.pendingNamespaces, this.namespaces, this.namespaceSpecificTags, prefix, attrs, body, this.out}));                
127                            }
128                    }
129                    
130                    /* (non-Javadoc)
131                     * @see groovy.lang.GroovyObject#getProperty(java.lang.String)
132                     */
133                    public Object getProperty(final String property) {
134                            this.prefix = property;
135                            return this;
136                    }
137                    
138                    /* (non-Javadoc)
139                     * @see groovy.lang.GroovyObject#setProperty(java.lang.String, java.lang.Object)
140                     */
141                    public void setProperty(String property, Object newValue) {
142                            if ("trigger".equals(property)) {
143                                    this.out = newValue;
144                                    this.root.call();
145                            } else {
146                                    super.setProperty(property, newValue);
147                            }
148                    }
149            }
150    }