/* ======================================================================
The Bodington System Software License, Version 1.0
  
Copyright (c) 2001 The University of Leeds.  All rights reserved.
  
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are
met:

1.  Redistributions of source code must retain the above copyright notice,
this list of conditions and the following disclaimer.

2.  Redistributions in binary form must reproduce the above copyright
notice, this list of conditions and the following disclaimer in the
documentation and/or other materials provided with the distribution.

3.  The end-user documentation included with the redistribution, if any,
must include the following acknowledgement:  "This product includes
software developed by the University of Leeds
(http://www.bodington.org/)."  Alternately, this acknowledgement may
appear in the software itself, if and wherever such third-party
acknowledgements normally appear.

4.  The names "Bodington", "Nathan Bodington", "Bodington System",
"Bodington Open Source Project", and "The University of Leeds" must not be
used to endorse or promote products derived from this software without
prior written permission. For written permission, please contact
d.gardner@leeds.ac.uk.

5.  The name "Bodington" may not appear in the name of products derived
from this software without prior written permission of the University of
Leeds.

THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
WARRANTIES, INCLUDING, BUT NOT LIMITED TO,  TITLE,  THE IMPLIED WARRANTIES 
OF QUALITY  AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO 
EVENT SHALL THE UNIVERSITY OF LEEDS OR ITS CONTRIBUTORS BE LIABLE FOR 
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE 
GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 
HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, 
STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN 
ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 
POSSIBILITY OF SUCH DAMAGE.
=========================================================

This software was originally created by the University of Leeds and may contain voluntary 
contributions from others.  For more information on the Bodington Open Source Project, please 
see http://bodington.org/

====================================================================== */

package org.bodington.assessment;

import java.util.logging.*;



import java.rmi.RemoteException;
import java.util.Vector;
import java.util.Enumeration;
import java.util.Hashtable;
import java.sql.Timestamp;
import java.io.*;

import org.xml.sax.*;
import org.xml.sax.helpers.*;
import javax.xml.parsers.ParserConfigurationException;  


import org.bodington.assessment.ims.QuestionnaireImsQtiHandler;
import org.bodington.database.PrimaryKey;
import org.bodington.server.*;
import org.bodington.text.BigString;
import org.bodington.server.realm.User;



public class QuestionnaireSessionImpl
	extends org.bodington.server.SingleResourceSession 
	implements QuestionnaireSession 
	{
	int max_ordinal=0;
	
	public QuestionnaireSessionImpl()
		throws RemoteException, BuildingServerException
		{
		super();
		}
	

	public Questionnaire getQuestionnaire()
		throws RemoteException, BuildingServerException
		{
		Questionnaire paper = Questionnaire.findQuestionnaire( getResource().getResourceId() );
		return paper;
		}
		
	
	public Vector getQuestionnaireQuestionIds()
		throws RemoteException, BuildingServerException
		{
		Vector list = new Vector();
		
		QuestionnaireQuestion question;
		Enumeration enum = QuestionnaireQuestion.findQuestionnaireQuestions( "resource_id = " + getResource().getResourceId().toString(), "ordinal" );
		while ( enum.hasMoreElements() )
			{
			question = (QuestionnaireQuestion)enum.nextElement();
			list.addElement( question.getQuestionnaireQuestionId() ); 
			if ( question.getOrdinal() > max_ordinal )
				max_ordinal = question.getOrdinal();
			}
		
		return list;
		}
	
	public Hashtable getQuestionnaireQuestions()
		throws RemoteException, BuildingServerException
		{
		Hashtable table = new Hashtable();
		
		QuestionnaireQuestion question;
		Enumeration enum = QuestionnaireQuestion.findQuestionnaireQuestions( "resource_id = " + getResource().getResourceId().toString() );
		while ( enum.hasMoreElements() )
			{
			question = (QuestionnaireQuestion)enum.nextElement();
			table.put( question.getQuestionnaireQuestionId(), question );
			if ( question.getOrdinal() > max_ordinal )
				max_ordinal = question.getOrdinal();
			}
		
		return table;
		}
	
	public Vector getQuestionnaireQuestionsInOrder()
		throws RemoteException, BuildingServerException
		{
		Vector list = new Vector();
		
		QuestionnaireQuestion question;
		Enumeration enum = QuestionnaireQuestion.findQuestionnaireQuestions( "resource_id = " + getResource().getResourceId().toString(), "ordinal" );
		while ( enum.hasMoreElements() )
			{
			question = (QuestionnaireQuestion)enum.nextElement();
			list.addElement( question ); 
			if ( question.getOrdinal() > max_ordinal )
				max_ordinal = question.getOrdinal();
			}
		
		return list;
		}
	
	public Vector getQuestionnaireResults()
		throws RemoteException, BuildingServerException
		{
		Vector list = new Vector();
		
		QuestionnaireResult result;
		Enumeration enum = QuestionnaireResult.findQuestionnaireResults( "resource_id = " + getResource().getResourceId().toString() );
		while ( enum.hasMoreElements() )
			{
			result = (QuestionnaireResult)enum.nextElement();
			list.addElement( result ); 
			}
		
		return list;
		}
	
	public Hashtable getStrings()
		throws RemoteException, BuildingServerException
		{
		return null;
		}

	public QuestionnaireQuestion createQuestion()
		throws RemoteException, BuildingServerException
		{
		QuestionnaireQuestion question = new QuestionnaireQuestion();
		
		question.setResourceId( getResource().getResourceId() );
		question.setIntroduction( "The text of the question goes here." );
		question.setStatement( 0, "I strongly agree." );
		question.setStatement( 1, "I agree." );
		question.setStatement( 2, "I partly agree, partly disagree" );
		question.setStatement( 3, "I disagree." );
		question.setStatement( 4, "I strongly disagree." );
		for ( int i=5; i<10; i++ )
			{
			question.setStatement( i, "-" );
			}
		question.setOrdinal( max_ordinal+100 );
		max_ordinal+=100;
		question.setFlags( (byte)0 );
		question.save();
		
		return question;
		}
	
	public void removeQuestion( PrimaryKey id )
		throws RemoteException, BuildingServerException
		{
		QuestionnaireQuestion question =  QuestionnaireQuestion.findQuestionnaireQuestion( id );
		if ( question == null )
			throw new BuildingServerException( "Question not found." );

		if ( !question.getResourceId().equals( getResource().getResourceId() ) )
			throw new BuildingServerException( "Question doesn't belong in this Questionnaire." );

		question.delete();
		}
	
	public void changeQuestion( PrimaryKey id, QuestionnaireQuestion model )
		throws RemoteException, BuildingServerException
		{
		QuestionnaireQuestion question =  QuestionnaireQuestion.findQuestionnaireQuestion( id );
		if ( question == null )
			throw new BuildingServerException( "Question not found." );

		if ( !question.getResourceId().equals( getResource().getResourceId() ) )
			throw new BuildingServerException( "Question doesn't belong in this Questionnaire." );

		question.setOrdinal( model.getOrdinal() );
		question.setFlags( model.getFlags() );
		question.save();

		if ( question.getOrdinal() > max_ordinal )
			max_ordinal = question.getOrdinal();
		}
	
	public void changeQuestionText( PrimaryKey qid, PrimaryKey tid, String new_text )
		throws RemoteException, BuildingServerException
		{
		QuestionnaireQuestion question =  QuestionnaireQuestion.findQuestionnaireQuestion( qid );
		if ( question == null )
			throw new BuildingServerException( "Question not found." );
			
		if ( !question.getResourceId().equals( getResource().getResourceId() ) )
			throw new BuildingServerException( "Question doesn't belong in this Questionnaire." );

		boolean found = false;
		found = found || tid.equals( question.getIntroductionBigStringId() );
		found = found || tid.equals( question.getStatementABigStringId() );
		found = found || tid.equals( question.getStatementBBigStringId() );
		found = found || tid.equals( question.getStatementCBigStringId() );
		found = found || tid.equals( question.getStatementDBigStringId() );
		found = found || tid.equals( question.getStatementEBigStringId() );
		found = found || tid.equals( question.getStatementFBigStringId() );
		found = found || tid.equals( question.getStatementGBigStringId() );
		found = found || tid.equals( question.getStatementHBigStringId() );
		found = found || tid.equals( question.getStatementIBigStringId() );
		found = found || tid.equals( question.getStatementJBigStringId() );

		if ( !found )
			throw new BuildingServerException( "Can't identify qhich part of question to edit." );

		BigString big = BigString.findBigString( tid );
		big.setString( new_text );
		big.save();
		question.setUnsaved();
		}
	

	public PrimaryKey record( PrimaryKey[] q_question_ids, int[] items, String[] comments )
		throws RemoteException, BuildingServerException
		{
		User user = (User)BuildingContext.getContext().getUser();
		Hashtable q_questions = this.getQuestionnaireQuestions();
		QuestionnaireResponse q_response;
		QuestionnaireQuestion q_question;
		QuestionnaireResult q_result = getQuestionnaireResult( user.getUserId() );
		if ( q_result == null )
			{
			q_result = new QuestionnaireResult();
			q_result.setResourceId( this.getResource().getResourceId() );
			q_result.setUserId( user.getUserId() );
			q_result.setWhenStarted( new Timestamp( System.currentTimeMillis() ) );
			q_result.setWhenSaved( new Timestamp( System.currentTimeMillis() ) );
			q_result.save();
			}
		Hashtable responses = getQuestionnaireResponses( q_result.getQuestionnaireResultId() );
		if ( responses == null ) responses = new Hashtable();

		q_result.setWhenSaved( new Timestamp( System.currentTimeMillis() ) );
		q_result.save();
		
		

		for ( int i=0; i< q_question_ids.length; i++ )
			{
			q_question = (QuestionnaireQuestion)q_questions.get( q_question_ids[i] );
			if ( q_question == null )
				throw new BuildingServerException( "Unable to find question data." );
			q_response = (QuestionnaireResponse)responses.get( q_question_ids[i] );
			if ( q_response == null )
				{
				q_response = new QuestionnaireResponse();
				q_response.setQuestionnaireResultId( q_result.getQuestionnaireResultId() );
				q_response.setQuestionnaireQuestionId( q_question_ids[i] );
				q_response.save();
				}
			if ( (items[i]+1) > q_question.getStatementCount() )
				items[i]=100;
			if ( items[i]<0 )
				items[i]=100;
				
			q_response.setItem( (byte)items[i] );
			q_response.setComment( comments[i] );
			q_response.save();
			}

		return q_result.getQuestionnaireResultId();
		}
	
	public QuestionnaireResult getQuestionnaireResult()
		throws RemoteException, BuildingServerException
		{
		User user = (User)BuildingContext.getContext().getUser();
		return getQuestionnaireResult( user.getUserId() );
		}
		
	public QuestionnaireResult getQuestionnaireResult( PrimaryKey uid )
		throws RemoteException, BuildingServerException
		{
		QuestionnaireResult entry = 
			QuestionnaireResult.findQuestionnaireResult( 
				"resource_id = " + getResource().getResourceId() + " AND user_id = " + uid );
		
		return entry;
		}
		
	public Hashtable getQuestionnaireResponses( PrimaryKey rid )
		throws RemoteException, BuildingServerException
		{
		QuestionnaireResult q_result = QuestionnaireResult.findQuestionnaireResult( rid );
		if ( q_result == null )
			return null;
		
		Hashtable table = new Hashtable();
		Enumeration enum = QuestionnaireResponse.findQuestionnaireResponses( "questionnaire_result_id = " + rid );
		QuestionnaireResponse q_response;
			
		while ( enum.hasMoreElements() )
			{
			q_response= (QuestionnaireResponse)enum.nextElement();
			table.put( q_response.getQuestionnaireQuestionId(), q_response );
			}

		return table;
		}

	public Vector getResultsOfUser()
		throws RemoteException, BuildingServerException
		{
		Vector list = new Vector();
		Enumeration enum = QuestionnaireResult.findQuestionnaireResults( 
					"resource_id = " + this.getResource().getResourceId() + " AND user_id = " + user_id,
					"when_done" );
			
		while ( enum.hasMoreElements() )
			list.addElement( enum.nextElement() );

		return list;
		}

	// returns enumeration of results found and if analysis is non null it is filled in with
	// statistical analysis.
	public Vector analyse( String where_clause, QuestionnaireAnalysis analysis )
		throws RemoteException, BuildingServerException
		{
		Vector list = new Vector();
		Enumeration enum;
		QuestionnaireResult q_result;
		QuestionnaireResponse q_response;
		QuestionnaireQuestion q_question;
		Hashtable responses;
		Vector questions=null;
		
		if ( where_clause == null )
			enum = QuestionnaireResult.findQuestionnaireResults( "resource_id = " + getResource().getResourceId() );
		else
			enum = QuestionnaireResult.findQuestionnaireResults( where_clause );

		if ( analysis != null )
			questions = getQuestionnaireQuestionsInOrder();	

		while ( enum.hasMoreElements() )
			{
			q_result = (QuestionnaireResult)enum.nextElement();
			list.addElement( q_result );
			
			if ( analysis == null )
				continue;
			
			//now add analysis...
			
			analysis.addQuestionnaireResult( q_result );
			responses = getQuestionnaireResponses( q_result.getQuestionnaireResultId() );
			
			for ( int i=0; i<questions.size(); i++ )
				{
				q_question = (QuestionnaireQuestion)questions.elementAt( i );
				q_response = (QuestionnaireResponse)responses.get( q_question.getQuestionnaireQuestionId() );
				if ( q_response != null )
					analysis.addQuestionnaireResponse( q_question, q_result, q_response );
				}
			
			}

		return list;
		}
	
	
	
	// export XML file containing questions and optionally results and responses
	public void exportXml( boolean include_results, boolean include_responses )
		throws RemoteException, BuildingServerException
	    {
	    int i, j, n;
   		Questionnaire paper = this.getQuestionnaire();
   		Hashtable qtable;
   		Vector qlist = this.getQuestionnaireQuestionsInOrder();
   		Vector rlist;
   		QuestionnaireQuestion question;
   		QuestionnaireResult result;
   		QuestionnaireResponse response;
	    Enumeration enum;
	    
	    
	    if ( paper == null || qlist == null )
            throw new BuildingServerException( "Unable to find questionnaire data." );
            
        try
            {
            File xmlfile = BuildingContext.createTempFile( "bod", ".xml" );
            
            PrintWriter xmlout = new PrintWriter( new FileWriter( xmlfile ) );
            
            xmlout.print( "<?xml version=\"1.0\" standalone=\"no\"?>" );
            xmlout.println();
            xmlout.print( "<!DOCTYPE questestinterop SYSTEM \"http://www.imsproject.org/xml/question/FinalQTIv101/xmla/dtds/QTIcorencdtd/IMS_QTIv1p01.dtd\">" );
            xmlout.println();
            xmlout.print( "<!-- Experimental IMS output -->" );
            
            xmlout.println();
            xmlout.println();

            xmlout.print( "<questestinterop>" );
            //xmlout.print( paper.getQuestionnaireId() );
            //xmlout.print( "\">" );

            xmlout.println();


            for ( i=0; i<qlist.size(); i++ )
                {
                question = (QuestionnaireQuestion)qlist.elementAt( i );

                // ask the question to export itself, allowing composite output
                question.exportXml( xmlout, false );
                }
           /* 
            if ( include_results )
                {
                qtable = this.getQuestionnaireQuestions();
                
                // not too bad to load all the results in one go because there is very
                // little data in the structure
    			rlist = this.getQuestionnaireResults();
                for ( i=0; i<rlist.size(); i++ )
                    {
                    result = (QuestionnaireResult)rlist.elementAt( i );

                    xmlout.print( "\t<QuestionnaireResult id=\"pk" );
                    xmlout.print( result.getQuestionnaireResultId() );
                    xmlout.print( "\">" );
                    xmlout.println();
                   
                    if ( include_responses )
                        {
                        enum = QuestionnaireResponse.findQuestionnaireResponses( "questionnaire_result_id = " + result.getQuestionnaireResultId() );
                        while ( enum.hasMoreElements() )
                            {
                            response = (QuestionnaireResponse)enum.nextElement();
                            question = (QuestionnaireQuestion)qtable.get( response.getQuestionnaireQuestionId() );
                            
                            xmlout.print( "\t\t<QuestionnaireResponse id=\"pk" );
                            xmlout.print( response.getQuestionnaireResponseId() );
                            xmlout.print( "\" questionnairequestionid=\"pk" );
                            xmlout.print( response.getQuestionnaireQuestionId() );
                            xmlout.print( "\">" );
                            xmlout.println();
                            
                            if ( question.getStatementCount()>0 )
                                {
                                xmlout.print( "\t\t\t<Item>" );
                                xmlout.print( response.getItem() );
                                xmlout.print( "</Item>" );
                                xmlout.println();
                                }
                                
                            if ( question.canComment() && response.getComment() !=null )
                                {
                                xmlout.print( "\t\t\t<Comment>" );
                                xmlout.print( response.getComment() );
                                xmlout.print( "</Comment>" );
                                xmlout.println();
                                }
                            
                            xmlout.print( "\t\t</QuestionnaireResponse>" );
                            xmlout.println();
                            }
                        }

                    xmlout.print( "\t</QuestionnaireResult>" );
                    xmlout.println();
                    }
                }
*/
            xmlout.print( "</questestinterop>" );

            xmlout.println();
            xmlout.println();
            
            xmlout.close();
            
            
            this.transferFile( xmlfile.getAbsolutePath(), "qti.xml", "application/octet-stream" );
            }
        catch ( IOException ioex )
            {
	    Logger.getLogger( "org.bodington" ).logp( 
		Level.SEVERE, 
		"QuestionnaireSessionImpl", 
		"exportXml", 
		ioex.getMessage(), 
		ioex );
            throw new BuildingServerException( "Problem creating XML file: " + ioex.getMessage() );
            }
	    }
	    

	
	
	public void importXmlQuestions( String file_name )
		throws RemoteException, BuildingServerException
	    {

        try
            {
            //make sure max_ordinal is updated
            getQuestionnaireQuestionIds();
            
            // Parse the input
            XMLReader reader = XMLReaderFactory.createXMLReader( 
            		BuildingContext.getProperty( "xmlrepository.driver", "org.apache.crimson.parser.XMLReaderImpl" ) );
            QuestionnaireImsQtiHandler handler = new QuestionnaireImsQtiHandler( reader );
            reader.setContentHandler( handler );
            
            reader.setProperty( "http://xml.org/sax/properties/lexical-handler", handler );
            reader.setEntityResolver( handler );
            
            File input_file = new File( file_name );
            if ( !input_file.exists() )
                throw new BuildingServerException( "Couldn't open the uploaded file." );
                
            reader.parse( "file:" + input_file );
            
            Vector questions = handler.getQuestions();
            QuestionnaireQuestion question=null, previous_question=null;
            
            for ( int i=0; i<questions.size(); i++ )
                {
                previous_question = question;
                question = (QuestionnaireQuestion)questions.elementAt( i );
                
                if (    previous_question != null &&
                        question.getStatementCount() == 0 && 
                        question.canComment() && 
                        question.getIntroduction().indexOf( "This is the comment box for the previous question." )>=0 )
                    {
                    // merge this question with previous
                    previous_question.setCanComment( true );
                    previous_question.save();
                    //remove current
                    questions.removeElementAt( i );
                    //fix index so we point at next question that has now moved down one place
                    i--;
                    continue;
                    }
                
                question.setResourceId( getResource().getResourceId() );
		        question.setOrdinal( max_ordinal+100 );
		        max_ordinal+=100;
		        question.save();
                }
            }
       catch (SAXParseException spe)
            {
            // Error generated by the parser
            Logger.getLogger( "org.bodington" ).warning ("\n** Parsing error" 
                + ", line " + spe.getLineNumber ()
                + ", uri " + spe.getSystemId ());
            Logger.getLogger( "org.bodington" ).warning("   " + spe.getMessage() );
            // Use the contained exception, if any
            if (spe.getException() != null)
		Logger.getLogger( "org.bodington" ).logp( 
		    Level.SEVERE, 
		    "QuestionnaireSessionImpl", 
		    "importXmlQuestions", 
		    spe.getException().getMessage(), 
		    spe.getException() );
	   else
		Logger.getLogger( "org.bodington" ).logp( 
		    Level.SEVERE, 
		    "QuestionnaireSessionImpl", 
		    "importXmlQuestions", 
		    spe.getMessage(), 
		    spe );
            throw new BuildingServerException( "Problem importing XML file: " + spe.getMessage() );
            }
        catch (Throwable t)
            {
	    Logger.getLogger( "org.bodington" ).logp( 
		Level.SEVERE, 
		"QuestionnaireSessionImpl", 
		"importXmlQuestions", 
		t.getMessage(), 
		t );
            throw new BuildingServerException( "Problem importing XML file: " + t.getMessage() );
            }
	    }
	    
	    
	}
	
