#include "COpeningBanner.h"
#include "CGameProject.h"

COpeningBanner::COpeningBanner(CGameProject *pProj)
    : mpProj(pProj)
{
    mWii = mpProj->IsWiiBuild();

    TString BannerPath = mpProj->DiscFilesystemRoot(false) + "opening.bnr";
    CFileInStream Banner(BannerPath, IOUtil::eBigEndian);

    if (Banner.IsValid())
    {
        mBannerData.resize(Banner.Size());
        Banner.ReadBytes(mBannerData.data(), mBannerData.size());
    }
}

TString COpeningBanner::EnglishGameName() const
{
    // opening.bnr stores the game name in a fixed-length buffer. Need to account for
    // this and prevent the string-reading function from overrunning the buffer
    CMemoryInStream Banner(mBannerData.data(), mBannerData.size(), IOUtil::eBigEndian);

    u32 CharSize = mWii ? 2 : 1;
    u32 MaxLen = MaxGameNameLength();

    std::vector<u8> NameBuffer((MaxLen + 1) * CharSize, 0);
    Banner.GoTo( mWii ? 0xB0 : 0x1860 );
    Banner.ReadBytes(NameBuffer.data(), MaxLen * CharSize);

    Banner.SetData(NameBuffer.data(), NameBuffer.size(), IOUtil::eBigEndian);
    return mWii ? Banner.ReadWString().ToUTF8() : Banner.ReadString();
}

void COpeningBanner::SetEnglishGameName(const TString& rkName)
{
        CMemoryOutStream Banner(mBannerData.data(), mBannerData.size(), IOUtil::eBigEndian);
    u32 PadCount = 0;

    u32 MaxLen = MaxGameNameLength();
    ASSERT(rkName.Size() <= MaxLen);

    if (mWii)
    {
        Banner.GoTo(0xB0);
        Banner.WriteWString(rkName.ToUTF16(), -1, false);
        PadCount = (MaxLen - rkName.Size()) * 2;
    }
    else
    {
        Banner.GoTo(0x1860);
        Banner.WriteString(rkName, -1, false);
        PadCount = MaxLen - rkName.Size();
    }

    for (u32 Pad = 0; Pad < PadCount; Pad++)
        Banner.WriteByte(0);
}

void COpeningBanner::Save()
{
    TString BannerPath = mpProj->DiscFilesystemRoot(false) + "opening.bnr";
    CFileOutStream Banner(BannerPath, IOUtil::eBigEndian);
    Banner.WriteBytes(mBannerData.data(), mBannerData.size());
}

u32 COpeningBanner::MaxGameNameLength() const
{
    return (mWii ? 21 : 64);
}