#include "../lib/io/io.h"
#include "../lib/fb/fb.h"
#include "../lib/tctl/tctl.h"
#include "../lib/aec/aec.h"
#include "../lib/ff/ff.h"
#include "../lib/arg/arg.h"
#include "../lib/cstr/cstr.h"

u8 centerx = 0;
u8 centery = 0;
i64 width = -1;
i64 height = -1;
u8 fill = 0;
u8 only_print = 0;


void set_only_print()
{
	only_print = 1;
}


void set_width(const char *buf)
{
	width = cstr_to_i64(buf);
}


void set_height(const char *buf)
{
	height = cstr_to_i64(buf);
}


void set_fill()
{
	fill = 1;
}


void set_centerx()
{
	centerx = 1;
}


void set_centery()
{
	centery = 1;
}


void draw_image(canvas_t canvas, image_t *img)
{
	canvas_pixel_t p;
	u64 offset_x = 0;
	u64 offset_y = 0;
	float scalex = 1;
	float scaley = 1;

	if (width != -1) {
		scalex = width;
		scalex /= img->width;
	}

	if (height != -1) {
		scaley = height;
		scaley /= img->height;
	}

	if (centerx) {
		offset_x = (canvas.width - img->width) / 2;
	}

	if (centery) {
		offset_y = (canvas.height - img->height) / 2;
	}

	for (u32 y = 0; y < img->height; ++y) {
		for (u32 x = 0; x < img->width; ++x) {
			p = image_pixel_to_canvas_pixel(img->data[x + y * img->width]);
			put_pixel(canvas, (x * scalex) + offset_x, (y * scaley) + offset_y, p);
		}
	}
}


void black_clear(canvas_t canvas)
{
	canvas_pixel_t p = { 0x00, 0x00, 0x00 };
	for (int x = 0; x < canvas.width; ++x) {
		for (int y = 0; y < canvas.height; ++y) {
			put_pixel(canvas, x, y, p);
		}
	}
}


int main(int argc, const char **argv)
{
	arg_register_flag("-p", &set_only_print);
	arg_register_flag("-c", &set_centerx);
	arg_register_flag("-z", &set_centery);
	arg_register_arg("-h", &set_height);
	arg_register_arg("-w", &set_width);
	arg_parse_arg(argc, argv);

	canvas_t canvas = open_framebuffer();
	termios_t term;
	termios_t orig;

	image_t *img = new_image_from_fd(STDIN_FD);

	tcgetattr(STDOUT_FD, &term);
	orig = term;
	setcanonical(&term, 0);
	setecho(&term, 0);
	cursor_enabled(0);
	tcsetattr(STDOUT_FD, &term);

	if (!img) {
		free_image(img);
		close_framebuffer(canvas);
		cursor_enabled(1);
		tcsetattr(STDOUT_FD, &orig);
		wf(STDERR_FD, "error: cannot read image\n");
		return -1;
	}

	if (width == -1) {
		width = img->width;
	}

	if (height == -1) {
		height = img->width;
	}

	char key = 0;

	while (key != 'q') {
		draw_image(canvas, img);
		if (!only_print)
			read(STDERR_FD, &key, 1);
		else
			break;
	}

	if (!only_print)
		black_clear(canvas);

	free_image(img);
	close_framebuffer(canvas);
	cursor_enabled(1);
	tcsetattr(STDOUT_FD, &orig);

	if (!only_print) {
		move_cursor(1, 1);
		clear_screen();
	} else {
		move_cursor(0, height / 14);
	}

	return 0;
}