UriMediaTypeHandler.java

/*
 * Copyright (c) 2012, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.wso2.carbon.governance.registry.extensions.handlers;

import org.apache.axiom.om.OMElement;
import org.apache.axiom.om.util.AXIOMUtil;
import org.apache.axiom.om.xpath.AXIOMXPath;
import org.apache.commons.logging.Log;
import org.jaxen.SimpleNamespaceContext;
import org.wso2.carbon.governance.registry.extensions.handlers.utils.HandlerConstants;
import org.wso2.carbon.registry.core.Registry;
import org.wso2.carbon.registry.core.Resource;
import org.wso2.carbon.registry.core.exceptions.RegistryException;
import org.wso2.carbon.registry.core.jdbc.handlers.Handler;
import org.wso2.carbon.registry.core.jdbc.handlers.RequestContext;
import org.wso2.carbon.registry.core.utils.RegistryUtils;

import java.util.Iterator;
import java.util.List;

public class UriMediaTypeHandler extends Handler {
    private static final Log log = org.apache.commons.logging.LogFactory.getLog(UriMediaTypeHandler.class);

    @Override
    public void put(RequestContext requestContext) throws org.wso2.carbon.registry.core.exceptions.RegistryException {
        Registry registry = requestContext.getRegistry();
        Resource resource = requestContext.getResource();
        String resourcePath = requestContext.getResourcePath().getPath();
        String fileUri = null;
        String type = null;

        try {
            String newContent;
            if(resource.getContent() instanceof String){
                newContent = (String) resource.getContent();
            } else {
                newContent = RegistryUtils.decodeBytes((byte[])resource.getContent());
            }

            if(registry.resourceExists(resourcePath)){
                return;
            }

            OMElement docElement = AXIOMUtil.stringToOM(newContent);
            SimpleNamespaceContext simpleNamespaceContext = new SimpleNamespaceContext();
            simpleNamespaceContext.addNamespace("pre",docElement.getNamespace().getNamespaceURI());
            AXIOMXPath expression = new AXIOMXPath("//pre:overview");
            expression.setNamespaceContext(simpleNamespaceContext);
            List<OMElement> overview = expression.selectNodes(docElement);
            Iterator<OMElement> elements = overview.get(0).getChildElements();

            while (elements.hasNext()){
                OMElement element = elements.next();
                String localName = element.getLocalName();
                if("uri".equals(localName)){
                    fileUri = element.getText();
                } else if ("type".equals(localName)){
                    type = element.getText();
                }
            }

        } catch (Exception e) {
            String msg = "Failed to parse content of URI " + fileUri + ".";
            log.error(msg, e);
            throw new RegistryException(msg, e);
        }

        if(HandlerConstants.WSDL.equals(type)){
            WsdlUriHandler wsdlUriHandler = new WsdlUriHandler();
            wsdlUriHandler.importResource(requestContext, fileUri);
        } else if(HandlerConstants.XSD.equals(type)){
            SchemaUriHandler schemaUriHandler = new SchemaUriHandler();
            schemaUriHandler.importResource(requestContext, fileUri);
        } else if(HandlerConstants.POLICY.equals(type)){
            PolicyUriHandler policyUriHandler = new PolicyUriHandler();
            policyUriHandler.importResource(requestContext, fileUri);
        }

    }
}