5 import javax.servlet.*;
6 import javax.servlet.http.*;
8 import org.ibex.util.*;
9 import org.ibex.util.Collections;
12 import org.prevayler.*;
14 public class Servlet extends HttpServlet {
17 private Prevayler prevayler;
19 private ServletContext cx = null;
21 public void destroy() { try {
22 synchronized(this.getClass()) {
23 Prevayler privatePrevayler = prevayler;
24 if (prevayler == null) return;
26 Prevalence.destroy(cx, prevayler);
28 } catch (Exception e) { e.printStackTrace(); } }
30 public void init(ServletConfig sc) throws ServletException {
31 cx = sc.getServletContext();
32 prevayler = Prevalence.getPrevayler(cx);
33 prevalent = (JS)prevayler.prevalentSystem();
36 public void doPost(HttpServletRequest request, HttpServletResponse response) throws IOException { doGet(request, response); }
37 public void doGet(HttpServletRequest rq, HttpServletResponse rs) throws IOException {
38 String path = cx.getRealPath(rq.getServletPath());
39 Servlet.Scope scope = new Servlet.Scope(cx, rq, rs, prevayler);
40 rs.getWriter().write("beginning output...");
41 try { Template.wrap(Template.parse(path, scope), scope).out(rs.getWriter()); }
42 catch (Exception e) { e.printStackTrace(); System.out.println("e = "+e); }
43 rs.getWriter().write("...output ends.");
46 public static class Scope extends Template.Scope {
47 private final ServletContext cx;
48 private final HttpServletRequest request;
49 private final HttpServletResponse response;
50 private final Prevayler prevayler;
52 public Scope(ServletContext cx, HttpServletRequest rq, HttpServletResponse rs, Prevayler p) {
53 super(null); this.cx = cx; request = rq; response = rs; prevayler = p;
56 public String getLocalPath() { return cx.getRealPath("/") + "/WEB-INF/"; }
57 public void transaction(JS t) {
58 try { prevayler.execute(new Prevalence.JSTransaction(t)); }
59 catch (Exception e) { e.printStackTrace(); throw new RuntimeException(e); }
62 private JS session = new JS() {
63 public Object get(Object key) throws JSExn {
64 //#switch(JS.toString(key))
65 case "created": return new JSDate(request.getSession(true).getCreationTime());
66 case "accessed": return new JSDate(request.getSession(true).getLastAccessedTime());
67 case "invalidate": return METHOD;
69 return super.get(key);
71 public void put(Object key, Object val) throws JSExn {
72 //#switch(JS.toString(key))
73 case "created": throw new JSExn("can not set session.created");
74 case "accessed": throw new JSExn("can not set session.accessed");
75 case "invalidate": throw new JSExn("can not set session.invalidate");
79 public Object callMethod(Object method, final Object a, final Object b, Object c, Object[] rest, int nargs)
81 //#switch(JS.toString(method))
82 case "invalidate": request.getSession(true).invalidate(); return null;
84 return super.callMethod(method, a, b, c, rest, nargs);
87 private JS params = new JS() {
88 private List keys = null;
89 public Object get(Object key) { return request.getParameter(JS.toString(key)); }
90 public Collection keys() {
91 return keys == null ? keys = Collections.list(request.getParameterNames()) : keys; }
93 private JS cookies = new JS() {
95 public Object get(Object key) { return request.getCookie(JS.toString(key)); }
96 public Enumeration keys() { return request.getCookieNames(); }
99 private JS sessionAttributes = new JS() {
100 private List keys = null;
101 public Object get(Object key) {
102 return request.getSession(true).getAttribute(JS.toString(key)); }
103 public void put(Object key, Object val) {
104 if (val == null) request.getSession(true).removeAttribute(JS.toString(key));
105 else request.setAttribute(JS.toString(key), val); }
106 public Collection keys() {
107 return keys == null ? keys = Collections.list(request.getSession(true).getAttributeNames()) : keys; }
109 private JS requestHeader = new JS() {
110 private List keys = null;
111 public Object get(Object key) { return request.getHeader(JS.toString(key)); }
112 public Collection keys() {
113 return keys == null ? keys = Collections.list(request.getHeaderNames()) : keys; }
115 private JS responseHeader = new JS() {
116 public void put(Object key, Object val) {
117 response.setHeader(JS.toString(key), JS.toString(val)); }
121 /** lets us put multi-level get/put/call keys all in the same method */
122 private class Sub extends JS {
124 Sub(Object key) { this.key = key; }
125 public void put(Object key, Object val) throws JSExn {
126 Scope.this.put(JS.toString(this.key) + "." + JS.toString(key), val); }
127 public Object get(Object key) throws JSExn {
128 return Scope.this.get(JS.toString(this.key) + "." + JS.toString(key)); }
129 public Object call(Object a0, Object a1, Object a2, Object[] rest, int nargs) throws JSExn {
130 return Scope.this.callMethod(this.key, a0, a1, a2, rest, nargs);
132 public Object callMethod(Object method, Object a0, Object a1, Object a2, Object[] rest, int nargs) throws JSExn {
133 return Scope.this.callMethod(JS.toString(this.key) + "."
134 + JS.toString(method), a0, a1, a2, rest, nargs);
137 private Sub getSub(String key) { return new Sub(key); }
139 public Object callMethod(Object method, final Object a, final Object b, Object c, Object[] rest, int nargs) throws JSExn {
141 case "session.invalidate": request.getSession(true).invalidate(); return null;
143 String path = JS.toString(a);
144 if (path.indexOf("..") != -1) throw new JSExn("cannot use .. in paths");
145 File f = new File(cx.getRealPath("/") + File.separatorChar + path);
146 if (!f.isDirectory()) return null;
147 String[] contents = f.list();
148 JSArray ret = new JSArray(contents.length);
149 for(int i=0; i<contents.length; i++) ret.add(contents[i]);
154 public Object get(Object key) throws JSExn {
157 case "arg": return null;
158 case "prevalent": return prevayler.prevalentSystem();
159 case "request": return getSub("request");
160 case "request.user": return request.getRemoteUser();
161 case "request.header": return requestHeader;
162 case "request.method": return request.getMethod();
163 case "request.remote": return getSub("request.remote");
164 case "request.remote.ip": return request.getRemoteAddr();
165 case "request.remote.host": return request.getRemoteHost();
166 case "request.ssl": return new Boolean(request.isSecure());
167 case "request.path": return request.getPathInfo();
168 case "response": return getSub("response");
169 case "response.header": return responseHeader;
170 case "session": return session;
171 case "page": return getSub("page");
172 //case "page.lastmodified": return new JSDate(new File(path).lastModified()); FIXME
173 case "context": return getSub("context");
174 case "context.list": return METHOD;
175 case "params": return params;
176 case "cookie": return cookies;
177 case "xt.date": return new JSDate(); // TODO: discuss
181 public void put(Object key, Object val) throws JSExn {
183 //#switch(JS.toString(key))
184 case "transaction": transaction((JS)val);
185 case "response.code": response.setStatus(JS.toInt(val));
186 case "response.redirect": response.sendRedirect(JS.toString(val));
187 case "response.contentType": response.setContentType(JS.toString(val));
189 } catch (IOException e) {