#include "control.h"

#include <unistd.h>

#include "alloc.h"
#include "buffer.h"
#include "error.h"
#include "getln.h"
#include "logmsg.h"
#include "open.h"
#include "scan.h"
#include "stralloc.h"


static char inbuf[2048];
static stralloc line = {0};
static stralloc me = {0};
static int meok = 0;

/** @file  control.c
*/

static void striptrailingwhitespace(stralloc *sa)
{
  while (sa->len > 0) {
    switch (sa->s[sa->len - 1]) {
      case '\n':
      case ' ':
      case '\t': --sa->len; break;
      default:   return;
    }
  }
}

int control_init(void)
{
  int r;

  r = control_readline(&me, "control/me");
  if (r == 1) meok = 1;
  return r;
}

int control_rldef(stralloc *sa, char *fn, int flagme, char *def)
{
  int r;

  r = control_readline(sa, fn);
  if (r) return r;
  if (flagme && meok) return stralloc_copy(sa, &me) ? 1 : -1;
  if (def) return stralloc_copys(sa, def) ? 1 : -1;
  return r;
}

int control_readline(stralloc *sa, char *fn)
{
  buffer b;
  int fd;
  int match;

  fd = open_read(fn);
  if (fd == -1) {
    if (errno == ENOENT) return 0;
    return -1;
  }

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

  if (getln(&b, sa, &match, '\n') == -1) {
    close(fd);
    return -1;
  }

  striptrailingwhitespace(sa);

  close(fd);
  return 1;
}

int control_readint(unsigned long *i, char *fn)
{
  unsigned long u;

  switch (control_readline(&line, fn)) {
    case 0:  return 0;
    case -1: return -1;
  }
  if (!stralloc_0(&line)) return -1;
  if (!scan_ulong(line.s, &u)) return 0;
  *i = u;

  return 1;
}

int control_readfile(stralloc *sa, char *fn, int flagme)
{
  buffer b;
  int fd;
  int match;

  if (!stralloc_copys(sa, "")) return -1;

  fd = open_read(fn);
  if (fd == -1) {
    if (errno == ENOENT) {
      if (flagme && meok) {
        if (!stralloc_copy(sa, &me)) return -1;
        if (!stralloc_0(sa)) return -1;
        return 1;
      }
      return 0;
    }
    return -1;
  }

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

  for (;;) {
    if (getln(&b, &line, &match, '\n') == -1) break;
    if (!match && !line.len) {
      close(fd);
      return 1;
    }
    striptrailingwhitespace(&line);
    if (!stralloc_0(&line)) break;
    if (line.s[0])
      if (line.s[0] != '#')
        if (!stralloc_cat(sa, &line)) break;
    if (!match) {
      close(fd);
      return 1;
    }
  }

  close(fd);
  return -1;
}