/*
 *  Copyright (C) 2021-2024 Savoir-faire Linux Inc.
 *
 *  Author: Olivier Dion <olivier.dion@savoirfairelinux.com>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA.
 */

#pragma once

#include <cstdint>
#include <filesystem>

#include "logger.h"

static inline SCM
to_guile(bool b)
{
    return scm_from_bool(b);
}

static inline SCM
to_guile(const std::string& str)
{
    return scm_from_utf8_string(str.c_str());
}

static inline SCM
to_guile(uint8_t x)
{
    return scm_from_uint8(x);
}

static inline SCM
to_guile(uint16_t x)
{
    return scm_from_uint16(x);
}

static inline SCM
to_guile(uint32_t x)
{
    return scm_from_uint32(x);
}

static inline SCM
to_guile(uint64_t x)
{
    return scm_from_uint64(x);
}

static inline SCM
to_guile(int8_t x)
{
    return scm_from_int8(x);
}

static inline SCM
to_guile(int16_t x)
{
    return scm_from_int16(x);
}

static inline SCM
to_guile(int32_t x)
{
    return scm_from_int32(x);
}

static inline SCM
to_guile(int64_t x)
{
    return scm_from_int64(x);
}

static inline SCM
to_guile(double x)
{
    return scm_from_double(x);
}

/* Forward declarations since we call to_guile() recursively for containers */
template<typename T> static inline SCM to_guile(const std::vector<T>& values);
template<typename K, typename V> static inline SCM to_guile(const std::map<K, V>& map);

template<typename T>
static inline SCM
to_guile(const std::vector<T>& values)
{
    SCM vec = scm_c_make_vector(values.size(), SCM_UNDEFINED);

    for (size_t i = 0; i < values.size(); ++i) {
        SCM_SIMPLE_VECTOR_SET(vec, i, to_guile(values[i]));
    }

    return vec;
}

template<typename K, typename V>
static inline SCM
to_guile(const std::map<K, V>& map)
{
    SCM assoc = SCM_EOL;

    for (auto const& [key, value] : map) {
        SCM pair = scm_cons(to_guile(key), to_guile(value));
        assoc = scm_cons(pair, assoc);
    }

    return assoc;
}

template<typename... Args>
static inline SCM
pack_to_guile(Args... args)
{
    SCM lst = SCM_EOL;
    std::vector<SCM> values = {to_guile(args)...};

    while (values.size()) {
        lst = scm_cons(values.back(), lst);
        values.pop_back();
    }

    return lst;
}

template<typename... Args>
static inline SCM
apply_to_guile(SCM body_proc, Args... args)
{
    if (scm_is_false(scm_procedure_p(body_proc))) {
        scm_wrong_type_arg_msg("apply_to_guile", 0, body_proc, "procedure");
    }

    SCM arglst = pack_to_guile(args...);

    return scm_apply_0(body_proc, arglst);
}

struct from_guile
{
    SCM value;

    from_guile(SCM val)
        : value(val)
    { }

    operator bool()
    {
        return scm_to_bool(value);
    }

    operator uint8_t()
    {
        return scm_to_uint8(value);
    }

    operator uint16_t()
    {
        return scm_to_uint16(value);
    }

    operator uint32_t()
    {
        return scm_to_uint32(value);
    }

    operator uint64_t()
    {
        return scm_to_uint64(value);
    }

    operator int8_t()
    {
        return scm_to_int8(value);
    }

    operator int16_t()
    {
        return scm_to_int16(value);
    }

    operator int32_t()
    {
        return scm_to_int32(value);
    }

    operator int64_t()
    {
        return scm_to_int64(value);
    }

    operator std::string()
    {
        char* str_raw = scm_to_locale_string(value);
        std::string ret(str_raw);
        free(str_raw);

        return ret;
    }

    operator std::filesystem::path()
    {
        char* str_raw = scm_to_locale_string(value);
        std::string ret(str_raw);
        free(str_raw);

        return std::filesystem::path(ret);
    }

    template<typename T>
    operator std::vector<T>()
    {
        std::vector<T> ret;

        ret.reserve(SCM_SIMPLE_VECTOR_LENGTH(value));

        for (size_t i = 0; i < SCM_SIMPLE_VECTOR_LENGTH(value); ++i) {
            SCM val = SCM_SIMPLE_VECTOR_REF(value, i);

            ret.emplace_back(from_guile(val));
        }

        return ret;
    }

    template<typename K, typename V>
    operator std::map<K, V>()
    {
        std::map<K, V> ret;

        while (not scm_is_null(value)) {
            SCM pair = scm_car(value);

            K key = from_guile(scm_car(pair));
            V val = from_guile(scm_cdr(pair));

            ret[key] = val;

            value = scm_cdr(value);
        }

        return ret;
    }
};

