/*
 *  rhctl
 *
 *  Copyright (C) 2009-2014 Christian Pointner <equinox@helsinki.at>
 *
 *  This file is part of rhctl.
 *
 *  rhctl is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  any later version.
 *
 *  rhctl is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with rhctl. If not, see <http://www.gnu.org/licenses/>.
 */

#include "datatypes.h"

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>

#include "log.h"
#include "sig_handler.h"
#include "options.h"

#include "daemon.h"
#include "utils.h"

int process_cmd(const char* cmd, int fd, cmd_t **cmd_q, client_t* client_lst, options_t* opt)
{
  return 0;
}

int process_data(int src_fd, int dest_fd)
{
  char* buffer[100];
  int ret = read(src_fd, buffer, 100);
  if(!ret)
    return 2;
  if(ret == -1 && errno == EAGAIN)
    return 0;
  if(ret < 0)
    return ret;

  log_printf(DEBUG, "read %d bytes from fd (%d)", ret, src_fd);

  int len = ret;
  int offset = 0;
  for(;;) {
    ret = write(dest_fd, &buffer[offset], len - offset);
    if(ret < 0) {

      if(errno != EINTR)
        return ret;

      ret = 0;
    }

    offset += ret;
    if(offset+1 >= len)
      break;
  }
  return 0;
}

int main_loop(int cmd_fd, options_t* opt)
{
  log_printf(NOTICE, "entering main loop");

  fd_set readfds, tmpfds;
  FD_ZERO(&readfds);
  FD_SET(0, &readfds);
  FD_SET(cmd_fd, &readfds);
  int max_fd = cmd_fd;

  int sig_fd = signal_init();
  if(sig_fd < 0)
    return -1;
  FD_SET(sig_fd, &readfds);
  max_fd = (max_fd < sig_fd) ? sig_fd : max_fd;

  int return_value = 0;

  while(!return_value) {
    memcpy(&tmpfds, &readfds, sizeof(tmpfds));

    int ret = select(max_fd+1, &tmpfds, NULL, NULL, NULL);
    if(ret == -1 && errno != EINTR) {
      log_printf(ERROR, "select returned with error: %s", strerror(errno));
      return_value = -1;
      break;
    }
    if(ret == -1 || !ret)
      continue;

    if(FD_ISSET(sig_fd, &tmpfds))
      if(signal_handle())
        return_value = 1;

    if(FD_ISSET(0, &tmpfds))
      return_value = process_data(0, cmd_fd);

    if(FD_ISSET(cmd_fd, &tmpfds))
      return_value = process_data(cmd_fd, 1);
  }

  signal_stop();
  return return_value;
}

int main(int argc, char* argv[])
{
  log_init();

  options_t opt;
  int ret = options_parse(&opt, argc, argv);
  if(ret) {
    if(ret > 0) {
      fprintf(stderr, "syntax error near: %s\n\n", argv[ret]);
    }
    if(ret == -2) {
      fprintf(stderr, "memory error on options_parse, exiting\n");
    }

    if(ret != -2)
      options_print_usage();

    options_clear(&opt);
    log_close();
    exit(ret);
  }
  string_list_element_t* tmp = opt.log_targets_.first_;
  if(!tmp) {
    log_add_target("stderr:2");
  }
  else {
    while(tmp) {
      ret = log_add_target(tmp->string_);
      if(ret) {
        switch(ret) {
        case -2: fprintf(stderr, "memory error on log_add_target, exitting\n"); break;
        case -3: fprintf(stderr, "unknown log target: '%s', exitting\n", tmp->string_); break;
        case -4: fprintf(stderr, "this log target is only allowed once: '%s', exitting\n", tmp->string_); break;
        default: fprintf(stderr, "syntax error near: '%s', exitting\n", tmp->string_); break;
        }

        options_clear(&opt);
        log_close();
        exit(ret);
      }
      tmp = tmp->next_;
    }
  }
  log_printf(NOTICE, "just started...");
  if(options_parse_post(&opt)) {
    options_clear(&opt);
    log_close();
    exit(-1);
  }

  int cmd_fd = connect_command_socket(opt.command_sock_);
  if(cmd_fd < 0) {
    options_clear(&opt);
    log_close();
    exit(-1);
  }

  ret = main_loop( cmd_fd, &opt);

  close(cmd_fd);

  if(!ret)
    log_printf(NOTICE, "normal shutdown");
  else if(ret < 0)
    log_printf(NOTICE, "shutdown after error (code %d)", ret);
  else
    log_printf(NOTICE, "shutdown after signal");

  options_clear(&opt);
  log_close();

  return ret;
}