5 import javax.servlet.*;
6 import javax.servlet.http.*;
8 import org.ibex.util.Collections;
11 import org.prevayler.*;
13 public class Servlet extends HttpServlet {
15 private Prevayler prevayler;
17 private ServletContext cx = null;
19 public void destroy() { try {
20 synchronized(this.getClass()) {
21 Prevayler privatePrevayler = prevayler;
22 if (prevayler == null) return;
24 Prevalence.destroy(cx, prevayler);
26 } catch (Exception e) { e.printStackTrace(); } }
28 public void init(ServletConfig sc) throws ServletException {
29 cx = sc.getServletContext();
30 prevayler = Prevalence.getPrevayler(cx);
31 prevalent = (JS)prevayler.prevalentSystem();
34 public void doPost(HttpServletRequest request, HttpServletResponse response) throws IOException { doGet(request, response); }
35 public void doGet(HttpServletRequest rq, HttpServletResponse rs) throws IOException {
36 String src = rq.getServletPath();
37 Servlet.Scope scope = new Servlet.Scope(cx, rq, rs, prevayler);
38 PrintWriter w = new PrintWriter(rs.getWriter());
42 StringWriter buffer = new StringWriter();
43 Template t = Template.parse(cx.getRealPath(src), scope);
44 Template.wrap(t, scope).out(buffer);
46 rs.setContentType("text/xml");
47 w.write(buffer.toString());
49 } catch (Template.RedirectSignal r) {
53 } catch (Template.Signal s) {
55 } catch (JSLeaf.Exn e) {
56 w.print("\n"+src+": ");
57 w.println(e.getMessage());
58 System.out.println(e);
59 } catch (FileNotFoundException e) {
60 w.print("\nfile not found: ");
61 w.println(e.getMessage());
62 System.out.println(e);
63 } catch (IOException e) {
64 w.print("\nio error: ");
65 w.println(e.getMessage());
67 } catch (Exception e) {
68 w.print("Unexpected error occurred.");
69 System.out.println("Unexpected Exception:");
74 public static class Scope extends Template.Scope {
75 private final ServletContext cx;
76 private final HttpServletRequest request;
77 private final HttpServletResponse response;
78 private final Prevayler prevayler;
80 public Scope(ServletContext cx, HttpServletRequest rq, HttpServletResponse rs, Prevayler p) {
81 super(null); this.cx = cx; request = rq; response = rs; prevayler = p;
84 public String getLocalPath() { return cx.getRealPath("/") + "/WEB-INF/"; }
85 public void transaction(JS t) {
86 try { prevayler.execute(new Prevalence.JSTransaction(t)); }
87 catch (Exception e) { e.printStackTrace(); throw new RuntimeException(e); }
90 private JS session = new JS() {
91 public Object get(Object key) throws JSExn {
92 //#switch(JS.toString(key))
93 case "created": return new JSDate(request.getSession(true).getCreationTime());
94 case "accessed": return new JSDate(request.getSession(true).getLastAccessedTime());
95 case "invalidate": return METHOD;
97 return super.get(key);
99 public Object put(Object key, Object val) throws JSExn {
100 //#switch(JS.toString(key))
101 case "created": throw new JSExn("can not set session.created");
102 case "accessed": throw new JSExn("can not set session.accessed");
103 case "invalidate": throw new JSExn("can not set session.invalidate");
105 return super.put(key, val);
107 public Object callMethod(Object method, final Object a, final Object b, Object c, Object[] rest, int nargs)
109 //#switch(JS.toString(method))
110 case "invalidate": request.getSession(true).invalidate(); return null;
112 return super.callMethod(method, a, b, c, rest, nargs);
115 private JS params = new JS() {
116 private List keys = null;
117 public Object get(Object key) { return request.getParameter(JS.toString(key)); }
118 public Collection keys() {
119 return keys == null ? keys = Collections.list(request.getParameterNames()) : keys; }
121 private JS cookies = new JS() {
123 public Object get(Object key) { return request.getCookie(JS.toString(key)); }
124 public Enumeration keys() { return request.getCookieNames(); }
127 private JS sessionAttributes = new JS() {
128 private List keys = null;
129 public Object get(Object key) {
130 return request.getSession(true).getAttribute(JS.toString(key)); }
131 public Object put(Object key, Object val) {
132 if (val == null) request.getSession(true).removeAttribute(JS.toString(key));
133 else request.setAttribute(JS.toString(key), val);
135 public Collection keys() {
136 return keys == null ? keys = Collections.list(request.getSession(true).getAttributeNames()) : keys; }
138 private JS requestHeader = new JS() {
139 private List keys = null;
140 public Object get(Object key) { return request.getHeader(JS.toString(key)); }
141 public Collection keys() {
142 return keys == null ? keys = Collections.list(request.getHeaderNames()) : keys; }
144 private JS responseHeader = new JS() {
145 public Object put(Object key, Object val) {
146 response.setHeader(JS.toString(key), JS.toString(val)); return null; }
150 /** lets us put multi-level get/put/call keys all in the same method */
151 private class Sub extends JS {
153 Sub(Object key) { this.key = key; }
154 public Object put(Object key, Object val) throws JSExn {
155 return Scope.this.put(JS.toString(this.key) + "." + JS.toString(key), val); }
156 public Object get(Object key) throws JSExn {
157 return Scope.this.get(JS.toString(this.key) + "." + JS.toString(key)); }
158 public Object call(Object a0, Object a1, Object a2, Object[] rest, int nargs) throws JSExn {
159 return Scope.this.callMethod(this.key, a0, a1, a2, rest, nargs);
161 public Object callMethod(Object method, Object a0, Object a1, Object a2, Object[] rest, int nargs) throws JSExn {
162 return Scope.this.callMethod(JS.toString(this.key) + "."
163 + JS.toString(method), a0, a1, a2, rest, nargs);
166 private Sub getSub(String key) { return new Sub(key); }
168 public Object callMethod(Object method, final Object a, final Object b, Object c, Object[] rest, int nargs) throws JSExn {
170 case "session.invalidate": request.getSession(true).invalidate(); return null;
172 String path = JS.toString(a);
173 if (path.indexOf("..") != -1) throw new JSExn("cannot use .. in paths");
174 File f = new File(cx.getRealPath("/") + File.separatorChar + path);
175 if (!f.isDirectory()) return null;
176 String[] contents = f.list();
177 JSArray ret = new JSArray(contents.length);
178 for(int i=0; i<contents.length; i++) ret.add(contents[i]);
183 public Object get(Object key) throws JSExn {
186 case "arg": return null;
187 case "prevalent": return prevayler.prevalentSystem();
188 case "request": return getSub("request");
189 case "request.user": return request.getRemoteUser();
190 case "request.header": return requestHeader;
191 case "request.method": return request.getMethod();
192 case "request.remote": return getSub("request.remote");
193 case "request.remote.ip": return request.getRemoteAddr();
194 case "request.remote.host": return request.getRemoteHost();
195 case "request.ssl": return new Boolean(request.isSecure());
196 case "request.path": return request.getPathInfo();
197 case "response": return getSub("response");
198 case "response.header": return responseHeader;
199 case "session": return session;
200 case "page": return getSub("page");
201 //case "page.lastmodified": return new JSDate(new File(path).lastModified()); FIXME
202 case "context": return getSub("context");
203 case "context.list": return METHOD;
204 case "params": return params;
205 case "cookie": return cookies;
206 case "xt.date": return new JSDate(); // TODO: discuss
210 public Object put(Object key, Object val) throws JSExn {
212 //#switch(JS.toString(key))
213 case "response.code": response.setStatus(JS.toInt(val));
214 case "response.redirect": response.sendRedirect(JS.toString(val));
215 case "response.contentType": response.setContentType(JS.toString(val));
218 } catch (IOException e) {