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