// Copyright 2009 Dolphin Emulator Project
// Licensed under GPLv2+
// Refer to the license.txt file included.

#pragma once

#include <cstddef>
#include <string>
#include <unordered_map>
#include <vector>

#include "Common/CommonTypes.h"
#include "Common/NandPaths.h"
#include "DiscIO/Volume.h"

namespace DiscIO
{
bool AddTicket(u64 title_id, const std::vector<u8>& ticket);

class CNANDContentData
{
public:
	virtual void Open() { };
	virtual const std::vector<u8> Get() = 0;
	virtual bool GetRange(u32 start, u32 size, u8* buffer) = 0;
	virtual void Close() { };
};

class CNANDContentDataFile final : public CNANDContentData
{
public:
	CNANDContentDataFile(const std::string& filename) : m_filename(filename) { };

	void Open() override;
	const std::vector<u8> Get() override;
	bool GetRange(u32 start, u32 size, u8* buffer) override;
	void Close() override;
private:
	void EnsureOpen();

	const std::string m_filename;
	std::unique_ptr<File::IOFile> m_file;
};
class CNANDContentDataBuffer final : public CNANDContentData
{
public:
	CNANDContentDataBuffer(const std::vector<u8>& buffer) : m_buffer(buffer) { };

	const std::vector<u8> Get() override { return m_buffer; };
	bool GetRange(u32 start, u32 size, u8* buffer) override;
private:
	const std::vector<u8> m_buffer;
};

struct SNANDContent
{
	u32 m_ContentID;
	u16 m_Index;
	u16 m_Type;
	u32 m_Size;
	u8 m_SHA1Hash[20];
	u8 m_Header[36]; //all of the above

	std::unique_ptr<CNANDContentData> m_Data;
};

// Instances of this class must be created by CNANDContentManager
class CNANDContentLoader final
{
public:
	CNANDContentLoader(const std::string& content_name);
	virtual ~CNANDContentLoader();

	bool IsValid() const { return m_Valid; }
	void RemoveTitle() const;
	u64 GetTitleID() const { return m_TitleID; }
	u16 GetIosVersion() const { return m_IosVersion; }
	u32 GetBootIndex() const { return m_BootIndex; }
	size_t GetContentSize() const { return m_Content.size(); }
	const SNANDContent* GetContentByIndex(int index) const;
	const u8* GetTMDView() const { return m_TMDView; }
	const u8* GetTMDHeader() const { return m_TMDHeader; }
	const std::vector<u8>& GetTicket() const { return m_Ticket; }

	const std::vector<SNANDContent>& GetContent() const { return m_Content; }

	u16 GetTitleVersion() const { return m_TitleVersion; }
	u16 GetNumEntries() const { return m_NumEntries; }
	DiscIO::IVolume::ECountry GetCountry() const;
	u8 GetCountryChar() const { return m_Country; }

	enum
	{
		TMD_VIEW_SIZE       = 0x58,
		TMD_HEADER_SIZE     = 0x1E4,
		CONTENT_HEADER_SIZE = 0x24,
		TICKET_SIZE         = 0x2A4
	};

private:
	bool Initialize(const std::string& name);
	void InitializeContentEntries(const std::vector<u8>& tmd, const std::vector<u8>& decrypted_title_key, const std::vector<u8>& data_app);

	static std::vector<u8> AESDecode(const u8* key, u8* iv, const u8* src, u32 size);
	static std::vector<u8> GetKeyFromTicket(const std::vector<u8>& ticket);

	bool m_Valid;
	bool m_IsWAD;
	std::string m_Path;
	u64 m_TitleID;
	u16 m_IosVersion;
	u32 m_BootIndex;
	u16 m_NumEntries;
	u16 m_TitleVersion;
	u8 m_TMDView[TMD_VIEW_SIZE];
	u8 m_TMDHeader[TMD_HEADER_SIZE];
	std::vector<u8> m_Ticket;
	u8 m_Country;

	std::vector<SNANDContent> m_Content;
};


// we open the NAND Content files too often... let's cache them
class CNANDContentManager
{
public:
	static CNANDContentManager& Access() { static CNANDContentManager instance; return instance; }
	u64 Install_WiiWAD(const std::string& fileName);

	const CNANDContentLoader& GetNANDLoader(const std::string& content_path);
	const CNANDContentLoader& GetNANDLoader(u64 title_id, Common::FromWhichRoot from);
	bool RemoveTitle(u64 title_id, Common::FromWhichRoot from);
	void ClearCache();

private:
	CNANDContentManager() {}
	~CNANDContentManager();

	CNANDContentManager(CNANDContentManager const&) = delete;
	void operator=(CNANDContentManager const&) = delete;

	std::unordered_map<std::string, std::unique_ptr<CNANDContentLoader>> m_map;
};

class CSharedContent
{
public:
	static CSharedContent& AccessInstance() { static CSharedContent instance; return instance; }

	std::string GetFilenameFromSHA1(const u8* hash);
	std::string AddSharedContent(const u8* hash);
	void UpdateLocation();

private:
	CSharedContent();
	virtual ~CSharedContent();

	CSharedContent(CSharedContent const&) = delete;
	void operator=(CSharedContent const&) = delete;

#pragma pack(push,1)
	struct SElement
	{
		u8 FileName[8];
		u8 SHA1Hash[20];
	};
#pragma pack(pop)

	u32 m_LastID;
	std::string m_ContentMap;
	std::vector<SElement> m_Elements;
};

class cUIDsys
{
public:
	static cUIDsys& AccessInstance() { static cUIDsys instance; return instance; }

	u32 GetUIDFromTitle(u64 title_id);
	void AddTitle(u64 title_id);
	void GetTitleIDs(std::vector<u64>& title_ids, bool owned = false);
	void UpdateLocation();

private:
	cUIDsys();
	virtual ~cUIDsys();

	cUIDsys(cUIDsys const&) = delete;
	void operator=(cUIDsys const&) = delete;

#pragma pack(push,1)
	struct SElement
	{
		u8 titleID[8];
		u8 UID[4];
	};
#pragma pack(pop)

	u32 m_LastUID;
	std::string m_UidSys;
	std::vector<SElement> m_Elements;
};

}