src/java/ibex/xt/Servlet.java

   1 package ibex.xt;
   2
   3 import ibex.js.*;
   4 import ibex.util.*;
   5 import java.util.*;
   6
   7 import java.io.*;
   8 import java.net.*;
   9 import javax.servlet.*;
  10 import javax.servlet.http.*;
  11
  12 import org.prevayler.*;
  13 import org.prevayler.implementation.snapshot.*;
  14
  15 public class Servlet extends HttpServlet {
  16
  17     private String path;
  18     private Prevayler prevayler;
  19     private JS prevalent;
  20     private ServletContext cx = null;
  21
  22     public void destroy() { try {
  23         synchronized(this.getClass()) {
  24             Prevayler privatePrevayler = prevayler;
  25             if (prevayler == null) return;
  26             prevayler = null;
  27             Prevalence.destroy(cx, prevayler);
  28         }
  29     } catch (Exception e) { e.printStackTrace(); } }
  30
  31     public void init(ServletConfig sc) throws ServletException {
  32         cx = sc.getServletContext();
  33         prevayler = Prevalence.getPrevayler(cx);
  34         prevalent = (JS)prevayler.prevalentSystem();
  35     }
  36
  37     public void doPost(HttpServletRequest request, HttpServletResponse response) throws IOException { doGet(request, response); }
  38     public void doGet(HttpServletRequest rq, HttpServletResponse rs) throws IOException {
  39         String path = cx.getRealPath(rq.getServletPath());
  40         Servlet.Scope scope = new Servlet.Scope(cx, rq, rs, prevayler);
  41         try { Template.wrap(Template.parse(path, scope), scope).toXML(rs.getWriter()); }
  42         catch (Exception e) { e.printStackTrace(); System.out.println("e = "+e); }
  43     }
  44
  45     public static class Scope extends Template.Scope {
  46         private final ServletContext cx;
  47         private final HttpServletRequest request;
  48         private final HttpServletResponse response;
  49         private final Prevayler prevayler;
  50
  51         public Scope(ServletContext cx, HttpServletRequest rq, HttpServletResponse rs, Prevayler p) {
  52             super(null); this.cx = cx; request = rq; response = rs; prevayler = p;
  53         }
  54
  55         public String getLocalPath() { return cx.getRealPath("/") + "/WEB-INF/"; }
  56         public void transaction(JS t) {
  57             try { prevayler.execute(new Prevalence.JSTransaction(t)); }
  58             catch (Exception e) { e.printStackTrace(); throw new RuntimeException(e); }
  59         }
  60
  61         private JS params = new JS() {
  62             private List keys = null;
  63             public Object get(Object key) { return request.getParameter(JS.toString(key)); }
  64             public Collection keys() {
  65                 return keys == null ? keys = Collections.list(request.getParameterNames()) : keys; }
  66         };
  67         private JS cookies = new JS() {
  68                 /*
  69                 public Object get(Object key) { return request.getCookie(JS.toString(key)); }
  70                 public Enumeration keys() { return request.getCookieNames(); }
  71                 */
  72             };
  73         private JS sessionAttributes = new JS() {
  74             private List keys = null;
  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 Collection keys() {
  81                 return keys == null ? keys = Collections.list(request.getSession(true).getAttributeNames()) : keys; }
  82         };
  83         private JS requestHeader = new JS() {
  84             private List keys = null;
  85             public Object get(Object key) { return request.getHeader(JS.toString(key)); }
  86             public Collection keys() {
  87                 return keys == null ? keys = Collections.list(request.getHeaderNames()) : keys; }
  88         };
  89         private JS responseHeader = new JS() {
  90             public void put(Object key, Object val) {
  91                 response.setHeader(JS.toString(key), JS.toString(val)); }
  92         };
  93
  94
  95         /** lets us put multi-level get/put/call keys all in the same method */
  96         private class Sub extends JS {
  97             Object key;
  98             Sub(Object key) { this.key = key; }
  99             public void put(Object key, Object val) throws JSExn {
 100                 Scope.this.put(JS.toString(this.key) + "." + JS.toString(key), val); }
 101             public Object get(Object key) throws JSExn {
 102                 return Scope.this.get(JS.toString(this.key) + "." + JS.toString(key)); }
 103             public Object call(Object a0, Object a1, Object a2, Object[] rest, int nargs) throws JSExn {
 104                 return Scope.this.callMethod(this.key, a0, a1, a2, rest, nargs);
 105             }
 106             public Object callMethod(Object method, Object a0, Object a1, Object a2, Object[] rest, int nargs) throws JSExn {
 107                 return Scope.this.callMethod(JS.toString(this.key) + "."
 108                                                          + JS.toString(method), a0, a1, a2, rest, nargs);
 109             }
 110         }
 111         private Sub getSub(String key) { return new Sub(key); }
 112
 113         public Object callMethod(Object method, final Object a, final Object b, Object c, Object[] rest, int nargs) throws JSExn {
 114             //#switch(method)
 115             case "session.invalidate":    request.getSession(true).invalidate(); return null;
 116             case "context.list":
 117                 String path = JS.toString(a);
 118                 if (path.indexOf("..") != -1) throw new JSExn("cannot use .. in paths");
 119                 File f = new File(cx.getRealPath("/") + File.separatorChar + path);
 120                 if (!f.isDirectory()) return null;
 121                 String[] contents = f.list();
 122                 JSArray ret = new JSArray(contents.length);
 123                 for(int i=0; i<contents.length; i++) ret.add(contents[i]);
 124                 return ret;
 125             //#end
 126             return null;
 127         }
 128         public Object get(Object key) throws JSExn {
 129             //#switch(key)
 130             case "body":
 131             case "arg":                   return null;
 132             case "prevalent":             return prevayler.prevalentSystem();
 133             case "request":               return getSub("request");
 134             case "request.user":          return request.getRemoteUser();
 135             case "request.header":        return requestHeader;
 136             case "request.method":        return request.getMethod();
 137             case "request.remote":        return getSub("request.remote");
 138             case "request.remote.ip":     return request.getRemoteAddr();
 139             case "request.remote.host":   return request.getRemoteHost();
 140             case "request.ssl":           return new Boolean(request.isSecure());
 141             case "request.path":          return request.getPathInfo();
 142             case "response":              return getSub("response");
 143             case "response.header":       return responseHeader;
 144             case "session":               return getSub("session");
 145             case "session.attr":          return sessionAttributes;
 146             case "session.created":       return new JSDate(request.getSession(true).getCreationTime());
 147             case "session.accessed":      return new JSDate(request.getSession(true).getLastAccessedTime());
 148             case "session.invalidate":    return METHOD;
 149             case "page":                  return getSub("page");
 150             //case "page.lastmodified":     return new JSDate(new File(path).lastModified()); FIXME
 151             case "context":               return getSub("context");
 152             case "context.list":          return METHOD;
 153             case "params":                return params;
 154             case "cookie":                return cookies;
 155             case "xt.date":               return new JSDate(); // TODO: discuss
 156             //#end
 157             return null;
 158         }
 159         public void put(Object key, Object val) throws JSExn {
 160             try {
 161             //#switch(JS.toString(key))
 162             case "transaction":           transaction((JS)val);
 163             case "response.code":         response.setStatus(JS.toInt(val));
 164             case "response.redirect":     response.sendRedirect(JS.toString(val));
 165             case "response.contentType":  response.setContentType(JS.toString(val));
 166             //#end
 167             } catch (IOException e) {
 168                 throw new JSExn(e);
 169             }
 170         }
 171     }
 172
 173 }