#pragma once

#include <QString>
#include <unordered_set>

#include "Library.h"

static const std::unordered_set<std::string> MANAGED_AGENTS = { "moe.yushi:authlibinjector" };
static const std::map<std::string, std::string> AGENT_CLASS_TO_MANAGED_AGENT = { { "moe.yushi.authlibinjector.Premain",
                                                                                   "moe.yushi:authlibinjector" } };

class Agent;

using AgentPtr = std::shared_ptr<Agent>;

class Agent {
   public:
    Agent(LibraryPtr library, const QString& argument)
    {
        m_library = library;
        m_argument = argument;
    }

   public: /* methods */
    LibraryPtr library() { return m_library; }
    QString argument() { return m_argument; }

   protected: /* data */
    /// The library pointing to the jar this Java agent is contained within
    LibraryPtr m_library;

    /// The argument to the Java agent, passed after an = if present
    QString m_argument;
};