#pragma once
#include "pch.h"
#include "NES/BaseMapper.h"
#include "NES/Mappers/Homebrew/FlashSST39SF040.h"
#include "Shared/Emulator.h"
#include "Shared/BatteryManager.h"
#include "Utilities/Patches/IpsPatcher.h"

class Cheapocabra : public BaseMapper
{
private:
	unique_ptr<FlashSST39SF040> _flash;
	uint8_t _prgReg = 0;
	vector<uint8_t> _orgPrgRom;

protected:
	uint16_t GetPrgPageSize() override { return 0x8000; }
	uint16_t GetChrPageSize() override { return 0x2000; }
	uint32_t GetWorkRamSize() override { return 0; }
	uint32_t GetSaveRamSize() override { return 0; }
	uint16_t RegisterStartAddress() override { return 0x5000; }
	uint16_t RegisterEndAddress() override { return 0x5FFF; }
	uint32_t GetChrRamSize() override { return 0x4000; }
	uint32_t GetNametableCount() override { return 16; }
	bool AllowRegisterRead() override { return true; }

	void InitMapper() override
	{
		AddRegisterRange(0x7000, 0x7FFF, MemoryOperation::Any);
		AddRegisterRange(0x8000, 0xFFFF, MemoryOperation::Any);

		_flash.reset(new FlashSST39SF040(_prgRom, _prgSize));
		
		WriteRegister(0x5000, GetPowerOnByte());

		_orgPrgRom = vector<uint8_t>(_prgRom, _prgRom + _prgSize);
		ApplySaveData();
	}

	void Serialize(Serializer& s) override
	{
		BaseMapper::Serialize(s);
		
		SV(_flash);
		SV(_prgReg);
		
		SerializeRomDiff(s, _orgPrgRom);
	}

	void ApplySaveData()
	{
		if(_console->GetNesConfig().DisableFlashSaves) {
			return;
		}

		LoadRomPatch(_orgPrgRom);
	}

	void SaveBattery() override
	{
		if(_console->GetNesConfig().DisableFlashSaves) {
			return;
		}

		SaveRom(_orgPrgRom);
	}

	void UpdateRegister(uint8_t value)
	{
		_prgReg = value & 0x0F;
		SelectPrgPage(0, _prgReg);

		SelectChrPage(0, (value >> 4) & 0x01);
		for(int i = 0; i < 8; i++) {
			SetNametable(i, ((value & 0x20) ? 8 : 0) + i);
		}
	}

	uint8_t ReadRegister(uint16_t addr) override
	{
		if(addr < 0x8000) {
			//"Note that reading from the register effectively writes the value of open bus"
			UpdateRegister(_console->GetMemoryManager()->GetOpenBus());
		} else {
			int16_t value = _flash->Read(addr);
			if(value >= 0) {
				return (uint8_t)value;
			}
		}
		return BaseMapper::InternalReadRam(addr);
	}

	void WriteRegister(uint16_t addr, uint8_t value) override
	{
		if(addr < 0x8000) {
			UpdateRegister(value);
		} else {
			_flash->Write((_prgReg << 15) | (addr & 0x7FFF), value);
		}
	}
};