#include <unistd.h>

#include "buffer.h"
#include "logmsg.h"
#include "open.h"
#include "stralloc.h"

#include "headerbody.h"
#include "hfield.h"
#include "maildir.h"
#include "prioq.h"

#define WHO "maildirwatch"

static void die_nomem()
{
  logmsg(WHO, 111, FATAL, "out of memory");
}

stralloc recipient = {0};
stralloc sender = {0};
stralloc fromline = {0};
stralloc text = {0};

static void addtext(char *s, int n)
{
  if (!stralloc_catb(&text, s, n)) die_nomem();
  if (text.len > 158) text.len = 158;
}

static void dobody(stralloc *h)
{
  addtext(h->s, h->len);
}

static void doheader(stralloc *h)
{
  int i;
  switch (hfield_known(h->s, h->len)) {
    case H_SUBJECT:
      i = hfield_skipname(h->s, h->len);
      addtext(h->s + i, h->len - i);
      break;
    case H_DELIVEREDTO:
      i = hfield_skipname(h->s, h->len);
      if (i < h->len)
        if (!stralloc_copyb(&recipient, h->s + i, h->len - i - 1)) die_nomem();
      break;
    case H_RETURNPATH:
      i = hfield_skipname(h->s, h->len);
      if (i < h->len)
        if (!stralloc_copyb(&sender, h->s + i, h->len - i - 1)) die_nomem();
      break;
    case H_FROM:
      if (!stralloc_copyb(&fromline, h->s, h->len - 1)) die_nomem();
      break;
  }
}

static void finishheader() {}

stralloc filenames = {0};
prioq pq = {0};

char inbuf[BUFFER_INSIZE];
buffer bi;

int main()
{
  struct prioq_elt pe;
  int fd;
  int i;

  if (maildir_chdir() == -1) logmsg(WHO, 111, FATAL, "Can't change to maildir");

  for (;;) {
    maildir_clean(&filenames);
    if (maildir_scan(&pq, &filenames, 1, 0) == -1) logmsg(WHO, 111, FATAL, "Can't read maildir");

    buffer_putsflush(buffer_1, "\033[;H\033[;J");

    while (prioq_min(&pq, &pe)) {
      prioq_delmin(&pq);

      fd = open_read(filenames.s + pe.id);
      if (fd == -1) continue;
      buffer_init(&bi, read, fd, inbuf, sizeof(inbuf));

      if (!stralloc_copys(&sender, "?")) die_nomem();
      if (!stralloc_copys(&recipient, "?")) die_nomem();
      if (!stralloc_copys(&fromline, "")) die_nomem();
      if (!stralloc_copys(&text, "")) die_nomem();
      if (headerbody(&bi, doheader, finishheader, dobody) == -1)
        logmsg(WHO, 111, FATAL, "trouble reading new message");

      for (i = 0; i < fromline.len; ++i)
        if ((fromline.s[i] < 32) || (fromline.s[i] > 126)) fromline.s[i] = '/';
      for (i = 0; i < sender.len; ++i)
        if ((sender.s[i] < 32) || (sender.s[i] > 126)) sender.s[i] = '?';
      for (i = 0; i < recipient.len; ++i)
        if ((recipient.s[i] < 32) || (recipient.s[i] > 126)) recipient.s[i] = '?';
      for (i = 0; i < text.len; ++i)
        if ((text.s[i] < 32) || (text.s[i] > 126)) text.s[i] = '/';
      buffer_puts(buffer_1, "FROM ");
      buffer_put(buffer_1, sender.s, sender.len);
      buffer_puts(buffer_1, " TO <");
      buffer_put(buffer_1, recipient.s, recipient.len);
      buffer_puts(buffer_1, ">\n");
      if (fromline.len) {
        buffer_puts(buffer_1, "\033[1m");
        buffer_put(buffer_1, fromline.s, fromline.len);
        buffer_puts(buffer_1, "\033[0m\n");
      }
      buffer_put(buffer_1, text.s, text.len);
      buffer_puts(buffer_1, "\n\n");

      close(fd);
    }

    buffer_flush(buffer_1);
    sleep(30);
  }
}