WriterAbstract::closeWriter()

Passed

Pull Request — develop_3.0 (#491)

by Adrien

created 2017-11-05 13:05 UTC

↳ Parent: WriterAbstract

Size

Total Lines

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
dl	0
loc	1
ccs	0
cts	0
cp	0
c	0
b	0
f	0
nc	1

<?php

namespace Box\Spout\Writer;

use Box\Spout\Common\Creator\HelperFactory;
use Box\Spout\Common\Exception\InvalidArgumentException;
use Box\Spout\Common\Exception\IOException;
use Box\Spout\Common\Exception\SpoutException;
use Box\Spout\Common\Helper\GlobalFunctionsHelper;
use Box\Spout\Common\Manager\OptionsManagerInterface;
use Box\Spout\Writer\Common\Entity\Options;
use Box\Spout\Writer\Common\Entity\Row;
use Box\Spout\Writer\Common\Entity\Style\Style;
use Box\Spout\Writer\Exception\WriterAlreadyOpenedException;
use Box\Spout\Writer\Exception\WriterNotOpenedException;

/**
 * Class WriterAbstract
 *
 * @abstract
 */
abstract class WriterAbstract implements WriterInterface
{
    /** @var string Path to the output file */
    protected $outputFilePath;

    /** @var resource Pointer to the file/stream we will write to */
    protected $filePointer;

    /** @var bool Indicates whether the writer has been opened or not */
    protected $isWriterOpened = false;

    /** @var GlobalFunctionsHelper Helper to work with global functions */
    protected $globalFunctionsHelper;

    /** @var HelperFactory $helperFactory */
    protected $helperFactory;

    /** @var OptionsManagerInterface Writer options manager */
    protected $optionsManager;

    /** @var string Content-Type value for the header - to be defined by child class */
    protected static $headerContentType;

    /**
     * @param OptionsManagerInterface $optionsManager
     * @param GlobalFunctionsHelper $globalFunctionsHelper
     * @param HelperFactory $helperFactory
     */
    public function __construct(
        OptionsManagerInterface $optionsManager,
        GlobalFunctionsHelper $globalFunctionsHelper,
        HelperFactory $helperFactory
    ) {
        $this->optionsManager = $optionsManager;
        $this->globalFunctionsHelper = $globalFunctionsHelper;
        $this->helperFactory = $helperFactory;
    }

    /**
     * Opens the streamer and makes it ready to accept data.
     *
     * @throws IOException If the writer cannot be opened
     * @return void
     */
    abstract protected function openWriter();

    /**
     * Adds a row to the currently opened writer.
     *
     * @param Row $row The row containing cells and styles
     * @throws WriterNotOpenedException If the workbook is not created yet
     * @throws IOException If unable to write data
     * @return void
     */
    abstract protected function addRowToWriter(Row $row);

    /**
     * Closes the streamer, preventing any additional writing.
     *
     * @return void
     */
    abstract protected function closeWriter();

    /**
     * {@inheritdoc}
     */
    public function setDefaultRowStyle(Style $defaultStyle)
    {
        $this->optionsManager->setOption(Options::DEFAULT_ROW_STYLE, $defaultStyle);

        return $this;
    }

    /**
     * {@inheritdoc}
     */
    public function openToFile($outputFilePath)
    {
        $this->outputFilePath = $outputFilePath;

        $this->filePointer = $this->globalFunctionsHelper->fopen($this->outputFilePath, 'wb+');
        $this->throwIfFilePointerIsNotAvailable();

        $this->openWriter();
        $this->isWriterOpened = true;

        return $this;
    }

    /**
     * @codeCoverageIgnore
     * {@inheritdoc}
     */
    public function openToBrowser($outputFileName)
    {
        $this->outputFilePath = $this->globalFunctionsHelper->basename($outputFileName);

        $this->filePointer = $this->globalFunctionsHelper->fopen('php://output', 'w');
        $this->throwIfFilePointerIsNotAvailable();

        // Clear any previous output (otherwise the generated file will be corrupted)
        // @see https://github.com/box/spout/issues/241
        $this->globalFunctionsHelper->ob_end_clean();

        // Set headers
        $this->globalFunctionsHelper->header('Content-Type: ' . static::$headerContentType);
        $this->globalFunctionsHelper->header('Content-Disposition: attachment; filename="' . $this->outputFilePath . '"');

        /*
         * When forcing the download of a file over SSL,IE8 and lower browsers fail
         * if the Cache-Control and Pragma headers are not set.
         *
         * @see http://support.microsoft.com/KB/323308
         * @see https://github.com/liuggio/ExcelBundle/issues/45
         */
        $this->globalFunctionsHelper->header('Cache-Control: max-age=0');
        $this->globalFunctionsHelper->header('Pragma: public');

        $this->openWriter();
        $this->isWriterOpened = true;

        return $this;
    }

    /**
     * Checks if the pointer to the file/stream to write to is available.
     * Will throw an exception if not available.
     *
     * @throws IOException If the pointer is not available
     * @return void
     */
    protected function throwIfFilePointerIsNotAvailable()
    {
        if (!$this->filePointer) {
            throw new IOException('File pointer has not be opened');
        }
    }

    /**
     * Checks if the writer has already been opened, since some actions must be done before it gets opened.
     * Throws an exception if already opened.
     *
     * @param string $message Error message
     * @throws WriterAlreadyOpenedException If the writer was already opened and must not be.
     * @return void
     */
    protected function throwIfWriterAlreadyOpened($message)
    {
        if ($this->isWriterOpened) {
            throw new WriterAlreadyOpenedException($message);
        }
    }

    /**
     * {@inheritdoc}
     */
    public function addRow(Row $row)
    {
        if ($this->isWriterOpened) {
            // empty $dataRow should not add an empty line
            if ($row->hasCells()) {
                try {
                    $this->addRowToWriter($row);
                } catch (SpoutException $e) {
                    // if an exception occurs while writing data,
                    // close the writer and remove all files created so far.
                    $this->closeAndAttemptToCleanupAllFiles();

                    // re-throw the exception to alert developers of the error
                    throw $e;
                }
            }
        } else {
            throw new WriterNotOpenedException('The writer needs to be opened before adding row.');
        }

        return $this;
    }

    /**
     * {@inheritdoc}
     */
    public function addRows(array $rows)
    {
        foreach ($rows as $row) {
            if (!$row instanceof Row) {
                $this->closeAndAttemptToCleanupAllFiles();
                throw new InvalidArgumentException('The input should be an array of Row');
            }

            $this->addRow($row);
        }

        return $this;
    }

    /**
     * {@inheritdoc}
     */
    public function close()
    {
        if (!$this->isWriterOpened) {
            return;
        }

        $this->closeWriter();

        if (is_resource($this->filePointer)) {
            $this->globalFunctionsHelper->fclose($this->filePointer);
        }

        $this->isWriterOpened = false;
    }

    /**
     * Closes the writer and attempts to cleanup all files that were
     * created during the writing process (temp files & final file).
     *
     * @return void
     */
    private function closeAndAttemptToCleanupAllFiles()
    {
        // close the writer, which should remove all temp files
        $this->close();

        // remove output file if it was created
        if ($this->globalFunctionsHelper->file_exists($this->outputFilePath)) {
            $outputFolderPath = dirname($this->outputFilePath);
            $fileSystemHelper = $this->helperFactory->createFileSystemHelper($outputFolderPath);
            $fileSystemHelper->deleteFile($this->outputFilePath);
        }
    }
}


1		<?php
2
3		namespace Box\Spout\Writer;
4
5		use Box\Spout\Common\Creator\HelperFactory;
6		use Box\Spout\Common\Exception\InvalidArgumentException;
7		use Box\Spout\Common\Exception\IOException;
8		use Box\Spout\Common\Exception\SpoutException;
9		use Box\Spout\Common\Helper\GlobalFunctionsHelper;
10		use Box\Spout\Common\Manager\OptionsManagerInterface;
11		use Box\Spout\Writer\Common\Entity\Options;
12		use Box\Spout\Writer\Common\Entity\Row;
13		use Box\Spout\Writer\Common\Entity\Style\Style;
14		use Box\Spout\Writer\Exception\WriterAlreadyOpenedException;
15		use Box\Spout\Writer\Exception\WriterNotOpenedException;
16
17		/**
18		* Class WriterAbstract
19		*
20		* @abstract
21		*/
22		abstract class WriterAbstract implements WriterInterface
23		{
24		/** @var string Path to the output file */
25		protected $outputFilePath;
26
27		/** @var resource Pointer to the file/stream we will write to */
28		protected $filePointer;
29
30		/** @var bool Indicates whether the writer has been opened or not */
31		protected $isWriterOpened = false;
32
33		/** @var GlobalFunctionsHelper Helper to work with global functions */
34		protected $globalFunctionsHelper;
35
36		/** @var HelperFactory $helperFactory */
37		protected $helperFactory;
38
39		/** @var OptionsManagerInterface Writer options manager */
40		protected $optionsManager;
41
42		/** @var string Content-Type value for the header - to be defined by child class */
43		protected static $headerContentType;
44
45		/**
46		* @param OptionsManagerInterface $optionsManager
47		* @param GlobalFunctionsHelper $globalFunctionsHelper
48		* @param HelperFactory $helperFactory
49		*/
50	94	public function __construct(
51		OptionsManagerInterface $optionsManager,
52		GlobalFunctionsHelper $globalFunctionsHelper,
53		HelperFactory $helperFactory
54		) {
55	94	$this->optionsManager = $optionsManager;
56	94	$this->globalFunctionsHelper = $globalFunctionsHelper;
57	94	$this->helperFactory = $helperFactory;
58	94	}
59
60		/**
61		* Opens the streamer and makes it ready to accept data.
62		*
63		* @throws IOException If the writer cannot be opened
64		* @return void
65		*/
66		abstract protected function openWriter();
67
68		/**
69		* Adds a row to the currently opened writer.
70		*
71		* @param Row $row The row containing cells and styles
72		* @throws WriterNotOpenedException If the workbook is not created yet
73		* @throws IOException If unable to write data
74		* @return void
75		*/
76		abstract protected function addRowToWriter(Row $row);
77
78		/**
79		* Closes the streamer, preventing any additional writing.
80		*
81		* @return void
82		*/
83		abstract protected function closeWriter();
84
85		/**
86		* {@inheritdoc}
87		*/
88	2	public function setDefaultRowStyle(Style $defaultStyle)
89		{
90	2	$this->optionsManager->setOption(Options::DEFAULT_ROW_STYLE, $defaultStyle);
91
92	2	return $this;
93		}
94
95		/**
96		* {@inheritdoc}
97		*/
98	83	public function openToFile($outputFilePath)
99		{
100	83	$this->outputFilePath = $outputFilePath;
101
102	83	$this->filePointer = $this->globalFunctionsHelper->fopen($this->outputFilePath, 'wb+');
103	83	$this->throwIfFilePointerIsNotAvailable();
104
105	80	$this->openWriter();
106	80	$this->isWriterOpened = true;
107
108	80	return $this;
109		}
110
111		/**
112		* @codeCoverageIgnore
113		* {@inheritdoc}
114		*/
115		public function openToBrowser($outputFileName)
116		{
117		$this->outputFilePath = $this->globalFunctionsHelper->basename($outputFileName);
118
119		$this->filePointer = $this->globalFunctionsHelper->fopen('php://output', 'w');
120		$this->throwIfFilePointerIsNotAvailable();
121
122		// Clear any previous output (otherwise the generated file will be corrupted)
123		// @see https://github.com/box/spout/issues/241
124		$this->globalFunctionsHelper->ob_end_clean();
125
126		// Set headers
127		$this->globalFunctionsHelper->header('Content-Type: ' . static::$headerContentType);
128		$this->globalFunctionsHelper->header('Content-Disposition: attachment; filename="' . $this->outputFilePath . '"');
129
130		/*
131		* When forcing the download of a file over SSL,IE8 and lower browsers fail
132		* if the Cache-Control and Pragma headers are not set.
133		*
134		* @see http://support.microsoft.com/KB/323308
135		* @see https://github.com/liuggio/ExcelBundle/issues/45
136		*/
137		$this->globalFunctionsHelper->header('Cache-Control: max-age=0');
138		$this->globalFunctionsHelper->header('Pragma: public');
139
140		$this->openWriter();
141		$this->isWriterOpened = true;
142
143		return $this;
144		}
145
146		/**
147		* Checks if the pointer to the file/stream to write to is available.
148		* Will throw an exception if not available.
149		*
150		* @throws IOException If the pointer is not available
151		* @return void
152		*/
153	83	protected function throwIfFilePointerIsNotAvailable()
154		{
155	83	if (!$this->filePointer) {
156	3	throw new IOException('File pointer has not be opened');
157		}
158	80	}
159
160		/**
161		* Checks if the writer has already been opened, since some actions must be done before it gets opened.
162		* Throws an exception if already opened.
163		*
164		* @param string $message Error message
165		* @throws WriterAlreadyOpenedException If the writer was already opened and must not be.
166		* @return void
167		*/
168	49	protected function throwIfWriterAlreadyOpened($message)
169		{
170	49	if ($this->isWriterOpened) {
171	5	throw new WriterAlreadyOpenedException($message);
172		}
173	44	}
174
175		/**
176		* {@inheritdoc}
177		*/
178	75	public function addRow(Row $row)
179		{
180	75	if ($this->isWriterOpened) {
181		// empty $dataRow should not add an empty line
182	65	if ($row->hasCells()) {
183		try {
184	65	$this->addRowToWriter($row);
185	4	} catch (SpoutException $e) {
186		// if an exception occurs while writing data,
187		// close the writer and remove all files created so far.
188	4	$this->closeAndAttemptToCleanupAllFiles();
189
190		// re-throw the exception to alert developers of the error
191	65	throw $e;
192		}
193		}
194		} else {
195	10	throw new WriterNotOpenedException('The writer needs to be opened before adding row.');
196		}
197
198	63	return $this;
199		}
200
201		/**
202		* {@inheritdoc}
203		*/
204	63	public function addRows(array $rows)
205		{
206	63	foreach ($rows as $row) {
207	63	if (!$row instanceof Row) {
208	2	$this->closeAndAttemptToCleanupAllFiles();
209	2	throw new InvalidArgumentException('The input should be an array of Row');
210		}
211
212	61	$this->addRow($row);
213		}
214
215	55	return $this;
216		}
217
218		/**
219		* {@inheritdoc}
220		*/
221	74	public function close()
222		{
223	74	if (!$this->isWriterOpened) {
224	4	return;
225		}
226
227	73	$this->closeWriter();
228
229	73	if (is_resource($this->filePointer)) {
230	73	$this->globalFunctionsHelper->fclose($this->filePointer);
231		}
232
233	73	$this->isWriterOpened = false;
234	73	}
235
236		/**
237		* Closes the writer and attempts to cleanup all files that were
238		* created during the writing process (temp files & final file).
239		*
240		* @return void
241		*/
242	6	private function closeAndAttemptToCleanupAllFiles()
243		{
244		// close the writer, which should remove all temp files
245	6	$this->close();
246
247		// remove output file if it was created
248	6	if ($this->globalFunctionsHelper->file_exists($this->outputFilePath)) {
249	5	$outputFolderPath = dirname($this->outputFilePath);
250	5	$fileSystemHelper = $this->helperFactory->createFileSystemHelper($outputFolderPath);
251	5	$fileSystemHelper->deleteFile($this->outputFilePath);
252		}
253	6	}
254		}
255

box / spout

Pull Request — develop_3.0 (#491)

WriterAbstract::closeWriter()

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like