implement the input widget

1 files changed, 114 insertions, 0 deletions

diff --git a/Library/UefiShellUfmCommandLib/widget/input.c b/Library/UefiShellUfmCommandLib/widget/input.c
new file mode 100644
index 0000000..7531321
--- /dev/null
+++ b/Library/UefiShellUfmCommandLib/widget/input.c
@@ -0,0 +1,114 @@
+#include <Library/UefiLib.h>
+#include <Library/DebugLib.h>
+#include <Library/MemoryAllocationLib.h>
+
+#include "input.h"
+
+struct widget_input *input_alloc(struct screen *scr, INT32 x, INT32 y,
+	INT32 width, INT32 attr, CONST CHAR16 *def_text)
+{
+	struct widget_input *in;
+
+	ASSERT(scr != NULL);
+
+	in = AllocateZeroPool(sizeof(struct widget_input));
+	if(!in)
+		return NULL;
+
+	in->win = newwin(scr, width, 1, x, y);
+	if(!in->win) {
+		input_release(in);
+		return NULL;
+	}
+	wattrset(in->win, attr);
+	whline(in->win, 0, 0, 0, attr, 0);
+	mvwprintf(in->win, 0, 0, def_text);
+
+	in->point = 0;
+	in->max_size = width;
+	in->buf_len = 0;
+	in->buffer = AllocateZeroPool((in->max_size + 1) * sizeof(CHAR16));
+	if(!in->buffer) {
+		input_release(in);
+		return NULL;
+	}
+
+	return in;
+}
+
+VOID input_release(struct widget_input *in)
+{
+	ASSERT(in != NULL);
+
+	if(in->win)
+		delwin(in->win);
+	if(in->buffer)
+		FreePool(in->buffer);
+
+	FreePool(in);
+}
+
+VOID input_set_point(struct widget_input *in, INTN pos)
+{
+	in->point = pos;
+	if(pos < 0)
+		in->point = 0;
+	else if(pos > in->buf_len)
+		in->point = in->buf_len;
+
+	wmove(in->win, in->point, 0);
+	input_update(in);
+}
+
+VOID input_handle_char(struct widget_input *in, EFI_INPUT_KEY key)
+{
+	UINTN i;
+
+	if(key.ScanCode == SCAN_LEFT)
+		input_set_point(in, in->point - 1);
+	else if(key.ScanCode == SCAN_RIGHT)
+		input_set_point(in, in->point + 1);
+	else {
+		switch(key.UnicodeChar) {
+
+		case 0x0:
+			break;
+		case CHAR_BACKSPACE: {
+			if(in->buf_len == 0)
+				break;
+
+			for(i = in->point; i < in->buf_len; i++)
+				in->buffer[i - 1] = in->buffer[i];
+			in->buffer[i - 1] = CHAR_NULL;
+			mvwaddch(in->win, i - 1, 0, L' ', -1);
+			mvwprintf(in->win, in->point - 1, 0, in->buffer + in->point - 1);
+
+			in->buf_len--;
+			input_set_point(in, in->point - 1);
+		}
+		case CHAR_TAB: 
+		case CHAR_LINEFEED:
+		case CHAR_CARRIAGE_RETURN:
+			break;
+		default: {
+			if(in->buf_len >= (in->max_size-1))
+				break;
+				
+			for(i = in->max_size-1; i > in->point; i--)
+				in->buffer[i] = in->buffer[i - 1];
+			in->buffer[in->point] = key.UnicodeChar;
+			wprintf(in->win, in->buffer + in->point);
+
+			in->buf_len++;
+			input_set_point(in, in->point + 1);
+		}
+
+		}
+	}
+}
+
+VOID input_update(struct widget_input *in)
+{
+	wrefresh(in->win);
+}
+