--- a
+++ b/src/main/java/net/timbusproject/dpes/pi/kbserver/beans/QueryBean.java
@@ -0,0 +1,124 @@
+/**
+ * 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 net.timbusproject.dpes.pi.kbserver.beans;
+
+import com.hp.hpl.jena.query.QuerySolution;
+import org.sba_research.timbus.kb.FormatMigrationOption;
+import org.sba_research.timbus.kb.ToolKnowledgeBase;
+import org.sba_research.timbus.kb.timbus.TimbusToolKnowledgeBase;
+import org.sbaresearch.owl.JenaQueryFacade;
+
+import javax.faces.application.FacesMessage;
+import javax.faces.bean.ApplicationScoped;
+import javax.faces.bean.ManagedBean;
+import javax.faces.context.FacesContext;
+import java.io.Serializable;
+import java.util.ArrayList;
+import java.util.List;
+
+import static ch.lambdaj.Lambda.extract;
+import static ch.lambdaj.Lambda.filter;
+import static ch.lambdaj.Lambda.on;
+import static org.hamcrest.Matchers.startsWith;
+
+@ManagedBean
+@ApplicationScoped
+public class QueryBean implements Serializable {
+ public static final String ONTOLOGY_FILE_NAME = "/toolKB_instance.owl";
+
+ private final ToolKnowledgeBase toolKnowledgeBase;
+ private final JenaQueryFacade queryFacade;
+ private String queryString;
+ private final List<String> formatIDs;
+ private final List<String> toolNames;
+
+ public List<String> getQueryResult() {
+ return queryResult;
+ }
+
+ public void setQueryResult(List<String> queryResult) {
+ this.queryResult = queryResult;
+ }
+
+ private List<String> queryResult = new ArrayList<String>();
+
+ public QueryBean() {
+ queryFacade = new JenaQueryFacade(JenaQueryFacade.getModelFromPath(getClass().getResource(ONTOLOGY_FILE_NAME).getFile(), true));
+ toolKnowledgeBase = new TimbusToolKnowledgeBase(queryFacade);
+ formatIDs = toolKnowledgeBase.getAllFormatIDs();
+ toolNames = getAllToolNames();
+ }
+
+ public String getQueryString() {
+ return queryString;
+ }
+
+ public void setQueryString(String queryString) {
+ this.queryString = queryString;
+ }
+
+ public void toolReplacement() {
+ try {
+ queryResult = toolKnowledgeBase.getAlternativeTools(queryString);
+ } catch (Exception e) {
+ handleError(e);
+ }
+ }
+
+ private void handleError(Exception e) {
+ FacesContext.getCurrentInstance().addMessage(null, new FacesMessage(FacesMessage.SEVERITY_ERROR, "Query Error", e.getMessage()));
+ queryResult = new ArrayList<>();
+ }
+
+ public void formatReplacement() {
+ try {
+ List<FormatMigrationOption> convertFromMigrationOptions = toolKnowledgeBase.getConvertFromMigrationOptions(queryString);
+ queryResult = extract(convertFromMigrationOptions, on(FormatMigrationOption.class).getFormat());
+ } catch (Exception e) {
+ handleError(e);
+ }
+ }
+
+ public void rawQuery() {
+ try {
+ List<QuerySolution> query = queryFacade.query(queryString);
+ queryResult = extract(query, on(QuerySolution.class).toString());
+ } catch (Exception e) {
+ handleError(e);
+ }
+ }
+
+ public List<String> completeQuery(String query) {
+ List<String> result = filter(startsWith(query), formatIDs);
+ result.addAll(filter(startsWith(query), toolNames));
+ return result;
+ }
+
+ // TODO: move to kblib
+ public List<String> getAllToolNames() {
+ String queryString = "" +
+ "PREFIX kb: <http://timbus.teco.edu/ontologies/preservationIdentifier/toolKB.owl#>\n" +
+ "\n" +
+ "SELECT ?tool\n" +
+ "{\n" +
+ " ?tool a kb:Tool .\n" +
+ "}\n" +
+ "GROUP BY ?tool";
+ return JenaQueryFacade.removeNamespace(JenaQueryFacade.extractColumn(queryFacade.query(queryString), "tool"));
+ }
+}