AssetManifest.h

#pragma once
#include <map>
#include "UrlLoaders.h"

namespace ParaEngine
{
    class CUrlProcessorUserData;

    struct AssetFileEntry
    {
    public:
        typedef boost::signals2::signal<void(int, AssetFileEntry*)>  SyncFile_Callback_t;

        enum AssetFileStatus {
            AssetFileStatus_Unknown = 0,
            AssetFileStatus_Downloaded = 1,
            AssetFileStatus_Downloading = 2,
            AssetFileStatus_Failed = 3,
        };
        enum AssetFileTypeEnum {
            AssetFileType_default = 0,
            AssetFileType_dds_file,
            AssetFileType_x_file,
            AssetFileType_ui_texture,
            AssetFileType_script,
            AssetFileType_audio,
            AssetFileType_big_file,
            AssetFileType_web_file,
        };

        string m_localFileName;
        string m_url;
        
        bool m_bIsZipFile;
        int m_nDownloadCount;
        int m_nFileSize;
        AssetFileTypeEnum m_file_type;

        SyncFile_Callback_t* m_sync_callback;

        AssetFileStatus m_nStatus;
    public:
        AssetFileEntry();
        ~AssetFileEntry();

        bool DoesFileExist();

        inline AssetFileStatus GetStatus(){return m_nStatus;}

        inline void SetStatus(AssetFileStatus status){m_nStatus = status;}

        inline AssetFileTypeEnum GetFileType() {return m_file_type;}

        AssetFileTypeEnum SetFileType(const std::string& file_extension);

        ResourceRequestID GetRequestQueueID();

        bool IsDownloading();

        bool SyncFile();

        HRESULT SyncFile_Async(URL_LOADER_CALLBACK pFuncCallback=NULL, CUrlProcessorUserData* pUserData=NULL, bool bDeleteUserData=false);

        HRESULT SyncFile_Async(const SyncFile_Callback_t::slot_type& slot);

        bool CheckSyncFile();

        void SignalComplete(int nResult=0);

        bool HasReachedMaxRetryCount();
        
        bool AddDownloadCount();

        const string& GetUrl() {return m_url;};

        string GetAbsoluteUrl();

        bool IsUrlFileCompressed();

        bool CheckMD5AndSize(const char* buffer, int nSize);

        bool SaveToDisk(const char* buffer, int nSize, bool bCheckMD5=true);

        int GetFileSize() {return m_nFileSize;}

        inline const std::string& GetLocalFileName() { return m_localFileName; }
        std::string GetFullFilePath();
    };

    
    class CFileReplaceMap
    {
    public:
        CFileReplaceMap();

        /* load a replace file mapping. it is just file to file pairs. If the file is requested on the left during GetFile(filename), file on the right is returned instead. 
        * This function is very useful to temperarily change the 3D and 2D theme of the entire game world, in which only a text file needs to be updated. 
        * @param filename: the file map file. 
        * @param bReplaceExistingOnes: whether we will overwrite any previous calls to this function. 
        */
        CFileReplaceMap(const string& filename, bool bReplaceExistingOnes = true);

        ~CFileReplaceMap();

        /* load a replace file mapping. it is just file to file pairs. If the file is requested on the left during GetFile(filename), file on the right is returned instead. 
        * This function is very useful to temperarily change the 3D and 2D theme of the entire game world, in which only a text file needs to be updated. 
        * @param filename: the file map file. 
        * @param bReplaceExistingOnes: whether we will overwrite any previous calls to this function. 
        */
        void LoadReplaceFile(const string& filename, bool bReplaceExistingOnes = true);

        static CFileReplaceMap& GetSingleton();

        bool IsEmpty();

        bool ReplaceFile(string& inout);
    private:
        typedef std::map<string, string>  Asset_Replace_Map_Type;

        /* a replace file mapping. it is just file name to file name pairs. If the file is requested on the left during GetFile(filename), file on the right is returned instead. 
        * This function is very useful to temperarily change the 3D and 2D theme of the entire game world, in which only a text file needs to be updated. 
        */
        Asset_Replace_Map_Type m_replace_map;
    };

    class CAssetManifest
    {
    public:
        CAssetManifest(void);
        virtual ~CAssetManifest(void);
    public:
        static CAssetManifest& GetSingleton();

        /* load all "Assets_manifest*.txt" manifest file under the root directory */
        void LoadManifest();

        void PrepareCacheFolders();

        /* load a given manifest file. */
        void LoadManifestFile(const string& filename);

        bool DoesFileExist(const char* filename);

        bool DoesFileExist(const string& filename);

        int CheckSyncFile(const char* filename);

        AssetFileEntry* GetFile(const string& filename, bool bUseReplaceMap = true, bool bIgnoreLocalFile = false);

        void AddEntry(const char* filename);

        void CleanUp();

        void PrintStat();

        inline bool IsEnabled() {return m_bEnableManifest;}

        void EnableManifest(bool bEnable=true);
        
        bool IsUseLocalFileFirst() const;
        void SetUseLocalFileFirst(bool val);

    private:
        bool ParseAssetEntryStr(const string &sFilename, string &fileKey, string &md5, string& filesize);
    private:
        typedef std::map<string, AssetFileEntry*>  Asset_Manifest_Map_Type;
        
        Asset_Manifest_Map_Type m_files;

        bool m_bEnableManifest;
        bool m_bUseLocalFileFirst;
    };
}