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