io.polyapi.plugin.service.schema.PublicEnumRule Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of polyapi-maven-plugin Show documentation

Maven plugin to run handle Poly API functions.

There is a newer version: 0.15.3

package io.polyapi.plugin.service.schema;

import static java.util.Spliterator.ORDERED;
import static java.util.stream.StreamSupport.stream;

import java.util.Spliterators;

import org.jsonschema2pojo.Schema;
import org.jsonschema2pojo.rules.EnumRule;
import org.jsonschema2pojo.rules.RuleFactory;

import com.fasterxml.jackson.databind.JsonNode;
import com.sun.codemodel.JClassContainer;
import com.sun.codemodel.JType;

public class PublicEnumRule extends EnumRule {

    protected PublicEnumRule(RuleFactory ruleFactory) {
        super(ruleFactory);
    }

    @Override
    public JType apply(String nodeName, JsonNode node, JsonNode parent, JClassContainer container, Schema schema) {
        return stream(Spliterators.spliteratorUnknownSize(container.getPackage().classes(), ORDERED), false)
                .filter(definedClass -> definedClass.name().equalsIgnoreCase(nodeName))
                .findFirst()
                .orElseGet(() -> super.apply(nodeName, node, parent, container.getPackage(), schema));
    }

    @Override
    protected String getConstantName(String nodeName, String customName) {
        return super.getConstantName(nodeName.replace("-", "_"), customName);
    }
}