#include <stdio.h>
#include <stdlib.h>
#include <esl.h>
#include <signal.h>
#include <sys/select.h>

#include <histedit.h>
static char prompt_str[512] = "";
static char hostname[512] = "";

char *prompt(EditLine * e)
{
	if (*prompt_str == '\0') {
        gethostname(hostname, sizeof(hostname));
        snprintf(prompt_str, sizeof(prompt_str), "freeswitch@%s> ", hostname);
    }

    return prompt_str;

}

static EditLine *el;
static History *myhistory;
static HistEvent ev;
static char *hfile = NULL;
static int running = 1;
static int thread_running = 0;
static esl_mutex_t *global_mutex;

static void handle_SIGINT(int sig)
{
	if (sig);
	return;
}

static const char* COLORS[] = { ESL_SEQ_DEFAULT_COLOR, ESL_SEQ_FRED, ESL_SEQ_FRED, 
								ESL_SEQ_FRED, ESL_SEQ_FMAGEN, ESL_SEQ_FCYAN, ESL_SEQ_FGREEN, ESL_SEQ_FYELLOW };


static void *msg_thread_run(esl_thread_t *me, void *obj)
{

	esl_handle_t *handle = (esl_handle_t *) obj;

	thread_running = 1;

	while(thread_running && handle->connected) {
		fd_set rfds, efds;
		struct timeval tv = { 0, 10 * 1000 };
		int max, activity, i = 0;
		
		esl_mutex_lock(global_mutex);
		FD_ZERO(&rfds);
		FD_ZERO(&efds);
		FD_SET(handle->sock, &rfds);
		FD_SET(handle->sock, &efds);
	
		max = handle->sock + 1;
		
		if ((activity = select(max, &rfds, NULL, &efds, &tv)) < 0) {
			esl_mutex_unlock(global_mutex);
			goto done;
		}
		if (activity && FD_ISSET(handle->sock, &rfds)) {
			if (esl_recv(handle)) {
				running = thread_running = 0;
				esl_mutex_unlock(global_mutex);
				esl_log(ESL_LOG_WARNING, "Disconnected.\n");
				goto done;
			}

			if (handle->last_event) {
				const char *type = esl_event_get_header(handle->last_event, "content-type");
				int known = 0;

				if (!esl_strlen_zero(type)) {
					if (!strcasecmp(type, "log/data")) {
						int level = 0, tchannel = 0;
						const char *lname = esl_event_get_header(handle->last_event, "log-level");
						const char *channel = esl_event_get_header(handle->last_event, "text-channel");
						const char *file = esl_event_get_header(handle->last_event, "log-file");
					
						if (channel) {
							tchannel = atoi(channel);
						}

						if (lname) {
							level = atoi(lname);
						}

						if (tchannel == 0 || (file && !strcmp(file, "switch_console.c"))) {
							printf("%s%s%s", COLORS[level], handle->last_event->body, ESL_SEQ_DEFAULT_COLOR);
						}
						known++;
					} else if (!strcasecmp(type, "text/disconnect-notice")) {
						running = thread_running = 0;
						known++;
					}
				}
				
				if (!known) {
					printf("INCOMING DATA [%s]\n%s", type, handle->last_event->body);
				}
			}

		}

		esl_mutex_unlock(global_mutex);
		usleep(1000);
	}

 done:

	thread_running = 0;
	esl_log(ESL_LOG_DEBUG, "Thread Done\n");

	return NULL;
}

static int process_command(esl_handle_t *handle, const char *cmd) 
{
	if (
		!strcasecmp(cmd, "exit") ||
		!strcasecmp(cmd, "quit") ||
		!strcasecmp(cmd, "bye")
		) {
		return -1;
	}

	if (
		!strncasecmp(cmd, "event", 5) || 
		!strncasecmp(cmd, "noevent", 7) ||
		!strncasecmp(cmd, "nixevent", 8) ||
		!strncasecmp(cmd, "log", 3) || 
		!strncasecmp(cmd, "nolog", 5) || 
		!strncasecmp(cmd, "filter", 6)
		) {
		esl_mutex_lock(global_mutex);
		esl_send_recv(handle, cmd);			
		printf("%s\n", handle->last_reply);
		esl_mutex_unlock(global_mutex);
		goto end;
	}
	
	printf("Unknown command [%s]\n", cmd);

 end:

	return 0;

}

typedef struct {
	char host[128];
	char pass[128];
	esl_port_t port;
} cli_profile_t;

static cli_profile_t profiles[128] = { 0 };
static int pcount;

int main(int argc, char *argv[])
{
	esl_handle_t handle = {0};
	int count;
	const char *line;
	char cmd_str[1024] = "";
	char hfile[512] = "/tmp/fs_cli_history";
	char cfile[512] = "/tmp/fs_cli_config";
	char *home = getenv("HOME");
	esl_config_t cfg;
	cli_profile_t *profile = &profiles[0];
	
	strncpy(profiles[0].host, "localhost", sizeof(profiles[0].host));
	strncpy(profiles[0].pass, "ClueCon", sizeof(profiles[0].pass));
	profiles[0].port = 8021;
	pcount = 1;
	
	if (home) {
		snprintf(hfile, sizeof(hfile), "%s/.fs_cli_history", home);
		snprintf(cfile, sizeof(cfile), "%s/.fs_cli_config", home);
	}
	
	esl_mutex_create(&global_mutex);

	signal(SIGINT, handle_SIGINT);
	gethostname(hostname, sizeof(hostname));

	handle.debug = 0;

	
	if (esl_config_open_file(&cfg, cfile)) {
		char *var, *val;
		while (esl_config_next_pair(&cfg, &var, &val)) {
			
		}
		esl_config_close_file(&cfg);
	}

	
	if (esl_connect(&handle, profile->host, profile->port, profile->pass)) {
		printf("Error Connecting [%s]\n", handle.err);
		return -1;
	}


	if (handle.debug) {
		esl_global_set_default_logger(7);
	}
		
	esl_thread_create_detached(msg_thread_run, &handle);
	
	el = el_init(__FILE__, stdout, stdout, stdout);
	el_set(el, EL_PROMPT, &prompt);
	el_set(el, EL_EDITOR, "emacs");
	myhistory = history_init();

	if (myhistory == 0) {
		fprintf(stderr, "history could not be initialized\n");
		goto done;
	}

	history(myhistory, &ev, H_SETSIZE, 800);
	el_set(el, EL_HIST, history, myhistory);
	history(myhistory, &ev, H_LOAD, hfile);
	

	snprintf(cmd_str, sizeof(cmd_str), "log info\n\n");
	esl_mutex_lock(global_mutex);
	esl_send_recv(&handle, cmd_str);
	esl_mutex_unlock(global_mutex);

	while (running) {

		line = el_gets(el, &count);
		
		if (count > 1) {
			if (!esl_strlen_zero(line)) {
				char *cmd = strdup(line);
				char *p;
				const LineInfo *lf = el_line(el);
				char *foo = (char *) lf->buffer;
				if ((p = strrchr(cmd, '\r')) || (p = strrchr(cmd, '\n'))) {
					*p = '\0';
				}
				assert(cmd != NULL);
				history(myhistory, &ev, H_ENTER, line);

				if (!strncasecmp(cmd, "...", 3)) {
					goto done;
				} else if (*cmd == '/') {
					if (process_command(&handle, cmd + 1)) {
						running = 0;
					}
				} else {
					snprintf(cmd_str, sizeof(cmd_str), "api %s\n\n", cmd);
					esl_mutex_lock(global_mutex);
					esl_send_recv(&handle, cmd_str);
					printf("%s\n", handle.last_event->body);
					esl_mutex_unlock(global_mutex);
				}

				el_deletestr(el, strlen(foo) + 1);
				memset(foo, 0, strlen(foo));
				free(cmd);
			}
		}

		usleep(1000);

	}


 done:
	
	history(myhistory, &ev, H_SAVE, hfile);

	/* Clean up our memory */
	history_end(myhistory);
	el_end(el);

	esl_disconnect(&handle);
	
	thread_running = 0;

	esl_mutex_destroy(&global_mutex);

	return 0;
}