silverstripe / sspak

src/DataExtractor/CsvTableReader.php

<?php

namespace SilverStripe\SsPak\DataExtractor;

class CsvTableReader implements TableReader
{

	private $filename;
	private $handle;
	private $columns;

	function __construct($filename) {

It is generally recommended to explicitly declare the visibility for methods.

		$this->filename = $filename;
	}

	function getColumns() {

It is generally recommended to explicitly declare the visibility for methods.

		if (!$this->columns) {
			$this->initColumns();
		}
		return $this->columns;
	}

	function getIterator() {

It is generally recommended to explicitly declare the visibility for methods.

		$this->columns = null;
		$this->initColumns();

		while(($row = $this->getRow()) !== false) {
			yield $this->mapToColumns($row);
		}

		$this->close();
	}

	private function mapToColumns($row) {
		$record = [];
		foreach($row as $i => $value)
		{
			if(isset($this->columns[$i])) {
				$record[$this->columns[$i]] = $value;
			} else {
				throw new \LogicException("Row contains invalid column #$i\n" . var_export($row, true));
			}
		}
		return $record;
	}

	private function initColumns() {
		$this->open();
		$this->columns = $this->getRow();
	}

	private function getRow() {
		return fgetcsv($this->handle);
	}

	private function open() {
		if ($this->handle) {
			fclose($this->handle);
			$this->handle = null;
		}
		$this->handle = fopen($this->filename, 'r');
	}

	private function close() {
		if ($this->handle) {
			fclose($this->handle);
			$this->handle = null;
		}
	}
}