TabularDataStream.php

<?php

namespace frictionlessdata\datapackage\DataStreams;

use frictionlessdata\datapackage\Exceptions\DataStreamValidationException;
use frictionlessdata\datapackage\Exceptions\DataStreamOpenException;
use frictionlessdata\tableschema\DataSources\CsvDataSource;
use frictionlessdata\tableschema\Schema;
use frictionlessdata\tableschema\Table;
use frictionlessdata\tableschema\Exceptions\FieldValidationException;
use frictionlessdata\tableschema\Exceptions\DataSourceException;

class TabularDataStream extends BaseDataStream
{
    public $table;
    public $schema;

  /**
   * @throws \frictionlessdata\datapackage\Exceptions\DataStreamOpenException
   * @throws \Exception
   */
    public function __construct($dataSource, $dataSourceOptions = null)
    {
        parent::__construct($dataSource, $dataSourceOptions);
        $this->dataSourceOptions = (object) $this->dataSourceOptions;
        $schema = $this->dataSourceOptions->schema;
        $dialect = $this->dataSourceOptions->dialect;
        if (empty($schema)) {
            throw new \Exception('schema is required for tabular data stream');
        } else {
            try {
                $this->schema = new Schema($schema);
                $this->table = new Table($this->getDataSourceObject(), $this->schema, $dialect);
            } catch (\Exception $e) {
                throw new DataStreamOpenException('Failed to open tabular data source '.json_encode($dataSource).': '.json_encode($e->getMessage()));
            }
        }
    }

    protected function getDataSourceObject()
    {
        return new CsvDataSource($this->dataSource);
    }

    public function rewind():void
    {
        $this->table->rewind();
    }

    public function save($filename)
    {
        $this->table->save($filename);
    }

    /**
     * @return array
     *
     * @throws DataStreamValidationException
     */
    public function current():mixed
    {
        try {
            return $this->table->current();
        } catch (DataSourceException|FieldValidationException $e) {
            throw new DataStreamValidationException($e->getMessage());
        }
    }

    public function key():mixed
    {
        return $this->table->key();
    }

    public function next():void
    {
        $this->table->next();
    }

    public function valid(): bool
    {
        return $this->table->valid();
    }
}