#pragma once

#include <iostream>
#include <random>
#include <ctime>
#include <string>

using namespace std;

inline void ltrim(string &str) {
    str.erase(str.begin(), find_if(str.begin(), str.end(), [](unsigned char ch) {
        return !isspace(ch);
    }));
}

inline void rtrim(string &str) {
    str.erase(find_if(str.rbegin(), str.rend(), [](unsigned char ch) {
        return !isspace(ch);
    }).base(), str.end());
}

inline void trim(string &str) {
    rtrim(str);
    ltrim(str);
}

Segment determineSegment(string segment)
{
    if (segment == "local")
        return Segment::LCL;
    if (segment == "argument")
        return Segment::ARG;
    if (segment == "this")
        return Segment::THIS;
    if (segment == "that")
        return Segment::THAT;
    if (segment == "pointer")
        return Segment::POINTER;
    if (segment == "static")
        return Segment::STATIC;
    if (segment == "temp")
        return Segment::TEMP;
    if (segment == "constant")
        return Segment::CONSTANT;

    return Segment::CONSTANT;
}

string generateRandomLabel(int length) {
    random_device rd;
    mt19937 gen(rd());
    uniform_int_distribution<> dis('A', 'Z');
    string label;
    
    for (int i = 0; i < length; ++i) {
        label += static_cast<char>(dis(gen));
    }

    return label;
}

NoArgumentCommand determineNoArgumentCommand(string command)
{
    if (command == "add")
        return NoArgumentCommand::ADD;
    if (command == "sub")
        return NoArgumentCommand::SUB;
    if (command == "neg")
        return NoArgumentCommand::NEG;
    if (command == "eq")
        return NoArgumentCommand::EQ;
    if (command == "gt")
        return NoArgumentCommand::GT;
    if (command == "lt")
        return NoArgumentCommand::LT;
    if (command == "and")
        return NoArgumentCommand::AND;
    if (command == "or")
        return NoArgumentCommand::OR;
    if (command == "not")
        return NoArgumentCommand::NOT;
    if (command == "return")
        return NoArgumentCommand::RETURN;

    return NoArgumentCommand::NEG;
}

OneArgumentCommand determineOneArgumentCommand(string command)
{
    if (command == "label")
        return OneArgumentCommand::LABEL;
    if (command == "goto")
        return OneArgumentCommand::GOTO;
    if (command == "if-goto")
        return OneArgumentCommand::IFGOTO;

    return OneArgumentCommand::IFGOTO;
}

TwoArgumentCommand determineTwoArgumentCommand(string command)
{
    if (command == "push")
        return TwoArgumentCommand::PUSH;
    if (command == "pop")
        return TwoArgumentCommand::POP;
    if (command == "function")
        return TwoArgumentCommand::FUNCTION;
    if (command == "call")
        return TwoArgumentCommand::CALL;

    return TwoArgumentCommand::POP;
}