#pragma once

#include <revolution/types.h>

class JMapInfo;
class TalkMessageInfo;
class TalkNode;

struct MessageInfoBlock {
    u32 mMagic;
    u32 mBlockSize;
    u16 mItemCount;
    u16 mItemSize;
    u32 _C;
};

struct MessageDataBlock {
    u32 mMagic;
    u32 mBlockSize;
};

struct MessageFlowBlock {
    u32 mMagic;
    u32 mBlockSize;
    u16 mNodeCount;
    u16 _A;
    u32 _C;
};

struct MessageFLI1Block {
    u32 mMagic;
    u32 mBlockSize;
};

class MessageData {
public:
    MessageData(const char*);

    bool getMessageDirect(TalkMessageInfo*, const char*) const;
    bool getMessage(TalkMessageInfo*, u16, u16) const;
    TalkNode* findNode(const char*) const;
    TalkNode* getNode(u32) const;
    TalkNode* getBranchNode(u32) const;
    bool isValidBranchNode(u32) const;
    u8* getMessageInfoTool(int) const;
    s32 findMessageIndex(const char*) const;

    JMapInfo* mIDTable;            // 0x0
    MessageInfoBlock* mInfoBlock;  // 0x4
    MessageDataBlock* mDataBlock;  // 0x8
    u32 _C;
    MessageFlowBlock* mFlowBlock;  // 0x10
    u16* _14;
    u8* _18;
    MessageFLI1Block* mFLI1Block;  // 0x1C
};

class MessageHolder {
public:
    MessageHolder();

    void initSceneData();
    void destroySceneData();
    void initSystemData();
    void initGameData();

    /* 0x00 */ MessageData* mSystemMessageData;
    /* 0x04 */ MessageData* mGameMessageData;
    /* 0x08 */ MessageData* mSceneMessageData;
};

class MessageSystem {
public:
    class Node {};

    struct FlowNodeBranch {};

    struct FlowNodeEvent {
        /* 0x00 */ u8 mFlowType;
        /* 0x01 */ u8 mEventType;
        /* 0x02 */ u16 mBranchID;
        /* 0x04 */ u32 mArg;
    };

    static bool getSystemMessageDirect(TalkMessageInfo*, const char*);
    static bool getGameMessageDirect(TalkMessageInfo*, const char*);
    static bool getLayoutMessageDirect(TalkMessageInfo*, const char*);
    static MessageData* getSceneMessageData();
};