#include <kernel/syscalls.h>
#include <kernel/file.h>
#include <kernel/abi.h>
#include <kernel/alloc.h>
#include <errno.h>
#include <logging.h>

#define O_CREAT	0100
#define O_EXCL 0200
#define O_NOCTTY 0400
#define O_TRUNC	01000
#define O_APPEND 02000
#define O_NONBLOCK 04000
#define O_DSYNC 010000
#define O_ASYNC 020000
#define O_LARGEFILE 0100000
#define O_DIRECTORY 0200000
#define O_NOFOLLOW 0400000
#define O_NOATIME 01000000
#define O_CLOEXEC 02000000
#define O_SYNC 04010000
#define O_RSYNC 04010000
#define O_TMPFILE 020000000

syscallret_t syscall_openat(context_t *context, int dirfd, const char *path, int flags, mode_t mode) {
	syscallret_t ret = {
		.ret = -1,
		.errno = 0
	};

	// transform O_RDONLY, O_WRONLY, O_RDWR into FILE_READ and FILE_WRITE
	++flags;

	vnode_t *dirnode = NULL;
	file_t *dirfile;

	size_t pathsize = strlen(path); // TODO u_strlen
	char *pathbuf = alloc(pathsize + 1);
	strcpy(pathbuf, path); // TODO u_strcpy

	if (*pathbuf == '/') {
		dirnode = sched_getroot();
		dirfile = NULL;
	} else if (dirfd == AT_FDCWD) {
		dirnode = sched_getcwd();
		dirfile = NULL;
	} else {
		dirfile = fd_get(dirfd);
		if (dirfile == NULL) {
			ret.errno = EBADF;
			goto cleanup;
		}

		dirnode = dirfile->vnode;
		if (dirnode->type != V_TYPE_DIR) {
			ret.errno = ENOTDIR;
			goto cleanup;
		}
	}

	// TODO pass CLOEXEC to fd
	file_t *newfile = NULL;
	int newfd;
	ret.errno = fd_new(0, &newfile, &newfd);
	if (ret.errno)
		goto cleanup;

	vnode_t *vnode = NULL;
	ret.errno = vfs_open(dirnode, pathbuf, fileflagstovnodeflags(flags), &vnode);

	if (ret.errno == ENOENT && (flags & O_CREAT)) {
		vattr_t attr = {
			.mode = mode,
			.gid = _cpu()->thread->proc->cred.gid,
			.uid = _cpu()->thread->proc->cred.uid
		};

		ret.errno = vfs_create(dirnode, pathbuf, &attr, V_TYPE_REGULAR, &vnode);
	}

	if (ret.errno)
		goto cleanup;

	if ((flags & O_DIRECTORY) && vnode->type != V_TYPE_DIR) {
		ret.errno = ENOTDIR;
		goto cleanup;
	}

	vattr_t attr;
	ret.errno = VOP_GETATTR(vnode, &attr, &_cpu()->thread->proc->cred);
	if (ret.errno)
		goto cleanup;

	// node refcount is already increased by open
	newfile->vnode = vnode;
	newfile->flags = flags;
	newfile->offset = 0;
	newfile->mode = attr.mode;

	ret.errno = 0;
	ret.ret = newfd;

	cleanup:
	if (vnode && ret.errno) {
		vfs_close(vnode, fileflagstovnodeflags(flags));
		VOP_RELEASE(vnode);
	}

	if (newfile) {
		if (ret.errno) {
			__assert(fd_close(newfd) == 0);
		} else {
			fd_release(newfile);
		}
	}

	if (pathbuf)
		free(pathbuf);

	if (dirfile) {
		fd_release(dirfile);
	} else if (dirnode){
		VOP_RELEASE(dirnode);
	}

	return ret;
}