#pragma once
#ifndef PROJECT_H
#define PROJECT_H

#include "map.h"
#include "blockdata.h"
#include "wildmoninfo.h"
#include "parseutil.h"
#include "orderedjson.h"
#include "regionmap.h"

#include <QStringList>
#include <QList>
#include <QVector>
#include <QPair>
#include <QStandardItem>
#include <QVariant>
#include <QFileSystemWatcher>

class Project : public QObject
{
    Q_OBJECT
public:
    Project(QObject *parent = nullptr);
    ~Project();

    Project(const Project &) = delete;
    Project & operator = (const Project &) = delete;

    inline QWidget *parentWidget() const { return static_cast<QWidget *>(parent()); }

public:
    QString root;
    QStringList mapNames;
    QStringList groupNames;
    QMap<QString, QStringList> groupNameToMapNames;
    QStringList healLocationSaveOrder;
    QMap<QString, QList<HealLocationEvent*>> healLocations;
    QMap<QString, QString> mapConstantsToMapNames;
    QString layoutsLabel;
    QStringList layoutIds;
    QStringList layoutIdsMaster;
    QMap<QString, Layout*> mapLayouts;
    QMap<QString, Layout*> mapLayoutsMaster;
    QMap<QString, int> gfxDefines;
    QString defaultSong;
    QStringList songNames;
    QStringList itemNames;
    QStringList flagNames;
    QStringList varNames;
    QStringList speciesNames;
    QStringList movementTypes;
    QStringList mapTypes;
    QStringList mapBattleScenes;
    QStringList weatherNames;
    QStringList coordEventWeatherNames;
    QStringList secretBaseIds;
    QStringList bgEventFacingDirections;
    QStringList trainerTypes;
    QStringList globalScriptLabels;
    QStringList mapSectionIdNamesSaveOrder;
    QStringList mapSectionIdNames;
    QMap<uint32_t, QString> encounterTypeToName;
    QMap<uint32_t, QString> terrainTypeToName;
    QMap<QString, QMap<QString, uint16_t>> metatileLabelsMap;
    QMap<QString, uint16_t> unusedMetatileLabels;
    QMap<QString, uint32_t> metatileBehaviorMap;
    QMap<uint32_t, QString> metatileBehaviorMapInverse;
    ParseUtil parser;
    QFileSystemWatcher fileWatcher;
    QSet<QString> modifiedFiles;
    bool usingAsmTilesets;
    QSet<QString> disabledSettingsNames;
    int pokemonMinLevel;
    int pokemonMaxLevel;
    int maxEncounterRate;
    bool wildEncountersLoaded;

    void set_root(QString);

    void clearMaps();
    void clearTilesetCache();
    void clearMapLayouts();
    void clearEventGraphics();
    void clearHealLocations();

    bool sanityCheck();
    bool load();

    Map* loadMap(const QString &mapName);

    // Note: This does not guarantee the map is loaded.
    Map* getMap(const QString &mapName) { return this->maps.value(mapName); }

    bool isMapLoaded(const Map *map) const { return map && isMapLoaded(map->name()); }
    bool isMapLoaded(const QString &mapName) const { return this->loadedMapNames.contains(mapName); }
    bool isLayoutLoaded(const Layout *layout) const { return layout && isLayoutLoaded(layout->id); }
    bool isLayoutLoaded(const QString &layoutId) const { return this->loadedLayoutIds.contains(layoutId); }

    QMap<QString, Tileset*> tilesetCache;
    Tileset* loadTileset(QString, Tileset *tileset = nullptr);
    Tileset* getTileset(QString, bool forceLoad = false);
    QStringList primaryTilesetLabels;
    QStringList secondaryTilesetLabels;
    QStringList tilesetLabelsOrdered;

    Blockdata readBlockdata(QString, bool *ok = nullptr);
    bool loadBlockdata(Layout *);
    bool loadLayoutBorder(Layout *);

    void saveTextFile(QString path, QString text);
    void appendTextFile(QString path, QString text);
    void deleteFile(QString path);

    bool readMapGroups();
    void addNewMapGroup(const QString &groupName);
    QString mapNameToMapGroup(const QString &mapName) const;
    QString getMapConstant(const QString &mapName, const QString &defaultValue = QString()) const;
    QString getMapLayoutId(const QString &mapName, const QString &defaultValue = QString()) const;
    QString getMapLocation(const QString &mapName, const QString &defaultValue = QString()) const;

    struct NewMapSettings {
        QString name;
        QString group;
        bool canFlyTo;
        Layout::Settings layout;
        MapHeader header;
    };
    NewMapSettings newMapSettings;
    Layout::Settings newLayoutSettings;

    void initNewMapSettings();
    void initNewLayoutSettings();

    Map *createNewMap(const Project::NewMapSettings &mapSettings, const Map* toDuplicate = nullptr);
    Layout *createNewLayout(const Layout::Settings &layoutSettings, const Layout* toDuplicate = nullptr);
    Tileset *createNewTileset(QString name, bool secondary, bool checkerboardFill);
    bool isIdentifierUnique(const QString &identifier) const;
    bool isValidNewIdentifier(const QString &identifier) const;
    QString toUniqueIdentifier(const QString &identifier) const;
    QString getProjectTitle() const;
    QString getNewHealLocationName(const Map* map) const;

    bool readWildMonData();
    OrderedMap<QString, OrderedMap<QString, WildPokemonHeader>> wildMonData;

    QString wildMonTableName;
    QVector<EncounterField> wildMonFields;
    QVector<QString> encounterGroupLabels;

    bool readSpeciesIconPaths();
    QString getDefaultSpeciesIconPath(const QString &species);
    QPixmap getSpeciesIcon(const QString &species);

    bool addNewMapsec(const QString &idName, const QString &displayName = QString());
    void removeMapsec(const QString &idName);
    QString getMapsecDisplayName(const QString &idName) const { return this->locationData.value(idName).displayName; }
    void setMapsecDisplayName(const QString &idName, const QString &displayName);

    bool hasUnsavedChanges();
    bool hasUnsavedDataChanges = false;

    bool readMapJson(const QString &mapName, QJsonDocument * out);
    bool loadMapEvent(Map *map, QJsonObject json, Event::Type defaultType = Event::Type::None);
    bool loadMapData(Map*);
    bool readMapLayouts();
    Layout *loadLayout(QString layoutId);
    bool loadLayout(Layout *);
    bool loadMapLayout(Map*);
    bool loadLayoutTilesets(Layout *);
    void loadTilesetAssets(Tileset*);
    void loadTilesetMetatileLabels(Tileset*);
    void readTilesetPaths(Tileset* tileset);

    void saveAll();
    void saveGlobalData();
    void saveLayout(Layout *);
    void saveLayoutBlockdata(Layout *);
    void saveLayoutBorder(Layout *);
    void writeBlockdata(QString, const Blockdata &);
    void saveMap(Map *map, bool skipLayout = false);
    void saveConfig();
    void saveMapLayouts();
    void saveMapGroups();
    void saveRegionMapSections();
    void saveWildMonData();
    void saveHealLocations();
    void saveTilesets(Tileset*, Tileset*);
    void saveTilesetMetatileLabels(Tileset*, Tileset*);
    void appendTilesetLabel(const QString &label, const QString &isSecondaryStr);
    bool readTilesetLabels();
    bool readTilesetMetatileLabels();
    bool readRegionMapSections();
    bool readItemNames();
    bool readFlagNames();
    bool readVarNames();
    bool readMovementTypes();
    bool readInitialFacingDirections();
    bool readMapTypes();
    bool readMapBattleScenes();
    bool readWeatherNames();
    bool readCoordEventWeatherNames();
    bool readSecretBaseIds();
    bool readBgEventFacingDirections();
    bool readTrainerTypes();
    bool readMetatileBehaviors();
    bool readHealLocations();
    bool readMiscellaneousConstants();
    bool readEventScriptLabels();
    bool readObjEventGfxConstants();
    bool readSongNames();
    bool readEventGraphics();
    bool readFieldmapProperties();
    bool readFieldmapMasks();
    QMap<QString, QMap<QString, QString>> readObjEventGfxInfo();

    QPixmap getEventPixmap(const QString &gfxName, const QString &movementName);
    QPixmap getEventPixmap(const QString &gfxName, int frame, bool hFlip);
    QPixmap getEventPixmap(Event::Group group);
    void loadEventPixmap(Event *event, bool forceLoad = false);

    QString fixPalettePath(QString path);
    QString fixGraphicPath(QString path);

    static QString getScriptFileExtension(bool usePoryScript);
    QString getScriptDefaultString(bool usePoryScript, QString mapName) const;
    QStringList getEventScriptsFilePaths() const;
    void insertGlobalScriptLabels(QStringList &scriptLabels) const;

    QString getDefaultPrimaryTilesetLabel() const;
    QString getDefaultSecondaryTilesetLabel() const;
    void updateTilesetMetatileLabels(Tileset *tileset);
    QString buildMetatileLabelsText(const QMap<QString, uint16_t> defines);
    QString findMetatileLabelsTileset(QString label);

    static QString getExistingFilepath(QString filepath);
    void applyParsedLimits();

    void setRegionMapEntries(const QHash<QString, MapSectionEntry> &entries);
    QHash<QString, MapSectionEntry> getRegionMapEntries() const;

    QSet<QString> getTopLevelMapFields() const;

    static QString getEmptyMapDefineName();
    static QString getDynamicMapDefineName();
    static QString getDynamicMapName();
    static QString getEmptySpeciesName();
    static int getNumTilesPrimary();
    static int getNumTilesTotal();
    static int getNumMetatilesPrimary();
    static int getNumMetatilesTotal();
    static int getNumPalettesPrimary();
    static int getNumPalettesTotal();
    static int getMaxMapDataSize();
    static int getDefaultMapDimension();
    static int getMaxMapWidth();
    static int getMaxMapHeight();
    static int getMapDataSize(int width, int height);
    static bool mapDimensionsValid(int width, int height);
    bool calculateDefaultMapSize();
    int getMaxEvents(Event::Group group);
    static QString getEmptyMapsecName();
    static QString getMapGroupPrefix();

private:
    QMap<QString, qint64> modifiedFileTimestamps;
    QMap<QString, QString> facingDirections;
    QHash<QString, QString> speciesToIconPath;
    QHash<QString, Map*> maps;

    // Fields for preserving top-level JSON data that Porymap isn't expecting.
    QJsonObject customLayoutsData;
    QJsonObject customMapSectionsData;
    QJsonObject customMapGroupsData;
    QJsonObject customHealLocationsData;
    OrderedJson::object customWildMonData;
    OrderedJson::object customWildMonGroupData;
    OrderedJson::array extraEncounterGroups;

    // Maps/layouts represented in these sets have been fully loaded from the project.
    // If a valid map name / layout id is not in these sets, a Map / Layout object exists
    // for it in Project::maps / Project::mapLayouts, but it has been minimally populated
    // (i.e. for a map layout it only has the data read from layouts.json, none of its assets
    // have been loaded, and for a map it only has the data needed to identify it in the map
    // list, none of the rest of its data in map.json).
    QSet<QString> loadedMapNames;
    QSet<QString> loadedLayoutIds;

    const QRegularExpression re_gbapalExtension;
    const QRegularExpression re_bppExtension;

    struct EventGraphics
    {
        QString filepath;
        bool loaded = false;
        QImage spritesheet;
        int spriteWidth = -1;
        int spriteHeight = -1;
        bool inanimate = false;
    };
    QMap<QString, EventGraphics*> eventGraphicsMap;

    // The extra data that can be associated with each MAPSEC name.
    struct LocationData
    {
        MapSectionEntry map;
        QString displayName;
        QJsonObject custom;
    };
    QHash<QString, LocationData> locationData;

    void updateLayout(Layout *);

    void setNewLayoutBlockdata(Layout *layout);
    void setNewLayoutBorder(Layout *layout);

    void ignoreWatchedFileTemporarily(QString filepath);
    void recordFileChange(const QString &filepath);
    void resetFileCache();

    QString findSpeciesIconPath(const QStringList &names) const;

    int maxEventsPerGroup;
    int maxObjectEvents;
    static int num_tiles_primary;
    static int num_tiles_total;
    static int num_metatiles_primary;
    static int num_pals_primary;
    static int num_pals_total;
    static int max_map_data_size;
    static int default_map_dimension;

signals:
    void fileChanged(const QString &filepath);
    void mapLoaded(Map *map);
    void mapCreated(Map *newMap, const QString &groupName);
    void layoutCreated(Layout *newLayout);
    void tilesetCreated(Tileset *newTileset);
    void mapGroupAdded(const QString &groupName);
    void mapSectionAdded(const QString &idName);
    void mapSectionDisplayNameChanged(const QString &idName, const QString &displayName);
    void mapSectionIdNamesChanged(const QStringList &idNames);
    void mapsExcluded(const QStringList &excludedMapNames);
    void eventScriptLabelsRead();
};

#endif // PROJECT_H