1 // Copyright 2000-2005 the Contributors, as shown in the revision logs.
2 // Licensed under the Apache Public Source License 2.0 ("the License").
3 // You may not use this file except in compliance with the License.
7 import org.ibex.util.*;
12 import javax.servlet.*;
13 import javax.servlet.http.*;
15 public class Template extends Node.Stream.Filter implements Node.Stream.Functor {
17 static Template newTemplate(Servlet.ServletScope servletscope, JSScope scope, String str) {
19 File f = new File(str);
20 if (!f.exists()) f = new File(str + ".xt");
21 if (!f.exists()) f = new File(str + ".xml");
22 return new Template(servletscope, scope, new InputStreamReader(new FileInputStream(f)));
23 } catch (Exception e) { throw new RuntimeException(e); }
26 static JSScope copyNodeToScope(Node n, JSScope scope) {
28 for(int i=0; i<n.numattrs; i++) {
29 scope.declare(n.attrs[i*2]);
30 scope.put(n.attrs[i*2], n.attrs[i*2+1]);
33 } catch (Exception e) { throw new RuntimeException(e); }
36 private class JSRewriter extends Node.Stream {
37 private Node.Stream in;
38 private JSScope scope;
39 public JSRewriter(Node.Stream in, JSScope scope) { this.in = in; this.scope = scope; }
40 protected boolean _read(Node n) { if (!in.read(n)) return false; transform(n, scope); return true; }
43 public static Node transform(Node n, JSScope scope) {
44 if (n.cdata != null) n.cdata = eval(n.cdata, scope).toString();
45 else for(int i=1; i<n.numattrs*2; i+=2) n.attrs[i] = eval(n.attrs[i], scope).toString();
49 private static Object eval(String s, JSScope scope) {
50 if (s == null) return null;
51 StringBuffer ret = new StringBuffer();
52 for(boolean first = true; s.indexOf("${") != -1; first = false) {
53 ret.append(s.substring(0, s.indexOf("${")));
54 String s2 = s.substring(s.indexOf("${")+2);
55 Object app = exec("return (" + s2.substring(0, s2.indexOf('}')) + ");\n", scope);
56 s = s.substring(s.indexOf('}') + 1);
57 //if (first && s.trim().length() == 0) return app;
58 if (!(app == null || app instanceof String || app instanceof Number || app instanceof Boolean))
59 throw new RuntimeException("javascripts within ${...} can only return strings, numbers, and booleans; not a " +
60 app.getClass().getName());
61 ret.append(app == null ? "null" : app.toString());
64 return ret.toString();
67 public static Object exec(String s, JSScope scope) {
69 return JS.eval(JS.cloneWithNewParentScope(JS.fromReader("input", 0, new StringReader(s)), scope));
70 } catch (Exception e) {
72 throw new RuntimeException(e);
76 private JSScope scope;
77 private Servlet.ServletScope servletscope;
78 private Node.Stream children;
79 public Node.Stream wrap(Node.Stream children) { this.children = children; return this; }
80 public Template(Servlet.ServletScope servletscope, JSScope scope, Reader template) {
81 super(new Node.Stream.FromXML(template));
83 this.servletscope = servletscope;
85 public boolean _read(Node n) { boolean ret = __read(n); if (ret) transform(n, scope); return ret; }
86 public boolean __read(final Node n) {
87 if (!upstreamRead(n)) return false;
88 if (n.cdata != null) return true;
89 final String uri = n.uri;
90 final String name = n.name;
91 if (uri.indexOf(':') == -1)
92 throw new RuntimeException("uri does not contain a colon: " + uri + " (tag name " + name + ")");
93 final String method = uri.substring(0, uri.indexOf(':'));
94 final String rest = uri.substring(uri.indexOf(':')+1);
95 if (uri.equals("http://www.w3.org/1999/xhtml")) { return true;
96 } else if (method.equals("webinf")) {
97 return graft(newTemplate(servletscope, copyNodeToScope(transform(n, scope), new JSScope(servletscope)),
98 servletscope.getRealPath("/") + "/WEB-INF/" + rest + name), n).upstreamRead(n);
99 } else if (uri.equals("http://xt.ibex.org/")) {
103 return graft("true".equals(n.attr("if")) ? new DropTag() : new DropAll(), n).upstreamRead(n);
104 case "js": return graft(new JsTag(scope), n).upstreamRead(n);
105 case "foreach": return graft(new ForEach(n, scope), n).upstreamRead(n);
107 if (children == null) return true;
108 graft(new Node.Stream.ConstantFunctor(children), n);
110 return upstreamRead(n);
113 } else if (method.equals("java")) {
114 try { return graft((Node.Stream.Functor)Class.forName(rest).newInstance(), n).upstreamRead(n); }
115 catch (Exception e) { throw new RuntimeException(e); }
117 throw new RuntimeException("Unknown namespace URI " + uri);
120 private class ForEach extends Node.Stream.Filter implements Node.Stream.Functor {
121 private Node[] nodes = null;
122 private Vec array = new Vec();
123 private JSScope scope;
124 public ForEach(Node n, JSScope s) {
125 super(Node.Stream.NULL);
126 Vec v = ((JSArray)exec("return (" + n.attr("in").toString() + ");", this.scope = s)).toVec();
127 while(true) { Object o = v.pop(); if (o == null) break; array.push(o); }
129 public Node.Stream wrap(Node.Stream kids) {
130 Vec nodes = new Vec();
131 Node n2 = new Node();
132 while(kids.read(n2)) nodes.addElement(new Node(n2));
133 nodes.copyInto(this.nodes = new Node[nodes.size()]);
136 protected boolean _read(Node n) {
137 if (upstreamRead(n)) return true;
138 if (array.size() == 0) return false;
139 JSScope scope2 = new JSScope(scope);
140 try { scope2.declare("x"); scope2.put("x", array.pop()); } catch (JSExn e) { throw new RuntimeException(e); }
141 return graft(new ConstantFunctor(new JSRewriter(new Node.Stream() {
143 protected boolean _read(Node n) {
144 if (i>=nodes.length) return false;
145 n.copyFrom(nodes[i++]);
147 } }, scope2)), n).upstreamRead(n);
151 private class DropTag implements Node.Stream.Functor {
152 public Node.Stream wrap(Node.Stream kids) {
156 private class DropAll implements Node.Stream.Functor {
157 public Node.Stream wrap(Node.Stream kids) {
158 return new Node.Stream() { public boolean _read(Node n) { return false; } };
161 private class JsTag implements Node.Stream.Functor {
163 public JsTag(JSScope scope) { this.scope = scope; }
164 public Node.Stream wrap(final Node.Stream s) {
165 return new Node.Stream() {
166 protected boolean _read(Node n) {
167 boolean ret = s.read(n);
168 if (ret && n.cdata != null) {
169 System.err.println("exec("+n.cdata+")");
170 exec(n.cdata, scope);