package io.apicurio.registry.wsdl.rules.validity;

import io.apicurio.registry.content.TypedContent;
import io.apicurio.registry.rules.validity.ValidityLevel;
import io.apicurio.registry.xml.rules.validity.XmlContentValidator;
import io.apicurio.registry.rules.violation.RuleViolationException;
import io.apicurio.registry.types.RuleType;
import io.apicurio.registry.wsdl.util.WSDLReaderAccessor;
import io.apicurio.registry.xml.util.DocumentBuilderAccessor;
import org.w3c.dom.Document;

import java.io.InputStream;
import java.util.Map;

public class WsdlContentValidator extends XmlContentValidator {

    /**
     * Constructor.
     */
    public WsdlContentValidator() {
    }

    /**
     * @see io.apicurio.registry.rules.validity.ContentValidator#validate(ValidityLevel, TypedContent, Map)
     */
    @Override
    public void validate(ValidityLevel level, TypedContent content,
            Map<String, TypedContent> resolvedReferences) throws RuleViolationException {
        if (level == ValidityLevel.SYNTAX_ONLY || level == ValidityLevel.FULL) {
            try (InputStream stream = content.getContent().stream()) {
                Document wsdlDoc = DocumentBuilderAccessor.getDocumentBuilder().parse(stream);
                if (level == ValidityLevel.FULL) {
                    // validate that its a valid schema
                    WSDLReaderAccessor.getWSDLReader().readWSDL(null, wsdlDoc);
                }
            } catch (Exception e) {
                throw new RuleViolationException("Syntax violation for WSDL Schema artifact.",
                        RuleType.VALIDITY, level.name(), e);
            }
        }
    }
}
