/*

 * main.c

 *

 * Copyright (C) 2005-2010 Mikael Berthe <mikael@lilotux.net>

 *

 * This program 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 2 of the License, or (at

 * your option) any later version.

 *

 * This program 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 this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307

 * USA

 */

#include <stdio.h>

#include <stdlib.h>

#include <unistd.h>

#include <string.h>

#include <signal.h>

#include <termios.h>

#include <sys/types.h>

#include <sys/wait.h>

#include <glib.h>

#include <config.h>

#include <poll.h>

#include "caps.h"

#include "screen.h"

#include "settings.h"

#include "roster.h"

#include "commands.h"

#include "histolog.h"

#include "hooks.h"

#include "utils.h"

#include "pgp.h"

#include "otr.h"

#include "fifo.h"

#include "xmpp.h"

#include "help.h"

#include "events.h"

#ifdef MODULES_ENABLE

# include "compl.h"

# include "modules.h"

#endif

#ifdef ENABLE_HGCSET

# include "hgcset.h"

#endif

#ifndef WAIT_ANY

# define WAIT_ANY -1

#endif

static unsigned int terminate_ui;

GMainContext *main_context;

static struct termios *backup_termios;

char *mcabber_version(void)

{

  char *ver;

#ifdef HGCSET

  ver = g_strdup_printf("%s (%s)", PACKAGE_VERSION, HGCSET);

#else

  ver = g_strdup(PACKAGE_VERSION);

#endif

  return ver;

}

static void mcabber_terminate(const char *msg)

{

  fifo_deinit();

  xmpp_disconnect();

  scr_terminate_curses();

  // Restore term settings, if needed.

  if (backup_termios)

    tcsetattr(fileno(stdin), TCSAFLUSH, backup_termios);

  if (msg)

    fprintf(stderr, "%s\n", msg);

  printf("Bye!\n");

  exit(EXIT_SUCCESS);

}

void sig_handler(int signum)

{

  if (signum == SIGCHLD) {

    int status;

    pid_t pid;

    do {

      pid = waitpid (WAIT_ANY, &status, WNOHANG);

      // Check the exit status value if 'eventcmd_checkstatus' is set

      if (settings_opt_get_int("eventcmd_checkstatus")) {

        if (pid > 0) {

          // exit status 2 -> beep

          if (WIFEXITED(status) && WEXITSTATUS(status) == 2) {

            scr_beep();

          }

        }

      }

    } while (pid > 0);

    signal(SIGCHLD, sig_handler);

  } else if (signum == SIGTERM) {

    mcabber_terminate("Killed by SIGTERM");

  } else if (signum == SIGINT) {

    mcabber_terminate("Killed by SIGINT");

#ifdef USE_SIGWINCH

  } else if (signum == SIGWINCH) {

    ungetch(KEY_RESIZE);

#endif

  } else {

    scr_LogPrint(LPRINT_LOGNORM, "Caught signal: %d", signum);

  }

}

//  ask_password(what)

// Return the password, or NULL.

// The string must be freed after use.

static char *ask_password(const char *what)

{

  char *password, *p;

  size_t passsize = 128;

  struct termios orig, new;

  password = g_new0(char, passsize);

  /* Turn echoing off and fail if we can't. */

  if (tcgetattr(fileno(stdin), &orig) != 0) return NULL;

  backup_termios = &orig;

  new = orig;

  new.c_lflag &= ~ECHO;

  if (tcsetattr(fileno(stdin), TCSAFLUSH, &new) != 0) return NULL;

  /* Read the password. */

  printf("Please enter %s: ", what);

  if (fgets(password, passsize, stdin) == NULL) return NULL;

  /* Restore terminal. */

  tcsetattr(fileno(stdin), TCSAFLUSH, &orig);

  printf("\n");

  backup_termios = NULL;

  for (p = (char*)password; *p; p++)

    ;

  for ( ; p > (char*)password ; p--)

    if (*p == '\n' || *p == '\r') *p = 0;

  return password;

}

static void credits(void)

{

  const char *v_fmt = "MCabber %s -- Email: mcabber [at] lilotux [dot] net\n";

  char *v = mcabber_version();

  printf(v_fmt, v);

  scr_LogPrint(LPRINT_NORMAL|LPRINT_NOTUTF8, v_fmt, v);

  g_free(v);

}

static void compile_options(void)

{

  puts("Installation data directory: " DATA_DIR "\n");

#ifdef HAVE_UNICODE

  puts("Compiled with unicode support.");

#endif

#ifdef MODULES_ENABLE

  printf("Compiled with modules support. (API %s:%d-%d)\n",

         MCABBER_BRANCH, MCABBER_API_MIN, MCABBER_API_VERSION);

# ifdef PKGLIB_DIR

  puts("Modules directory: " PKGLIB_DIR);

# endif

#endif

#ifdef HAVE_GPGME

  puts("Compiled with GPG support.");

#endif

#ifdef HAVE_LIBOTR

  puts("Compiled with OTR support.");

#endif

#ifdef WITH_ENCHANT

  puts("Compiled with Enchant support.");

#endif

#ifdef WITH_ASPELL

  puts("Compiled with Aspell support.");

#endif

#ifdef ENABLE_DEBUG

  puts("Compiled with debugging support.");

#endif

}

static void main_init_pgp(void)

{

#ifdef HAVE_GPGME

  const char *pk, *pp;

  char *typed_passwd = NULL;

  char *p;

  bool pgp_invalid = FALSE;

  bool pgp_agent;

  int retries;

  p = getenv("GPG_AGENT_INFO");

  pgp_agent = (p && strchr(p, ':'));

  pk = settings_opt_get("pgp_private_key");

  pp = settings_opt_get("pgp_passphrase");

  if (settings_opt_get("pgp_passphrase_retries"))

    retries = settings_opt_get_int("pgp_passphrase_retries");

  else

    retries = 2;

  if (!pk) {

    scr_LogPrint(LPRINT_LOGNORM, "WARNING: unknown PGP private key");

    pgp_invalid = TRUE;

  } else if (!(pp || pgp_agent)) {

    // Request PGP passphrase

    pp = typed_passwd = ask_password("your PGP passphrase");

  }

  gpg_init(pk, pp);

  // Erase password from the settings array

  if (pp) {

    memset((char*)pp, 0, strlen(pp));

    if (typed_passwd)

      g_free(typed_passwd);

    else

      settings_set(SETTINGS_TYPE_OPTION, "pgp_passphrase", NULL);

  }

  if (!pgp_agent && pk && pp && gpg_test_passphrase()) {

    // Let's check the pasphrase

    int i;

    for (i = 1; retries < 0 || i <= retries; i++) {

      typed_passwd = ask_password("your PGP passphrase"); // Ask again...

      if (typed_passwd) {

        gpg_set_passphrase(typed_passwd);

        memset(typed_passwd, 0, strlen(typed_passwd));

        g_free(typed_passwd);

      }

      if (!gpg_test_passphrase())

        break; // Ok

    }

    if (i > retries)

      pgp_invalid = TRUE;

  }

  if (pgp_invalid)

    scr_LogPrint(LPRINT_LOGNORM, "WARNING: PGP key/pass invalid");

#else /* not HAVE_GPGME */

  scr_LogPrint(LPRINT_LOGNORM, "WARNING: not compiled with PGP support");

#endif /* HAVE_GPGME */

}

void mcabber_set_terminate_ui(void)

{

  terminate_ui = TRUE;

}

typedef struct {

  GSource source;

  GPollFD pollfd;

} mcabber_source_t;

static gboolean mcabber_source_prepare(GSource *source, gint *timeout)

{

  *timeout = -1;

  return FALSE;

}

static gboolean mcabber_source_check(GSource *source)

{

  mcabber_source_t *mc_source = (mcabber_source_t *) source;

  gushort revents = mc_source->pollfd.revents;

  if (revents)

    return TRUE;

  return FALSE;

}

static gboolean keyboard_activity(void)

{

  keycode kcode;

  if (terminate_ui) {

    return FALSE;

  }

  scr_do_update();

  scr_getch(&kcode);

  while (kcode.value != ERR) {

    scr_process_key(kcode);

    scr_getch(&kcode);

  }

  scr_check_auto_away(FALSE);

  return TRUE;

}

static gboolean mcabber_source_dispatch(GSource *source, GSourceFunc callback,

                                        gpointer udata) {

  return keyboard_activity();

}

static GSourceFuncs mcabber_source_funcs = {

  mcabber_source_prepare,

  mcabber_source_check,

  mcabber_source_dispatch,

  NULL,

  NULL,

  NULL

};

int main(int argc, char **argv)

{

  char *configFile = NULL;

  const char *optstring;

  int optval, optval2;

  int ret;

  credits();

  signal(SIGTERM, sig_handler);

  signal(SIGINT,  sig_handler);

  signal(SIGCHLD, sig_handler);

#ifdef USE_SIGWINCH

  signal(SIGWINCH, sig_handler);

#endif

  signal(SIGPIPE, SIG_IGN);

  /* Parse command line options */

  while (1) {

    int c = getopt(argc, argv, "hVf:");

    if (c == -1) {

      break;

    } else

      switch (c) {

      case 'h':

      case '?':

        printf("Usage: %s [-h|-V|-f mcabberrc_file]\n\n", argv[0]);

        return (c == 'h' ? 0 : -1);

      case 'V':

        compile_options();

        return 0;

      case 'f':

        configFile = g_strdup(optarg);

        break;

      }

  }

  if (optind < argc) {

    fprintf(stderr, "Usage: %s [-h|-V|-f mcabberrc_file]\n\n", argv[0]);

    return -1;

  }

  /* Initialize command system, roster and default key bindings */

  cmd_init();

  roster_init();

  settings_init();

  scr_init_bindings();

  caps_init();

#ifdef MODULES_ENABLE

  compl_init_system();

  modules_init();

#endif

  /* Initialize charset */

  scr_init_locale_charset();

  ut_init_debug();

  help_init();

  /* Parsing config file... */

  ret = cfg_read_file(configFile, TRUE);

  /* free() configFile if it has been allocated during options parsing */

  g_free(configFile);

  /* Leave if there was an error in the config. file */

  if (ret == -2)

    exit(EXIT_FAILURE);

  /* Display configuration settings */

  {

    const char *p;

    if ((p = settings_opt_get("server")) != NULL)

      scr_log_print(LPRINT_NORMAL, "Server: %s", p);

    if ((p = settings_opt_get("jid")) != NULL) {

      scr_log_print(LPRINT_NORMAL, "User JID: %s", p);

    } else if (settings_opt_get("username")) {

      /* TODO: remove after 0.10.2/3 */

      scr_log_print(LPRINT_NORMAL, "** ERROR: The JID is missing, but "

                    "the variable 'username' is defined in your "

                    "configuration file.\n"

                    "** Please update your configuration file and set "

                    "the 'jid' variable.");

    }

    if (settings_opt_get("ssl_verify")) {  // Deprecated option

      /* TODO: remove after 0.10.2/3 */