//! Unix-specific system calls and utilities.

use std::intrinsics;

/// File access modes.

pub record OpenFlags(i64);

/// Open file for reading only.

pub const O_RDONLY: OpenFlags = OpenFlags(0);

/// Open file for writing only.

pub const O_WRONLY: OpenFlags = OpenFlags(1);

/// Open file for reading and writing.

pub const O_RDWR: OpenFlags = OpenFlags(2);

/// Create file if it doesn't exist.

pub const O_CREAT: OpenFlags = OpenFlags(64);

/// Truncate file to zero length.

pub const O_TRUNC: OpenFlags = OpenFlags(512);

/// Standard file descriptor for stdin.

pub const STDIN: i64 = 0;

/// Standard file descriptor for stdout.

pub const STDOUT: i64 = 1;

/// Standard file descriptor for stderr.

pub const STDERR: i64 = 2;

/// Special value representing current working directory for `openat()`.

const AT_FDCWD: i64 = -100;

/// Opens a file at the given path and returns a file descriptor.

/// Returns a negative value on error.

pub fn open(path: *[u8], flags: OpenFlags) -> i64 {

    return intrinsics::ecall(56, AT_FDCWD, path.ptr as i64, *flags, 0);

}

/// Opens a file at the given path with mode, returns a file descriptor.

pub fn openOpts(path: *[u8], flags: OpenFlags, mode: i64) -> i64 {

    return intrinsics::ecall(56, AT_FDCWD, path.ptr as i64, *flags, mode);

}

/// Reads from a file descriptor into the provided buffer.

/// Returns the number of bytes read, or a negative value on error.

pub fn read(fd: i64, buf: *mut [u8]) -> i64 {

    return intrinsics::ecall(63, fd, buf.ptr as i64, buf.len as i64, 0);

}

/// Reads from a file descriptor until EOF or buffer is full.

/// Returns the total number of bytes read, or a negative value on error.

pub fn readToEnd(fd: i64, buf: *mut [u8]) -> i64 {

    let mut total: u32 = 0;

    while total < buf.len {

        let chunk = &mut buf[total..];

        let n = read(fd, chunk);

        if n < 0 {

            return n;

        }

        if n == 0 {

            break;

        }

        total += n as u32;

    }

    return total as i64;

}

/// Writes to a file descriptor from the provided buffer.

/// Returns the number of bytes written, or a negative value on error.

pub fn write(fd: i64, buf: *[u8]) -> i64 {

    return intrinsics::ecall(64, fd, buf.ptr as i64, buf.len as i64, 0);

}

/// Closes a file descriptor.

/// Returns `0` on success, or a negative value on error.

pub fn close(fd: i64) -> i64 {

    return intrinsics::ecall(57, fd, 0, 0, 0);

}

/// Reads the entire contents of a file at the given path into the provided buffer.

/// Returns a slice containing the data read, or `nil` on error.

pub fn readFile(path: *[u8], buf: *mut [u8]) -> ?*[u8] {

    let fd = open(path, O_RDONLY);

    if fd < 0 {

        return nil;

    }

    let n = readToEnd(fd, buf);

    if close(fd) < 0 {

        return nil;

    }

    if n < 0 {

        return nil;

    }

    return &buf[..n as u32];

}

/// Exit the current process with the given status code.

pub fn exit(status: i64) {

    intrinsics::ecall(93, status, 0, 0, 0);

}

/// Writes the entire contents of a buffer to a file at the given path.

/// Creates the file if it doesn't exist, truncates if it does.

/// Returns `true` on success.

pub fn writeFile(path: *[u8], data: *[u8]) -> bool {

    let flags = OpenFlags(*O_WRONLY | *O_CREAT | *O_TRUNC);

    let fd = openOpts(path, flags, 420); // 0644 in octal.

    if fd < 0 {

        return false;

    }

    let mut written: u32 = 0;

    while written < data.len {

        let chunk = &data[written..];

        let n = write(fd, chunk);

        if n <= 0 {

            close(fd);

            return false;

        }

        written += n as u32;

    }

    close(fd);

    return true;

}