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         new Template(servletscope, servletscope, xmlreader).wrap(null).toXML(response.getWriter());
  48     }
  49
  50     public class ServletScope extends JS.Obj {
  51         HttpServletRequest request;
  52         HttpServletResponse response;
  53         ServletContext cx;
  54         public String getRealPath(String s) { return cx.getRealPath(s); }
  55         public ServletScope(ServletRequest request, ServletResponse response, ServletContext cx) {
  56             super();
  57             this.request = (HttpServletRequest)request;
  58             this.response = (HttpServletResponse)response;
  59             this.cx = cx;
  60         }
  61         private JS params = new JS.Obj() {
  62                 public JS get(JS key) throws JSExn { return JSU.S(request.getParameter(JSU.toString(key))); }
  63                 public Enumeration keys() throws JSExn { return new JS.Enumeration.JavaStringEnumeration(request.getParameterNames()); }
  64             };
  65         private JS cookies = new JS.Obj() {
  66                 /*
  67                 public Object get(Object key) { return request.getCookie(JSU.toString(key)); }
  68                 public Enumeration keys() { return request.getCookieNames(); }
  69                 */
  70             };
  71         private JS sessionAttributes = new JS.Obj() {
  72                 public JS get(JS key) throws JSExn { return JSU.S(request.getSession(true).getAttribute(JSU.toString(key)).toString()); }
  73                 public void put(JS key, JS val) throws JSExn {
  74                     if (val == null) request.getSession(true).removeAttribute(JSU.toString(key));
  75                     else request.setAttribute(JSU.toString(key), val); }
  76                 public Enumeration keys() throws JSExn { return new JS.Enumeration.JavaStringEnumeration(request.getSession(true).getAttributeNames()); }
  77             };
  78         private JS requestHeader = new JS.Obj() {
  79                 public JS get(JS key) throws JSExn { return JSU.S(request.getHeader(JSU.toString(key))); }
  80                 public Enumeration keys() throws JSExn { return new JS.Enumeration.JavaStringEnumeration(request.getHeaderNames()); }
  81             };
  82         private JS responseHeader = new JS.Obj() {
  83                 public void put(JS key, JS val) throws JSExn { response.setHeader(JSU.toString(key), JSU.toString(val)); }
  84             };
  85
  86
  87         /** lets us put multi-level get/put/call keys all in the same method */
  88         private class Sub extends JS.Obj {
  89             JS key;
  90             Sub(JS key) { this.key = key; }
  91             public void put(JS key, JS val) throws JSExn {
  92                 ServletScope.this.put(JSU.S(JSU.toString(this.key) + "." + JSU.toString(key)), val); }
  93             public JS get(JS key) throws JSExn {
  94                 return ServletScope.this.get(JSU.S(JSU.toString(this.key) + "." + JSU.toString(key))); }
  95             public JS call(JS method, JS[] args) throws JSExn {
  96                 return ServletScope.this.call(JSU.S(JSU.toString(this.key) + "." + JSU.toString(method)), args);
  97             }
  98         }
  99         private Sub getSub(String key) { return new Sub(JSU.S(key)); }
 100
 101         public JS call(JS method, JS[] args) throws JSExn {
 102             //#switch(JSU.toString(method))
 103             case "session.invalidate":    request.getSession(true).invalidate(); return null;
 104             case "context.list":
 105                 String path = JSU.toString(args[0]);
 106                 if (path.indexOf("..") != -1) throw new JSExn("cannot use .. in paths");
 107                 File f = new File(cx.getRealPath("/") + File.separatorChar + path);
 108                 if (!f.isDirectory()) return null;
 109                 String[] contents = f.list();
 110                 JSArray ret = new JSArray(contents.length);
 111                 for(int i=0; i<contents.length; i++) ret.put(JSU.N(ret.size()), JSU.S(contents[i]));
 112                 return ret;
 113             //#end
 114             return null;
 115         }
 116         public JS get(JS key) throws JSExn {
 117             //#switch(JSU.toString(key))
 118             case "body":
 119             case "arg":                   return null;
 120             case "prevalent":             return (JS)prevayler.prevalentSystem();
 121             case "request":               return getSub("request");
 122             case "request.user":          return JSU.S(request.getRemoteUser());
 123             case "request.header":        return requestHeader;
 124             case "request.method":        return JSU.S(request.getMethod());
 125             case "request.remote":        return getSub("request.remote");
 126             case "request.remote.ip":     return JSU.S(request.getRemoteAddr());
 127             case "request.remote.host":   return JSU.S(request.getRemoteHost());
 128             case "request.ssl":           return JSU.B(request.isSecure());
 129             case "request.path":          return JSU.S(request.getPathInfo());
 130             case "response":              return getSub("response");
 131             case "response.header":       return responseHeader;
 132             case "session":               return getSub("session");
 133             case "session.attr":          return sessionAttributes;
 134             case "session.created":       return new JSDate(request.getSession(true).getCreationTime());
 135             case "session.accessed":      return new JSDate(request.getSession(true).getLastAccessedTime());
 136             case "session.invalidate":    return METHOD;
 137             case "page":                  return getSub("page");
 138             case "page.lastmodified":     return new JSDate(new File(path).lastModified());
 139             case "context":               return getSub("context");
 140             case "context.list":          return METHOD;
 141             case "params":                return params;
 142             case "cookie":                return cookies;
 143             //#end
 144             return null;
 145         }
 146         public void put(JS key, JS val) throws JSExn {
 147             try {
 148             //#switch(JSU.toString(key))
 149             case "transaction":
 150                 try { prevayler.execute(new Prevalence.JSTransaction((JS)val));
 151                 } catch (Exception e) { e.printStackTrace(); throw new RuntimeException(e); }
 152             case "response.code":         response.setStatus(JSU.toInt(val));
 153             case "response.redirect":     response.sendRedirect(JSU.toString(val));
 154             case "response.contentType":  response.setContentType(JSU.toString(val));
 155             //#end
 156             } catch (IOException e) { throw new JSExn(JSU.S(e.toString())); }
 157         }
 158     }
 159
 160 }