← All changes | classes/Visualizer/Source/Csv.php +20 -70 3.10.2 → 2.1.4 View file →

		@@ -46,10 +46,10 @@
46	46	*
47	47	* @access public
48	48	* @param string $filename The path to the file.
49	49	*/
50		- public function __construct( $filename = '' ) {
51		- $this->_filename = trim( (string) $filename );
	50	+ public function __construct( $filename = null ) {
	51	+ $this->_filename = trim( $filename );
52	52	}
53	53
54	54	/**
55	55	* Fetches series information.
		@@ -61,42 +61,31 @@
61	61	*/
62	62	private function _fetchSeries( &$handle ) {
63	63	// read column titles
64	64	$labels = fgetcsv( $handle, 0, VISUALIZER_CSV_DELIMITER, VISUALIZER_CSV_ENCLOSURE );
65		- $types = null;
	65	+ $labels = array_map( 'utf8_encode', $labels );
	66	+ // read series types
	67	+ $types = fgetcsv( $handle, 0, VISUALIZER_CSV_DELIMITER, VISUALIZER_CSV_ENCLOSURE );
66	68
67		- if ( false !== strpos( $this->_filename, 'tqx=out:csv' ) ) {
68		- $attributes = $this->_fetchSeriesForGoogleQueryLanguage( $labels );
69		- if ( ! $attributes['abort'] ) {
70		- $labels = $attributes['labels'];
71		- $types = $attributes['types'];
72		- }
73		- }
74		-
75		- if ( is_null( $types ) ) {
76		- // read series types
77		- $types = fgetcsv( $handle, 0, VISUALIZER_CSV_DELIMITER, VISUALIZER_CSV_ENCLOSURE );
78		- }
79		-
80		- $labels = array_filter( $labels );
81		- $types = array_filter( $types );
82		-
83	69	if ( ! $labels || ! $types ) {
84		- $this->_error = esc_html__( 'File should have a heading row (1st row) and a data type row (2nd row). Please try again.', 'visualizer' );
85	70	return false;
86	71	}
87	72
	73	+ // if no types were setup, re read labels and empty types array
88	74	$types = array_map( 'trim', $types );
89	75	if ( ! self::_validateTypes( $types ) ) {
90		- $this->_error = esc_html__( 'Invalid data types detected in the data type row (2nd row). Please try again.', 'visualizer' );
91		- return false;
	76	+ // re open the file
	77	+ fclose( $handle );
	78	+ $handle = $this->_get_file_handle();
	79	+
	80	+ // re read the labels and empty types array
	81	+ $labels = fgetcsv( $handle, 0, VISUALIZER_CSV_DELIMITER, VISUALIZER_CSV_ENCLOSURE );
	82	+ $labels = array_map( 'utf8_encode', $labels );
	83	+ $types = array();
92	84	}
93	85
94	86	for ( $i = 0, $len = count( $labels ); $i < $len; $i++ ) {
95		- $default_type = $i === 0 ? 'string' : 'number';
96		-
97		- $labels[ $i ] = $this->toUTF8( $labels[ $i ] );
98		-
	87	+ $default_type = $i == 0 ? 'string' : 'number';
99	88	$this->_series[] = array(
100	89	'label' => $labels[ $i ],
101	90	'type' => isset( $types[ $i ] ) ? $types[ $i ] : $default_type,
102	91	);
		@@ -117,9 +106,12 @@
117	106	protected function _get_file_handle( $filename = false ) {
118	107	// set line endings auto detect mode
119	108	ini_set( 'auto_detect_line_endings', true );
120	109	// open file and return handle
121		- return fopen( $filename ? $filename : $this->_filename, 'rb' );
	110	+ $fc = iconv( 'ISO-8859-1', 'utf-8', file_get_contents( $filename ? $filename : $this->_filename ) );
	111	+ $tmp = tempnam( sys_get_temp_dir(), rand() );
	112	+ file_put_contents( $tmp, $fc );
	113	+ return fopen( $tmp, 'rb' );
122	114	}
123	115
124	116	/**
125	117	* Fetches information from source, parses it and builds series and data arrays.
		@@ -131,9 +123,8 @@
131	123	*/
132	124	public function fetch() {
133	125	// if filename is empty return false
134	126	if ( empty( $this->_filename ) ) {
135		- $this->_error = esc_html__( 'No file provided. Please try again.', 'visualizer' );
136	127	return false;
137	128	}
138	129
139	130	// read file and fill arrays
		@@ -144,10 +135,10 @@
144	135	return false;
145	136	}
146	137
147	138	// fetch data
148		- // phpcs:ignore WordPress.CodeAnalysis.AssignmentInCondition.FoundInWhileCondition
149	139	while ( ( $data = fgetcsv( $handle, 0, VISUALIZER_CSV_DELIMITER, VISUALIZER_CSV_ENCLOSURE ) ) !== false ) {
	140	+
150	141	$this->_data[] = $this->_normalizeData( $data );
151	142	}
152	143
153	144	// close file handle
		@@ -168,46 +159,5 @@
168	159	public function getSourceName() {
169	160	return __CLASS__;
170	161	}
171	162
172		- /**
173		- * Adds support for QueryLanguage https://developers.google.com/chart/interactive/docs/querylanguage
174		- * where the user can provide something like /gviz/tq?tq=select%20A%2C%20B%20&tqx=out:csv after the URL of the raw spreadsheet
175		- * to get the subset of data specified by the query
176		- * this will conflate the heading and the type into one value viz. for heading XXX and type string, the value will become "XXX string"
177		- * so we need to split them apart logically
178		- * also the $types variable now contains the first row because the header and the type got conflated
179		- */
180		- private function _fetchSeriesForGoogleQueryLanguage( $labels, $types = array() ) {
181		- $new_labels = array();
182		- $new_types = array();
183		- $abort = false;
184		- foreach ( $labels as $label ) {
185		- // get the index of the last space
186		- $index = strrpos( $label, ' ' );
187		- if ( $index === false ) {
188		- // no space here? something has gone wrong; abort the entire process.
189		- $abort = true;
190		- break;
191		- }
192		- $type = trim( substr( $label, $index + 1 ) );
193		- if ( ! self::_validateTypes( array( $type ) ) ) {
194		- // some other data type? abort the entire process.
195		- $abort = true;
196		- break;
197		- }
198		- $label = substr( $label, 0, $index );
199		- $new_labels[] = $label;
200		- $new_types[] = $type;
201		- }
202		- if ( ! $abort ) {
203		- $labels = $new_labels;
204		- $types = $new_types;
205		- }
206		-
207		- return array(
208		- 'abort' => $abort,
209		- 'labels' => $labels,
210		- 'types' => $types,
211		- );
212		- }
213	163	}