package com.turkraft.springfilter.transformer.processor;

import com.fasterxml.jackson.databind.JsonNode;
import com.turkraft.springfilter.helper.FieldTypeResolver;
import com.turkraft.springfilter.language.NotInOperator;
import com.turkraft.springfilter.parser.node.FieldNode;
import com.turkraft.springfilter.parser.node.InfixOperationNode;
import com.turkraft.springfilter.transformer.FilterJsonNodeTransformer;
import com.turkraft.springfilter.transformer.TransformerUtils;
import org.springframework.stereotype.Component;

@Component
public class NotInOperationJsonNodeProcessor implements
        FilterInfixOperationProcessor<FilterJsonNodeTransformer, JsonNode> {

    protected final FieldTypeResolver fieldTypeResolver;
    protected final TransformerUtils transformerUtils;

    public NotInOperationJsonNodeProcessor(
            FieldTypeResolver fieldTypeResolver,
            TransformerUtils transformerUtils) {
        this.fieldTypeResolver = fieldTypeResolver;
        this.transformerUtils = transformerUtils;
    }

    @Override
    public Class<FilterJsonNodeTransformer> getTransformerType() {
        return FilterJsonNodeTransformer.class;
    }

    @Override
    public Class<NotInOperator> getDefinitionType() {
        return NotInOperator.class;
    }

    @Override
    public JsonNode process(FilterJsonNodeTransformer transformer,
                            InfixOperationNode source) {

        transformer.registerTargetType(source, Boolean.class);

        if (source.getLeft() instanceof FieldNode fieldNode) {
            transformer.registerTargetType(source.getRight(),
                    fieldTypeResolver.resolve(transformer.getEntityType(), fieldNode.getName()).getClazz());
        } else if (source.getRight() instanceof FieldNode fieldNode) {
            transformer.registerTargetType(source.getLeft(),
                    fieldTypeResolver.resolve(transformer.getEntityType(), fieldNode.getName()).getClazz());
        }

        JsonNode result = transformer.getObjectMapper().createObjectNode().set("$and",
                transformer.getObjectMapper().createArrayNode()
                        .add(transformer.getObjectMapper().createObjectNode()
                                .set("$isArray", transformer.transform(source.getRight())))
                        .add(transformer.getObjectMapper().createObjectNode()
                                .set("$not", transformer.getObjectMapper().createObjectNode().set("$in",
                                        transformer.getObjectMapper().createArrayNode()
                                                .add(transformer.transform(source.getLeft()))
                                                .add(transformer.transform(source.getRight()))))));

        return transformerUtils.wrapArrays(transformer, result, source);

    }
}