<?php
/*
 * Abstracting data
 *
 * This class abstracting data for further use
 *
 * @author awesome.ug, Author <support@awesome.ug>
 * @package Questions/Core
 * @version 1.0.0
 * @since 1.0.0
 * @license GPL 2

  Copyright 2015 awesome.ug (support@awesome.ug)

  This program is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License, version 2, as 
  published by the Free Software Foundation.

  This program 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 General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

 */
 
if ( !defined( 'ABSPATH' ) ) exit;
 
class Questions_AbstractData{
	
	/**
	 * Prepare data to be used with charts
	 * @param array $response_array
	 * @return array $ordered_data
	 * @since 1.0.0
	 */
	public static function order_for_charting( $response_array ){
		
		global $wpdb, $questions_global;

		$ordered_data = array();

		$ordered_data[ 'questions' ] = array();
		$ordered_data[ 'data' ] = array();
		
		// Getting every entery of one 
		foreach( $response_array AS $key => $line ):
			$merged_data = array();

			$sql = $wpdb->prepare( "SELECT type FROM {$questions_global->tables->questions} WHERE id = %s", $key );
			$result = $wpdb->get_row( $sql );

			$element_class = 'Questions_FormElement_' . $result->type;
			
			if( !class_exists( $element_class ) )
				continue;
			
			$element = new $element_class( $key );
			
			if( !$element->is_analyzable )
				continue;
			
			$ordered_data[ 'questions' ][ $key ] = $line[ 'question' ];
			
			// Fill up missed answers with 0
			$sql = $wpdb->prepare( "SELECT * FROM {$questions_global->tables->answers} WHERE question_id = %s", $key );
			$results = $wpdb->get_results( $sql );
			
			$voted_answers = array_keys( $merged_data );
			foreach( $results AS $result ):
			 	$merged_data[ $result->answer ] = 0;
			endforeach;
			
			// Adding voted data
            if( !array_key_exists( 'responses', $response_array[ $key ] ) )
                continue;

			$responses = $response_array[ $key ][ 'responses' ];
			
			foreach( $responses AS $response ):
				if( !$element->answer_is_multiple ):
					$merged_data[ $response ] += 1;
				else:
					foreach( $response AS $answer_option => $answer ):
						if( $answer == __( 'Yes', 'questions-locale' ) )
							$merged_data[ $answer_option ] += 1;
					endforeach;
				endif;
			endforeach;
			
			$ordered_data[ 'data' ][ $key ] = $merged_data;
		endforeach;
		
		return $ordered_data;		
	}

	/**
	 * Getting all questions for in an array
	 * @param array $response_array Response array of a survey
	 * @return array $headlines Ordered data prepared to be used in lines
	 * @since 1.0.0
	 */
	public static function get_headlines( $response_array ){
		// Only starting if there is any data		
		if( is_array( $response_array ) ):
			$headlines = array();
			
			/**
			 * Getting Headlines by running each element
			 */
			foreach( $response_array AS $question_id => $question ):
				
				// If element uses sections
				if( array_key_exists( 'sections', $question ) && TRUE == $question[ 'sections' ] ):
					foreach( $question[ 'responses' ] AS $response ):
						$i = 0;
						
						foreach( $response AS $key => $value ):
							if( is_array( $value ) ):
								foreach( $value AS $key2 => $key1 ):
									$column = $question[ 'question' ] . ' (' . $key . ' / ' . $key2 . ')';
									$headlines[ $question_id . '-' . $i++ ] = self::filter_string( $column ); 
								endforeach;
							else:	
								$column = $question[ 'question' ] . ' (' . $key . ')';
								$headlines[ $question_id . '-' . $i++ ] = self::filter_string( $column ); 
							endif;
						endforeach;
						
						break;
					endforeach;
				
				// If there are more answers than one posssible (e.g. Multiple Choice)
				elseif( array_key_exists( 'array', $question ) && TRUE == $question[ 'array' ] ):
					
					if( array_key_exists( 'responses', $question ) ):
						foreach( $question[ 'responses' ] AS $response ):
							$i = 0;
							foreach( $response AS $key => $value ):
								$column = $question[ 'question' ] . ' (' . $key . ')';
								$headlines[ $question_id . '-' . $i++ ] = self::filter_string( $column ); 
							endforeach;
							break;
						endforeach;
					endif;
					
				// If there is only one value for one element possible
				else:
					$headlines[ $question_id ] = self::filter_string( $question[ 'question' ] ); 
				endif;
			endforeach;
			
			return $headlines;
		else:
			return FALSE;
		endif;
	}
	
	/**
	 * Prepare data for printing content in lines (e.g. for CSV export)
	 * @param array $response_array Response array of a survey
	 * @return array $lines Ordered data prepared to be used in lines
	 * @since 1.0.0
	 */
	public static function get_lines( $response_array ){
		$headlines = self::get_headlines( $response_array );
		
		// Only starting if there is any data		
		if( is_array( $response_array ) ):
			$results = array();
			
			/**
			 * Getting content by running each element
			 */
			foreach( $response_array AS $question_id => $question ):
				
				// If element uses sections
				if( array_key_exists( 'sections', $question ) && TRUE == $question[ 'sections' ] ):
					
					foreach( $question[ 'responses' ] AS $response_id => $response ):
						$i = 0;
						
						foreach( $response AS $key => $value ):
							if( is_array( $value ) ):
								foreach( $value AS $key2 => $key1 ):
									$results[ $response_id ][ $question_id . '-' . $i++ ] = self::filter_string( $key1 ); 
								endforeach;
							else:
								$results[ $response_id ][ $question_id . '-' . $i++ ] = self::filter_string( $value ); 
							endif;
						endforeach;
						
					endforeach;
					
				// If there are more answers than one posssible (e.g. Multiple Choice)
				elseif( array_key_exists( 'array', $question ) && TRUE == $question[ 'array' ] ):


                    if( is_array( $question[ 'responses' ] ) && 0 != count( $question[ 'responses' ] ) ):
                        // Running response of each user
                        foreach( $question[ 'responses' ] AS $response_id => $response ):
                            $i = 0;

                            // Running each answer of response
                            foreach( $response AS $key => $value ):
                                $results[ $response_id ][ $question_id . '-' . $i++ ] = self::filter_string( $value );
                            endforeach;

                        endforeach;
                    endif;
					
				// If there is only one value for one element possible
				else:

					if( array_key_exists( 'responses', $question ) ):
						foreach( $question[ 'responses' ]  AS $response_id => $value ):
							$results[ $response_id ][ $question_id ] = self::filter_string( $value ); 
						endforeach;
					endif;
					
				endif;
				
			endforeach;
			
			// Cleaning up results
			$cleaned_up_results = array();
			
			// Running each headline (column)
			foreach( $headlines AS $headline_key => $headline_text ): 
				
				foreach( $results AS $response_id => $response ):
					
					if( !array_key_exists( $headline_key, $response ) ):
						$cleaned_up_results[ $response_id ][ $headline_key ] = '-'; // Filling up missing columns in results
					else:
						$cleaned_up_results[ $response_id ][ $headline_key ] = $response[ $headline_key ]; // Filling with response
					endif;
				
				endforeach;
				
			endforeach;
			
			return $cleaned_up_results;
		else:
			return FALSE;
		endif;
	}

	/**
	 * Filtering not wanted chars
	 * @param string $string The string to filter
	 * @return string $string The filtered string
	 * @since 1.0.0
	 */
	public static function filter_string( $string ){
		if( '' ==  $string )
			return '-';
		
		$string = self::remove_new_lines( $string );
		$string = str_replace( ';', '#', $string );
        $string = stripslashes( $string );
		
		return $string;
	}
	
	/**
	 * Removing new lines
	 * @param string $string The string to filter
	 * @return string $string The filtered string
	 * @since 1.0.0
	 */
	public static function remove_new_lines( $string ){
		return trim( preg_replace( '/\s\s+/', ' ', $string ) );
	}
}