#include "TableEvalValue.hpp"
#include "../Iterator/SimpleKeyIterator.hpp"
#include "NumericEvalValue.hpp"
#include "../Iterator/NumericalKeyIterator.hpp"

Porygon::Evaluation::Iterator * Porygon::Evaluation::TableEvalValue::GetKeyIterator() const {
    return new TableKeyIterator(this);
}

Porygon::Evaluation::EvalValue *
Porygon::Evaluation::TableEvalValue::UnaryOperation(Porygon::Binder::BoundUnaryOperation operation) const {
    if (operation == Porygon::Binder::BoundUnaryOperation::Count){
        return new NumericEvalValue(static_cast<int64_t>(this->_table->size()));
    }
    return EvalValue::UnaryOperation(operation);
}

void Porygon::Evaluation::TableEvalValue::SetIndexValue(const Porygon::Utilities::HashedString *key,
                                                        const Porygon::Evaluation::EvalValue *value) const {
    auto insert = _table->insert({*key, value});
    if (!insert.second) {
        _table->at(*key).ClearAssign(value);
    }

}

Porygon::Evaluation::Iterator *Porygon::Evaluation::NumericTableEvalValue::GetKeyIterator() const {
    return new NumericalKeyIterator(this);
}

Porygon::Evaluation::EvalValue *Porygon::Evaluation::NumericTableEvalValue::Clone() const {
    return new NumericTableEvalValue(_table, _hash);
}