#include <socks/tty.h>
#include <socks/printk.h>

static void putchar(struct device *tty, int c)
{
	struct tty_device *ttydev = TTY_DEVICE(tty);
	struct tty_driver *tty_driver = TTY_DRIVER(tty->dev_owner);
	struct tty_driver_ops *ops = tty_driver->tty_ops;

	if (!ops->tty_putc) {
		return;
	}

	if (tty_driver->tty_type == TTY_DRIVER_SIMPLE) {
		ops->tty_putc(tty, c, -1, -1, 0);
		return;
	}

	ops->tty_putc(tty, c, ttydev->tty_xcur, ttydev->tty_ycur, ttydev->tty_curattrib);
	ttydev->tty_xcur++;

	if (ttydev->tty_xcur >= ttydev->tty_xcells) {
		ttydev->tty_xcur = 0;
		ttydev->tty_ycur++;
	}

	if (ttydev->tty_ycur >= ttydev->tty_ycells) {
		if (ops->tty_scroll) {
			ops->tty_scroll(tty, TTY_SCROLL_DOWN, 1);
		} else {
			ttydev->tty_xcur = 0;
			ttydev->tty_ycur = 0;
		}
	}

	ops->tty_move_cursor(tty, ttydev->tty_xcur, ttydev->tty_ycur);
}

kern_status_t tty_read(struct device *tty, void *buf, size_t max, size_t *nr_read, socks_flags_t flags)
{
	printk("tty_read");
	return KERN_OK;
}

kern_status_t tty_write(struct device *tty, const void *buf, size_t len, size_t *nr_written, socks_flags_t flags)
{
	size_t r = 0;
	const char *s = buf;
	for (size_t i = 0; i < len; i++) {
		putchar(tty, s[i]);
		r++;
	}

	*nr_written = r;
	return KERN_OK;
}