/* vim: tabstop=4 shiftwidth=4 noexpandtab
 * This file is part of ToaruOS and is released under the terms
 * of the NCSA / University of Illinois License - see LICENSE.md
 * Copyright (C) 2014-2017 K. Lange
 *
 * gsudo - graphical implementation of sudo
 *
 * probably even less secure than the original
 *
 */

#include <stdio.h>
#include <stdint.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>

#include <toaru/auth.h>
#include <toaru/yutani.h>
#include <toaru/graphics.h>
#include <toaru/sdf.h>

#define main __main_unused
#include "sudo.c"
#undef main

#define FONT_SIZE_MAIN 20
#define FONT_SIZE_PASSWD 25

static yutani_t * yctx;
static gfx_context_t * ctx;
static yutani_window_t * window;

static int graphical_callback(char * username, char * password, int fails, char * argv[]) {
	int i = 0;
	while (1) {
		draw_fill(ctx, rgba(0,0,0,200));
		int h = ctx->height-1;
		int w = ctx->width-1;
		draw_line(ctx, 0,0,0,h, rgb(255,0,0));
		draw_line(ctx, w,w,0,h, rgb(255,0,0));
		draw_line(ctx, 0,w,0,0, rgb(255,0,0));
		draw_line(ctx, 0,w,h,h, rgb(255,0,0));

		char prompt_message[512];
		sprintf(prompt_message, "Enter password for '%s'", username);
		draw_sdf_string(ctx, (ctx->width - draw_sdf_string_width(prompt_message, FONT_SIZE_MAIN, SDF_FONT_THIN)) / 2, 20, prompt_message, FONT_SIZE_MAIN, rgb(255, 255, 255), SDF_FONT_THIN);

		sprintf(prompt_message, "requested by %s", argv[1]);
		draw_sdf_string(ctx, (ctx->width - draw_sdf_string_width(prompt_message, FONT_SIZE_MAIN, SDF_FONT_THIN)) / 2, 150, prompt_message, FONT_SIZE_MAIN, rgb(255, 255, 255), SDF_FONT_THIN);

		if (fails) {
			sprintf(prompt_message, "Try again. %d failures.", fails);
			draw_sdf_string(ctx, (ctx->width - draw_sdf_string_width(prompt_message, FONT_SIZE_MAIN, SDF_FONT_THIN)) / 2, 50, prompt_message, FONT_SIZE_MAIN, rgb(255, 0, 0), SDF_FONT_THIN);

		}

		char password_circles[512] = {0};;
		strcpy(password_circles, "");
		for (unsigned int i = 0; i < strlen(password) && i < 512/4; ++i) {
			strcat(password_circles, "\007");
		}
		draw_sdf_string(ctx, (ctx->width - draw_sdf_string_width(password_circles, FONT_SIZE_PASSWD, SDF_FONT_THIN)) / 2, 80, password_circles, FONT_SIZE_PASSWD, rgb(255, 255, 255), SDF_FONT_THIN);

		flip(ctx);
		yutani_flip(yctx, window);

		yutani_msg_t * msg = yutani_poll(yctx);

		switch (msg->type) {
			case YUTANI_MSG_KEY_EVENT:
				{
					struct yutani_msg_key_event * ke = (void*)msg->data;
					if (ke->event.action == KEY_ACTION_DOWN) {
						if (ke->event.keycode == KEY_ESCAPE) {
							return 1;
						}
						if (ke->event.keycode == '\n') {
							return 0;
						} else  if (ke->event.key == 8) {
							if (i > 0) i--;
							password[i] = '\0';
						} else if (ke->event.key) {
							password[i] = ke->event.key;
							password[i+1] = '\0';
							i++;
						}
					}
				}
				break;
			case YUTANI_MSG_WINDOW_CLOSE:
			case YUTANI_MSG_SESSION_END:
				return 1;
		}
	}
}

int main(int argc, char ** argv) {

	if (argc < 2) {
		return 1;
	}

	yctx = yutani_init();

	if (!yctx) {
		fprintf(stderr, "%s: could not connect to compositor\n", argv[0]);
		return 1;
	}

	int width = 300;
	int height = 200;

	int left = (yctx->display_width - width) / 2;
	int top = (yctx->display_height - height) / 2;

	window = yutani_window_create(yctx, width, height);
	yutani_window_move(yctx, window, left, top);
	yutani_window_advertise(yctx, window, "gsudo");

	ctx = init_graphics_yutani_double_buffer(window);

	return sudo_loop(graphical_callback, argv);
}