/*-
 * Copyright (c) 2024 Netflix, Inc.
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

/* Copied from a file that likely shoulve have had this at the top */
/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright 2020 Toomas Soome
 * Copyright 2020 RackTop Systems, Inc.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include "lua.h"
#include "lauxlib.h"
#include "lutils.h"
#include <gfx_fb.h>
#include <pnglite.h>

/*
 * put image using terminal coordinates.
 */
static int
lua_term_putimage(lua_State *L)
{
        const char *name;
        png_t png;
        uint32_t x1, y1, x2, y2, f;
        int nargs, ret = 0, error;

        nargs = lua_gettop(L);
        if (nargs != 6) {
                lua_pushboolean(L, 0);
                return 1;
        }

        name = luaL_checkstring(L, 1);
        x1 = luaL_checknumber(L, 2);
        y1 = luaL_checknumber(L, 3);
        x2 = luaL_checknumber(L, 4);
        y2 = luaL_checknumber(L, 5);
        f = luaL_checknumber(L, 6);

        x1 = gfx_state.tg_origin.tp_col + x1 * gfx_state.tg_font.vf_width;
        y1 = gfx_state.tg_origin.tp_row + y1 * gfx_state.tg_font.vf_height;
        if (x2 != 0) {
                x2 = gfx_state.tg_origin.tp_col +
                    x2 * gfx_state.tg_font.vf_width;
        }
        if (y2 != 0) {
                y2 = gfx_state.tg_origin.tp_row +
                    y2 * gfx_state.tg_font.vf_height;
        }

        if ((error = png_open(&png, name)) != PNG_NO_ERROR) {
                if (f & FL_PUTIMAGE_DEBUG)
                        printf("%s\n", png_error_string(error));
        } else {
                if (gfx_fb_putimage(&png, x1, y1, x2, y2, f) == 0)
                        ret = 1;
                (void) png_close(&png);
        }
        lua_pushboolean(L, ret);
        return 1;
}

static int
lua_fb_putimage(lua_State *L)
{
        const char *name;
        png_t png;
        uint32_t x1, y1, x2, y2, f;
        int nargs, ret = 0, error;

        nargs = lua_gettop(L);
        if (nargs != 6) {
                lua_pushboolean(L, 0);
                return 1;
        }

        name = luaL_checkstring(L, 1);
        x1 = luaL_checknumber(L, 2);
        y1 = luaL_checknumber(L, 3);
        x2 = luaL_checknumber(L, 4);
        y2 = luaL_checknumber(L, 5);
        f = luaL_checknumber(L, 6);

        if ((error = png_open(&png, name)) != PNG_NO_ERROR) {
                if (f & FL_PUTIMAGE_DEBUG)
                        printf("%s\n", png_error_string(error));
        } else {
                if (gfx_fb_putimage(&png, x1, y1, x2, y2, f) == 0)
                        ret = 1;
                (void) png_close(&png);
        }
        lua_pushboolean(L, ret);
        return 1;
}

static int
lua_fb_setpixel(lua_State *L)
{
        uint32_t x, y;
        int nargs;

        nargs = lua_gettop(L);
        if (nargs != 2) {
                lua_pushnil(L);
                return 1;
        }

        x = luaL_checknumber(L, 1);
        y = luaL_checknumber(L, 2);
        gfx_fb_setpixel(x, y);
        return 0;
}

static int
lua_fb_line(lua_State *L)
{
        uint32_t x0, y0, x1, y1, wd;
        int nargs;

        nargs = lua_gettop(L);
        if (nargs != 5) {
                lua_pushnil(L);
                return 1;
        }

        x0 = luaL_checknumber(L, 1);
        y0 = luaL_checknumber(L, 2);
        x1 = luaL_checknumber(L, 3);
        y1 = luaL_checknumber(L, 4);
        wd = luaL_checknumber(L, 5);
        gfx_fb_line(x0, y0, x1, y1, wd);
        return 0;
}

static int
lua_fb_bezier(lua_State *L)
{
        uint32_t x0, y0, x1, y1, x2, y2, width;
        int nargs;

        nargs = lua_gettop(L);
        if (nargs != 7) {
                lua_pushnil(L);
                return 1;
        }

        x0 = luaL_checknumber(L, 1);
        y0 = luaL_checknumber(L, 2);
        x1 = luaL_checknumber(L, 3);
        y1 = luaL_checknumber(L, 4);
        x2 = luaL_checknumber(L, 5);
        y2 = luaL_checknumber(L, 6);
        width = luaL_checknumber(L, 7);
        gfx_fb_bezier(x0, y0, x1, y1, x2, y2, width);
        return 0;
}

static int
lua_fb_drawrect(lua_State *L)
{
        uint32_t x0, y0, x1, y1, fill;
        int nargs;

        nargs = lua_gettop(L);
        if (nargs != 5) {
                lua_pushnil(L);
                return 1;
        }

        x0 = luaL_checknumber(L, 1);
        y0 = luaL_checknumber(L, 2);
        x1 = luaL_checknumber(L, 3);
        y1 = luaL_checknumber(L, 4);
        fill = luaL_checknumber(L, 5);
        gfx_fb_drawrect(x0, y0, x1, y1, fill);
        return 0;
}

static int
lua_term_drawrect(lua_State *L)
{
        uint32_t x0, y0, x1, y1;
        int nargs;

        nargs = lua_gettop(L);
        if (nargs != 4) {
                lua_pushnil(L);
                return 1;
        }

        x0 = luaL_checknumber(L, 1);
        y0 = luaL_checknumber(L, 2);
        x1 = luaL_checknumber(L, 3);
        y1 = luaL_checknumber(L, 4);
        gfx_term_drawrect(x0, y0, x1, y1);
        return 0;
}

#define REG_SIMPLE(n)   { #n, lua_ ## n }
static const struct luaL_Reg gfxlib[] = {
        REG_SIMPLE(fb_bezier),
        REG_SIMPLE(fb_drawrect),
        REG_SIMPLE(fb_line),
        REG_SIMPLE(fb_putimage),
        REG_SIMPLE(fb_setpixel),
        REG_SIMPLE(term_drawrect),
        REG_SIMPLE(term_putimage),
        { NULL, NULL },
};

int
luaopen_gfx(lua_State *L)
{
        luaL_newlib(L, gfxlib);
        return 1;
}

void
gfx_interp_ref(void)
{
}

static void
gfx_init_md(lua_State *L)
{
        luaL_requiref(L, "gfx", luaopen_gfx, 1);
        lua_pop(L, 1);  /* Remove lib */

        /*
         * Add in the compatibility references in the loader table.  Doing it with
         * a pseudo-embedded script is easier than the raw calls.
         */
        if (luaL_dostring(L,
                "loader.fb_bezier = gfx.fb_bezier\n"
                "loader.fb_drawrect = gfx.fb_drawrect\n"
                "loader.fb_line = gfx.fb_line\n"
                "loader.fb_putimage = gfx.fb_putimage\n"
                "loader.fb_setpixel = gfx.fb_setpixel\n"
                "loader.term_drawrect = gfx.term_drawrect\n"
                "loader.term_putimage = gfx.term_putimage") != 0) {
                lua_pop(L, 1);
                const char *errstr = lua_tostring(L, -1);
                errstr = errstr == NULL ? "unknown" : errstr;
                printf("Error adding compat loader bindings: %s.\n", errstr);
        }
}

LUA_COMPILE_SET(gfx_init_md);