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