PollForReportImpl.java

package org.openiot.sdum.core.api.impl.PollForReport;


/**
 *    Copyright (c) 2011-2014, OpenIoT
 *    
 *    This file is part of OpenIoT.
 *
 *    OpenIoT is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU Lesser General Public License as published by
 *    the Free Software Foundation, version 3 of the License.
 *
 *    OpenIoT is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU Lesser General Public License for more details.
 *
 *    You should have received a copy of the GNU Lesser General Public License
 *    along with OpenIoT.  If not, see <http://www.gnu.org/licenses/>.
 *
 *     Contact: OpenIoT mailto: info@openiot.eu
 */

import java.io.StringReader;
import java.util.ArrayList;
import java.util.Set;

import org.openiot.commons.sdum.serviceresultset.model.PresentationAttr;
import org.openiot.commons.sdum.serviceresultset.model.RequestPresentation;
import org.openiot.commons.sdum.serviceresultset.model.SdumServiceResultSet;
import org.openiot.commons.sdum.serviceresultset.model.Widget;
import org.openiot.commons.sparql.protocoltypes.model.QueryRequest;
import org.openiot.commons.sparql.protocoltypes.model.QueryResult;
import org.openiot.commons.sparql.result.model.Binding;
import org.openiot.commons.sparql.result.model.Head;
import org.openiot.commons.sparql.result.model.Literal;
import org.openiot.commons.sparql.result.model.Result;
import org.openiot.commons.sparql.result.model.Results;
import org.openiot.commons.sparql.result.model.Sparql;
import org.openiot.commons.sparql.result.model.Variable;
import org.openiot.commons.util.PropertyManagement;
import org.openiot.sdum.core.utils.sparql.SesameSPARQLClient;

import org.openrdf.query.BindingSet;
import org.openrdf.query.QueryEvaluationException;
import org.openrdf.query.TupleQueryResult;
import org.openrdf.repository.RepositoryException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.w3c.dom.Document;
import org.xml.sax.InputSource;

/**
 * 
 * @author Stavros Petris (spet) e-mail: spet@ait.edu.gr
 * @author Nikos Kefalakis (nkef) e-mail: nkef@ait.edu.gr
 */

public class PollForReportImpl 
{	
	
	private static class Queries
	{
		public static class QueryData
		{
			private String id;
			private String queryString;			
			
			public String getId() {
				return id;
			}
			public void setId(String id) {
				this.id = id;
			}
			public String getQueryString() {
				return queryString;
			}
			public void setQueryString(String queryString) {
				this.queryString = queryString;
			}			
		}
		public static class ServicePresentationData 
		{
			private String widgetPreID;
			private String widgetID;
			private String widgetAttrID;
			private String widgetAttrName;
			private String widgetAttrDesc;
			
			
			public String getWidgetPreID() {
				return widgetPreID;
			}
			public void setWidgetPreID(String widgetPreID) {
				this.widgetPreID = widgetPreID;
			}
			
			public String getWidgetID() {
				return widgetID;
			}
			public void setWidgetID(String widgetID) {
				this.widgetID = widgetID;
			}
			
			public String getWidgetAttrID() {
				return widgetAttrID;
			}
			public void setWidgetAttrID(String widgetAttrID) {
				this.widgetAttrID = widgetAttrID;
			}
			
			public String getWidgetAttrName() {
				return widgetAttrName;
			}
			public void setWidgetAttrName(String widgetAttrName) {
				this.widgetAttrName = widgetAttrName;
			}
			
			public String getWidgetAttrDesc() {
				return widgetAttrDesc;
			}
			public void setWidgetAttrDesc(String widgetAttrDesc) {
				this.widgetAttrDesc = widgetAttrDesc;
			}
		}
				
		public static ArrayList<QueryData> parseOSMOQueryData(TupleQueryResult qres)
		{
			ArrayList<QueryData> queryDataList = new ArrayList<QueryData>();
			
			try 
			{
				while (qres.hasNext())
				{
					BindingSet b = qres.next();
					Set names = b.getBindingNames();
					QueryData queryData = new QueryData();
					
					for (Object n : names)
					{						
						if(((String) n).equalsIgnoreCase("queryID"))
						{
							String str = (b.getValue((String) n)==null) ? null : b.getValue((String) n).stringValue();
							queryData.setId(str);
							System.out.print("srvcStatus id: "+queryData.getId()+" ");	
						}
						else if(((String) n).equalsIgnoreCase("queryString"))
						{
							String str = (b.getValue((String) n)==null) ? null : b.getValue((String) n).stringValue();
							queryData.setQueryString(str);
							System.out.print("srvcStatusTime : "+queryData.getQueryString()+" ");	
						}
					}
					queryDataList.add(queryData);					
				}//while
				return queryDataList;
			} 
			catch (QueryEvaluationException e)			
			{				
				e.printStackTrace();
				return null;
			}
			catch (Exception e)			
			{				
				e.printStackTrace();
				return null;
			}
		}
		public static ArrayList<ServicePresentationData> parseWPresentationFromService(TupleQueryResult qres)
		{
			ArrayList<ServicePresentationData> srvcPreDataList = new ArrayList<ServicePresentationData>();
			try
			{
				while (qres.hasNext())
				{
					BindingSet b = qres.next();
					Set names = b.getBindingNames();
					ServicePresentationData srvcPreData = new ServicePresentationData();
					
					for (Object n : names)
					{						
						if(((String) n).equalsIgnoreCase("widgetPreID"))
						{
							String str = (b.getValue((String) n)==null) ? null : b.getValue((String) n).stringValue();
							srvcPreData.setWidgetPreID(str);
							System.out.print("widgetPreID : "+str+" ");	
						}
						else if(((String) n).equalsIgnoreCase("widgetID"))
						{
							String str = (b.getValue((String) n)==null) ? null : b.getValue((String) n).stringValue();
							srvcPreData.setWidgetID(str);
							System.out.print("widgetID : "+str+" ");	
						}
						else if(((String) n).equalsIgnoreCase("widgetAttrID"))
						{
							String str = (b.getValue((String) n)==null) ? null : b.getValue((String) n).stringValue();
							srvcPreData.setWidgetAttrID(str);
							System.out.print("widgetAttrID : "+str+" ");	
						}
						else if(((String) n).equalsIgnoreCase("widgetAttrName"))
						{
							String str = (b.getValue((String) n)==null) ? null : b.getValue((String) n).stringValue();
							srvcPreData.setWidgetAttrName(str);
							System.out.print("widgetAttrName : "+str+" ");	
						}
						else if(((String) n).equalsIgnoreCase("widgetAttrDesc"))
						{
							String str = (b.getValue((String) n)==null) ? null : b.getValue((String) n).stringValue();
							srvcPreData.setWidgetAttrDesc(str);
							System.out.print("widgetAttrDesc : "+str+" ");	
						}
					}
					srvcPreDataList.add(srvcPreData);					
				}//while
				return srvcPreDataList;
			} 
			catch (QueryEvaluationException e)			
			{				
				e.printStackTrace();
				return null;
			}
			catch (Exception e)			
			{				
				e.printStackTrace();
				return null;
			}
		}
		
		
		public static String getQueryListOfOSMO(String openiotFunctionalGraph,String osmoID)
		{
			StringBuilder update = new StringBuilder();			
			
			String str=("SELECT ?queryID ?queryString " 
					+"from <"+openiotFunctionalGraph+"> "
					+"WHERE "
					+"{"
					+"?queryID <http://openiot.eu/ontology/ns/queryString> ?queryString . "
					+"<"+osmoID+"> <http://openiot.eu/ontology/ns/query> ?queryID . "
					+"}");
			
			update.append(str);
			return update.toString();
		}
		public static String getWPresentationFromOSMO(String openiotFunctionalGraph,String serviceID)
		{
			StringBuilder update = new StringBuilder();	        
			
			String str=("SELECT ?widgetPreID ?widgetID  ?widgetAttrID  ?widgetAttrName ?widgetAttrDesc " 
					+"from <"+openiotFunctionalGraph+"> "
					+"WHERE "
					+"{"
					+"?widgetAttrID <http://openiot.eu/ontology/ns/widgeAttrDescription> ?widgetAttrDesc . "
					+"?widgetAttrID <http://openiot.eu/ontology/ns/widgetAttrName> ?widgetAttrName . "
					+"?widgetPreID <http://openiot.eu/ontology/ns/widgetAttribute> ?widgetAttrID . "
					+"?widgetPreID <http://openiot.eu/ontology/ns/widget> ?widgetID . "
					+"?widgetPreID <http://openiot.eu/ontology/ns/widgetPresOf> <"+serviceID+"> ."
					+"}");	
			
			update.append(str);
			return update.toString();
		}
	}
	
	final static Logger logger = LoggerFactory.getLogger(PollForReportImpl.class);
	
	private String openiotFunctionalGraph;
	//
	private String serviceID;
	private SdumServiceResultSet sdumServiceResultSet=null;
	
	//cosntructor
	public PollForReportImpl(String serviceID) 
	{		
		PropertyManagement propertyManagement = new PropertyManagement();
		openiotFunctionalGraph = propertyManagement.getSdumLsmFunctionalGraph();
		
		logger.debug("Recieved Parameters: serviceID= {}", serviceID);
		
		this.serviceID=serviceID;
		
		pollForReport();		
	}	
	
	
	public SdumServiceResultSet getSdumServiceResultSet() 
	{
		return sdumServiceResultSet;
	}
	
	
	//helper methods
	
	private void pollForReport() 
	{
		SesameSPARQLClient sparqlCl = null;
		try {
			sparqlCl = new SesameSPARQLClient();
		} catch (RepositoryException e) {			
			logger.error("Init sparql repository error. ",e);
			return;
		}
		
		TupleQueryResult qres = sparqlCl.sparqlToQResult(Queries.getQueryListOfOSMO(openiotFunctionalGraph,serviceID));
		ArrayList<Queries.QueryData> queryDataList = Queries.parseOSMOQueryData(qres);
		
		SdumServiceResultSet sdumServiceResultSet = new SdumServiceResultSet();
		
		for (Queries.QueryData queryData : queryDataList)
		{
			try
			{
				TupleQueryResult qresultOfOSMOQuery = sparqlCl.sparqlToQResult(queryData.getQueryString());
				
				Sparql sparql = new Sparql();
				
				Head head = new Head();
				for (String value : qresultOfOSMOQuery.getBindingNames()) 
				{			
					Variable var = new Variable();
					var.setName(value);
					head.getVariable().add(var);
				}
				
				sparql.setHead(head);
							
				Results sparqlResults = new Results();	
				while (qresultOfOSMOQuery.hasNext())
				{				
					BindingSet b = qresultOfOSMOQuery.next();
					Set<String> names = b.getBindingNames();				
					
					Result sparqlResult = new Result();
					for (String n : names)
					{
						Binding sparqlResultBinding = new Binding();
						sparqlResultBinding.setName(n);
						
						Literal literalValue = new Literal();
						literalValue.setContent(b.getValue((String) n).stringValue());
						sparqlResultBinding.setLiteral(literalValue );
	//					sparqlResultBinding.setBnode("BnodeValue");		
	//					sparqlResultBinding.setUri("UriValue");
						
						sparqlResult.getBinding().add(sparqlResultBinding);
					}
					
					sparqlResults.getResult().add(sparqlResult);				
				}//while
				
				sparql.setResults(sparqlResults);
				
				QueryResult queryResult = new QueryResult();
				queryResult.setSparql(sparql);
				
				sdumServiceResultSet.getQueryResult().add(queryResult);			
			} 
			catch (QueryEvaluationException e)			
			{				
				e.printStackTrace();			
			}
			catch (Exception e)			
			{				
				e.printStackTrace();			
			}
			
		}//for
		
		///////////////////////////////////
		
		TupleQueryResult qres3 = sparqlCl.sparqlToQResult(Queries.getWPresentationFromOSMO(openiotFunctionalGraph,this.serviceID));				
		ArrayList<Queries.ServicePresentationData> srvcPreDatas = Queries.parseWPresentationFromService(qres3);
		
		ArrayList<String> distinctWidgetP = new ArrayList<String>();
		
		//Fill the RequestPresentation
		RequestPresentation requestPresentation = new RequestPresentation();
		
		for (Queries.ServicePresentationData srvcPreData : srvcPreDatas) 
		{
			Widget widget = new Widget();
			widget.setWidgetID(srvcPreData.getWidgetID());
			
			int idx = checkExists(widget.getWidgetID(),requestPresentation);
			if(idx==-1){
				requestPresentation.getWidget().add(widget);
			}
			else{
				widget = requestPresentation.getWidget().get(idx);
			}
			
			PresentationAttr presentationAttr = new PresentationAttr();
			presentationAttr.setName(srvcPreData.getWidgetAttrName());
			presentationAttr.setValue(srvcPreData.getWidgetAttrDesc());
			
			widget.getPresentationAttr().add(presentationAttr);			
		}
		
		sdumServiceResultSet.setRequestPresentation(requestPresentation);
		
		this.sdumServiceResultSet = sdumServiceResultSet;
	}
	
	private int checkExists(String widgetID,RequestPresentation requestPresentation)
	{
		if(requestPresentation.getWidget().isEmpty())
		{
			return -1;
		}
		else
		{
			for (int i=0; i<requestPresentation.getWidget().size(); i++) 
			{
				if(widgetID.equals(requestPresentation.getWidget().get(i).getWidgetID()))
				{
					return i;
				}				
			}
			
			return -1;
		}
	}
}