#include <libk/stdio.h>
#include <libk/string.h>
#include <graphics.h>
#include <serial.h>

mutex_t stdio_lock;

void print_char(char c)
{
	if (main_fb.x * 8 >= main_fb.width) {
		main_fb.x = 0;
		main_fb.y++;
	}
	if (main_fb.y * 16 >= main_fb.height) {
		main_fb.x = 0;
		main_fb.y = 0;
	}
	if (c == '\n') {
		write_serial('\n');
		main_fb.x = 0;
		main_fb.y++;
		return;
	}
	if (c == '\b') {
		if (main_fb.x != 0) {
			main_fb.x--;
		}
		return;
	}
	fb_draw_character(main_fb, c, (int32_t)main_fb.x * 8,
			  (int32_t)main_fb.y * 16);
	main_fb.x++;

	write_serial(c);
}

void print_string(const char *s)
{
	size_t i;
	for (i = 0; i < strlen(s); i++) {
		print_char(s[i]);
	}
}

void print_int(uint64_t num)
{
	char a[100];
	itos(num, a);
	print_string(a);
}

void print_hex(uint64_t num)
{
	char a[100];
	itoh(num, a);
	print_string(a);
}

void printf(const char *s, ...)
{
	va_list list;
	va_start(list, s);

	vprintf(s, list);

	va_end(list);
}

void vprintf(const char *s, va_list list)
{
	lock(stdio_lock);
	size_t count = 0;
	size_t i;
	for (i = 0; i < strlen(s); i++)
		if (s[i] == '%')
			count++;

	for (i = 0; i < strlen(s); i++) {
		if (s[i] == '%') {
			i++;
			if (s[i] == 'c')
				print_char((char)va_arg(list, uint32_t));
			else if (s[i] == 's')
				print_string(va_arg(list, char *));
			else if (s[i] == 'd')
				print_int((uint64_t)va_arg(list, uint64_t));
			else if (s[i] == 'x')
				print_hex((uint64_t)va_arg(list, uint64_t));
			else {
				print_string("Wrong format using printf\n");
				return;
			}
		} else
			print_char(s[i]);
	}
	unlock(stdio_lock);
}