append sorta works; dates are broken
[org.ibex.mail.git] / src / org / ibex / mail / Query.java
1 package org.ibex.mail;
2 import java.util.*;
3 import org.ibex.mail.target.*;
4
5 /**
6  *  [immutable] This class encapsulates a query against a mailbox.
7  *
8  *  The default implementation of Mailbox.query() simply retrieves
9  *  each message and calls Query.match() on it to filter out any
10  *  messages which do not match.
11  *
12  *  Production-quality Mailbox implementations will want to inspect
13  *  the Query object passed to Mailbox.query() to perform more
14  *  efficient queries.
15  */
16 public class Query {
17
18     public static Query not(Query q)                { return new Query(NOT, new Query[] { q },0,0,0,null,null,null,null,null); }
19     public static Query and(Query q1, Query q2)     { return new Query(AND,new Query[]{q1,q2},0,0,0,null,null,null,null,null); }
20     public static Query and(Query[] q)              { return new Query(AND, q,0,0,0,null, null, null, null, null ); }
21     public static Query or(Query q1, Query q2)      { return new Query(OR,new Query[] {q1,q2},0,0,0,null,null, null, null, null); }
22     public static Query or(Query[] q)               { return new Query(OR, q, 0, 0, 0, null, null, null, null, null); }
23     public static Query uid(int min, int max)       { return new Query(UID, null, min, max, 0, null, null, null, null, null); }
24     public static Query messagenum(int min,int max) { return new Query(NUM,null,min,max,0,null,null,null,null, null); }
25     public static Query sent(Date e, Date l)        { return new Query(SENT,null,0,0,0,null,null,e,l,null); }
26     public static Query arrival(Date e, Date l)     { return new Query(ARRIVAL,null,0,0,0,null,null, e,l,null);}
27     public static Query header(String k, String v)  { return new Query(HEADER, null, 0, 0, 0, k, v, null, null, null);}
28     public static Query size(int min, int max)      { return new Query(SIZE, null, min, max, 0, null, null, null, null, null);}
29     public static Query body(String text)           { return new Query(BODY, null, 0, 0, 0, null, text, null, null, null);}
30     public static Query full(String text)           { return new Query(FULL, null, 0, 0, 0, null, text, null, null, null);}
31     public static Query uid(int[] set)              { return new Query(UID,  null, 0 ,0 ,0, null, null, null, null, set);}
32     public static Query num(int[] set)              { return new Query(NUM,  null, 0 ,0 ,0, null, null, null, null, set);}
33     public static Query all()                       { return new Query(ALL, null, 0, 0, 0, null, null, null, null, null); }
34     public static Query deleted()                   { return new Query(DELETED, null, 0, 0, 0, null, null, null, null, null); }
35     public static Query seen()                      { return new Query(SEEN, null, 0, 0, 0, null, null, null, null, null); }
36     public static Query flagged()                   { return new Query(FLAGGED, null, 0, 0, 0, null, null, null, null, null); }
37     public static Query draft()                     { return new Query(DRAFT, null, 0, 0, 0, null, null, null, null, null); }
38     public static Query answered()                  { return new Query(ANSWERED, null, 0, 0, 0, null, null, null, null, null); }
39     public static Query recent()                    { return new Query(RECENT, null, 0, 0, 0, null, null, null, null, null); }
40
41     private Query(int type, Query[] q,int min,int max, int flags, String key, String text, Date earliest, Date latest, int[] set) {
42         this.type = type; this.q = q; this.min = min; this.max = max; this.flags = flags; this.key = key; this.text = text;
43         this.earliest = earliest; this.latest = latest; this.set = set; }
44
45     public static final int ALL        = 0;
46     public static final int NOT        = 1;
47     public static final int AND        = 2;
48     public static final int OR         = 3;
49     public static final int UID        = 4;
50     public static final int SENT       = 5;
51     public static final int ARRIVAL    = 6;
52     public static final int HEADER     = 7;
53     public static final int SIZE       = 8;
54     public static final int BODY       = 9;
55     public static final int FULL       = 10;
56     public static final int NUM        = 11;
57     public static final int DELETED    = 12;
58     public static final int SEEN       = 13;
59     public static final int FLAGGED    = 14;
60     public static final int DRAFT      = 15;
61     public static final int ANSWERED   = 16;
62     public static final int RECENT     = 17;
63
64     public final int type;
65     public final Query[] q;
66     public final int min;
67     public final int max;
68     public final int flags;
69     public final String key;
70     public final String text;
71     public final Date earliest;
72     public final Date latest;
73     public final int[] set;
74
75     public boolean match(Mailbox.Iterator it) {
76         switch(type) {
77             case ALL:        return true;
78             case NOT:        return !q[0].match(it);
79             case OR:         for(int i=0; i<q.length; i++) if (q[i].match(it)) return true; return false;
80             case AND:        for(int i=0; i<q.length; i++) if (!q[i].match(it)) return false; return true;
81             case UID:        if (set != null){for(int i=0; i<set.length; i++) if (set[i] == it.uid()) return true; return false; }
82                              else return it.uid() >= min && it.uid() <= max;
83             case NUM:        if (set != null){for(int i=0; i<set.length; i++) if (set[i] == it.uid()) return true; return false; }
84                              else return it.uid() >= min && it.uid() <= max;
85             case SENT:       return (latest==null||it.cur().date.before(latest)) && 
86                                     (earliest==null||it.cur().date.after(earliest));
87             case ARRIVAL:    return (latest == null || it.cur().arrival.before(latest)) &&
88                                     (earliest == null || it.cur().arrival.after(earliest));
89             case SIZE:       return it.cur().rfc822size() >= min && it.cur().rfc822size() <= max;
90             case HEADER:     return it.cur().headers.get(key) != null && ((String)it.cur().headers.get(key)).indexOf(text) != -1;
91             case BODY:       return it.cur().body.indexOf(text) != -1;
92             case FULL:       return it.cur().body.indexOf(text) != -1 || it.cur().allHeaders.indexOf(text) != -1;
93             case DELETED:    return it.deleted();
94             case SEEN:       return it.seen();
95             case FLAGGED:    return it.flagged();
96             case DRAFT:      return it.draft();
97             case ANSWERED:   return it.answered();
98             case RECENT:     return it.recent();
99             default:         throw new Error("this should not happen");
100         }
101     }
102 }