src/java/org/ibex/xt/Servlet.java

   1 package org.ibex.xt;
   2
   3 import java.io.*;
   4 import java.util.*;
   5 import javax.servlet.*;
   6 import javax.servlet.http.*;
   7
   8 import org.ibex.util.Collections;
   9 import org.ibex.js.*;
  10
  11 import org.prevayler.*;
  12
  13 public class Servlet extends HttpServlet {
  14
  15     private Prevayler prevayler;
  16     private JS prevalent;
  17     private ServletContext cx = null;
  18
  19     public void destroy() { try {
  20         synchronized(this.getClass()) {
  21             Prevayler privatePrevayler = prevayler;
  22             if (prevayler == null) return;
  23             prevayler = null;
  24             Prevalence.destroy(cx, prevayler);
  25         }
  26     } catch (Exception e) { e.printStackTrace(); } }
  27
  28     public void init(ServletConfig sc) throws ServletException {
  29         cx = sc.getServletContext();
  30         prevayler = Prevalence.getPrevayler(cx);
  31         prevalent = (JS)prevayler.prevalentSystem();
  32     }
  33
  34     public void doPost(HttpServletRequest request, HttpServletResponse response) throws IOException { doGet(request, response); }
  35     public void doGet(HttpServletRequest rq, HttpServletResponse rs) throws IOException {
  36         String src = rq.getServletPath();
  37         Servlet.Scope scope = new Servlet.Scope(cx, rq, rs, prevayler);
  38         PrintWriter w = new PrintWriter(rs.getWriter());
  39         try {
  40             while (src != null) {
  41                 try {
  42                     StringWriter buffer = new StringWriter();
  43                     Template t = Template.parse(cx.getRealPath(src), scope);
  44                     Template.wrap(t, scope).out(buffer);
  45
  46                     rs.setContentType("text/xml");
  47                     w.write(buffer.toString());
  48                     src = null;
  49                 } catch (Template.RedirectSignal r) {
  50                     src = r.getTarget();
  51                 }
  52             }
  53         } catch (Template.Signal s) {
  54             // exit silently
  55         } catch (JSLeaf.Exn e) {
  56             w.print("\n"+src+": ");
  57             w.println(e.getMessage());
  58             System.out.println(e);
  59         } catch (FileNotFoundException e) {
  60             w.print("\nfile not found: ");
  61             w.println(e.getMessage());
  62             System.out.println(e);
  63         } catch (IOException e) {
  64             w.print("\nio error: ");
  65             w.println(e.getMessage());
  66             e.printStackTrace();
  67         } catch (Exception e) {
  68             w.print("Unexpected error occurred.");
  69             System.out.println("Unexpected Exception:");
  70             e.printStackTrace();
  71         }
  72     }
  73
  74     public static class Scope extends Template.Scope {
  75         private final ServletContext cx;
  76         private final HttpServletRequest request;
  77         private final HttpServletResponse response;
  78         private final Prevayler prevayler;
  79
  80         public Scope(ServletContext cx, HttpServletRequest rq, HttpServletResponse rs, Prevayler p) {
  81             super(null); this.cx = cx; request = rq; response = rs; prevayler = p;
  82         }
  83
  84         public String getLocalPath() { return cx.getRealPath("/") + "/WEB-INF/"; }
  85         public void transaction(JS t) {
  86             try { prevayler.execute(new Prevalence.JSTransaction(t)); }
  87             catch (Exception e) { e.printStackTrace(); throw new RuntimeException(e); }
  88         }
  89
  90         private JS session = new JS() {
  91             public Object get(Object key) throws JSExn {
  92                 //#switch(JS.toString(key))
  93                 case "created":       return new JSDate(request.getSession(true).getCreationTime());
  94                 case "accessed":      return new JSDate(request.getSession(true).getLastAccessedTime());
  95                 case "invalidate":    return METHOD;
  96                 //#end
  97                 return super.get(key);
  98             }
  99             public void put(Object key, Object val) throws JSExn {
 100                 //#switch(JS.toString(key))
 101                 case "created":       throw new JSExn("can not set session.created");
 102                 case "accessed":      throw new JSExn("can not set session.accessed");
 103                 case "invalidate":    throw new JSExn("can not set session.invalidate");
 104                 //#end
 105                 super.put(key, val);
 106             }
 107             public Object callMethod(Object method, final Object a, final Object b, Object c, Object[] rest, int nargs)
 108                 throws JSExn {
 109                 //#switch(JS.toString(method))
 110                 case "invalidate":    request.getSession(true).invalidate(); return null;
 111                 //#end
 112                 return super.callMethod(method, a, b, c, rest, nargs);
 113             }
 114         };
 115         private JS params = new JS() {
 116             private List keys = null;
 117             public Object get(Object key) { return request.getParameter(JS.toString(key)); }
 118             public Collection keys() {
 119                 return keys == null ? keys = Collections.list(request.getParameterNames()) : keys; }
 120         };
 121         private JS cookies = new JS() {
 122                 /*
 123                 public Object get(Object key) { return request.getCookie(JS.toString(key)); }
 124                 public Enumeration keys() { return request.getCookieNames(); }
 125                 */
 126             };
 127         private JS sessionAttributes = new JS() {
 128             private List keys = null;
 129             public Object get(Object key) {
 130                 return request.getSession(true).getAttribute(JS.toString(key)); }
 131             public void put(Object key, Object val) {
 132                 if (val == null) request.getSession(true).removeAttribute(JS.toString(key));
 133                 else request.setAttribute(JS.toString(key), val); }
 134             public Collection keys() {
 135                 return keys == null ? keys = Collections.list(request.getSession(true).getAttributeNames()) : keys; }
 136         };
 137         private JS requestHeader = new JS() {
 138             private List keys = null;
 139             public Object get(Object key) { return request.getHeader(JS.toString(key)); }
 140             public Collection keys() {
 141                 return keys == null ? keys = Collections.list(request.getHeaderNames()) : keys; }
 142         };
 143         private JS responseHeader = new JS() {
 144             public void put(Object key, Object val) {
 145                 response.setHeader(JS.toString(key), JS.toString(val)); }
 146         };
 147
 148
 149         /** lets us put multi-level get/put/call keys all in the same method */
 150         private class Sub extends JS {
 151             Object key;
 152             Sub(Object key) { this.key = key; }
 153             public void put(Object key, Object val) throws JSExn {
 154                 Scope.this.put(JS.toString(this.key) + "." + JS.toString(key), val); }
 155             public Object get(Object key) throws JSExn {
 156                 return Scope.this.get(JS.toString(this.key) + "." + JS.toString(key)); }
 157             public Object call(Object a0, Object a1, Object a2, Object[] rest, int nargs) throws JSExn {
 158                 return Scope.this.callMethod(this.key, a0, a1, a2, rest, nargs);
 159             }
 160             public Object callMethod(Object method, Object a0, Object a1, Object a2, Object[] rest, int nargs) throws JSExn {
 161                 return Scope.this.callMethod(JS.toString(this.key) + "."
 162                                                          + JS.toString(method), a0, a1, a2, rest, nargs);
 163             }
 164         }
 165         private Sub getSub(String key) { return new Sub(key); }
 166
 167         public Object callMethod(Object method, final Object a, final Object b, Object c, Object[] rest, int nargs) throws JSExn {
 168             //#switch(method)
 169             case "session.invalidate":    request.getSession(true).invalidate(); return null;
 170             case "context.list":
 171                 String path = JS.toString(a);
 172                 if (path.indexOf("..") != -1) throw new JSExn("cannot use .. in paths");
 173                 File f = new File(cx.getRealPath("/") + File.separatorChar + path);
 174                 if (!f.isDirectory()) return null;
 175                 String[] contents = f.list();
 176                 JSArray ret = new JSArray(contents.length);
 177                 for(int i=0; i<contents.length; i++) ret.add(contents[i]);
 178                 return ret;
 179             //#end
 180             return null;
 181         }
 182         public Object get(Object key) throws JSExn {
 183             //#switch(key)
 184             case "body":
 185             case "arg":                   return null;
 186             case "prevalent":             return prevayler.prevalentSystem();
 187             case "request":               return getSub("request");
 188             case "request.user":          return request.getRemoteUser();
 189             case "request.header":        return requestHeader;
 190             case "request.method":        return request.getMethod();
 191             case "request.remote":        return getSub("request.remote");
 192             case "request.remote.ip":     return request.getRemoteAddr();
 193             case "request.remote.host":   return request.getRemoteHost();
 194             case "request.ssl":           return new Boolean(request.isSecure());
 195             case "request.path":          return request.getPathInfo();
 196             case "response":              return getSub("response");
 197             case "response.header":       return responseHeader;
 198             case "session":               return session;
 199             case "page":                  return getSub("page");
 200             //case "page.lastmodified":     return new JSDate(new File(path).lastModified()); FIXME
 201             case "context":               return getSub("context");
 202             case "context.list":          return METHOD;
 203             case "params":                return params;
 204             case "cookie":                return cookies;
 205             case "xt.date":               return new JSDate(); // TODO: discuss
 206             //#end
 207             return null;
 208         }
 209         public void put(Object key, Object val) throws JSExn {
 210             try {
 211             //#switch(JS.toString(key))
 212             case "response.code":         response.setStatus(JS.toInt(val));
 213             case "response.redirect":     response.sendRedirect(JS.toString(val));
 214             case "response.contentType":  response.setContentType(JS.toString(val));
 215             //#end
 216             } catch (IOException e) {
 217                 throw new JSExn(e);
 218             }
 219         }
 220     }
 221
 222 }