src/java/org/ibex/xt/Servlet.java

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