/*
* Copyright (c) 2017 Calvin Rose
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to
* deal in the Software without restriction, including without limitation the
* rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
* sell copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
* IN THE SOFTWARE.
*/


#include <dst/dst.h>
#include <stdio.h>

int dst_sys_print(DstValue *argv, int32_t argn) {
    int32_t i;
    for (i = 0; i < argn; ++i) {
        int32_t j, len;
        const uint8_t *vstr = dst_to_string(argv[i]);
        len = dst_string_length(vstr);
        for (j = 0; j < len; ++j) {
            putc(vstr[j], stdout);
        }
    }
    putc('\n', stdout);
    return 0;
}

int dst_sys_asm(DstValue *argv, int32_t argn) {
    DstAssembleOptions opts;
    DstAssembleResult res;
    if (argn < 1) {
        dst_vm_fiber->ret = dst_cstringv("expected assembly source");
        return 1;
    }
    opts.source = argv[0];
    opts.sourcemap = (argn >= 2 && dst_checktype(argv[1], DST_TUPLE))
        ? dst_unwrap_tuple(argv[1])
        : NULL;
    opts.flags = 0;
    res = dst_asm(opts);
    if (res.status == DST_ASSEMBLE_OK) {
        dst_vm_fiber->ret = dst_wrap_function(dst_asm_func(res));
        return 0;
    } else {
        dst_vm_fiber->ret = dst_wrap_string(res.error);
        return 1;
    }
}

int dst_sys_tuple(DstValue *argv, int32_t argn) {
    dst_vm_fiber->ret = dst_wrap_tuple(dst_tuple_n(argv, argn));
    return 0;
}

int dst_sys_array(DstValue *argv, int32_t argn) {
    DstArray *array = dst_array(argn);
    array->count = argn;
    memcpy(array->data, argv, argn * sizeof(DstValue));
    dst_vm_fiber->ret = dst_wrap_array(array);
    return 0;
}

int dst_sys_table(DstValue *argv, int32_t argn) {
    int32_t i;
    DstTable *table = dst_table(argn/2);
    if (argn & 1) {
        dst_vm_fiber->ret = dst_cstringv("expected even number of arguments");
        return 1;
    }
    for (i = 0; i < argn; i += 2) {
        dst_table_put(table, argv[i], argv[i + 1]);
    }
    dst_vm_fiber->ret = dst_wrap_table(table);
    return 0;
}

int dst_sys_struct(DstValue *argv, int32_t argn) {
    int32_t i;
    DstValue *st = dst_struct_begin(argn/2);
    if (argn & 1) {
        dst_vm_fiber->ret = dst_cstringv("expected even number of arguments");
        return 1;
    }
    for (i = 0; i < argn; i += 2) {
        dst_struct_put(st, argv[i], argv[i + 1]);
    }
    dst_vm_fiber->ret = dst_wrap_struct(dst_struct_end(st));
    return 0;
}

int dst_sys_get(DstValue *argv, int32_t argn) {
    int32_t i;
    DstValue ds;
    if (argn < 1) {
        dst_vm_fiber->ret = dst_cstringv("expected at least 1 argument");
        return 1;
    }
    ds = argv[0];
    for (i = 1; i < argn; i++) {
        ds = dst_get(ds, argv[i]);
        if (dst_checktype(ds, DST_NIL))
            break;
    }
    dst_vm_fiber->ret = ds;
    return 0;
}

int dst_sys_put(DstValue *argv, int32_t argn) {
    DstValue ds, key, value;
    if (argn < 3) {
        dst_vm_fiber->ret = dst_cstringv("expected at least 3 arguments");
        return 1;
    }
    if(dst_sys_get(argv, argn - 2))
        return 1;
    ds = dst_vm_fiber->ret;
    key = argv[argn - 2];
    value = argv[argn - 1];
    dst_put(ds, key, value);
    return 0;
}

const DstCFunction dst_vm_syscalls[256] = {
    dst_sys_print,
    dst_sys_asm,
    dst_sys_tuple,
    dst_sys_array,
    dst_sys_struct,
    dst_sys_table,
    dst_sys_get,
    dst_sys_put,
    NULL
};