#include "remove_where.h"
#include <vector>
#include <iostream>
#include <cstdlib>

template <typename T>
void PrintVector(const std::vector<T>& vec)
{
    std::cout << "-----" << std::endl;
    for(const auto& x : vec)
    {
        std::cout << x << std::endl;
    }
}

int main(int argc, char* argv[])
{
    if(argc < 3)
    {
        std::cout << "Usage: " << argv[0] << " (gt|lt|ge|le|ne|eq) operand N N N N ..." << std::endl;
        return 1;
    }

    std::vector<int> vec;
    int operand = atoi(argv[2]);
    std::string oper(argv[1]);

    for(int i = 3; i < argc; ++i)
    {
        vec.push_back(atoi(argv[i]));
    }

    PrintVector(vec);

    if(oper == "gt")
    {
        RemoveWhere(vec) > operand;
    }
    else if(oper == "lt")
    {
        RemoveWhere(vec) < operand;
    }
    else if(oper == "ge")
    {
        RemoveWhere(vec) >= operand;
    }
    else if(oper == "le")
    {
        RemoveWhere(vec) <= operand;
    }
    else if(oper == "eq")
    {
        RemoveWhere(vec) == operand;
    }
    else if(oper == "ne")
    {
        RemoveWhere(vec) != operand;
    }
    else
    {
        std::cout << "Operator " << oper << " not recognized." << std::endl;
        return 1;
    }

    PrintVector(vec);

    return 0;
}