massive cleanup, almost there!
[org.ibex.mail.git] / src / org / ibex / mail / Message.java
1 package org.ibex.mail;
2 import org.ibex.crypto.*;
3 import org.ibex.js.*;
4 import org.ibex.util.*;
5 import org.ibex.mail.protocol.*;
6 import java.util.*;
7 import java.net.*;
8 import java.io.*;
9
10 // soft line limit (suggested): 78 chars /  hard line limit: 998 chars
11 // folded headers: can insert CRLF anywhere that whitespace appears (before the whitespace)
12 // date/time parsing: see spec, 3.3
13
14 // FEATURE: PGP-signature-parsing
15 // FEATURE: mailing list header parsing
16 // FEATURE: delivery status notification (and the sneaky variety)
17 // FEATURE: threading as in http://www.jwz.org/doc/threading.html
18 // FEATURE: lazy body
19
20 /** 
21  *  [immutable] This class encapsulates a message "floating in the
22  *  ether": RFC2822 data but no storage-specific flags or other
23  *  metadata.
24  */
25 public class Message extends JSReflection {
26
27     public final String allHeaders;           // pristine headers
28     public final Hashtable headers;           // hash of headers (not including resent's and traces)
29     public final String body;                 // entire body
30     public final int lines;                   // lines in the body
31
32     public final Date date;
33     public final Address to;
34     public final Address from;        // if multiple From entries, this is sender
35     public final Address replyto;     // if none provided, this is equal to sender
36     public final String subject;
37     public final String messageid;
38     public final Address[] cc;
39     public final Address[] bcc;
40     public final Hashtable[] resent;
41     public final Trace[] traces;
42
43     public final Address envelopeFrom;
44     public final Address[] envelopeTo;
45
46     public final Date arrival;         // when the message first arrived at this machine; IMAP "internal date message attr"
47     
48     // FIXME: need to be able to read back in the EnvelopeFrom / EnvelopeTo fields
49     public void dump(OutputStream os) throws IOException {
50         Writer w = new OutputStreamWriter(os);
51         w.write(allHeaders);
52         w.write("X-org.ibex.mail-envelopeFrom: " + envelopeFrom + "\r\n");
53         w.write("X-org.ibex.mail-envelopeTo: "); for(int i=0; i<envelopeTo.length; i++) w.write(envelopeTo[i] + " "); w.write("\r\n");
54         w.write(body);
55         w.flush();
56     }
57
58     public class Trace {
59         final String returnPath;
60         final Element[] elements;
61         public Trace(LineReader lr) throws Trace.Malformed, IOException {
62             String retPath = lr.readLine();
63             if (!retPath.startsWith("Return-Path:")) throw new Trace.Malformed("trace did not start with Return-Path header");
64             returnPath = retPath.substring(12).trim();
65             Vec el = new Vec();
66             while(true) {
67                 String s = lr.readLine();
68                 if (s == null) break;
69                 if (!s.startsWith("Received:")) { lr.pushback(s); break; }
70                 s = s.substring(9).trim();
71                 el.addElement(new Element(s));
72             }
73             elements = new Element[el.size()];
74             el.copyInto(elements);
75         }
76         public class Element {
77              String fromDomain;
78              String fromIP;
79              String toDomain;
80              String forWhom;
81              Date date;
82             public Element(String fromDomain, String fromIP, String toDomain, String forWhom, Date date) {
83                 this.fromDomain=fromDomain; this.fromIP=fromIP; this.toDomain=toDomain; this.forWhom=forWhom; this.date=date; }
84             public Element(String s) throws Trace.Malformed {
85                 StringTokenizer st = new StringTokenizer(s);
86                 if (!st.nextToken().equals("FROM")) throw new Trace.Malformed("trace did note have a FROM element: " + s);
87                 fromDomain = st.nextToken();
88                 if (!st.nextToken().equals("BY")) throw new Trace.Malformed("trace did note have a BY element: " + s);
89                 toDomain = st.nextToken();
90                 // FIXME not done yet
91             }
92         }
93         public class Malformed extends Message.Malformed { public Malformed(String s) { super(s); } }
94     }
95
96     public static class Malformed extends MailException.Malformed { public Malformed(String s) { super(s); } }
97
98     public Message(Address envelopeFrom, Address[] envelopeTo, String s, Date arrival)
99         { this(envelopeFrom, envelopeTo, new LineReader(new StringReader(s)), arrival); }
100     public Message(Address envelopeFrom, Address[] envelopeTo, LineReader rs) { this(envelopeFrom, envelopeTo, rs, null); }
101     public Message(Address envelopeFrom, Address[] envelopeTo, LineReader rs, Date arrival) {
102         try {
103             this.arrival = arrival == null ? new Date() : arrival;
104             this.headers = new CaseInsensitiveHash();
105             Vec envelopeToHeader = new Vec();
106             String key = null;
107             StringBuffer all = new StringBuffer();
108             Date date = null;
109             Address to = null, from = null, replyto = null;
110             String subject = null, messageid = null;
111             Vec cc = new Vec(), bcc = new Vec(), resent = new Vec(), traces = new Vec();
112             for(String s = rs.readLine(); s != null && !s.equals(""); s = rs.readLine()) {
113                 all.append(s);
114                 all.append("\r\n");
115                 if (s.length() == 0 || Character.isSpace(s.charAt(0))) {
116                     if (key == null) throw new Malformed("Message began with a blank line; no headers");
117                     ((CaseInsensitiveHash)headers).add(key, headers.get(key) + s);
118                     continue;
119                 }
120                 if (s.indexOf(':') == -1) throw new Malformed("Header line does not contain colon: " + s);
121                 key = s.substring(0, s.indexOf(':'));
122                 for(int i=0; i<key.length(); i++)
123                     if (key.charAt(i) < 33 || key.charAt(i) > 126)
124                         throw new Malformed("Header key \""+key+"\" contains invalid character \"" + key.charAt(i) + "\"");
125                 String val = s.substring(s.indexOf(':') + 1).trim();
126                 while(val.length() > 0 && Character.isSpace(val.charAt(0))) val = val.substring(1);
127                 if (key.startsWith("Resent-")) {
128                     if (key.startsWith("Resent-From")) resent.addElement(new Hashtable());
129                     ((Hashtable)resent.lastElement()).put(key.substring(7), val);
130                 } else if (key.startsWith("Return-Path")) {
131                     rs.pushback(s); traces.addElement(new Trace(rs));
132                 } else if (key.equals("X-org.ibex.mail.headers.envelopeFrom")) {
133                     if (envelopeFrom == null) envelopeFrom = new Address(val);
134                 } else if (key.equals("X-org.ibex.mail.headers.envelopeTo")) {
135                     if (envelopeTo == null) envelopeToHeader.addElement(new Address(val));
136                 } else {
137                     // just append it to the previous one; valid for Comments/Keywords
138                     if (headers.get(key) != null) val = headers.get(key) + " " + val;
139                     ((CaseInsensitiveHash)headers).add(key, val);
140                 }            
141             }
142             if (envelopeTo == null) envelopeTo = new Address[envelopeToHeader.size()];
143             envelopeToHeader.copyInto(envelopeTo);
144
145             // FIXME what if all are null?
146             this.to           = headers.get("To") == null   ? envelopeTo[0] : new Address((String)headers.get("To"));
147             this.from         = headers.get("From") == null ? envelopeFrom  : new Address((String)headers.get("From"));
148             this.envelopeFrom = envelopeFrom == null        ? this.from                 : envelopeFrom;
149             this.envelopeTo   = envelopeTo == null          ? new Address[] { this.to } : envelopeTo;
150
151             this.date      = new Date(); // FIXME (Date)headers.get("Date");
152             this.replyto   = headers.get("Reply-To") == null ? null : new Address((String)headers.get("Reply-To"));
153             this.subject   = (String)headers.get("Subject");
154             this.messageid = (String)headers.get("Message-Id");
155             if (headers.get("Cc") != null) {
156                 StringTokenizer st = new StringTokenizer((String)headers.get("Cc"));
157                 this.cc = new Address[st.countTokens()];
158                 for(int i=0; i<this.cc.length; i++) this.cc[i] = new Address(st.nextToken());
159             } else {
160                 this.cc = new Address[0];
161             }
162             if (headers.get("Bcc") != null) {
163                 StringTokenizer st = new StringTokenizer((String)headers.get("Bcc"));
164                 this.bcc = new Address[st.countTokens()];
165                 for(int i=0; i<this.bcc.length; i++) this.bcc[i] = new Address(st.nextToken());
166             } else {
167                 this.bcc = new Address[0];
168             }
169             resent.copyInto(this.resent = new Hashtable[resent.size()]);
170             traces.copyInto(this.traces = new Trace[traces.size()]);
171             allHeaders = all.toString();
172             StringBuffer body = new StringBuffer();
173             int lines = 0;
174             for(String s = rs.readLine();; s = rs.readLine()) { if (s == null) break; lines++; body.append(s + "\r\n"); }
175             this.lines = lines;
176             this.body = body.toString();
177         } catch (IOException e) { throw new MailException.IOException(e); }
178     }
179
180     // http://www.jwz.org/doc/mid.html
181     private static final Random random = new Random();
182     public static String generateFreshMessageId() {
183         StringBuffer ret = new StringBuffer();
184         ret.append('<');
185         ret.append(Base36.encode(System.currentTimeMillis()));
186         ret.append('.');
187         ret.append(Base36.encode(random.nextLong()));
188         ret.append('.');
189         try { ret.append(InetAddress.getLocalHost().getHostName()); } catch (UnknownHostException e) { /* DELIBERATE */ }
190         ret.append('>');
191         return ret.toString();
192     }
193
194     public int rfc822size() { return allHeaders.length() + 2 /* CRLF */ + body.length(); }  // FIXME: double check this
195
196     public String summary() {
197         return
198             "          Subject: " + subject + "\n" +
199             "     EnvelopeFrom: " + envelopeFrom + "\n" +
200             "       EnvelopeTo: " + envelopeTo + "\n" +
201             "        MessageId: " + messageid;
202     }
203
204     //  use null-sender for error messages (don't send errors to the null addr)
205     public Message bounce(String reason) { throw new RuntimeException("bounce not implemented"); }  // FIXME!
206
207     private static class CaseInsensitiveHash extends Hashtable {
208         public Object get(Object o) { return (o instanceof String) ? super.get(((String)o).toLowerCase()) : super.get(o); }
209         public Object put(Object k, Object v) { throw new Error("you cannot write to a CaseInsensitiveHash"); }
210         void add(Object k, Object v) { if (k instanceof String) super.put(((String)k).toLowerCase(), v); else super.put(k, v); }
211     }
212
213 }