ref: 5d357160775432816ceb80bfabc98375f7db73a5
dir: /mbox.c/
#include <u.h>
#include <libc.h>
#include <bio.h>
#include <thread.h>
#include <plumb.h>
#include <ctype.h>
#include <regexp.h>
#include "mail.h"
typedef struct Fn Fn;
struct Fn {
char *name;
void (*fn)(char **, int);
};
enum {
Cevent,
Cseemail,
Cshowmail,
Csendmail,
Nchan,
};
char *maildir = "/mail/fs";
char *mailbox = "mbox";
Mesg dead = {.messageid="", .hash=42};
Reprog *addrpat;
Reprog *mesgpat;
int threadsort = 1;
int sender;
int plumbsendfd;
int plumbseemailfd;
int plumbshowmailfd;
int plumbsendmailfd;
Channel *cwait;
Mbox mbox;
static void showmesg(Biobuf*, Mesg*, int, int);
static void
plumbloop(Channel *ch, int fd)
{
Plumbmsg *m;
while(1){
if((m = plumbrecv(fd)) == nil)
threadexitsall("plumber gone");
sendp(ch, m);
}
}
static void
plumbshowmail(void*)
{
threadsetname("plumbshow %s", mbox.path);
plumbloop(mbox.show, plumbshowmailfd);
}
static void
plumbseemail(void*)
{
threadsetname("plumbsee %s", mbox.path);
plumbloop(mbox.see, plumbseemailfd);
}
static void
plumbsendmail(void*)
{
threadsetname("plumbsend %s", mbox.path);
plumbloop(mbox.send, plumbsendmailfd);
}
static void
eventread(void*)
{
Event *ev;
threadsetname("mbevent %s", mbox.path);
while(1){
ev = emalloc(sizeof(Event));
if(winevent(&mbox, ev) == -1)
break;
sendp(mbox.event, ev);
}
sendp(mbox.event, nil);
threadexits(nil);
}
static int
ideq(Mesg *a, Mesg *b)
{
if(a->messageid == nil || b->messageid == nil)
return 0;
return strcmp(a->messageid, b->messageid) == 0;
}
static int
cmpmesg(void *pa, void *pb)
{
Mesg *a, *b;
a = *(Mesg**)pa;
b = *(Mesg**)pb;
return b->time - a->time;
}
static int
rcmpmesg(void *pa, void *pb)
{
Mesg *a, *b;
a = *(Mesg**)pa;
b = *(Mesg**)pb;
return a->time - b->time;
}
static int
mesglineno(Mesg *msg, int *depth)
{
Mesg *p, *m;
int i, o, n, d;
o = 0;
d = 0;
n = 1;
/* Walk up to root, counting depth in the thread */
p = msg;
while(p->parent != nil){
m = p;
p = p->parent;
for(i = 0; i < p->nchild; i++){
if(p->child[i] == m)
break;
o += p->child[i]->nsub + 1;
}
if(!(p->state & Sdummy)){
o++;
d++;
}
}
/* Find the thread in the thread list */
for(i = 0; i < mbox.nmesg; i++){
m = mbox.mesg[i];
if(m == p)
break;
if(m->state & Stoplev){
n += mbox.mesg[i]->nsub;
if(!(m->state & Sdummy))
n++;
}
}
if(depth != nil)
*depth = d;
assert(n + o < mbox.nmesg);
return n + o;
}
static int
addchild(Mesg *p, Mesg *m)
{
Mesg *q;
assert(m->parent == nil);
for(q = p; q != nil; q = q->parent){
if(ideq(m, q)){
fprint(2, "wonky message replies to self\n");
return 0;
}
if(m->time > q->time)
q->time = m->time;
}
for(q = p; q != nil; q = q->parent)
q->nsub++;
p->child = erealloc(p->child, ++p->nchild*sizeof(Mesg*));
p->child[p->nchild - 1] = m;
qsort(p->child, p->nchild, sizeof(Mesg*), rcmpmesg);
m->parent = p;
return 1;
}
static int
slotfor(Mesg *m)
{
int i;
for(i = 0; i < mbox.nmesg; i++)
if(cmpmesg(&mbox.mesg[i], &m) >= 0)
break;
return i;
}
static void
removeid(Mesg *m)
{
Mesg *e;
int i;
/* Dummies don't go in the table */
if(m->state & Sdummy)
return;
i = m->hash % mbox.hashsz;
while(1){
e = mbox.hash[i];
if(e == nil)
return;
if(e != &dead && e->hash == m->hash && strcmp(e->messageid, m->messageid) == 0){
mbox.hash[i] = &dead;
mbox.ndead++;
}
i = (i + 1) % mbox.hashsz;
}
}
Mesg*
lookupid(char *msgid)
{
u32int h, i;
Mesg *e;
if(msgid == nil || strlen(msgid) == 0)
return nil;
h = strhash(msgid);
i = h % mbox.hashsz;
while(1){
e = mbox.hash[i];
if(e == nil)
return nil;
if(e != &dead && e->hash == h && strcmp(e->messageid, msgid) == 0)
return e;
i = (i + 1) % mbox.hashsz;
}
}
static void
addmesg(Mesg *m, int ins)
{
Mesg *o, *e, **oldh;
int i, oldsz, idx;
/*
* on initial load, it's faster to append everything then sort,
* but on subsequent messages it's better to just put it in the
* right place; we don't want to shuffle the already-sorted
* messages.
*/
if(mbox.nmesg == mbox.mesgsz){
mbox.mesgsz *= 2;
mbox.mesg = erealloc(mbox.mesg, mbox.mesgsz*sizeof(Mesg*));
}
if(ins)
idx = slotfor(m);
else
idx = mbox.nmesg;
memmove(&mbox.mesg[idx + 1], &mbox.mesg[idx], (mbox.nmesg - idx)*sizeof(Mesg*));
mbox.mesg[idx] = m;
mbox.nmesg++;
if(m->messageid == nil)
return;
/* grow hash table, or squeeze out deadwood */
if(mbox.hashsz <= 2*(mbox.nmesg + mbox.ndead)){
oldsz = mbox.hashsz;
oldh = mbox.hash;
if(mbox.hashsz <= 2*mbox.nmesg)
mbox.hashsz *= 2;
mbox.ndead = 0;
mbox.hash = emalloc(mbox.hashsz*sizeof(Mesg*));
for(i = 0; i < oldsz; i++){
if((o = oldh[i]) == nil)
continue;
mbox.hash[o->hash % mbox.hashsz] = o;
}
free(oldh);
}
i = m->hash % mbox.hashsz;
while(1){
e = mbox.hash[i % mbox.hashsz];
if(e == nil || e == &dead)
break;
i = (i + 1) % mbox.hashsz;
}
mbox.hash[i] = m;
}
static Mesg *
placeholder(char *msgid, vlong time, int ins)
{
Mesg *m;
m = emalloc(sizeof(Mesg));
m->state |= Sdummy|Stoplev;
m->messageid = estrdup(msgid);
m->hash = strhash(msgid);
m->time = time;
addmesg(m, ins);
return m;
}
static Mesg*
change(char *name, char *digest)
{
Mesg *m;
char *f;
if((m = mesglookup(name, digest)) == nil)
return nil;
if((f = rslurp(m, "flags", nil)) == nil)
return nil;
free(m->mflags);
m->mflags = f;
m->flags = 0;
if(strchr(m->mflags, 'd')) m->flags |= Fdel;
if(strchr(m->mflags, 's')) m->flags |= Fseen;
if(strchr(m->mflags, 'a')) m->flags |= Fresp;
return m;
}
static Mesg*
delete(char *name, char *digest)
{
Mesg *m;
if((m = mesglookup(name, digest)) == nil)
return nil;
m->flags |= Fdel;
return m;
}
static Mesg*
load(char *name, char *digest, int ins)
{
Mesg *m, *p;
if(strncmp(name, mbox.path, strlen(mbox.path)) == 0)
name += strlen(mbox.path);
if((m = mesgload(name)) == nil)
goto error;
if(digest != nil && strcmp(digest, m->digest) != 0)
goto error;
/* if we already have a dummy, populate it */
p = lookupid(m->messageid);
if(p != nil && (p->state & Sdummy)){
m->child = p->child;
m->nchild = p->nchild;
m->nsub = p->nsub;
mesgclear(p);
memcpy(p, m, sizeof(*p));
free(m);
m = p;
}else
addmesg(m, ins);
if(!threadsort || m->inreplyto == nil){
m->state |= Stoplev;
return m;
}
p = lookupid(m->inreplyto);
if(p == nil)
p = placeholder(m->inreplyto, m->time, ins);
addchild(p, m);
return m;
error:
mesgfree(m);
return nil;
}
void
mbredraw(Mesg *m, int add, int rec)
{
Biobuf *bfd;
int ln, depth;
ln = mesglineno(m, &depth);
fprint(mbox.addr, "%d%s", ln, add?"-#0":"");
bfd = bwindata(&mbox, OWRITE);
showmesg(bfd, m, depth, rec);
Bterm(bfd);
/* highlight the redrawn message */
fprint(mbox.addr, "%d%s", ln, add ? "-#0" : "");
fprint(mbox.ctl, "dot=addr\n");
}
static void
mbload(void)
{
int i, n, fd;
Dir *d;
mbox.mesgsz = 128;
mbox.hashsz = 128;
mbox.mesg = emalloc(mbox.mesgsz*sizeof(Mesg*));
mbox.hash = emalloc(mbox.hashsz*sizeof(Mesg*));
mbox.path = esmprint("%s/%s/", maildir, mailbox);
cleanname(mbox.path);
n = strlen(mbox.path);
if(mbox.path[n - 1] != '/')
mbox.path[n] = '/';
if((fd = open(mbox.path, OREAD)) == -1)
sysfatal("%s: open: %r", mbox.path);
while(1){
n = dirread(fd, &d);
if(n == -1)
sysfatal("%s read: %r", mbox.path);
if(n == 0)
break;
for(i = 0; i < n; i++)
if(strcmp(d[i].name, "ctl") != 0)
load(d[i].name, nil, 0);
free(d);
}
qsort(mbox.mesg, mbox.nmesg, sizeof(Mesg*), cmpmesg);
}
static void
showmesg(Biobuf *bfd, Mesg *m, int depth, int recurse)
{
char *sep, *flag, *dots;
int i, width;
if(!(m->state & Sdummy)){
dots = "";
flag = "★";
sep = depth ? "\t" : "";
width = depth ? Subjlen - 4 : Subjlen;
if(m->flags & Fseen) flag = " ";
if(m->flags & Fresp) flag = "←";
if(m->flags & Fdel) flag = "∉";
if(m->flags & Ftodel) flag = "∉";
if(utflen(m->subject) > Subjlen){
width -= 3;
dots = "...";
}
Bprint(bfd, "%-6s\t%s %s%*.*s%s\t<%s>\n",
m->name,
flag, sep, -width, width,
m->subject,
dots,
m->from);
depth++;
}
if(recurse && mbox.view != Vflat)
for(i = 0; i < m->nchild; i++)
showmesg(bfd, m->child[i], depth, recurse);
}
static void
mark(char **f, int nf, char *fstr, int flags, int add)
{
char *sel, *p, *q, *e, *path;
int i, q0, q1, fd;
Mesg *m;
if(flags == 0)
return;
wingetsel(&mbox, &q0, &q1);
if(nf == 0){
sel = winreadsel(&mbox);
for(p = sel; p != nil; p = e){
if((e = strchr(p, '\n')) != nil)
*e++ = 0;
if(!matchmesg(&mbox, p))
continue;
if((q = strchr(p, '/')) != nil)
q[1] = 0;
if((m = mesglookup(p, nil)) != nil){
if(add)
m->flags |= flags;
else
m->flags &= ~flags;
if(fstr != nil && strlen(fstr) != 0){
path = estrjoin(mbox.path, "/", m->name, "/flags", nil);
if((fd = open(path, OWRITE)) != -1){
fprint(fd, fstr);
close(fd);
}
free(path);
}
mbredraw(m, 0, 0);
}
}
free(sel);
}else for(i = 0; i < nf; i++){
if((m = mesglookup(f[i], nil)) != nil){
m->flags |= flags;
mbredraw(m, 0, 0);
}
}
winsetsel(&mbox, q0, q1);
}
static void
mbmark(char **f, int nf)
{
int add, flg;
if(nf == 0){
fprint(2, "missing fstr");
return;
}
if((flg = mesgflagparse(f[0], &add)) == -1){
fprint(2, "Mark: invalid flags %s\n", f[0]);
return;
}
mark(f+1, nf-1, f[0], flg, add);
}
static void
removemesg(Mesg *m)
{
Mesg *c, *p, *pp;
int i, j;
/* remove child, preserving order */
j = 0;
p = m->parent;
if(p != nil){
for(i = 0; p && i < p->nchild; i++){
if(p->child[i] != m)
j++;
p->child[j] = p->child[i];
}
p->nchild--;
for(pp = p; pp != nil; pp = pp->parent)
pp->nsub--;
}
/* reparent children */
for(i = 0; i < m->nchild; i++){
c = m->child[i];
c->parent = nil;
if(p != nil)
addchild(p, c);
else
c->state |= Stoplev;
}
}
static void
mbflush(char **, int)
{
int i, j, ln, fd;
char *path;
Mesg *m;
i = 0;
path = estrjoin(maildir, "/ctl", nil);
fd = open(path, OWRITE);
free(path);
if(fd == -1)
sysfatal("open mbox: %r");
while(i < mbox.nmesg){
m = mbox.mesg[i];
if((m->state & Sopen) || !(m->flags & (Fdel|Ftodel))){
i++;
continue;
}
ln = mesglineno(m, nil);
fprint(mbox.addr, "%d,%d", ln, ln+m->nsub);
write(mbox.data, "", 0);
if(m->flags & Ftodel)
fprint(fd, "delete %s %d", mailbox, atoi(m->name));
removemesg(m);
removeid(m);
for(j = 0; j < m->nchild; j++)
mbredraw(m->child[j], 1, 1);
memmove(&mbox.mesg[i], &mbox.mesg[i+1], (mbox.nmesg - i)*sizeof(Mesg*));
mbox.nmesg--;
mesgfree(m);
}
close(fd);
fprint(mbox.ctl, "clean\n");
}
static void
mbcompose(char **, int)
{
compose("", nil, 0, 0);
}
static void
delmesg(char **f, int nf)
{
mark(f, nf, nil, Ftodel, 1);
}
static void
undelmesg(char **f, int nf)
{
mark(f, nf, nil, Ftodel, 0);
}
static void
showlist(void)
{
Biobuf *bfd;
Mesg *m;
int i;
bfd = bwinopen(&mbox, "data", OWRITE);
for(i = 0; i < mbox.nmesg; i++){
m = mbox.mesg[i];
if(mbox.view == Vflat || m->state & (Sdummy|Stoplev))
showmesg(bfd, m, 0, 1);
}
Bterm(bfd);
fprint(mbox.addr, "0");
fprint(mbox.ctl, "dot=addr\n");
fprint(mbox.ctl, "show\n");
}
static void
quitall(char **, int)
{
Mesg *m;
Comp *c;
if(mbox.nopen > 0 && !mbox.canquit){
fprint(2, "Del: %d open messages\n", mbox.nopen);
mbox.canquit = 1;
return;
}
for(m = mbox.openmesg; m != nil; m = m->qnext)
fprint(m->ctl, "del\n");
for(c = mbox.opencomp; c != nil; c = c->qnext)
fprint(c->ctl, "del\n");
fprint(mbox.ctl, "del\n");
threadexitsall(nil);
}
/*
* shuffle a message to the right location
* in the list without doing a full sort.
*/
static void
reinsert(Mesg *m)
{
int i, idx;
idx = slotfor(m);
for(i = idx; i < mbox.nmesg; i++)
if(mbox.mesg[i] == m)
break;
memmove(&mbox.mesg[idx + 1], &mbox.mesg[idx], (i - idx)*sizeof(Mesg*));
mbox.mesg[idx] = m;
}
static void
changemesg(Plumbmsg *pm)
{
char *digest, *action;
Mesg *m, *r;
int ln, nr;
digest = plumblookup(pm->attr, "digest");
action = plumblookup(pm->attr, "mailtype");
// fprint(2, "changing message %s, %s %s\n", action, pm->data, digest);
if(strcmp(action, "new") == 0){
if((m = load(pm->data, digest, 1)) == nil)
return;
for(r = m; r->parent != nil; r = r->parent)
/* nothing */;
/* Bump whole thread up in list */
if(r->nsub > 0){
ln = mesglineno(r, nil);
nr = r->nsub-1;
if(!(r->state & Sdummy))
nr++;
/*
* We can end up with an empty container
* in an edge case.
*
* Imagine we have a dummy message with
* a child, and that child gets deleted,
* then a new message comes in replying
* to that dummy.
*
* In this case, r->nsub == 1 due to the
* newly added message, so nr=0.
* in that case, skip the redraw, and
* reinsert the dummy in the right place.
*/
if(nr > 0){
fprint(mbox.addr, "%d,%d", ln, ln+nr-1);
write(mbox.data, "", 0);
}
reinsert(r);
}
mbredraw(r, 1, 1);
}else if(strcmp(action, "delete") == 0){
if((m = delete(pm->data, digest)) != nil)
mbredraw(m, 0, 0);
}else if(strcmp(action, "modify") == 0){
if((m = change(pm->data, digest)) != nil)
mbredraw(m, 0, 0);
}
}
static void
viewmesg(Plumbmsg *pm)
{
Mesg *m;
m = mesgopen(pm->data, plumblookup(pm->attr, "digest"));
if(m != nil){
fprint(mbox.addr, "%d", mesglineno(m, nil));
fprint(mbox.ctl, "dot=addr\n");
fprint(mbox.ctl, "show\n");
}
}
static void
redraw(char **, int)
{
fprint(mbox.addr, ",");
showlist();
}
static void
nextunread(char **, int)
{
fprint(mbox.ctl, "addr=dot\n");
fprint(mbox.addr, "/^[0-9]+\\/ *\t★.*");
fprint(mbox.ctl, "dot=addr\n");
fprint(mbox.ctl, "show\n");
}
Fn mboxfn[] = {
{"Put", mbflush},
{"Mail", mbcompose},
{"Delmesg", delmesg},
{"Undelmesg", undelmesg},
{"Del", quitall},
{"Redraw", redraw},
{"Next", nextunread},
{"Mark", mbmark},
#ifdef NOTYET
{"Filter", filter},
{"Get", mbrefresh},
#endif
{nil}
};
static void
doevent(Event *ev)
{
char *f[32];
int nf;
Fn *p;
if(ev->action != 'M')
return;
switch(ev->type){
case 'l':
case 'L':
if(matchmesg(&mbox, ev->text))
mesgopen(ev->text, nil);
else
winreturn(&mbox, ev);
break;
case 'x':
case 'X':
if((nf = tokenize(ev->text, f, nelem(f))) == 0)
return;
for(p = mboxfn; p->fn != nil; p++)
if(strcmp(p->name, f[0]) == 0 && p->fn != nil){
p->fn(&f[1], nf - 1);
break;
}
if(p->fn == nil)
winreturn(&mbox, ev);
else if(p->fn != quitall)
mbox.canquit = 0;
break;
}
}
static void
execlog(void*)
{
Waitmsg *w;
while(1){
w = recvp(cwait);
if(w->msg[0] != 0)
fprint(2, "%d: %s\n", w->pid, w->msg);
free(w);
}
}
static void
mbmain(void*)
{
Event *ev;
Plumbmsg *psee, *pshow, *psend;
Alt a[] = {
[Cevent] = {mbox.event, &ev, CHANRCV},
[Cseemail] = {mbox.see, &psee, CHANRCV},
[Cshowmail] = {mbox.show, &pshow, CHANRCV},
[Csendmail] = {mbox.send, &psend, CHANRCV},
[Nchan] = {nil, nil, CHANEND},
};
threadsetname("mbox %s", mbox.path);
wininit(&mbox, mbox.path);
wintagwrite(&mbox, "Put Mail Delmesg Undelmesg Next ");
showlist();
fprint(mbox.ctl, "clean\n");
procrfork(eventread, nil, Stack, RFNOTEG);
while(1){
switch(alt(a)){
case Cevent:
doevent(ev);
free(ev);
break;
case Cseemail:
changemesg(psee);
plumbfree(psee);
break;
case Cshowmail:
viewmesg(pshow);
plumbfree(pshow);
break;
case Csendmail:
compose(psend->data, nil, 0, 0);
plumbfree(psend);
break;
}
}
}
static void
usage(void)
{
fprint(2, "usage: %s [-T] [-f mailfs] [mbox]\n", argv0);
exits("usage");
}
void
threadmain(int argc, char **argv)
{
mbox.view = Vgroup;
ARGBEGIN{
case 'f':
maildir = EARGF(usage());
break;
case 'T':
mbox.view = Vflat;
break;
case 's':
sender++;
break;
default:
usage();
break;
}ARGEND;
if(argc > 1)
usage();
if(argc == 1)
mailbox = argv[0];
doquote = needsrcquote;
quotefmtinstall();
tmfmtinstall();
addrpat = regcomp("[^ \t]*@[^ \t]*\\.[^ \t]*");
mesgpat = regcomp("[0-9]+(/.*)?");
cwait = threadwaitchan();
/* open these early so we won't miss messages while loading */
mbox.event = chancreate(sizeof(Event*), 0);
mbox.see = chancreate(sizeof(Plumbmsg*), 0);
mbox.show = chancreate(sizeof(Plumbmsg*), 0);
mbox.send = chancreate(sizeof(Plumbmsg*), 0);
plumbsendfd = plumbopen("send", OWRITE|OCEXEC);
plumbseemailfd = plumbopen("seemail", OREAD|OCEXEC);
plumbshowmailfd = plumbopen("showmail", OREAD|OCEXEC);
mbload();
procrfork(plumbseemail, nil, Stack, RFNOTEG);
procrfork(plumbshowmail, nil, Stack, RFNOTEG);
/* avoid duplicate sends when multiple mailboxes are open */
if(sender || strcmp(mailbox, "mbox") == 0){
plumbsendmailfd = plumbopen("sendmail", OREAD|OCEXEC);
procrfork(plumbsendmail, nil, Stack, RFNOTEG);
}
threadcreate(execlog, nil, Stack);
threadcreate(mbmain, nil, Stack);
threadexits(nil);
}