#include <kernel/mem/alloc.h>
#include <kernel/mem/virt.h>
#include <kernel/panic.h>
#include <kernel/proc.h>
#include <kernel/vfs/request.h>
#include <kernel/vfs/root.h>

int vfs_request_create(struct vfs_request req_) {
	struct vfs_request *req;
	assert(process_current->state == PS_RUNNING);
	process_current->state = PS_WAITS4FS;
	process_current->waits4fs.queue_next = NULL;

	// the request is owned by the caller
	process_current->waits4fs.req = req_;
	req = &process_current->waits4fs.req;

	if (!req->backend)
		return vfs_request_finish(req, -1);

	switch (req->backend->type) {
		case VFS_BACK_ROOT:
			return vfs_root_handler(req);
		case VFS_BACK_USER:
			if (req->backend->handler
					&& req->backend->handler->state == PS_WAITS4REQUEST)
			{
				vfs_request_accept(req);
			} else {
				// backend isn't ready yet, join the queue
				struct process **iter = &req->backend->queue;
				while (*iter != NULL)
					iter = &(*iter)->waits4fs.queue_next;
				*iter = process_current;
			}
			return -1; // isn't passed to the caller process anyways
		default:
			panic_invalid_state();
	}
}

int vfs_request_accept(struct vfs_request *req) {
	struct process *handler = req->backend->handler;
	struct fs_wait_response res = {0};
	int len;
	assert(handler);
	assert(handler->state == PS_WAITS4REQUEST);
	assert(!handler->handled_req);

	len = min(req->input.len, handler->awaited_req.max_len);

	// wouldn't it be kinda nice to have a fake kernel "process"?
	if (!virt_cpy(handler->pages, handler->awaited_req.buf, 
				req->input.kern ? NULL : req->caller->pages, req->input.buf, len))
		goto fail; // can't copy buffer

	res.len      = len;
	res.capacity = req->output.len;
	res.id       = req->id;
	res.offset   = req->offset;
	res.op       = req->type;

	if (!virt_cpy_to(handler->pages,
				handler->awaited_req.res, &res, sizeof res))
		goto fail; // can't copy response struct

	handler->state = PS_RUNNING;
	handler->handled_req = req;
	regs_savereturn(&handler->regs, 0);
	return 0;
fail:
	panic_unimplemented(); // TODO
}

int vfs_request_finish(struct vfs_request *req, int ret) {
	if (req->type == VFSOP_OPEN && ret >= 0) {
		// open() calls need special handling
		// we need to wrap the id returned by the VFS in a handle passed to
		// the client
		handle_t handle = process_find_handle(req->caller);
		if (handle < 0)
			panic_invalid_state(); // we check for free handles before the open() call
		req->caller->handles[handle] = (struct handle){
			.type = HANDLE_FILE,
			.file = {
				.backend = req->backend,
				.id = ret,
			},
		};
		ret = handle;
	}

	if (req->input.kern)  kfree(req->input.buf_kern);

	assert(req->caller->state = PS_WAITS4FS);
	req->caller->state = PS_RUNNING;
	regs_savereturn(&req->caller->regs, ret);
	return ret;
}