5 import org.ibex.js.JSArray;
6 import org.ibex.js.JSDate;
7 import org.ibex.js.JSExn;
12 import javax.servlet.*;
13 import javax.servlet.http.*;
15 import org.prevayler.*;
16 import org.prevayler.implementation.snapshot.*;
18 public class Servlet extends HttpServlet {
21 private Prevayler prevayler;
23 private ServletContext cx = null;
25 public void destroy() { try {
26 synchronized(this.getClass()) {
27 Prevayler privatePrevayler = prevayler;
28 if (prevayler == null) return;
30 Prevalence.destroy(cx, prevayler);
32 } catch (Exception e) { e.printStackTrace(); } }
34 public void init(ServletConfig sc) throws ServletException {
35 cx = sc.getServletContext();
36 prevayler = Prevalence.getPrevayler(cx);
37 prevalent = (JS)prevayler.prevalentSystem();
40 public void doPost(HttpServletRequest request, HttpServletResponse response) throws IOException { doGet(request, response); }
41 public void doGet(HttpServletRequest rq, HttpServletResponse rs) throws IOException {
42 String path = cx.getRealPath(rq.getServletPath());
43 Servlet.Scope scope = new Servlet.Scope(cx, rq, rs, prevayler);
44 try { Template.wrap(Template.parse(path, scope), scope).toXML(rs.getWriter()); }
45 catch (Exception e) { e.printStackTrace(); System.out.println("e = "+e); }
48 public static class Scope extends Template.Scope {
49 private final ServletContext cx;
50 private final HttpServletRequest request;
51 private final HttpServletResponse response;
52 private final Prevayler prevayler;
54 public Scope(ServletContext cx, HttpServletRequest rq, HttpServletResponse rs, Prevayler p) {
55 super(null); this.cx = cx; request = rq; response = rs; prevayler = p;
58 public String getLocalPath() { return cx.getRealPath("/") + "/WEB-INF/"; }
59 public void transaction(JS t) {
60 try { prevayler.execute(new Prevalence.JSTransaction(t)); }
61 catch (Exception e) { e.printStackTrace(); throw new RuntimeException(e); }
64 private JS params = new JS() {
65 public Object get(Object key) { return request.getParameter(JS.toString(key)); }
66 public Enumeration keys() { return request.getParameterNames(); }
68 private JS cookies = new JS() {
70 public Object get(Object key) { return request.getCookie(JS.toString(key)); }
71 public Enumeration keys() { return request.getCookieNames(); }
74 private JS sessionAttributes = new JS() {
75 public Object get(Object key) {
76 return request.getSession(true).getAttribute(JS.toString(key)); }
77 public void put(Object key, Object val) {
78 if (val == null) request.getSession(true).removeAttribute(JS.toString(key));
79 else request.setAttribute(JS.toString(key), val); }
80 public Enumeration keys() { return request.getSession(true).getAttributeNames(); }
82 private JS requestHeader = new JS() {
83 public Object get(Object key) { return request.getHeader(JS.toString(key)); }
84 public Enumeration keys() { return request.getHeaderNames(); }
86 private JS responseHeader = new JS() {
87 public void put(Object key, Object val) {
88 response.setHeader(JS.toString(key), JS.toString(val)); }
92 /** lets us put multi-level get/put/call keys all in the same method */
93 private class Sub extends JS {
95 Sub(Object key) { this.key = key; }
96 public void put(Object key, Object val) throws JSExn {
97 Scope.this.put(JS.toString(this.key) + "." + JS.toString(key), val); }
98 public Object get(Object key) throws JSExn {
99 return Scope.this.get(JS.toString(this.key) + "." + JS.toString(key)); }
100 public Object call(Object a0, Object a1, Object a2, Object[] rest, int nargs) throws JSExn {
101 return Scope.this.callMethod(this.key, a0, a1, a2, rest, nargs);
103 public Object callMethod(Object method, Object a0, Object a1, Object a2, Object[] rest, int nargs) throws JSExn {
104 return Scope.this.callMethod(JS.toString(this.key) + "."
105 + JS.toString(method), a0, a1, a2, rest, nargs);
108 private Sub getSub(String key) { return new Sub(key); }
110 public Object callMethod(Object method, final Object a, final Object b, Object c, Object[] rest, int nargs) throws JSExn {
112 case "session.invalidate": request.getSession(true).invalidate(); return null;
114 String path = JS.toString(a);
115 if (path.indexOf("..") != -1) throw new JSExn("cannot use .. in paths");
116 File f = new File(cx.getRealPath("/") + File.separatorChar + path);
117 if (!f.isDirectory()) return null;
118 String[] contents = f.list();
119 JSArray ret = new JSArray(contents.length);
120 for(int i=0; i<contents.length; i++) ret.addElement(contents[i]);
125 public Object get(Object key) throws JSExn {
128 case "arg": return null;
129 case "prevalent": return prevayler.prevalentSystem();
130 case "request": return getSub("request");
131 case "request.user": return request.getRemoteUser();
132 case "request.header": return requestHeader;
133 case "request.method": return request.getMethod();
134 case "request.remote": return getSub("request.remote");
135 case "request.remote.ip": return request.getRemoteAddr();
136 case "request.remote.host": return request.getRemoteHost();
137 case "request.ssl": return new Boolean(request.isSecure());
138 case "request.path": return request.getPathInfo();
139 case "response": return getSub("response");
140 case "response.header": return responseHeader;
141 case "session": return getSub("session");
142 case "session.attr": return sessionAttributes;
143 case "session.created": return new JSDate(request.getSession(true).getCreationTime());
144 case "session.accessed": return new JSDate(request.getSession(true).getLastAccessedTime());
145 case "session.invalidate": return METHOD;
146 case "page": return getSub("page");
147 //case "page.lastmodified": return new JSDate(new File(path).lastModified()); FIXME
148 case "context": return getSub("context");
149 case "context.list": return METHOD;
150 case "params": return params;
151 case "cookie": return cookies;
152 case "xt.date": return new JSDate(); // TODO: discuss
156 public void put(Object key, Object val) throws JSExn {
158 //#switch(JS.toString(key))
159 case "transaction": transaction((JS)val);
160 case "response.code": response.setStatus(JS.toInt(val));
161 case "response.redirect": response.sendRedirect(JS.toString(val));
162 case "response.contentType": response.setContentType(JS.toString(val));
164 } catch (IOException e) {