<?php

namespace Export;

/**
 * ExportData is the base class for exporters to specific file formats. See other
 * classes below.
 *
 * php-export-data by Eli Dickinson, http://github.com/elidickinson/php-export-data
 */
abstract class ExportData
{
    protected $exportTo; // Set in constructor to one of 'browser', 'file', 'string'
    protected $stringData; // stringData so far, used if export string mode
    protected $tempFile; // handle to temp file (for export file mode)
    protected $tempFilename; // temp file name and path (for export file mode)
    public $filename; // file mode: the output file name; browser mode: file name for download; string mode: not used
    protected $encoding;

    public function __construct($exportTo = 'browser', $filename = 'exportdata')
    {
        if (!in_array($exportTo, ['browser', 'file', 'string'])) {
            throw new Exception("{$exportTo} is not a valid ExportData export type");
        }
        $this->exportTo = $exportTo;
        $this->filename = $filename;
    }

    public function initialize()
    {
        global $cfg;
        switch ($this->exportTo) {
            case 'browser':
                $this->sendHttpHeaders();
                break;
            case 'string':
                $this->stringData = '';
                break;
            case 'file':
                $this->tempFilename = tempnam($cfg['Path']['data'], 'exportdata');
                $this->tempFile = fopen($this->tempFilename, 'w');
                break;
        }

        $this->write($this->generateHeader());
    }

    public function addRow($row)
    {
        $this->write($this->generateRow($row));
    }

    public function loadOrdersData($order_ids)
    {
        foreach ($order_ids as $id) {
            $this->addRow($this->getOrder($id));
        }
    }

    public function finalize()
    {
        $this->write($this->generateFooter());

        switch ($this->exportTo) {
            case 'browser':
                flush();
                break;
            case 'string':
                // do nothing
                break;
            case 'file':
                // close temp file and move it to correct location
                fclose($this->tempFile);
                rename($this->tempFilename, $this->filename);
                break;
        }
    }

    public function getString()
    {
        return $this->stringData;
    }

    abstract public function sendHttpHeaders();

    protected function write($data)
    {
        if (!empty($data)) {
            switch ($this->exportTo) {
                case 'browser':
                    echo $data;
                    break;
                case 'string':
                    $this->stringData .= $data;
                    break;
                case 'file':
                    fwrite($this->tempFile, $data);
                    break;
            }
        }
    }

    protected function generateHeader()
    {
        // can be overridden by subclass to return any data that goes at the top of the exported file
    }

    protected function generateFooter()
    {
        // can be overridden by subclass to return any data that goes at the bottom of the exported file
    }

    // In subclasses generateRow will take $row array and return string of it formatted for export type
    abstract protected function generateRow($row);

    protected function getOrder($id)
    {
        $order = new \Order();

        $order->createFromDB($id);
        $order->fetchItems();

        return $order;
    }

    public function setCoding($encoding)
    {
        $this->encoding = $encoding;
    }

    public function encodeData($string)
    {
        return iconv('utf-8', $this->encoding.'//TRANSLIT', $string);
    }
}