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     public static final JS METHOD = new JS.Method();
  21     private ServletScope servletscope = null;
  22     private String path;
  23     private Prevayler prevayler;
  24     private JS prevalent;
  25     private ServletContext cx = null;
  26
  27     public void destroy() { try {
  28         synchronized(this.getClass()) {
  29             Prevayler privatePrevayler = prevayler;
  30             if (prevayler == null) return;
  31             prevayler = null;
  32             Prevalence.destroy(cx, prevayler);
  33         }
  34     } catch (Exception e) { e.printStackTrace(); } }
  35
  36     public void init(ServletConfig sc) throws ServletException {
  37         cx = sc.getServletContext();
  38         prevayler = Prevalence.getPrevayler(cx);
  39         prevalent = (JS)prevayler.prevalentSystem();
  40     }
  41
  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);
  49         w.flush();
  50     }
  51
  52     public class ServletScope extends JS.Obj {
  53         HttpServletRequest request;
  54         HttpServletResponse response;
  55         ServletContext cx;
  56         public String getRealPath(String s) { return cx.getRealPath(s); }
  57         public ServletScope(ServletRequest request, ServletResponse response, ServletContext cx) {
  58             super();
  59             this.request = (HttpServletRequest)request;
  60             this.response = (HttpServletResponse)response;
  61             this.cx = cx;
  62         }
  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()); }
  66             };
  67         private JS cookies = new JS.Obj() {
  68                 /*
  69                 public Object get(Object key) { return request.getCookie(JSU.toString(key)); }
  70                 public Enumeration keys() { return request.getCookieNames(); }
  71                 */
  72             };
  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()); }
  79             };
  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()); }
  83             };
  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)); }
  86             };
  87
  88
  89         /** lets us put multi-level get/put/call keys all in the same method */
  90         private class Sub extends JS.Obj {
  91             JS key;
  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);
  99             }
 100         }
 101         private Sub getSub(String key) { return new Sub(JSU.S(key)); }
 102
 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;
 106             case "context.list":
 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]));
 114                 return ret;
 115             //#end
 116             return null;
 117         }
 118         public JS get(JS key) throws JSExn {
 119             //#switch(JSU.toString(key))
 120             case "body":
 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;
 145             //#end
 146             return null;
 147         }
 148         public void put(JS key, JS val) throws JSExn {
 149             try {
 150             //#switch(JSU.toString(key))
 151             case "transaction":
 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));
 157             //#end
 158             } catch (IOException e) { throw new JSExn(JSU.S(e.toString())); }
 159         }
 160     }
 161
 162 }