/*

 * main.c

 *

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

 * Parts of this file come from Cabber <cabber@ajmacias.com>

 *

 * 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 "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"

#ifdef ENABLE_HGCSET

# include "hgcset.h"

#endif

#ifndef WAIT_ANY

# define WAIT_ANY -1

#endif

static unsigned int terminate_ui;

GMainLoop *main_loop = NULL;

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_TerminateCurses();

  // 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_LOGNORM|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 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("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("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;

}

gboolean mcabber_loop()

{

  keycode kcode;

  if (terminate_ui) {

    g_main_loop_quit(main_loop);

    return FALSE;

  }

  scr_DoUpdate();

  scr_Getch(&kcode);

  while (kcode.value != ERR) {

    process_key(kcode);

    scr_DoUpdate();

    scr_Getch(&kcode);

  }

  scr_CheckAutoAway(FALSE);

  if (update_roster)

    scr_DrawRoster();

  hk_mainloop();

  return TRUE;

}

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();

  /* Initialize charset */

  scr_InitLocaleCharSet();

  /* 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);

  optstring = settings_opt_get("tracelog_file");

  if (optstring)

    ut_InitDebug(settings_opt_get_int("tracelog_level"), optstring);

  /* If no password is stored, we ask for it before entering

     ncurses mode -- unless the username is unknown. */

  if (settings_opt_get("jid") && !settings_opt_get("password")) {

    const char *p;

    char *pwd;

    p = settings_opt_get("server");

    if (p)

      printf("Server: %s\n", p);

    p = settings_opt_get("jid");

    if (p)

      printf("User JID: %s\n", p);

    pwd = ask_password("Jabber password");

    settings_set(SETTINGS_TYPE_OPTION, "password", pwd);

    g_free(pwd);

  }

  /* Initialize PGP system

     We do it before ncurses initialization because we may need to request

     a passphrase. */

  if (settings_opt_get_int("pgp"))

    main_init_pgp();

  /* Initialize N-Curses */

  scr_LogPrint(LPRINT_DEBUG, "Initializing N-Curses...");

  scr_InitCurses();

  scr_DrawMainWindow(TRUE);

  optval   = (settings_opt_get_int("logging") > 0);

  optval2  = (settings_opt_get_int("load_logs") > 0);

  if (optval || optval2)

    hlog_enable(optval, settings_opt_get("logging_dir"), optval2);

#if defined(WITH_ENCHANT) || defined(WITH_ASPELL)

  /* Initialize spelling */

  if (settings_opt_get_int("spell_enable")) {

    spellcheck_init();

  }

#endif

  optstring = settings_opt_get("events_command");

  if (optstring)

    hk_ext_cmd_init(optstring);

  if (settings_opt_get_int("hide_offline_buddies") > 0) { // XXX Deprecated

    scr_RosterDisplay("ofdna");

    scr_LogPrint(LPRINT_LOGNORM,

                 "* Warning: 'hide_offline_buddies' is deprecated.");

  } else {

    optstring = settings_opt_get("roster_display_filter");

    if (optstring)

      scr_RosterDisplay(optstring);

    // Empty filter isn't allowed...

    if (!buddylist_get_filter())

      scr_RosterDisplay("*");

  }