Open Source Projects Europe

/**

 * Copyright (c) 2013/2014 Verein zur Foerderung der IT-Sicherheit in Oesterreich (SBA).

 * The work has been developed in the TIMBUS Project and the above-mentioned are Members of the TIMBUS Consortium.

 * TIMBUS is supported by the European Union under the 7th Framework Programme for research and technological

 * development and demonstration activities (FP7/2007-2013) under grant agreement no. 269940.

 *

 * 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, including without

 * limitation, any warranties or conditions of TITLE, NON-INFRINGEMENT, MERCHANTIBITLY, or FITNESS FOR A PARTICULAR

 * PURPOSE. In no event and under no legal theory, whether in tort (including negligence), contract, or otherwise,

 * unless required by applicable law or agreed to in writing, shall any Contributor be liable for damages, including

 * any direct, indirect, special, incidental, or consequential damages of any character arising as a result of this

 * License or out of the use or inability to use the Work.

 * See the License for the specific language governing permissions and limitation under the License.

 */

package org.sbaresearch.owl;

import ch.lambdaj.function.convert.Converter;

import ch.lambdaj.function.convert.PropertyExtractor;

import com.hp.hpl.jena.query.*;

import com.hp.hpl.jena.rdf.model.Model;

import org.apache.commons.validator.routines.UrlValidator;

import org.semanticweb.owlapi.apibinding.OWLManager;

import org.semanticweb.owlapi.model.IRI;

import java.util.List;

import static ch.lambdaj.Lambda.*;

public class JenaQueryFacade {

    private final Model model;

    public JenaQueryFacade(Model model) {

        this.model = model;

    }

    public List<QuerySolution> query(String queryString) throws InvalidQueryException {

        if (queryString == null || queryString.isEmpty()) {

            throw new IllegalArgumentException("Query is not valid: " + queryString);

        }

        try {

            QueryExecution queryExecution = QueryExecutionFactory.create(QueryFactory.create(queryString), model);

            ResultSet results = queryExecution.execSelect();

            return ResultSetFormatter.toList(results);

        } catch (QueryParseException e) {

            throw new InvalidQueryException(e);

        }

    }

    public static List<String> extractColumn(List<QuerySolution> solution, String columnName) {

        return extract(solution, on(QuerySolution.class).get(columnName).toString());

    }

    public static List<String> removeInColumn(List<String> items, final String needle) {

        // this does not work because of limitations in lambdaj: https://code.google.com/p/lambdaj/wiki/KnownLimitations

        //return extract(items, on(String.class).replace(needle, ""));

        return convert(items, new PropertyExtractor<String, String>(needle) {

            @Override

            public String convert(String o) {

                return o.replaceAll(needle, "");

            }

        });

    }

    public static List<String> removeNamespace(List<String> items) {

        return convert(items, new Converter<String, String>() {

            @Override

            public String convert(String o) {

                return OwlApiFacade.getFragment(IRI.create(o));

            }

        });

    }

    public static Model getModelFromPath(String path, boolean ignoreMissingImports) throws UnableToReadModelException {

        try {

            // fallback, jena seems to have issues with ontologies in the owl format

            OwlApiFacade api = new OwlApiFacade(OWLManager.createOWLOntologyManager(), OWLManager.getOWLDataFactory());

            UrlValidator defaultValidator = new UrlValidator();

            if (!defaultValidator.isValid(path)) path = "file://" + path;

            api.load(path);

            return api.getJenaModel(ignoreMissingImports);

        } catch (Exception f) {

            throw new UnableToReadModelException(f);

        }

    }

}