//  SPDX-License-Identifier: MIT
//
//  ES-DE Frontend
//  FileData.h
//
//  Provides game file data structures and functions to access and sort this information.
//  Also provides functions to look up paths to media files and for launching games
//  (launching initiated in ViewController).
//

#ifndef ES_APP_FILE_DATA_H
#define ES_APP_FILE_DATA_H

#include "MetaData.h"
#include "SystemData.h"
#include "Window.h"
#include "utils/FileSystemUtil.h"
#include "utils/StringUtil.h"

#include <functional>
#include <unordered_map>

enum FileType {
    GAME = 1, // Cannot have children.
    FOLDER = 2,
    PLACEHOLDER = 3
};

// A tree node that holds information for a file.
class FileData
{
public:
    FileData(FileType type,
             const std::string& path,
             SystemEnvironmentData* envData,
             SystemData* system);

    virtual ~FileData();

    const std::string& getName() { return metadata.get("name"); }
    const std::string& getSortName();
    // Returns our best guess at the "real" name for this file.
    std::string getDisplayName() const { return Utils::FileSystem::getStem(mPath); }
    std::string getCleanName() const
    {
        return Utils::String::removeParenthesis(this->getDisplayName());
    }

    const bool getFavorite();
    const bool getKidgame();
    const bool getHidden();
    const bool getCountAsGame();
    const std::pair<unsigned int, unsigned int>& getGameCount() const { return mGameCount; }
    const bool getExcludeFromScraper();
    const std::vector<FileData*> getChildrenRecursive() const;
    FileType getType() const { return mType; }
    const std::string& getPath() const { return mPath; }
    FileData* getParent() const { return mParent; }
    const std::unordered_map<std::string, FileData*>& getChildrenByFilename() const
    {
        return mChildrenByFilename;
    }
    const std::vector<FileData*>& getChildren() const { return mChildren; }
    const std::vector<FileData*>& getFilteredChildren() const { return mFilteredChildren; }
    SystemData* getSystem() const { return mSystem; }
    SystemData* getSourceSystem()
    {
        return (mSourceFileData != nullptr ? mSourceFileData->mSystem : mSystem);
    }
    SystemEnvironmentData* getSystemEnvData() const { return mEnvData; }

    // These functions are used by GameSelectorComponent.
    const std::vector<FileData*>& getChildrenLastPlayed() const { return mChildrenLastPlayed; }
    const std::vector<FileData*>& getChildrenMostPlayed() const { return mChildrenMostPlayed; }
    void setUpdateChildrenLastPlayed(bool state) { mUpdateChildrenLastPlayed = state; }
    void setUpdateChildrenMostPlayed(bool state) { mUpdateChildrenMostPlayed = state; }
    void setUpdateListCallback(const std::function<void()>& func) { mUpdateListCallback = func; }

    const bool getOnlyFoldersFlag() const { return mOnlyFolders; }
    const bool getHasFoldersFlag() const { return mHasFolders; }
    static const std::string getROMDirectory();
    static const std::string getMediaDirectory();
    const std::string getMediafilePath(const std::string& subdirectory) const;
    const std::string getImagePath() const;
    const std::string get3DBoxPath() const;
    const std::string getBackCoverPath() const;
    const std::string getCoverPath() const;
    const std::string getFanArtPath() const;
    const std::string getMarqueePath() const;
    const std::string getPhysicalMediaPath() const;
    const std::string getMiximagePath() const;
    const std::string getScreenshotPath() const;
    const std::string getTitleScreenPath() const;
    const std::string getCustomImagePath() const;
    const std::string getVideoPath() const;
    const std::string getManualPath() const;

    const bool getDeletionFlag() const { return mDeletionFlag; }
    void setDeletionFlag(bool setting) { mDeletionFlag = setting; }
    const bool getNoLoad() const { return mNoLoad; }
    void setNoLoad(bool state) { mNoLoad = state; }
    const bool isPlaceHolder() const { return mType == PLACEHOLDER; }
    void refreshMetadata() { metadata = mSourceFileData->metadata; }

    const std::vector<FileData*>& getChildrenListToDisplay();
    std::vector<FileData*> getFilesRecursive(unsigned int typeMask,
                                             bool displayedOnly = false,
                                             bool countAllGames = true) const;
    std::vector<FileData*> getScrapeFilesRecursive(bool includeFolders,
                                                   bool excludeRecursively,
                                                   bool respectExclusions) const;

    void addChild(FileData* file);
    void removeChild(FileData* file);

    virtual std::string getKey() { return getFileName(); }
    const bool isArcadeAsset() const;
    const bool isArcadeGame() const;
    const std::string& getFullPath() const { return getPath(); }
    std::string getFileName() { return Utils::FileSystem::getFileName(getPath()); }
    virtual FileData* getSourceFileData() { return this; }
    const std::string& getSystemName() const { return mSystemName; }

    enum class findEmulatorResult {
        FOUND_FILE,
        FOUND_ANDROID_PACKAGE,
        NOT_FOUND,
        NO_RULES
    };

    void launchGame();
    const std::pair<std::string, FileData::findEmulatorResult> findEmulator(std::string& command,
                                                                            const bool preCommand);

    using ComparisonFunction = bool(const FileData* a, const FileData* b);
    struct SortType {
        ComparisonFunction* comparisonFunction;
        std::string description;
        SortType(ComparisonFunction* sortFunction, const std::string& sortDescription)
            : comparisonFunction(sortFunction)
            , description(sortDescription)
        {
        }
    };

    void sort(ComparisonFunction& comparator, std::pair<unsigned int, unsigned int>& gameCount);
    void sortFavoritesOnTop(ComparisonFunction& comparator,
                            std::pair<unsigned int, unsigned int>& gameCount);
    void sort(const SortType& type, bool mFavoritesOnTop = false);
    MetaDataList metadata;
    // Only count the games, a cheaper alternative to a full sort when that is not required.
    void countGames(std::pair<unsigned int, unsigned int>& gameCount);
    void updateLastPlayedList();
    void updateMostPlayedList();
    void setSortTypeString(std::string typestring) { mSortTypeString = typestring; }
    const std::string& getSortTypeString() const { return mSortTypeString; }
    const FileData::SortType& getSortTypeFromString(const std::string& desc) const;

    static inline std::vector<std::string> sImageExtensions {".png", ".jpg"};
    static inline std::vector<std::string> sVideoExtensions {".mp4", ".mkv", ".avi",
                                                             ".mp4", ".wmv", ".mov"};

protected:
    FileData* mSourceFileData;
    FileData* mParent;
    std::string mSystemName;
    std::string mSortTypeString = "";

private:
    FileType mType;
    std::string mPath;
    SystemEnvironmentData* mEnvData;
    SystemData* mSystem;
    std::unordered_map<std::string, FileData*> mChildrenByFilename;
    std::vector<FileData*> mChildren;
    std::vector<FileData*> mFilteredChildren;
    std::vector<FileData*> mChildrenLastPlayed;
    std::vector<FileData*> mChildrenMostPlayed;
    std::function<void()> mUpdateListCallback;

    // The pair includes all games, and favorite games.
    std::pair<unsigned int, unsigned int> mGameCount;
    bool mOnlyFolders;
    bool mHasFolders;
    bool mUpdateChildrenLastPlayed;
    bool mUpdateChildrenMostPlayed;
    // Used for flagging a game for deletion from its gamelist.xml file.
    bool mDeletionFlag;
    bool mNoLoad;
};

class CollectionFileData : public FileData
{
public:
    CollectionFileData(FileData* file, SystemData* system);
    ~CollectionFileData();
    FileData* getSourceFileData() { return mSourceFileData; }
    std::string getKey() { return getFullPath(); }
};

#endif // ES_APP_FILE_DATA_H