#pragma once
#include <vector>
#include <stdint.h>

namespace Cam
{
namespace Arguments
{

enum ERROR {
	IMPOSSIBLE = -1,
	NO_ERROR,
	ERROR_UNKNOWN_OPTION,
	ERROR_INCORRECT_TYPE,
	ERROR_MISSING_ARGUMENT,
	ERROR_MISSING_POSITIONAL_ARGUMENT,
	SPECIAL_CASE_HELP
};

enum OPTTYPE {
	FLAG = 0,
	STRING,
	INT
};

class Option
{
public:
	Option(const char* name, const char* description, OPTTYPE type);
	Option(const char* name, const char short_name, const char* description, OPTTYPE type);
	Option(const char short_name, const char* description, OPTTYPE type);
	bool found();
	void* data;
private:
	const char* m_name;
	const char m_short_name;
	const char* m_description;
	OPTTYPE m_type;
	bool m_found;

	friend class Parser;
};

class PositionalArgument
{
public:
	PositionalArgument(const char* name, bool required, OPTTYPE type = STRING);
	bool found();
	void* data;
private:
	const char* m_name;
	OPTTYPE m_type;
	bool m_req;
	bool m_found;

	friend class Parser;
};

class Parser
{
public:
	Parser(const char* program_name);
	void set_description(const char* description);
	void add_option(Option* opt);
	void add_positional_argument(PositionalArgument* arg);
	ERROR parse(int argc, char** argv);
private:
	ERROR print_help_message();
	ERROR handle_long_option(const char* option, const char* next_value);
	ERROR handle_short_option(const char option, const char* next_value);
	ERROR handle_positional_argument(const char* arg);
	ERROR get_option_data(Option* opt, const char* option_name, const char* data_str);
	ERROR unknown_option(const char* option);
	ERROR unknown_option(const char option);
	ERROR incorrect_type(const char* option, const char* got);
	ERROR missing_argument(const char* option);
	ERROR missing_positional_argument(const char* arg);
private:
	const char* m_program_name;
	const char* m_description;
	uint16_t m_opt_index;
	uint16_t m_pos_index;
	std::vector<Option*> m_options;
	std::vector<PositionalArgument*> m_positional_args;
};

} // namespace Arguments
} // namespace Cam