mindswap/wsdl/WSDLTranslator.java

//The MIT License
//
// Copyright (c) 2004 Evren Sirin
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to
// deal in the Software without restriction, including without limitation the
// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
// sell copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
// IN THE SOFTWARE.

package org.mindswap.wsdl;

import java.io.OutputStream;
import java.io.Writer;
import java.net.URI;

import org.mindswap.owl.OWLFactory;
import org.mindswap.owl.OWLKnowledgeBase;
import org.mindswap.owl.OWLOntology;
import org.mindswap.owl.OWLType;
import org.mindswap.owls.grounding.AtomicGrounding;
import org.mindswap.owls.grounding.Grounding;
import org.mindswap.owls.grounding.WSDLAtomicGrounding;
import org.mindswap.owls.process.AtomicProcess;
import org.mindswap.owls.process.Input;
import org.mindswap.owls.process.Output;
import org.mindswap.owls.profile.Profile;
import org.mindswap.owls.service.Service;
import org.mindswap.utils.URIUtils;

public class WSDLTranslator {
    private Service service;
    private OWLOntology ont;
    URI baseURI;


    public WSDLTranslator(WSDLOperation op, String serviceURI, String prefix) {
        OWLKnowledgeBase kb = OWLFactory.createKB();
        ont = kb.createOntology();

        baseURI = URIUtils.createURI(serviceURI);

        service = ont.createService(URIUtils.createURI(baseURI, prefix + "Service"));

        Profile profile = ont.createProfile(URIUtils.createURI(baseURI, prefix + "Profile"));
        AtomicProcess process = ont.createAtomicProcess(URIUtils.createURI(baseURI, prefix + "Process"));
        Grounding grounding = ont.createGrounding(URIUtils.createURI(baseURI, prefix + "Grounding"));
        WSDLAtomicGrounding ag = ont.createWSDLAtomicGrounding(URIUtils.createURI(baseURI, prefix + "AtomicProcessGrounding"));

        process.setLabel(process.getURI().getFragment());

        // set the links between structures
        service.setProfile(profile);
        service.setProcess(process);
        service.setGrounding(grounding);

        // add the WSDL details to the atomic grounding
        ag.setProcess(process);
        ag.setOperation(URI.create(op.getOperationName()));
        ag.setWSDL(op.getService().getFileURI());
        ag.setPortType(URI.create(op.getPortName()));
        ag.setInputMessage(URI.create(op.getInputMessageName()));
        ag.setOutputMessage(URI.create(op.getOutputMessageName()));

        // add the atomic process grounding to service grounding
        grounding.addGrounding(ag);
    }

    public void setServiceName(String serviceName) {
        service.getProfile().setServiceName(serviceName);
    }

    public void setTextDescription(String textDescription) {
        service.getProfile().setTextDescription(textDescription);
        System.out.println(service.getProfile().getTextDescription());
    }

    public void addInput(WSDLParameter param, String paramName, URI paramType, String xsltTransformation) {
        Profile profile = service.getProfile();
        AtomicProcess process = (AtomicProcess) service.getProcess();

        // create process param
        Input input = ont.createInput(URIUtils.createURI(baseURI, paramName));
        input.setLabel(paramName);

        OWLType type = ont.getType(paramType);
        input.setParamType(type == null ? ont.createClass(paramType) : type);

        // add the param to process and profile
        process.addInput(input);
        profile.addInput(input);

        AtomicGrounding grounding = process.getGrounding();
        // create grounding message map
        grounding.addMessageMap(input, param.getName(), xsltTransformation);
    }

    public void addOutput(WSDLParameter param, String paramName, URI paramType, String xsltTransformation) {
        Profile profile = service.getProfile();
        AtomicProcess process = (AtomicProcess) service.getProcess();

        // create process param
        Output output = ont.createOutput(URIUtils.createURI(baseURI, paramName));
        output.setLabel(paramName);

        OWLType type = ont.getType(paramType);
        output.setParamType(type == null ? ont.createClass(paramType) : type);

        // add the param to process and profile
        process.addOutput(output);
        profile.addOutput(output);

        AtomicGrounding grounding = process.getGrounding();
        // create grounding message map
        grounding.addMessageMap(output, param.getName(), xsltTransformation);
    }

    public void writeOWLS(Writer out) {
        ont.write(out, baseURI);
    }

    public void writeOWLS(OutputStream out) {
        ont.write(out, baseURI);
    }

    public Service getService() {
        return service;
    }
}