#include <Library/UefiLib.h>
#include <Library/DebugLib.h>
#include <Library/ShellLib.h>
#include <Library/MemoryAllocationLib.h>
#include <Library/PrintLib.h>
#include <Library/UefiBootServicesTableLib.h>

#include "dir.h"
#include "fs.h"

#include "menu-bar.h"
#include "command-bar.h"
#include "dialog-box.h"
#include "panel.h"
#include "filemanager.h"
#include "cmds.h"
#include "actions.h"

#define MENUBAR (fm_ctx.menubar)
#define CMDBAR (fm_ctx.cmdbar)
#define PANEL (fm_ctx.curpanel)
#define LEFT_PANEL (fm_ctx.left)
#define RIGHT_PANEL (fm_ctx.right)
#define UPANEL ((PANEL == LEFT_PANEL) ? RIGHT_PANEL : LEFT_PANEL)

#define FILECOUNT_LENGTH 20
STATIC CONST CHAR16 cp_title[] = L" Copy ";
STATIC CONST CHAR16 cp_label[] = L"Copy %u files/directories to:";
STATIC CONST CHAR16 mv_title[] = L" Move ";
STATIC CONST CHAR16 mv_label[] = L"Move %u files/directories to:";
STATIC CONST CHAR16 rm_title[] = L" Delete ";
STATIC CONST CHAR16 rm_label[] = L"Delete %u files/directories?";
STATIC CONST CHAR16 mkdir_title[] = L" Create a new directory ";
STATIC CONST CHAR16 mkdir_label[] = L"Enter directory name:";

STATIC EFI_STATUS shell_exec2(CONST CHAR16 *farg,
	CONST CHAR16 *sarg, EFI_STATUS *cmd_status)
{
	EFI_STATUS status;
	CHAR16 *cmd_line = NULL;
	UINTN size = 0;

	StrnCatGrow(&cmd_line, &size, farg, 0);
	StrnCatGrow(&cmd_line, &size, sarg, 0);
	status = ShellExecute(&gImageHandle, cmd_line, FALSE, NULL, cmd_status);
	SHELL_FREE_NON_NULL(cmd_line);
	return status;
}

STATIC VOID redraw()
{
	panel_refresh(LEFT_PANEL);
	panel_refresh(RIGHT_PANEL);
	menubar_refresh(MENUBAR);
	cmdbar_refresh(CMDBAR);
}

BOOLEAN jump_up(VOID)
{
	panel_move_cursor(PANEL, PANEL->curline - 1);
	return TRUE;
}

BOOLEAN jump_down(VOID)
{
	panel_move_cursor(PANEL, PANEL->curline + 1);
	return TRUE;
}

BOOLEAN change_panel(VOID)
{
	panel_set_active(PANEL, FALSE);
	PANEL = UPANEL;
	panel_set_active(PANEL, TRUE);
	return TRUE;
}

BOOLEAN execute(VOID)
{
	EFI_SHELL_FILE_INFO *file;
	EFI_STATUS cmd_status;

	if(!PANEL->cwd)
		return panel_cd_to(PANEL, PANEL->fsa->full_name[PANEL->curline-1]);
	
	file = dirl_getn(PANEL->dirs, PANEL->curline);
	if((file->Info->Attribute & EFI_FILE_DIRECTORY) != 0)
		return panel_cd_to(PANEL, file->FullName);

	shell_exec2(L"", file->FullName, &cmd_status);
	redraw();
	return TRUE;
}

BOOLEAN mark(VOID)
{
	panel_mark_file(PANEL, PANEL->curline);
	return TRUE;
}

BOOLEAN edit(VOID)
{
	EFI_SHELL_FILE_INFO *file;
	EFI_STATUS cmd_status;

	if(!PANEL->cwd)
		return FALSE;

	file = dirl_getn(PANEL->dirs, PANEL->curline);
	if((file->Info->Attribute & EFI_FILE_DIRECTORY) != 0)
		return FALSE;

	shell_exec2(L"edit ", file->FullName, &cmd_status);
	redraw();
	return TRUE;
}

BOOLEAN hexedit(VOID)
{
	EFI_SHELL_FILE_INFO *file;
	EFI_STATUS cmd_status;

	if(!PANEL->cwd)
		return FALSE;

	file = dirl_getn(PANEL->dirs, PANEL->curline);
	if((file->Info->Attribute & EFI_FILE_DIRECTORY) != 0)
		return FALSE;

	shell_exec2(L"hexedit ", file->FullName, &cmd_status);
	redraw();
	return TRUE;
}

BOOLEAN cp(VOID)
{
	struct dbox_ctx *dbox;
	UINTN i, size, line = PANEL->curline;
	CHAR16 *label;
	EFI_SHELL_FILE_INFO *file;
	BOOLEAN status_op;

	if(!PANEL->cwd)
		return FALSE;

	if(PANEL->marked < 1)
		return FALSE;

	size = (StrLen(cp_label) + FILECOUNT_LENGTH + 1) * sizeof(CHAR16);
	label = AllocatePool(size);
	if(!label)
		return FALSE;

	UnicodeSPrint(label, size, cp_label, PANEL->marked);
	dbox = dbox_alloc(fm_ctx.scr, cp_title, label, TRUE, UPANEL->cwd);

	dbox_refresh(dbox);
	status_op = dbox_handle(dbox);
	if(status_op) {
		for(i = 1; i <= PANEL->dirs->len; i++) {
			if(PANEL->dirs->marked[i-1] == FALSE)
				continue;

			file = dirl_getn(PANEL->dirs, i);
			copy_file(file->FullName, dbox->in->buffer); // NOT UPANEL->cwd!!!
		}
		panel_cd_to(PANEL, PANEL->cwd);
		panel_move_cursor(PANEL, (line > PANEL->dirs->len) ? PANEL->dirs->len : line);
	}
	
	redraw();
	FreePool(label);
	dbox_release(dbox);
	return TRUE;
}

BOOLEAN mv(VOID)
{
	struct dbox_ctx *dbox;
	UINTN i, size, line = PANEL->curline;
	CHAR16 *label;
	EFI_SHELL_FILE_INFO *file;
	EFI_STATUS status;
	BOOLEAN status_op;

	if(!PANEL->cwd)
		return FALSE;

	if(PANEL->marked < 1)
		return FALSE;

	size = (StrLen(mv_label) + FILECOUNT_LENGTH + 1) * sizeof(CHAR16);
	label = AllocatePool(size);
	if(!label)
		return FALSE;

	UnicodeSPrint(label, size, mv_label, PANEL->marked);
	dbox = dbox_alloc(fm_ctx.scr, mv_title, label, TRUE, UPANEL->cwd);

	dbox_refresh(dbox);
	status_op = dbox_handle(dbox);
	if(status_op) {
		for(i = 1; i <= PANEL->dirs->len; i++) {
			if(PANEL->dirs->marked[i-1] == FALSE)
				continue;

			file = dirl_getn(PANEL->dirs, i);
			status = copy_file(file->FullName, UPANEL->cwd); // NOT UPANEL->cwd!!!
			if(!EFI_ERROR(status))
				delete_file(file);
		}
		panel_cd_to(PANEL, PANEL->cwd);
		panel_move_cursor(PANEL, (line > PANEL->dirs->len) ? PANEL->dirs->len : line);
	}
	
	redraw();
	FreePool(label);
	dbox_release(dbox);
	return TRUE;
}

BOOLEAN mkdir(VOID)
{
	struct dbox_ctx *dbox;
	UINTN line = PANEL->curline;
	EFI_STATUS status;
	BOOLEAN status_op;

	if(!PANEL->cwd)
		return FALSE;

	dbox = dbox_alloc(fm_ctx.scr, mkdir_title, mkdir_label, TRUE, L"");
	if(!dbox)
		return FALSE;

	dbox_refresh(dbox);
	status_op = dbox_handle(dbox);
	if(status_op) {
		status = make_directory(dbox->in->buffer);
		if(status == EFI_SUCCESS) {
			panel_cd_to(PANEL, PANEL->cwd);
			panel_move_cursor(PANEL, line);
		}
	}

	redraw();
	dbox_release(dbox);
	return TRUE;
}

BOOLEAN rm(VOID)
{
	struct dbox_ctx *dbox;
	UINTN i, size, line = PANEL->curline;
	CHAR16 *label;
	EFI_SHELL_FILE_INFO *file;
	BOOLEAN status_op;

	if(!PANEL->cwd)
		return FALSE;

	if(PANEL->marked < 1)
		return FALSE;

	size = (StrLen(rm_label) + FILECOUNT_LENGTH + 1) * sizeof(CHAR16);
	label = AllocatePool(size);
	if(!label)
		return FALSE;

	UnicodeSPrint(label, size, rm_label, PANEL->marked);
	dbox = dbox_alloc(fm_ctx.scr, rm_title, label, FALSE, L"");

	dbox_refresh(dbox);
	status_op = dbox_handle(dbox);
	if(status_op) {
		for(i = 1; i <= PANEL->dirs->len; i++) {
			if(PANEL->dirs->marked[i-1] == FALSE)
				continue;

			file = dirl_getn(PANEL->dirs, i);
			delete_file(file);
		}
		panel_cd_to(PANEL, PANEL->cwd);
		panel_move_cursor(PANEL, (line > PANEL->dirs->len) ? PANEL->dirs->len : line);
	}
	
	redraw();
	FreePool(label);
	dbox_release(dbox);
	return TRUE;
}

BOOLEAN show_filesystems(VOID)
{
	return panel_cd_to(PANEL, NULL);
}

BOOLEAN do_nothing(VOID)
{
	return TRUE;
}

BOOLEAN quit(VOID)
{
	fm_ctx.flag_run = FALSE;
	return TRUE;
}