package com.networknt.schema;

import static org.junit.jupiter.api.Assertions.assertLinesMatch;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Optional;
import java.util.Set;

import org.junit.jupiter.api.Test;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.networknt.schema.SpecVersion.VersionFlag;
import com.networknt.schema.walk.JsonSchemaWalkListener;
import com.networknt.schema.walk.WalkEvent;
import com.networknt.schema.walk.WalkFlow;

class Issue724Test {

    @Test
    void test() throws JsonProcessingException {
        SchemaValidatorsConfig config = new SchemaValidatorsConfig();
        StringCollector stringCollector = new StringCollector();
        config.addKeywordWalkListener(stringCollector);

        String schema =
            "{\n"
                + "  \"$schema\": \"https://json-schema.org/draft/2020-12/schema\",\n"
                + "  \"type\" : \"object\",\n"
                + "  \"properties\" : {\n"
                + "    \"credit_card\": {\n"
                + "      \"type\" : \"string\"\n"
                + "    }\n"
                + "  },\n"
                + "  \"dependentSchemas\": {\n"
                + "    \"credit_card\": {\n"
                + "      \"properties\": {\n"
                + "        \"billing_address\": {\n"
                + "          \"type\" : \"string\"\n"
                + "        }\n"
                + "      },\n"
                + "      \"required\": [\"billing_address\"]\n"
                + "    }\n"
                + "  }\n"
                + "}\n";
        String data =
            "{\n"
                + "  \"credit_card\" : \"my_credit_card\",\n"
                + "  \"billing_address\" : \"my_billing_address\"\n"
                + "}\n";

        JsonSchema jsonSchema = JsonSchemaFactory.getInstance(VersionFlag.V202012).getSchema(schema, config);
        jsonSchema.walk(new ObjectMapper().readTree(data), /* shouldValidateSchema= */ false);

        System.out.println(stringCollector.strings);
        assertLinesMatch(Arrays.asList("my_credit_card", "my_billing_address"), stringCollector.strings);
    }

    static class StringCollector implements JsonSchemaWalkListener {
        final List<String> strings = new ArrayList<>();

        @Override
        public WalkFlow onWalkStart(WalkEvent walkEvent) {
            boolean isString =
                Optional.of(walkEvent.getSchema().getSchemaNode())
                    .map(jsonNode -> jsonNode.get("type"))
                    .map(JsonNode::asText)
                    .map(type -> type.equals("string"))
                    .orElse(false);

            if (isString) {
                this.strings.add(walkEvent.getInstanceNode().asText());
            }

            return WalkFlow.CONTINUE;
        }

        @Override
        public void onWalkEnd(WalkEvent walkEvent, Set<ValidationMessage> validationMessages) {
            // nothing to do here
        }
    }

}