src/org/ibex/xt/Servlet.java

   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.
   4
   5 package org.ibex.xt;
   6 import org.ibex.js.*;
   7 import org.ibex.util.*;
   8 import org.ibex.io.*;
   9 import java.io.*;
  10 import java.net.*;
  11 import java.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.*;
  17
  18 public class Servlet extends HttpServlet {
  19
  20     private ServletScope servletscope = null;
  21     private String path;
  22     private Prevayler prevayler;
  23     private JS prevalent;
  24     private ServletContext cx = null;
  25
  26     public void destroy() { try {
  27         synchronized(this.getClass()) {
  28             Prevayler privatePrevayler = prevayler;
  29             if (prevayler == null) return;
  30             prevayler = null;
  31             Prevalence.destroy(cx, prevayler);
  32         }
  33     } catch (Exception e) { e.printStackTrace(); } }
  34
  35     public void init(ServletConfig sc) throws ServletException {
  36         cx = sc.getServletContext();
  37         prevayler = Prevalence.getPrevayler(cx);
  38         prevalent = (JS)prevayler.prevalentSystem();
  39     }
  40
  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());
  47     }
  48
  49     public class ServletScope extends JSScope {
  50         HttpServletRequest request;
  51         HttpServletResponse response;
  52         ServletContext cx;
  53         public String getRealPath(String s) { return cx.getRealPath(s); }
  54         public ServletScope(ServletRequest request, ServletResponse response, ServletContext cx) {
  55             super(null);
  56             this.request = (HttpServletRequest)request;
  57             this.response = (HttpServletResponse)response;
  58             this.cx = cx;
  59         }
  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(); }
  63             };
  64         private JS cookies = new JS() {
  65                 /*
  66                 public Object get(Object key) { return request.getCookie(JS.toString(key)); }
  67                 public Enumeration keys() { return request.getCookieNames(); }
  68                 */
  69             };
  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(); }
  76             };
  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(); }
  80             };
  81         private JS responseHeader = new JS() {
  82                 public void put(Object key, Object val) { response.setHeader(JS.toString(key), JS.toString(val)); }
  83             };
  84
  85
  86         /** lets us put multi-level get/put/call keys all in the same method */
  87         private class Sub extends JS {
  88             Object key;
  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);
  96             }
  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);
 100             }
 101         }
 102         private Sub getSub(String key) { return new Sub(key); }
 103
 104         public Object callMethod(Object method, final Object a, final Object b, Object c, Object[] rest, int nargs) throws JSExn {
 105             //#switch(method)
 106             case "session.invalidate":    request.getSession(true).invalidate(); return null;
 107             case "context.list":
 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]);
 115                 return ret;
 116             //#end
 117             return null;
 118         }
 119         public Object get(Object key) throws JSExn {
 120             //#switch(key)
 121             case "body":
 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;
 146             //#end
 147             return null;
 148         }
 149         public void put(Object key, Object val) throws JSExn {
 150             try {
 151             //#switch(JS.toString(key))
 152             case "transaction":
 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));
 158             //#end
 159             } catch (IOException e) {
 160                 throw new JSExn(e);
 161             }
 162         }
 163     }
 164
 165 }