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 public static final JS METHOD = new JS.Method();
21 private ServletScope servletscope = null;
23 private Prevayler prevayler;
25 private ServletContext cx = null;
27 public void destroy() { try {
28 synchronized(this.getClass()) {
29 Prevayler privatePrevayler = prevayler;
30 if (prevayler == null) return;
32 Prevalence.destroy(cx, prevayler);
34 } catch (Exception e) { e.printStackTrace(); } }
36 public void init(ServletConfig sc) throws ServletException {
37 cx = sc.getServletContext();
38 prevayler = Prevalence.getPrevayler(cx);
39 prevalent = (JS)prevayler.prevalentSystem();
42 public void doPost(HttpServletRequest request, HttpServletResponse response) throws IOException { doGet(request, response); }
43 public void doGet(HttpServletRequest request, HttpServletResponse response) throws IOException {
44 servletscope = new ServletScope(request, response, cx);
45 path = cx.getRealPath(((HttpServletRequest)request).getServletPath());
46 Reader xmlreader = new InputStreamReader(new FileInputStream(path));
47 Writer w = response.getWriter();
48 new Template(servletscope, servletscope, xmlreader).wrap(null).toXML(w);
52 public class ServletScope extends JS.Obj {
53 HttpServletRequest request;
54 HttpServletResponse response;
56 public String getRealPath(String s) { return cx.getRealPath(s); }
57 public ServletScope(ServletRequest request, ServletResponse response, ServletContext cx) {
59 this.request = (HttpServletRequest)request;
60 this.response = (HttpServletResponse)response;
63 private JS params = new JS.Obj() {
64 public JS get(JS key) throws JSExn { return JSU.S(request.getParameter(JSU.toString(key))); }
65 public Enumeration keys() throws JSExn { return new JS.Enumeration.JavaStringEnumeration(request.getParameterNames()); }
67 private JS cookies = new JS.Obj() {
69 public Object get(Object key) { return request.getCookie(JSU.toString(key)); }
70 public Enumeration keys() { return request.getCookieNames(); }
73 private JS sessionAttributes = new JS.Obj() {
74 public JS get(JS key) throws JSExn { return JSU.S(request.getSession(true).getAttribute(JSU.toString(key)).toString()); }
75 public void put(JS key, JS val) throws JSExn {
76 if (val == null) request.getSession(true).removeAttribute(JSU.toString(key));
77 else request.setAttribute(JSU.toString(key), val); }
78 public Enumeration keys() throws JSExn { return new JS.Enumeration.JavaStringEnumeration(request.getSession(true).getAttributeNames()); }
80 private JS requestHeader = new JS.Obj() {
81 public JS get(JS key) throws JSExn { return JSU.S(request.getHeader(JSU.toString(key))); }
82 public Enumeration keys() throws JSExn { return new JS.Enumeration.JavaStringEnumeration(request.getHeaderNames()); }
84 private JS responseHeader = new JS.Obj() {
85 public void put(JS key, JS val) throws JSExn { response.setHeader(JSU.toString(key), JSU.toString(val)); }
89 /** lets us put multi-level get/put/call keys all in the same method */
90 private class Sub extends JS.Obj {
92 Sub(JS key) { this.key = key; }
93 public void put(JS key, JS val) throws JSExn {
94 ServletScope.this.put(JSU.S(JSU.toString(this.key) + "." + JSU.toString(key)), val); }
95 public JS get(JS key) throws JSExn {
96 return ServletScope.this.get(JSU.S(JSU.toString(this.key) + "." + JSU.toString(key))); }
97 public JS call(JS method, JS[] args) throws JSExn {
98 return ServletScope.this.call(JSU.S(JSU.toString(this.key) + "." + JSU.toString(method)), args);
101 private Sub getSub(String key) { return new Sub(JSU.S(key)); }
103 public JS call(JS method, JS[] args) throws JSExn {
104 //#switch(JSU.toString(method))
105 case "session.invalidate": request.getSession(true).invalidate(); return null;
107 String path = JSU.toString(args[0]);
108 if (path.indexOf("..") != -1) throw new JSExn("cannot use .. in paths");
109 File f = new File(cx.getRealPath("/") + File.separatorChar + path);
110 if (!f.isDirectory()) return null;
111 String[] contents = f.list();
112 JSArray ret = new JSArray(contents.length);
113 for(int i=0; i<contents.length; i++) ret.put(JSU.N(ret.size()), JSU.S(contents[i]));
118 public JS get(JS key) throws JSExn {
119 //#switch(JSU.toString(key))
121 case "arg": return null;
122 case "prevalent": return (JS)prevayler.prevalentSystem();
123 case "request": return getSub("request");
124 case "request.user": return JSU.S(request.getRemoteUser());
125 case "request.header": return requestHeader;
126 case "request.method": return JSU.S(request.getMethod());
127 case "request.remote": return getSub("request.remote");
128 case "request.remote.ip": return JSU.S(request.getRemoteAddr());
129 case "request.remote.host": return JSU.S(request.getRemoteHost());
130 case "request.ssl": return JSU.B(request.isSecure());
131 case "request.path": return JSU.S(request.getPathInfo());
132 case "response": return getSub("response");
133 case "response.header": return responseHeader;
134 case "session": return getSub("session");
135 case "session.attr": return sessionAttributes;
136 case "session.created": return new JSDate(request.getSession(true).getCreationTime());
137 case "session.accessed": return new JSDate(request.getSession(true).getLastAccessedTime());
138 case "session.invalidate": return METHOD;
139 case "page": return getSub("page");
140 case "page.lastmodified": return new JSDate(new File(path).lastModified());
141 case "context": return getSub("context");
142 case "context.list": return METHOD;
143 case "params": return params;
144 case "cookie": return cookies;
148 public void put(JS key, JS val) throws JSExn {
150 //#switch(JSU.toString(key))
152 try { prevayler.execute(new Prevalence.JSTransaction((JS)val));
153 } catch (Exception e) { e.printStackTrace(); throw new RuntimeException(e); }
154 case "response.code": response.setStatus(JSU.toInt(val));
155 case "response.redirect": response.sendRedirect(JSU.toString(val));
156 case "response.contentType": response.setContentType(JSU.toString(val));
158 } catch (IOException e) { throw new JSExn(JSU.S(e.toString())); }