src/org/ibex/mail/target/Script.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.mail.target;
   6 import org.ibex.js.*;
   7 import org.ibex.io.*;
   8 import org.ibex.util.*;
   9 import org.ibex.mail.*;
  10 import org.ibex.mail.filter.*;
  11 import org.ibex.mail.target.*;
  12 import java.io.*;
  13 import java.util.*;
  14
  15 public class Script extends JS.Obj implements Target {
  16
  17     private static final JS.Method METHOD = new JS.Method();
  18
  19     private static Script root = null;
  20     private static final String DEFAULT_CONF = File.separatorChar + "etc" + File.separatorChar + "org.ibex.mail.conf";
  21     public static Script root() {
  22         try {
  23             if (root == null) root = new Script(System.getProperty("ibex.mail.conf", DEFAULT_CONF));
  24             return root;
  25         } catch (Exception e) {
  26             Log.error(Script.class, e);
  27             return null;
  28         }
  29     }
  30
  31     final JS js;
  32     private Message m = null;
  33     private String filePath = null;
  34     public Script(String filePath) throws JSExn, IOException {
  35         this.filePath = filePath;
  36         js = JSU.cloneWithNewGlobalScope(JSU.fromReader(filePath, 1, new InputStreamReader(new FileInputStream(filePath))),
  37                                          new ScriptScope()); }
  38
  39     private class ScriptScope extends JS.Immutable {
  40         ScriptEnv env = new ScriptEnv();
  41         public JS get(JS name) throws JSExn {
  42             //#jsswitch(name)
  43             case "m": return m;
  44             case "ibex": return env;
  45             default: return null;
  46             //#end
  47             return null;
  48         }
  49     }
  50
  51     public void accept(Message m) throws IOException, MailException {
  52         try {
  53             new Script(filePath).reallyAccept(m);
  54         } catch (JSExn e) {
  55             Log.error(this, e);
  56             throw new MailException(e.toString());
  57         }
  58     }
  59
  60     private synchronized void reallyAccept(Message m) throws IOException, MailException, JSExn {
  61         this.m = m;
  62         try {
  63             Object ret = js.call(null, new JS[] { m });
  64             Log.debug(this, "configuration script returned " + ret);
  65             if (ret == null) throw new IOException("configuration script returned null");
  66             while (ret instanceof JSReflection.Wrapper) ret = ((JSReflection.Wrapper)ret).unwrap();
  67             if (ret instanceof Target)      ((Target)ret).accept(m);
  68             //else if (ret instanceof Filter) ((Filter)ret).process(m);
  69             else throw new IOException("configuration script returned a " + ret.getClass().getName());
  70         } catch (JSExn e) {
  71             Log.warn(this, e);
  72             throw new IOException("configuration script threw an exception");
  73         }
  74     }
  75
  76     // FIXME: this should extend org.ibex.core.Ibex
  77     public class ScriptEnv extends JS.Obj {
  78
  79         private PropertyFile prefs = null;
  80             /*
  81         static {
  82             try {
  83                 // FIXME
  84                 prefs = new PropertyFile(new File("/etc/org.ibex.mail.properties"));
  85             } catch (IOException e) {
  86                 Log.error(ScriptEnv.class, e);
  87             }
  88         }
  89             */
  90
  91         /** lets us put multi-level get/put/call keys all in the same method */
  92         private class Sub extends JS.Immutable {
  93             String key;
  94             Sub(String key) { this.key = key; }
  95             public void put(JS key, JS val) throws JSExn {
  96                 ScriptEnv.this.put(JSU.S(this.key + "." + JSU.toString(key)), val); }
  97             public JS get(JS key) throws JSExn { return ScriptEnv.this.get(JSU.S(this.key + "." + JSU.toString(key))); }
  98             public JS call(JS method, JS[] args) throws JSExn {
  99                 return ScriptEnv.this.call(JSU.S(this.key + "." + JSU.toString(method)), args);
 100             }
 101         }
 102         private Sub getSub(String s) { return new Sub(s); }
 103
 104         public JS get(JS name) throws JSExn {
 105             //#jsswitch(name)
 106             case "math": return ibexMath;
 107             case "string": return ibexString;
 108             case "date": return METHOD;
 109             case "regexp": return METHOD;
 110             case "log": return getSub("log");
 111             case "log.debug": return METHOD;
 112             case "log.info": return METHOD;
 113             case "log.warn": return METHOD;
 114             case "log.error": return METHOD;
 115             case "list": return getSub("list");
 116             case "url": return getSub("url");
 117             case "url.encode": return METHOD;
 118             case "mail": return getSub("mail");
 119             case "mail.forward": return METHOD;
 120             case "mail.forward2": return METHOD;
 121             case "mail.send": return METHOD;
 122             case "mail.attempt": return METHOD;
 123             case "mail.later": return Later.instance;
 124             case "mail.drop": return METHOD;
 125             case "mail.razor": return getSub("mail.razor");
 126             case "mail.razor.check": return METHOD;
 127             case "mail.dcc": return getSub("mail.dcc");
 128             case "mail.dcc.check": return METHOD;
 129             case "mail.bounce": return METHOD;
 130             case "mail.reject": return METHOD;
 131             case "mail.my": return getSub("mail.my");
 132             case "mail.dir": return METHOD;
 133             case "mail.shell": return METHOD;
 134             case "mail.my.prefs": try {
 135                     return new org.ibex.js.Directory(new File("/etc/org.ibex.mail.prefs"));
 136             } catch (IOException e) { throw new JSExn(e.toString()); }
 137             case "mail.my.mailbox":
 138                 FileBasedMailbox root = FileBasedMailbox.getFileBasedMailbox(Mailbox.STORAGE_ROOT, true);
 139                 return root.slash("user", true).slash("megacz", true).slash("newmail", true);
 140             case "mail.list": return METHOD;
 141             //#end
 142             return super.get(name);
 143         }
 144
 145         public JS call(JS name0, JS[] args) throws JSExn {
 146             final JS a = args.length >= 1 ? args[0] : null;
 147             final JS b = args.length >= 2 ? args[1] : null;
 148             final JS c = args.length >= 3 ? args[2] : null;
 149             final int nargs = args.length;
 150             String name = JSU.toString(name0);
 151             try {
 152                 if (name.equals("url.encode")) return JSU.S(java.net.URLEncoder.encode(JSU.toString(args[0])));
 153                 if (name.equals("mail.list")) return JSReflection.wrap(MailingList.getMailingList(JSU.toString(args[0])));
 154                 if (name.equals("mail.dir")) {
 155                     return new org.ibex.js.Directory(new File(JSU.toString(args[0])));
 156                 }
 157                 if (name.equals("mail.shell")) {
 158                     // FIXME: EEEEEVIL!
 159                     Log.warn("dbug", args[0].getClass().getName());
 160                     Log.warn("dbug", args[1].getClass().getName());
 161                     final Process p = Runtime.getRuntime().exec(JSU.toString(args[1]));
 162                     Message m = (Message)args[0];
 163                     new Thread() {
 164                         public void run() {
 165                             try {
 166                                 BufferedReader br = new BufferedReader(new InputStreamReader(p.getErrorStream()));
 167                                 String s = null;
 168                                 while((s = br.readLine())!=null)
 169                                     Log.warn("shell", s);
 170                             } catch (Exception e) { e.printStackTrace(); }
 171                         }
 172                     }.start();
 173                     OutputStream os = p.getOutputStream();
 174                     Stream stream = new Stream(os);
 175                     m.getStream().transcribe(stream);
 176                     stream.close();
 177                     p.waitFor();
 178                     return null;
 179                 }
 180                 if (name.equals("date")) { return new JSDate(args); }
 181                 if (name.equals("mail.send") || name.equals("send") || name.equals("mail.attempt") || name.equals("attempt")) {
 182                     boolean attempt = name.equals("mail.attempt") || name.equals("attempt");
 183                     JS m = (JS)a;
 184                     StringBuffer headers = new StringBuffer();
 185                     String body = "";
 186                     Address from = null, to = null, envelopeFrom = null, envelopeTo = null;
 187                     JS.Enumeration e = m.keys();
 188                     for(; e.hasNext();) {
 189                         JS key = (JS)e.next();
 190                         JS val = m.get(key) == null ? null : m.get(key);
 191                         if ("body".equalsIgnoreCase(JSU.toString(key))) body = JSU.toString(val);
 192                         else headers.append(JSU.toString(key) + ": " + JSU.toString(val) + "\r\n");
 193                         if ("from".equalsIgnoreCase(JSU.toString(key))) from = Address.parse(JSU.toString(val));
 194                         if ("to".equalsIgnoreCase(JSU.toString(key))) to = Address.parse(JSU.toString(val));
 195                         if ("envelopeFrom".equalsIgnoreCase(JSU.toString(key))) envelopeFrom = Address.parse(JSU.toString(val));
 196                         if ("envelopeTo".equalsIgnoreCase(JSU.toString(key))) envelopeTo = Address.parse(JSU.toString(val));
 197                     }
 198                     if (envelopeTo == null) envelopeTo = to;
 199                     if (envelopeFrom == null) envelopeFrom = from;
 200                     Message message =
 201                         Message.newMessage(new org.ibex.io.Fountain.StringFountain(headers.toString() + "\r\n" + body),
 202                                            envelopeFrom,
 203                                            envelopeTo
 204                                            );
 205
 206                     boolean ok = false;
 207                     try {
 208                         if (attempt) {
 209                             org.ibex.mail.protocol.SMTP.Outgoing.attempt(message);
 210                         } else {
 211                             org.ibex.mail.protocol.SMTP.Outgoing.accept(message);
 212                         }
 213                         ok = true;
 214                     } catch (Exception ex) {
 215                         if (!attempt) Log.warn(this, ex);
 216                     }
 217                     if (!ok && !attempt) throw new JSExn("SMTP server rejected message");
 218                     return JSU.B(ok);
 219                 }
 220                 if (name.equals("mail.razor.check")) {
 221                     Process p = Runtime.getRuntime().exec("razor-check");
 222                     ((Message)args[0]).getStream().transcribe(new Stream(p.getOutputStream()), true);
 223                     return JSU.N(p.waitFor());
 224                 }
 225                 if (name.equals("mail.dcc.check")) {
 226                     Process p = Runtime.getRuntime().exec(new String[] { "dccproc", "-H" });
 227                     ((Message)args[0]).getStream().transcribe(new Stream(p.getOutputStream()), true);
 228                     StringBuffer ret = new StringBuffer();
 229                     new Stream(p.getInputStream()).transcribe(ret);
 230                     p.waitFor();
 231                     return JSU.S(ret.toString());
 232                 }
 233                 if (name.equals("mail.drop")) {
 234                     return args.length==0 ? new Drop() : new Drop(JSU.toString(args[0]));
 235                 }
 236                 if (name.equals("mail.bounce")) {
 237                     return new JSTarget() {
 238                             public void accept(Message m) throws MailException {
 239                                 try {
 240                                     Message m2 = m.bounce(JSU.toString(a));
 241                                     org.ibex.mail.protocol.SMTP.Outgoing.accept(m2);
 242                                     Log.error(this, "BOUNCING! " + m2.summary());
 243                                 } catch (Exception e) {
 244                                     Log.warn(this, e);
 245                                 }
 246                             } };
 247                 }
 248                 if (name.equals("mail.forward2") || name.equals("forward2")) {
 249                     try {
 250                         Message m2 = Message.newMessage(new org.ibex.io.Fountain.StringFountain(m.toString()),
 251                                                         m.envelopeFrom,
 252                                                         new Address(JSU.toString(a)));
 253                         org.ibex.mail.protocol.SMTP.Outgoing.accept(m2);
 254                     } catch (Exception e) {
 255                         Log.warn(this, e);
 256                         throw new JSExn(e.toString());
 257                     }
 258                     return null;
 259                 }
 260                 if (name.equals("mail.forward") || name.equals("forward")) {
 261                     Message m2 = Message.newMessage(Script.this.m, Script.this.m.envelopeFrom, new Address(JSU.toString(a)));
 262                     org.ibex.mail.protocol.SMTP.Outgoing.attempt(m2, false);
 263                     return Drop.instance;
 264                 }
 265                 if (name.equals("mail.reject"))
 266                     return new Reject(JSU.toString(a));
 267                 if (name.equals("log.debug") || name.equals("debug")) {    JSU.debug(a== null ? "**null**" : JSU.toString(a)); return null; }
 268                 if (name.equals("log.info") || name.equals("info")) {     JSU.info(a== null ? "**null**" : JSU.toString(a)); return null; }
 269                 if (name.equals("log.warn") || name.equals("warn")) {     JSU.warn(a== null ? "**null**" : JSU.toString(a)); return null; }
 270                 if (name.equals("log.error") || name.equals("error")) {    JSU.error(a== null ? "**null**" : JSU.toString(a)); return null; }
 271                 switch (nargs) {
 272                 case 1:
 273                     if (name.equals("regexp")) {return new JSRegexp(a, null); }
 274                     break;
 275                 case 2:
 276                     if (name.equals("regexp")) {return new JSRegexp(a, b); }
 277                 }
 278             } catch (MailException e) { throw e;
 279             } catch (Exception e) {
 280                 Log.warn(this, "ibex."+name+"() threw: " + e);
 281                 Log.warn(this, e);
 282                 if (e instanceof JSExn) throw ((JSExn)e);
 283                 throw new JSExn("invalid argument for ibex object method "+JSU.toString(name0)+"()");
 284             }
 285             throw new JSExn("invalid number of arguments ("+nargs+") for ibex object method "+JSU.toString(name0)+"()");
 286         }
 287
 288         public final JSMath ibexMath = new JSMath() {
 289                 public JS get(JS name) throws JSExn {
 290                     // FIXME!!!
 291                     /*
 292                     case "isNaN": return gs.get(name);
 293                     case "isFinite": return gs.get(name);
 294                     case "NaN": return gs.get(name);
 295                     case "Infinity": return gs.get(name);
 296                     */
 297                     return null;
 298                 }
 299             };
 300
 301         public final JS ibexString = new JS.Immutable() {
 302                 public JS get(JS name) throws JSExn {
 303                     // FIXME!!!
 304                     /*
 305                     case "parseInt": return gs.get(JSU.S("parseInt"));
 306                     case "parseFloat": return gs.get(JSU.S("parseFloat"));
 307                     case "decodeURI": return gs.get(JSU.S("decodeURI"));
 308                     case "decodeURIComponent": return gs.get(JSU.S("decodeURIComponent"));
 309                     case "encodeURI": return gs.get(JSU.S("encodeURI"));
 310                     case "encodeURIComponent": return gs.get(JSU.S("encodeURIComponent"));
 311                     case "escape": return gs.get(JSU.S("escape"));
 312                     case "unescape": return gs.get(JSU.S("unescape"));
 313                     case "fromCharCode": return gs.get(JSU.S("stringFromCharCode"));
 314                     */
 315                     return null;
 316                 }
 317             };
 318     }
 319
 320     private static abstract class JSTarget extends JS.Obj implements Target { }
 321 }