#include "cmdlinelib.h"
#include <gpxe/command.h>
#include <gpxe/tables.h>
#include <console.h>
#include <malloc.h>
#include <string.h>
#include <stdarg.h>

static struct command cmd_start[0] __table_start ( commands );
static struct command cmd_end[0] __table_end ( commands );

void cmdl_setgetchar(cmd_line* cmd, cmdl_getchar_t in)
{
	cmd->getchar = in;
}

void cmdl_setputchar(cmd_line* cmd, cmdl_putchar_t in)
{
	cmd->putchar = in;
}

void cmdl_setprintf(cmd_line* cmd, cmdl_printf_t in)
{
	cmd->printf = in;
}
      
int cmdl_printf(cmd_line* cmd, const char *format, ...)
{
	int ret;
        char string[CMDL_BUFFER_SIZE];
        va_list ap;

        va_start(ap, format);
        ret = vsprintf(string, format, ap);
        cmdl_addstr(cmd, string);
        va_end(ap);
        return ret;
}

void cmdl_addstr(cmd_line* cmd, char* str)
{
	unsigned int i;
	for(i = 0; i < strlen(str); i++){
		cmdl_addchar(cmd, str[i]);
	}
}

void cmdl_setpropmt(cmd_line* cmd, char prompt[CMDL_PROMPT_SIZE])
{
	if(cmdl_check(cmd) && prompt != NULL){
		strncat(cmd->prompt, prompt, CMDL_PROMPT_SIZE);
	}
}

char *cmdl_getprompt(cmd_line* cmd)
{
	if(cmdl_check(cmd)){
		return cmd->prompt;
	}else{
		return "";
	}
}

char* cmdl_getbuffer(cmd_line* cmd){
	if(cmdl_check(cmd)){
		return cmd->buffer;
	}else{
		return "";
	}
}

static int cmdl_exit = 0;

static int exit_exec ( int argc __unused, char **argv __unused ) {
	cmdl_exit = 1;
	return 0;
}

struct command exit_command __command = {
	.name = "exit",
	.exec = exit_exec,
};

void cmdl_enterloop(cmd_line* cmd)
{
	cmdl_exit = 0;
	do {
		if(cmd->refresh){
			cmd->printf("%s %s", cmd->prompt, cmd->buffer);
			cmd->refresh = 0;
		}
//		cmd->printf("Got %d\n", cmd->getchar());
		cmdl_parsechar(cmd, cmd->getchar());
	} while ( ! cmdl_exit );
}

void cmdl_addreplace(cmd_line* cmd, char in)
{
	if(cmd->cursor < CMDL_BUFFER_SIZE - 2){
		cmd->buffer[cmd->cursor] = in;
		cmd->cursor++;
		cmd->putchar((int)in);
	}
}

void cmdl_addinsert(cmd_line* cmd, char in)
{
	int i;
	int to;
	if(cmd->cursor < CMDL_BUFFER_SIZE - 2 && cmd->cursor >= 0){
		if(strlen(cmd->buffer) < CMDL_BUFFER_SIZE - 2){
			to = strlen(cmd->buffer);
		}else{
			to = CMDL_BUFFER_SIZE - 2;
		}
			for(i=to; i > cmd->cursor; i--){
				cmd->buffer[i] = cmd->buffer[i-1];
			}
			cmd->buffer[cmd->cursor] = in;

			for(i=cmd->cursor; i < to; i++){
				cmd->putchar(cmd->buffer[i]);
			}
			
			for(i=cmd->cursor; i < to - 1; i++){
				cmd->putchar(CMDLK_BS);
			}
			cmd->cursor++;
			//cmdl_movecursor(cmd, CMDL_RIGHT);
	}
}

void cmdl_addchar(cmd_line* cmd, char in){
	if(cmd->insert){
		cmdl_addinsert(cmd, in);
	}else{
		cmdl_addreplace(cmd, in);
	}
}

void cmdl_parsechar(cmd_line* cmd, char in)
{
	if(cmdl_check(cmd)){
		if(in >= 32){
			cmdl_addchar(cmd, in);
		}else{
			switch(in){
				case CMDLK_BS:
					if(cmdl_movecursor(cmd, CMDL_LEFT)){
						cmdl_del(cmd);
					}
					break;

				case CMDLK_RETURN:
					cmd->putchar('\n');
					system ( cmd->buffer );
					cmdl_clearbuffer(cmd);
					cmd->refresh = 1;
					break;

				case CMDLK_BW:
					cmdl_movecursor(cmd, CMDL_LEFT);
					break;

				case CMDLK_FW:
					//cmdl_movecursor(cmd, CMDL_RIGHT);
					break;
				
				case CMDLK_TAB:
					cmdl_tabcomplete(cmd);
					break;

			}
		}
	}
}

void cmdl_tabcomplete(cmd_line *cmd)
{
	struct command *ccmd;
	int count=0;
	char* result[CMDL_MAX_TAB_COMPLETE_RESULT];

	for ( ccmd = cmd_start ; ccmd < cmd_end ; ccmd++ ) {
		if(!strncmp(ccmd->name, cmd->buffer, strlen(cmd->buffer))){
			if(count <= CMDL_MAX_TAB_COMPLETE_RESULT){
				result[count++] = (char*)(ccmd->name);
			}
		}
	}
	

	if( count == 1 ){
		cmdl_addstr(cmd, (char*)(result[0] + strlen(cmd->buffer)));
		cmd->tabstate = 0;
		cmdl_addchar(cmd, ' ');
	} else if( count > 1 ) {
		int i, i2, minlen=CMDL_BUFFER_SIZE, same=1;
		char last;

		for(i = 0; i < count; i ++) {
			if(minlen > (int)strlen( result[i] ) ){
				minlen = strlen(result[i]);
			}
		
		}
		if((int)strlen(cmd->buffer) < minlen){
			for(i = strlen(cmd->buffer); i < minlen; i++){
				last = result[0][i];
				for(i2 = 1; i2 < count; i2 ++) {
					if(result[i2][i] != last){
						same = 0;
						break;
					}
				}
				if(same){
					cmdl_addchar(cmd, last);
				}
				
			}
		}
		cmd->tabstate++;
	}
	
	if(count > 1 && cmd->tabstate > 1){
		int i;
		cmd->tabstate = 0;
		cmd->refresh = 1;
		cmd->putchar('\n');
		for(i = 0; i < count; i ++){
			cmd->printf("%s\t", result[i]);
		}
		cmd->putchar('\n');
	}

	

}

void cmdl_clearbuffer(cmd_line* cmd)
{
	if(cmdl_check(cmd)){
		int i;
		cmd->cursor = 0;
		for(i=0; i < CMDL_BUFFER_SIZE; i++){
			cmd->buffer[i] = 0;
		}
	}
}

int cmdl_movecursor(cmd_line* cmd, int direction)
{
	if(cmdl_check(cmd)){
		switch(direction){
			case CMDL_LEFT:
				if(cmd->cursor > 0){
					cmd->cursor--;
					cmd->putchar(CMDLK_BS);
				}else{
					return 0;
				}
				break;
			case CMDL_RIGHT:
				if(cmd->cursor < CMDL_BUFFER_SIZE - 2){
					cmd->cursor++;
					cmd->putchar(' ');
				}else{
					return 0;
				}
				break;
		}
	}
	return 1;
}

void cmdl_del(cmd_line* cmd)
{
	if(cmdl_check(cmd) && cmd->cursor < CMDL_BUFFER_SIZE - 2 && cmd->cursor >= 0){
		int i;
		for(i = cmd->cursor; i < (int)strlen(cmd->buffer); i++){
			cmd->buffer[i] = cmd->buffer[i + 1];
			if(!cmd->buffer[i]){
				cmd->putchar(' ');
			}else{
				cmd->putchar(cmd->buffer[i]);
			}
		}
		for(i = cmd->cursor; i < (int)strlen(cmd->buffer) + 1; i++){
			cmd->putchar(CMDLK_BS);
		}
	}
}


int cmdl_check(cmd_line* cmd)
{
	if(
		cmd != NULL && 
		cmd->buffer != NULL &&
		cmd->prompt != NULL &&
		cmd->cursor >= 0 && 
		cmd->cursor < CMDL_BUFFER_SIZE - 1 &&
		cmd->buffer[CMDL_BUFFER_SIZE - 1] == 0 &&
		cmd->prompt[CMDL_PROMPT_SIZE - 1] == 0
	){
		return 1;
	}else{
		return 0;
	}
}

cmd_line* cmdl_create()
{
	cmd_line* this;
	int i;
	
	/* Initiate the command line */
	
	this = (cmd_line*)malloc(sizeof(cmd_line));
	
	if(this == NULL){
		return NULL;
	}
	

	/* Allocate output buffer */
	
	/*this->output = (char*)malloc(CMDL_OUTPUT_SIZE);
	if(this->output == NULL){
		free(this);
		return NULL;
	}*/
	
/*	for(i = 0; i < CMDL_OUTPUT_SIZE; i++){
		this->output[i] = 0;
	}*/

	/* Allocate command line buffer */
	
	this->buffer = (char*)malloc(CMDL_BUFFER_SIZE);
	if(this->buffer == NULL){
		free(this);
		return NULL;
	}
	
	for(i = 0; i < CMDL_BUFFER_SIZE; i++){
		this->buffer[i] = 0;
	}
	
	/* Allocate prompt buffer */
	
	this->prompt = (char*)malloc(CMDL_PROMPT_SIZE);
	if(this->prompt == NULL){
		free(this);
		return NULL;
	}
	
	for(i = 0; i < CMDL_PROMPT_SIZE; i++){
		this->prompt[i] = 0;
	}
	
	/* Initiate cursor position etc.*/
	
	this->cursor = 0;
	//this->has_output = 0;
	this->refresh = 1;
	this->tabstate = 0;
	this->insert = 0;

	/* set callbacks to NULL */

	this->getchar = NULL;
	this->putchar = NULL;
	this->printf = NULL;

	/* List the commands */

	struct command *cmd;

	printf ( "Available commands: ");
	for ( cmd = cmd_start ; cmd < cmd_end ; cmd++ ) {
		printf("%s ", cmd->name);
	}
	printf("\n\n");

	return this;
}

void cmdl_free(cmd_line* cmd)
{
	free(cmd);
}