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.*;
14 import com.thoughtworks.xstream.*;
15 import org.prevayler.*;
16 import org.prevayler.implementation.snapshot.*;
18 public class Servlet extends HttpServlet {
20 private ServletScope servletscope = null;
22 private Prevayler prevayler;
24 private ServletContext cx = null;
26 public void destroy() { try {
27 synchronized(this.getClass()) {
28 Prevayler privatePrevayler = prevayler;
29 if (prevayler == null) return;
31 Prevalence.destroy(cx, prevayler);
33 } catch (Exception e) { e.printStackTrace(); } }
35 public void init(ServletConfig sc) throws ServletException {
36 cx = sc.getServletContext();
37 prevayler = Prevalence.getPrevayler(cx);
38 prevalent = (JS)prevayler.prevalentSystem();
41 public void doPost(HttpServletRequest request, HttpServletResponse response) throws IOException { doGet(request, response); }
42 public void doGet(HttpServletRequest request, HttpServletResponse response) throws IOException {
43 servletscope = new ServletScope(request, response, cx);
44 path = cx.getRealPath(((HttpServletRequest)request).getServletPath());
45 Reader xmlreader = new InputStreamReader(new FileInputStream(path));
46 new Template(servletscope, new JSScope(servletscope), xmlreader).wrap(null).toXML(response.getWriter());
49 public class ServletScope extends JSScope {
50 HttpServletRequest request;
51 HttpServletResponse response;
53 public String getRealPath(String s) { return cx.getRealPath(s); }
54 public ServletScope(ServletRequest request, ServletResponse response, ServletContext cx) {
56 this.request = (HttpServletRequest)request;
57 this.response = (HttpServletResponse)response;
60 private JS params = new JS() {
61 public Object get(Object key) { return request.getParameter(JS.toString(key)); }
62 public Enumeration keys() { return request.getParameterNames(); }
64 private JS cookies = new JS() {
66 public Object get(Object key) { return request.getCookie(JS.toString(key)); }
67 public Enumeration keys() { return request.getCookieNames(); }
70 private JS sessionAttributes = new JS() {
71 public Object get(Object key) { return request.getSession(true).getAttribute(JS.toString(key)); }
72 public void put(Object key, Object val) {
73 if (val == null) request.getSession(true).removeAttribute(JS.toString(key));
74 else request.setAttribute(JS.toString(key), val); }
75 public Enumeration keys() { return request.getSession(true).getAttributeNames(); }
77 private JS requestHeader = new JS() {
78 public Object get(Object key) { return request.getHeader(JS.toString(key)); }
79 public Enumeration keys() { return request.getHeaderNames(); }
81 private JS responseHeader = new JS() {
82 public void put(Object key, Object val) { response.setHeader(JS.toString(key), JS.toString(val)); }
86 /** lets us put multi-level get/put/call keys all in the same method */
87 private class Sub extends JS {
89 Sub(Object key) { this.key = key; }
90 public void put(Object key, Object val) throws JSExn {
91 ServletScope.this.put(JS.toString(this.key) + "." + JS.toString(key), val); }
92 public Object get(Object key) throws JSExn {
93 return ServletScope.this.get(JS.toString(this.key) + "." + JS.toString(key)); }
94 public Object call(Object a0, Object a1, Object a2, Object[] rest, int nargs) throws JSExn {
95 return ServletScope.this.callMethod(this.key, a0, a1, a2, rest, nargs);
97 public Object callMethod(Object method, Object a0, Object a1, Object a2, Object[] rest, int nargs) throws JSExn {
98 return ServletScope.this.callMethod(JS.toString(this.key) + "."
99 + JS.toString(method), a0, a1, a2, rest, nargs);
102 private Sub getSub(String key) { return new Sub(key); }
104 public Object callMethod(Object method, final Object a, final Object b, Object c, Object[] rest, int nargs) throws JSExn {
106 case "session.invalidate": request.getSession(true).invalidate(); return null;
108 String path = JS.toString(a);
109 if (path.indexOf("..") != -1) throw new JSExn("cannot use .. in paths");
110 File f = new File(cx.getRealPath("/") + File.separatorChar + path);
111 if (!f.isDirectory()) return null;
112 String[] contents = f.list();
113 JSArray ret = new JSArray(contents.length);
114 for(int i=0; i<contents.length; i++) ret.addElement(contents[i]);
119 public Object get(Object key) throws JSExn {
122 case "arg": return null;
123 case "prevalent": return prevayler.prevalentSystem();
124 case "request": return getSub("request");
125 case "request.user": return request.getRemoteUser();
126 case "request.header": return requestHeader;
127 case "request.method": return request.getMethod();
128 case "request.remote": return getSub("request.remote");
129 case "request.remote.ip": return request.getRemoteAddr();
130 case "request.remote.host": return request.getRemoteHost();
131 case "request.ssl": return request.isSecure();
132 case "request.path": return request.getPathInfo();
133 case "response": return getSub("response");
134 case "response.header": return responseHeader;
135 case "session": return getSub("session");
136 case "session.attr": return sessionAttributes;
137 case "session.created": return new JSDate(request.getSession(true).getCreationTime());
138 case "session.accessed": return new JSDate(request.getSession(true).getLastAccessedTime());
139 case "session.invalidate": return METHOD;
140 case "page": return getSub("page");
141 case "page.lastmodified": return new JSDate(new File(path).lastModified());
142 case "context": return getSub("context");
143 case "context.list": return METHOD;
144 case "params": return params;
145 case "cookie": return cookies;
149 public void put(Object key, Object val) throws JSExn {
151 //#switch(JS.toString(key))
153 try { prevayler.execute(new Prevalence.JSTransaction((JS)val));
154 } catch (Exception e) { e.printStackTrace(); throw new RuntimeException(e); }
155 case "response.code": response.setStatus(JS.toInt(val));
156 case "response.redirect": response.sendRedirect(JS.toString(val));
157 case "response.contentType": response.setContentType(JS.toString(val));
159 } catch (IOException e) {