/**
* 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.sba_research.timbus.kb;
import static ch.lambdaj.Lambda.convert;
import java.util.ArrayList;
import java.util.List;
import org.sbaresearch.owl.JenaQueryFacade;
import org.semanticweb.owlapi.model.IRI;
import org.slf4j.LoggerFactory;
import ch.lambdaj.function.convert.Converter;
import com.google.common.base.Stopwatch;
import com.hp.hpl.jena.query.QuerySolution;
/**
* @author Rudolf Mayer
* @author Johannes Binder
*/
public class PackageKnowledgeBase {
private static final org.slf4j.Logger LOG = LoggerFactory.getLogger(PackageKnowledgeBase.class);
private static final String PREFIX_RDFS = "PREFIX rdfs: <http://www.w3.org/2000/01/rdf-schema#>\n";
private static final String PREFIX_KBP = "PREFIX kbp: <http://timbus.teco.edu/ontologies/KB/PackageAlternativesUbuntu_12_10_amd64.owl#>\n";
private static final String PREFIX_CUDF = "PREFIX cudf: <http://timbus.teco.edu/ontologies/DSOs/CUDF.owl#>\n";
private final JenaQueryFacade queryFacade;
public PackageKnowledgeBase(JenaQueryFacade jenaQueryFacade) {
this.queryFacade = jenaQueryFacade;
}
/** Gets the names of all packages */
public List<String> getAllPackageNames() {
String queryString = PREFIX_CUDF + "\n" +
"SELECT ?package\n" +
"WHERE {\n" +
" ?package a cudf:Package .\n" +
"}\n" +
"GROUP BY ?package\n" +
"ORDER BY ?package\n";
return JenaQueryFacade.removeNamespace(JenaQueryFacade.extractColumn(queryFacade.query(queryString), "package"));
}
/** Gets the names of all virtual packages */
public List<String> getAllVirtualPackageNames() {
String queryString = PREFIX_CUDF + "\n" +
"SELECT ?virtualPackage\n" +
"WHERE {\n" +
" ?virtualPackage a cudf:VirtualPackage .\n" +
"}\n" +
"GROUP BY ?virtualPackage\n" +
"ORDER BY ?virtualPackage\n";
return JenaQueryFacade.removeNamespace(JenaQueryFacade.extractColumn(queryFacade.query(queryString), "virtualPackage"));
}
/** Gets all "package provides virtualPackage" relations */
public List<String> getAllVirtualPackageProvidedByRelations() {
LOG.debug("Starting query for all \"package provides virtualPackage\" relations");
// IMPORTANT: in the query below, we have to do FIRST the "package provides virtualPackage" condition
// If we don't do that, there seems to be first a too big projection (package x virtualPackage) created,
// which is filtered only afterwards. Query time then increases from 0.7 seconds to 1m40 seconds...
String queryString = (PREFIX_KBP + PREFIX_CUDF + "\n" +
"SELECT ( CONCAT ( STR(?package), \" provides \", STR(?virtualPackage) ) as ?provides ) ?virtualPackage ?package\n" +
"WHERE {\n" +
" ?package cudf:provides ?virtualPackage .\n" +
" ?virtualPackage a cudf:VirtualPackage .\n" +
" ?package a cudf:Package .\n" +
"}\n" +
"ORDER BY ?virtualPackage\n");
LOG.debug("Query: " + queryString);
Stopwatch timer = Stopwatch.createStarted();
List<QuerySolution> query = queryFacade.query(queryString);
LOG.debug("Query executed, got " + query.size() + " results, took " + timer.toString());
// process results
ArrayList<String> results = new ArrayList<>(query.size());
for (QuerySolution querySolution : query) {
String providesStr = querySolution.get("provides").toString();
String packageStr = querySolution.get("package").toString();
String virtualPackageStr = querySolution.get("virtualPackage").toString();
results.add(providesStr.replace(IRI.create(packageStr).getNamespace(), "") //
.replace(IRI.create(virtualPackageStr).getNamespace(), ""));
}
LOG.debug("Query results processed, total time " + timer.toString());
return results;
}
/** Gets all packages providing any of the virtual packages provided by packageName */
public List<String> getAlternativePackages(String packageName) {
String queryString = (PREFIX_KBP + PREFIX_CUDF + "\n" +
"SELECT ?alternative\n" +
"WHERE {\n" +
" kbp:[package] cudf:provides ?virtual .\n" +
" ?alternative cudf:provides ?virtual .\n" +
" ?alternative a cudf:Package .\n" +
"}\n" +
"GROUP BY ?alternative\n" +
"HAVING (?alternative != kbp:[package])").replace("[package]", packageName);
return JenaQueryFacade.removeNamespace(JenaQueryFacade.extractColumn(queryFacade.query(queryString), "alternative"));
}
/** Gets the packages that are a replacement for packageName, providing (at least) virtualPackage */
public List<String> getAlternativePackages(String packageName, String virtualPackage) {
String queryString = (PREFIX_KBP + PREFIX_CUDF + "\n" +
"SELECT ?alternative\n" +
"WHERE {\n" +
" ?alternative cudf:provides kbp:[virtual_package] .\n" +
" ?alternative a cudf:Package .\n" +
"}\n" +
"GROUP BY ?alternative\n" +
"HAVING (?alternative != kbp:[package])").replace("[package]", packageName).replace("[virtual_package]", virtualPackage);
return JenaQueryFacade.removeNamespace(JenaQueryFacade.extractColumn(queryFacade.query(queryString), "alternative"));
}
/** Gets the virtual packages that are provided by both packages passed */
public List<String> getProvidedVirtualPackages(String packageIRI1, String packageIRI2) {
String queryString = (PREFIX_KBP + PREFIX_CUDF + "\n" +
"SELECT ?provided\n" +
"WHERE {\n" +
" kbp:[package1] cudf:provides ?provided .\n" +
" kbp:[package2] cudf:provides ?provided .\n" +
"}\n" +
"GROUP BY ?provided\n" +
"ORDER BY ?provided").replace("[package1]", packageIRI1).replace("[package2]", packageIRI2);
List<QuerySolution> querySolution = queryFacade.query(queryString);
return JenaQueryFacade.removeNamespace(JenaQueryFacade.extractColumn(querySolution, "provided"));
}
/** Gets all virtual packages provided by the given package */
public List<String> getProvidedVirtualPackages(String packageIRI) {
String queryString = (PREFIX_KBP + PREFIX_CUDF + "\n" +
"SELECT ?virtual\n" +
"WHERE {\n" +
" kbp:[package] cudf:provides ?virtual .\n" +
" ?virtual a cudf:VirtualPackage .\n" +
"}\n" +
"GROUP BY ?virtual\n" +
"ORDER BY ?virtual").replace("[package]", packageIRI);
return JenaQueryFacade.removeNamespace(JenaQueryFacade.extractColumn(queryFacade.query(queryString), "virtual"));
}
/** Gets all virtual packages provided by the given package */
public List<String> getProvidersOf(String virtualPackageIRI) {
String queryString = (PREFIX_KBP + PREFIX_CUDF + "\n" +
"SELECT ?package\n" +
"WHERE {\n" +
" ?package cudf:provides kbp:[virtualPackage] .\n" +
" ?package a cudf:Package .\n" +
" ?virtual a cudf:VirtualPackage .\n" +
"}\n" +
"GROUP BY ?package\n" +
"ORDER BY ?package").replace("[virtualPackage]", virtualPackageIRI);
return JenaQueryFacade.removeNamespace(JenaQueryFacade.extractColumn(queryFacade.query(queryString), "package"));
}
/**
* Note: This expects the ontology to be the model where replacements should be performed, not the package knowledgebase.
* TODO: move to a separate class
*/
public List<String> getVirtualPackages(String packageIRI) {
String queryString = (PREFIX_CUDF + PREFIX_RDFS + "\n" +
"SELECT ?iri (str(?label) as ?strLabel)\n" +
"WHERE {\n" +
" ?iri cudf:providedBy <[packageIRI]> .\n" +
" ?iri a cudf:VirtualPackage .\n" +
" OPTIONAL { ?iri rdfs:label ?label }\n" +
"}\n" +
"GROUP BY ?iri ?label\n").replace("[packageIRI]", packageIRI);
return JenaQueryFacade.removeNamespace(extractIriOrLabel(queryFacade.query(queryString)));
}
/**
* TODO: move to jena facade
*/
public static List<String> extractIriOrLabel(List<QuerySolution> solution) {
return convert(solution, new Converter<QuerySolution, String>() {
@Override
public String convert(QuerySolution solution) {
String label = solution.get("strLabel").toString();
String iri = solution.get("iri").toString();
if (label != null && !label.isEmpty()) return label;
return iri;
}
});
}
}