start counting at 1
[org.ibex.mail.git] / src / org / ibex / mail / protocol / NNTP.java
index 150c8d0..048efba 100644 (file)
@@ -1,4 +1,247 @@
 package org.ibex.mail.protocol;
+import org.ibex.util.*;
+import org.ibex.io.*;
+import org.ibex.mail.*;
+import org.ibex.mail.target.*;
+import org.ibex.jinetd.*;
+import java.io.*;
+import java.net.*;
+import java.util.*;
+import java.text.*;
+
 /** NNTP send/recieve */
-public class NNTP extends MessageProtocol {
+public class NNTP {
+
+    // FIXME: command lines limited to 512 chars
+
+    public static class No  extends RuntimeException { int code = 0; }    // 4xx response codes
+    public static class Bad extends RuntimeException { int code = 0; }    // 5xx response codes
+
+    public static class Group {
+        public Group(String n, boolean p, int f, int l, int c) { this.name=n;this.post=p;this.first=f;this.last=l;this.count=c;}
+        public final String  name;
+        public final boolean post;
+        public final int     first;
+        public final int     last;
+        public final int     count;
+    }
+
+    public static class Article {
+        public Article(String messageid, int num, Message message) { this.message=message;this.messageid=messageid;this.num=num;}
+        public final String  messageid;
+        public final int     num;
+        public final Message message;
+    }
+
+    public static interface Server {
+        public Group    group(String s);
+        public boolean  ihave(String messageid);
+        public Article  next();
+        public Article  last();
+        public boolean  post(Message m);
+        public Article  article(String messageid,  boolean head, boolean body);
+        public Article  article(int    messagenum, boolean head, boolean body);
+        public Group[]  list();
+        public Group[]  newgroups(Date d, String[] distributions);
+        public String[] newnews(String[] groups, Date d, String[] distributions);
+    }
+
+    public static class MailboxWrapper implements Server {
+        private final Mailbox root;
+        private Mailbox current;
+        private int ptr = 0;
+        public MailboxWrapper(Mailbox root)     { this.root = root; }
+        public Group    group(String s)         { ptr = 0; setgroup(s); return getgroup(s); }
+        public boolean  ihave(String messageid) { /* FEATURE */ return false; }
+        public boolean  post(Message m)         { /* FEATURE */ return false; }
+        public Article  next()                  { return article(ptr++, false, false); }
+        public Article  last()                  { return article(ptr--, false, false); }
+        public Article  article(String i, boolean h, boolean b) { return article(Query.header("message-id",i),h,b); }
+        public Article  article(int    n, boolean h, boolean b) { ptr = n; return article(Query.messagenum(n,n),h,b); }
+        private Article article(Query q, boolean head, boolean body) {
+            Mailbox.Iterator it = current.iterator(q);
+            if (!it.next()) return null;
+            Message m = body ? it.cur() : it.head();
+            return new Article(m.messageid, it.num(), m);
+        }
+        public Group[]  list() { return list(root, ""); }
+        private Group[] list(Mailbox who, String prefix) {
+            Vec v = new Vec();
+            if (who == null) who = root;
+            String[] s = who.children();
+            for(int i=0; i<s.length; i++) {
+                v.addElement(new Group(prefix + s[i], true, 0, 0, 0)); // FIXME numbers
+                Group[] g2 = list(who.slash(s[i], false), prefix + s[i] + ".");
+                for(int j=0; j<g2.length; j++) v.addElement(g2[j]);
+            }
+            Group[] ret = new Group[v.size()];
+            v.copyInto(ret);
+            return ret;
+        }
+
+        private void setgroup(String s) {
+            current = root;
+            for(StringTokenizer st = new StringTokenizer(s, ".");
+                st.hasMoreTokens();
+                current = current.slash(st.nextToken(), false));
+        }
+        private Group getgroup(String s) {
+            Mailbox box = root;
+            for(StringTokenizer st = new StringTokenizer(s, "."); st.hasMoreTokens(); box = box.slash(st.nextToken(), false));
+            return new Group(s, true, 1, box.count(Query.all()), box.count(Query.all()));
+        }
+
+        public Group[]  newgroups(Date d, String[] distributions) { /* FEATURE */ return new Group[] { }; }
+        public String[] newnews(String[] groups, Date d, String[] distributions) { /* FIXME */  return null; }
+    }
+
+    public static class Listener implements Worker {
+        private Server api = null;
+        private Login login;
+        private Connection conn;
+        public Listener(Login l) { this.login = l; }
+
+        private void println(String s) { Log.warn("[nntp-write]", s); conn.println(s); }
+        private void println() { Log.warn("[nntp-write]", ""); conn.println(); }
+
+        private void article(String numOrMessageId, boolean head, boolean body) {
+            String s = numOrMessageId.trim();
+            Article a;
+            if (s.startsWith("<")) a = api.article(s.substring(0, s.length() - 1), head, body);
+            else                   a = api.article(Integer.parseInt(s), head, body);
+            if (a == null) {
+                println("423 No such article.");
+                return;
+            }
+            int code = (head && body) ? 220 : head ? 221 : body ? 222 : 223;
+            println(code + " " + a.num + " <" + a.messageid + "> get ready for some stuff...");
+            if (head) println(a.message.headers.raw);
+            if (head && body) println();
+            if (body) println(a.message.body);
+            println(".");
+        }
+        public void handleRequest(Connection conn) {
+            this.conn = conn;
+            conn.setTimeout(30 * 60 * 1000);
+            println("200 " + conn.vhost + " [" + NNTP.class.getName() + "]");
+            String user = null;
+            String pass = null;
+            Account account = login.anonymous();
+            this.api = account == null ? null : new MailboxWrapper(account.getMailbox(NNTP.class));
+            for(String line = conn.readln(); line != null; line = conn.readln()) try {
+                Log.warn("[nntp-read]", line);
+                StringTokenizer st = new StringTokenizer(line, " ");
+                String command = st.nextToken().toUpperCase();
+                if (command.equals("AUTHINFO")) {
+                    String uop = st.nextToken().toUpperCase();
+                    if (uop.equals("USER")) user = st.nextToken();
+                    else if (uop.equals("PASS")) pass = st.nextToken();
+                    // FIXME error here
+                }
+                if (this.api == null) {
+                    if (user == null) { println("480 Authentication required"); continue; }
+                    if (pass == null) { println("381 Password required"); continue; }
+                    account = login.login(user, pass);
+                    if (account == null) { println("502 Invalid"); continue; }
+                    this.api = new MailboxWrapper(account.getMailbox(NNTP.class));
+                    println("281 Good to go");
+                    continue;
+                }
+                if        (command.equals("ARTICLE"))   { article(st.hasMoreTokens() ? st.nextToken() : null, true,  true); 
+                } else if (command.equals("HEAD"))      { article(st.hasMoreTokens() ? st.nextToken() : null, true,  false); 
+                } else if (command.equals("MODE"))      { println("201 Hello, you can post.");
+                } else if (command.equals("BODY"))      { article(st.hasMoreTokens() ? st.nextToken() : null, false, true); 
+                } else if (command.equals("STAT"))      { article(st.hasMoreTokens() ? st.nextToken() : null, false, false); 
+                } else if (command.equals("HELP"))      { println("100 you are beyond help."); println(".");
+                } else if (command.equals("SLAVE"))     { println("220 I don't care");
+                } else if (command.equals("XOVER"))     {
+                    println("224 Overview information follows");
+                    MailboxWrapper api = (MailboxWrapper)this.api;
+                    String range = st.hasMoreTokens() ? st.nextToken() : (api.ptr+"-"+api.ptr);
+                    int start = Integer.parseInt(range.substring(0, range.indexOf('-')));
+                    int end   = Integer.parseInt(range.substring(range.indexOf('-') + 1));
+                    Mailbox.Iterator it = api.current.iterator(Query.messagenum(start, end));
+                    while(it.next()) {
+                        Message m = it.cur();
+                        println(it.num()+"\t"+m.subject+"\t"+m.from+"\t"+m.date+"\t"+m.messageid+"\t"+
+                                m.headers.gets("references") + "\t" + m.size() + "\t" + m.lines);
+                    }
+                    println(".");
+                } else if (command.equals("LAST"))      { Article a = api.last(); println("223 "+a.num+" "+a.messageid+" ok");
+                } else if (command.equals("NEXT"))      { Article a = api.next(); println("223 "+a.num+" "+a.messageid+" ok");
+                } else if (command.equals("QUIT"))      { println("205 Bye."); conn.close(); return; 
+                } else if (command.equals("GROUP"))     {
+                    Group g = api.group(st.nextToken().toLowerCase());
+                    println("221 " + g.count + " " + g.first + " " + g.last + " " + g.name);
+                } else if (command.equals("NEWGROUPS") || command.equals("NEWNEWS")) { 
+                    String groups = command.equals("NEWNEWS") ? st.nextToken() : null;
+                    String datetime = st.nextToken() + " " + st.nextToken();
+                    String gmt = st.nextToken();
+                    String distributions = gmt.equals("GMT") ? (st.hasMoreTokens() ? st.nextToken() : null) : gmt;
+                    while(st.hasMoreTokens()) distributions += " " + st.nextToken();
+                    // FIXME deal with GMT
+                    Date d = new Date();
+                    try {
+                        d = new SimpleDateFormat("YYMMDD HHMMSS").parse(gmt);
+                    } catch (ParseException pe) {
+                        Log.warn(this, pe);
+                    }
+                    distributions = distributions.trim();
+                    if (distributions.startsWith("<")) distributions = distributions.substring(1, distributions.length() - 1);
+
+                    st = new StringTokenizer(distributions, ",");
+                    String[] dists = new String[st.countTokens()];
+                    for(int i=0; st.hasMoreTokens(); i++) dists[i] = st.nextToken();
+
+                    if (command.equals("NEWGROUPS")) {
+                        Group[] g = api.newgroups(d, dists);
+                        println("231 list of groups follows");
+                        for(int i=0; i<g.length; i++)
+                            println(g[i].name + " " + g[i].last + " " + g[i].first + " " + (g[i].post ? "y" : "n"));
+                        println(".");
+                    } else {
+                        st = new StringTokenizer(groups, ",");
+                        String[] g = new String[st.countTokens()];
+                        for(int i=0; st.hasMoreTokens(); i++) g[i] = st.nextToken();
+                        String[] a = api.newnews(g, d, dists);
+                        println("230 list of article messageids follows");
+                        for(int i=0; i<a.length; i++) println(a[i]);
+                        println(".");
+                    }
+
+                } else if (command.equals("POST"))      { 
+                    // FIXME
+                    /*
+                    boolean postok = api.post();
+                    if (!postok) {
+                        println("440 no posting allowed");
+                    } else {
+                    */
+                        println("340 send the article");
+                        // FIME read the article here
+                        println("240 article posted ok");
+                        //}
+
+                } else if (command.equals("LIST"))      { 
+                    Group[] g = api.list();
+                    println("215 list of groups follows");
+                    for(int i=0; i<g.length; i++)
+                        println(g[i].name + " " + g[i].last + " " + g[i].first + " " + (g[i].post ? "y" : "n"));
+                    println(".");
+
+                } else if (command.equals("IHAVE"))     {
+                    boolean want = api.ihave(st.nextToken());
+                    if (!want) {
+                        println("435 No thanks");
+                    } else {
+                        println("335 Proceed");
+                        // FIXME read article here
+                        println("235 Got it");
+                    }
+                }
+            } catch (No n)  { println(n.code + " " + n.getMessage());
+            } catch (Bad b) { println(b.code + " " + b.getMessage()); Log.warn(this, b); }
+            conn.close();
+        }
+    }
 }