#include <sys/stat.h>
#include <unistd.h>

#include <stdio.h>  // rename

#include "buffer.h"
#include "case.h"
#include "cdbmake.h"
#include "exit.h"
#include "getln.h"
#include "logmsg.h"
#include "open.h"
#include "stralloc.h"

#ifdef USE_CONFIG
  #include "fehsqm-config.h"
#else
  #include "auto_qmail.h"
#endif

#define WHO "qmail-recipients"

void die_read()
{
  logmsg(WHO, 111, ERROR, "unable to read users/recipients");
}

void die_write()
{
  logmsg(WHO, 111, ERROR, "unable to write to users/recipients.tmp");
}

char inbuf[1024];
buffer b;

int fd;
int fdtemp;

struct cdb_make cdb;
stralloc line = {0};
stralloc key = {0};
int match;

int main()
{
  umask(033);
  if (chdir(auto_qmail) == -1) logmsg(WHO, 110, ERROR, B("unable to chdir to: ", auto_qmail));

  fd = open_read("users/recipients");
  if (fd == -1) die_read();

  buffer_init(&b, read, fd, inbuf, sizeof(inbuf));

  fdtemp = open_trunc("users/recipients.tmp");
  if (fdtemp == -1) die_write();

  if (cdb_make_start(&cdb, fdtemp) == -1) die_write();

  for (;;) {
    stralloc_copys(&key, ":");
    if (getln(&b, &line, &match, '\n') != 0) die_read();
    while (line.len) {
      if (line.s[line.len - 1] == ' ') {
        --line.len;
        continue;
      }
      if (line.s[line.len - 1] == '\n') {
        --line.len;
        continue;
      }
      if (line.s[line.len - 1] == '\t') {
        --line.len;
        continue;
      }
      if (line.s[0] != '#' && stralloc_cat(&key, &line)) {
        case_lowerb(key.s, key.len);
        if (cdb_make_add(&cdb, key.s, key.len, "", 0) == -1) die_write();
      }
      break;
    }
    if (!match) break;
  }

  if (cdb_make_finish(&cdb) == -1) die_write();
  if (fsync(fdtemp) == -1) die_write();
  if (close(fdtemp) == -1) die_write(); /* NFS stupidity */
  if (rename("users/recipients.tmp", "users/recipients.cdb") == -1)
    logmsg(WHO, 111, ERROR, "unable to move users/recipients.tmp to users/recipients.cdb");

  _exit(0);
}