#pragma once

#include <map>
#include <vector>
#include <string>

using namespace std;

class GameDatabase {
 private:
    std::map<ObjectId, GameObject, std::greater<ObjectId>> db;
 public:
    GameDatabase() = default;

    void Insert(ObjectId id, string name, size_t x, size_t y) {
        GameObject obj;
        obj.name = name;
        obj.id = id;
        obj.x = x;
        obj.y = y;
        db[id] = obj;
    }

    void Remove(ObjectId id) {
        map<ObjectId, GameObject>::iterator obj;
        obj = db.find(id);
        if (obj != db.end())
            db.erase(obj);
    }

    vector<GameObject> DataByName(string name) const {
        vector<GameObject> mass;
        map<ObjectId, GameObject>::const_iterator i;
        for (i = db.begin(); i != db.end(); i++) {
            if (i->second.name == name) {
                mass.push_back(i->second);
            }
        }
        return mass;
    }

    vector<GameObject> DataByPosition(size_t x, size_t y) const {
        vector<GameObject> mass;
        map<ObjectId, GameObject>::const_iterator i;
        for (i = db.begin(); i != db.end(); i++) {
            if (i->second.x == x && i->second.y == y) {
                mass.push_back(i->second);
            }
        }
        return mass;
    }

    vector<GameObject> Data() const {
        vector<GameObject> mass;
        map<ObjectId, GameObject>::const_iterator i;
        for (i = db.begin(); i != db.end(); i++) {
            mass.push_back(i->second);
        }
        return mass;
    }
};