#include <libintl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "ldminfo.h"
#include "ldmutils.h"
#include "ldmgreetercomm.h"
#include "plugin.h"

/*
 * get_userid
 *  Ask the greeter to check autologin or get username
 */
void get_userid(gchar **username) {
    gsize len, end;
    gchar *cmd;

    cmd = g_strconcat("prompt <b>", _("Username"), "</b>\nuserid\n", NULL);
    *username = ask_value_greeter(cmd);

    g_free(cmd);
}

/*
 * get_passwd
 */
void get_passwd(gchar **password) {
    gchar *cmd;

    cmd = g_strconcat("prompts <b>", _("Password"), "</b>\npasswd\n", NULL);
    *password = (gchar*) ask_value_greeter(cmd);

    set_message(g_strconcat("<b>", _("Verifying password.  Please wait."), "</b>", NULL));
    g_free(cmd);
}

/*
 * get_host
 */
void get_host(gchar **server) {
    gsize len, end;
    gchar *cmd = "hostname\n";
    gchar *host;

    host = (gchar*) ask_value_greeter(cmd);
    if(g_ascii_strncasecmp(host, "None", 4)) {
        if(*server)
            g_free(*server);
        *server = host;
    }
}

/*
 * get_language
 */
void get_language(gchar **language) {
    gchar *cmd = "language\n";
    gchar *lang;

    lang = (gchar*) ask_value_greeter(cmd);

    if(g_ascii_strncasecmp(lang, "None", 4)) {
        if(*language)
            g_free(*language);
        *language = lang;
        setenv("LDM_LANGUAGE", lang, 1);
    }
}

/*
 * get_session
 */
void get_session(gchar **session) {
    gchar *cmd = "session\n";
    gchar *sess;

    sess = (gchar*) ask_value_greeter(cmd);

    if (g_ascii_strncasecmp(sess, "None", 4)) {
        if (*session)
            g_free(*session);
        *session = sess;
    }
}

/*
 * get_Xsession
 *
 * Return to variable xsession the LDM_XSESSION info
 */
void __attribute__ ((visibility("default"))) get_Xsession(gchar **xsession, gconstpointer server) {
    ldminfo *curr_host = NULL;
    gchar *tmp_xsess = NULL;
    tmp_xsess = g_strdup(getenv("LDM_XSESSION"));

    if (!tmp_xsess || strlen(tmp_xsess) == 0) {
        curr_host = ldminfo_lookup(server);
        if(curr_host) {
            tmp_xsess = curr_host->xsession;
            if (!tmp_xsess || strlen(tmp_xsess) == 0) {
                tmp_xsess = g_strdup(getenv("LDM_DEFAULT_XSESSION"));
                if (!tmp_xsess || strlen(tmp_xsess) == 0)
                    die("ldm","no Xsession");
            }
        }
    }

    *xsession = tmp_xsess;
}

/*
 * set_session_env
 *
 * Set LDM_SESSION and LDM_XSESSION env variable with param
 * Then run xsession script
 */
void __attribute__ ((visibility("default"))) set_session_env(gchar *xsession, gchar *session) {
    if (g_strcmp0(session,"default") != 0 || getenv("LDM_SESSION") == NULL) {
        log_entry("ldm",7,"Export LDM_SESSION to environment: \"%s\"", session);
        setenv("LDM_SESSION", session, 1);
    }
    else {
        log_entry("ldm",7,"Using existing LDM_SESSION from environment: \"%s\"", getenv("LDM_SESSION"));
    }

    setenv("LDM_XSESSION", xsession, 1);

    /* Starting xsession script */
    rc_files("xsession");
}

/*
 * get_ltsp_cfg
 *  Check for LTSP-Cluster. If true, contact loadbalancer for an IP
 */
void __attribute__ ((visibility("default"))) get_ltsp_cfg(gchar **server) {
    /* Check for LTSP-Cluster, if true, contact the loadbalancer for an IP */
    if (access(LTSP_CLUSTER_CONFILE, F_OK) == 0) {
        FILE *fp;
        fp = popen("getltscfg-cluster -l ldm", "r");
        if (fp != NULL) {
            log_entry("ltsp-cluster",6,"IP before load-balancing: %s", *server);
            if (fgets(*server, PATH_MAX, fp) == NULL)
                log_entry("ltsp-cluster",4,"failed to get an IP from the load-balancer");

            log_entry("ltsp-cluster",6,"IP after loadbalancing: %s", *server);
        }
        if (pclose(fp) == -1)
            log_entry("ltsp-cluster",3,"load-balancing failed");
    }
}