Completed
Push — master ( 2d297e...16a2f9 )
by Adrien
02:10
created

getCellXMLFragmentForNonEmptyString()   A

Complexity

Conditions 3
Paths 3

Size

Total Lines 15

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 7
CRAP Score 3.0175

Importance

Changes 0
Metric Value
dl 0
loc 15
ccs 7
cts 8
cp 0.875
rs 9.7666
c 0
b 0
f 0
cc 3
nc 3
nop 1
crap 3.0175
1
<?php
2
3
namespace Box\Spout\Writer\XLSX\Manager;
4
5
use Box\Spout\Common\Entity\Cell;
6
use Box\Spout\Common\Entity\Row;
7
use Box\Spout\Common\Entity\Style\Style;
8
use Box\Spout\Common\Exception\InvalidArgumentException;
9
use Box\Spout\Common\Exception\IOException;
10
use Box\Spout\Common\Helper\Escaper\XLSX as XLSXEscaper;
11
use Box\Spout\Common\Helper\StringHelper;
12
use Box\Spout\Common\Manager\OptionsManagerInterface;
13
use Box\Spout\Writer\Common\Creator\InternalEntityFactory;
14
use Box\Spout\Writer\Common\Entity\Options;
15
use Box\Spout\Writer\Common\Entity\Worksheet;
16
use Box\Spout\Writer\Common\Helper\CellHelper;
17
use Box\Spout\Writer\Common\Manager\RowManager;
18
use Box\Spout\Writer\Common\Manager\Style\StyleMerger;
19
use Box\Spout\Writer\Common\Manager\WorksheetManagerInterface;
20
use Box\Spout\Writer\XLSX\Manager\Style\StyleManager;
21
22
/**
23
 * Class WorksheetManager
24
 * XLSX worksheet manager, providing the interfaces to work with XLSX worksheets.
25
 */
26
class WorksheetManager implements WorksheetManagerInterface
27
{
28
    /**
29
     * Maximum number of characters a cell can contain
30
     * @see https://support.office.com/en-us/article/Excel-specifications-and-limits-16c69c74-3d6a-4aaf-ba35-e6eb276e8eaa [Excel 2007]
31
     * @see https://support.office.com/en-us/article/Excel-specifications-and-limits-1672b34d-7043-467e-8e27-269d656771c3 [Excel 2010]
32
     * @see https://support.office.com/en-us/article/Excel-specifications-and-limits-ca36e2dc-1f09-4620-b726-67c00b05040f [Excel 2013/2016]
33
     */
34
    const MAX_CHARACTERS_PER_CELL = 32767;
35
36
    const SHEET_XML_FILE_HEADER = <<<'EOD'
37
<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
38
<worksheet xmlns="http://schemas.openxmlformats.org/spreadsheetml/2006/main" xmlns:r="http://schemas.openxmlformats.org/officeDocument/2006/relationships">
39
EOD;
40
41
    /** @var bool Whether inline or shared strings should be used */
42
    protected $shouldUseInlineStrings;
43
44
    /** @var RowManager Manages rows */
45
    private $rowManager;
46
47
    /** @var StyleManager Manages styles */
48
    private $styleManager;
49
50
    /** @var StyleMerger Helper to merge styles together */
51
    private $styleMerger;
52
53
    /** @var SharedStringsManager Helper to write shared strings */
54
    private $sharedStringsManager;
55
56
    /** @var XLSXEscaper Strings escaper */
57
    private $stringsEscaper;
58
59
    /** @var StringHelper String helper */
60
    private $stringHelper;
61
62
    /** @var InternalEntityFactory Factory to create entities */
63
    private $entityFactory;
64
65
    /**
66
     * WorksheetManager constructor.
67
     *
68
     * @param OptionsManagerInterface $optionsManager
69
     * @param RowManager $rowManager
70
     * @param StyleManager $styleManager
71
     * @param StyleMerger $styleMerger
72
     * @param SharedStringsManager $sharedStringsManager
73
     * @param XLSXEscaper $stringsEscaper
74
     * @param StringHelper $stringHelper
75
     * @param InternalEntityFactory $entityFactory
76
     */
77 39
    public function __construct(
78
        OptionsManagerInterface $optionsManager,
79
        RowManager $rowManager,
80
        StyleManager $styleManager,
81
        StyleMerger $styleMerger,
82
        SharedStringsManager $sharedStringsManager,
83
        XLSXEscaper $stringsEscaper,
84
        StringHelper $stringHelper,
85
        InternalEntityFactory $entityFactory
86
    ) {
87 39
        $this->shouldUseInlineStrings = $optionsManager->getOption(Options::SHOULD_USE_INLINE_STRINGS);
88 39
        $this->rowManager = $rowManager;
89 39
        $this->styleManager = $styleManager;
90 39
        $this->styleMerger = $styleMerger;
91 39
        $this->sharedStringsManager = $sharedStringsManager;
92 39
        $this->stringsEscaper = $stringsEscaper;
93 39
        $this->stringHelper = $stringHelper;
94 39
        $this->entityFactory = $entityFactory;
95 39
    }
96
97
    /**
98
     * @return SharedStringsManager
99
     */
100 35
    public function getSharedStringsManager()
101
    {
102 35
        return $this->sharedStringsManager;
103
    }
104
105
    /**
106
     * {@inheritdoc}
107
     */
108 39
    public function startSheet(Worksheet $worksheet)
109
    {
110 39
        $sheetFilePointer = fopen($worksheet->getFilePath(), 'w');
111 39
        $this->throwIfSheetFilePointerIsNotAvailable($sheetFilePointer);
112
113 39
        $worksheet->setFilePointer($sheetFilePointer);
114
115 39
        fwrite($sheetFilePointer, self::SHEET_XML_FILE_HEADER);
116 39
        fwrite($sheetFilePointer, '<sheetData>');
117 39
    }
118
119
    /**
120
     * Checks if the sheet has been sucessfully created. Throws an exception if not.
121
     *
122
     * @param bool|resource $sheetFilePointer Pointer to the sheet data file or FALSE if unable to open the file
123
     * @throws IOException If the sheet data file cannot be opened for writing
124
     * @return void
125
     */
126 39
    private function throwIfSheetFilePointerIsNotAvailable($sheetFilePointer)
127
    {
128 39
        if (!$sheetFilePointer) {
129
            throw new IOException('Unable to open sheet for writing.');
130
        }
131 39
    }
132
133
    /**
134
     * {@inheritdoc}
135
     */
136 32
    public function addRow(Worksheet $worksheet, Row $row)
137
    {
138 32
        if (!$this->rowManager->isEmpty($row)) {
139 32
            $this->addNonEmptyRow($worksheet, $row);
140
        }
141
142 31
        $worksheet->setLastWrittenRowIndex($worksheet->getLastWrittenRowIndex() + 1);
143 31
    }
144
145
    /**
146
     * Adds non empty row to the worksheet.
147
     *
148
     * @param Worksheet $worksheet The worksheet to add the row to
149
     * @param Row $row The row to be written
150
     * @throws IOException If the data cannot be written
151
     * @throws InvalidArgumentException If a cell value's type is not supported
152
     * @return void
153
     */
154 32
    private function addNonEmptyRow(Worksheet $worksheet, Row $row)
155
    {
156 32
        $rowStyle = $row->getStyle();
157 32
        $rowIndexOneBased = $worksheet->getLastWrittenRowIndex() + 1;
158 32
        $numCells = $row->getNumCells();
159
160 32
        $rowXML = '<row r="' . $rowIndexOneBased . '" spans="1:' . $numCells . '">';
161
162 32
        foreach ($row->getCells() as $columnIndexZeroBased => $cell) {
163 32
            $rowXML .= $this->applyStyleAndGetCellXML($cell, $rowStyle, $rowIndexOneBased, $columnIndexZeroBased);
164
        }
165
166 31
        $rowXML .= '</row>';
167
168 31
        $wasWriteSuccessful = fwrite($worksheet->getFilePointer(), $rowXML);
169 31
        if ($wasWriteSuccessful === false) {
170
            throw new IOException("Unable to write data in {$worksheet->getFilePath()}");
171
        }
172 31
    }
173
174
    /**
175
     * Applies styles to the given style, merging the cell's style with its row's style
176
     * Then builds and returns xml for the cell.
177
     *
178
     * @param Cell  $cell
179
     * @param Style $rowStyle
180
     * @param int   $rowIndexOneBased
181
     * @param int   $columnIndexZeroBased
182
     *
183
     * @throws InvalidArgumentException If the given value cannot be processed
184
     * @return string
185
     */
186 32
    private function applyStyleAndGetCellXML(Cell $cell, Style $rowStyle, $rowIndexOneBased, $columnIndexZeroBased)
187
    {
188
        // Apply row and extra styles
189 32
        $mergedCellAndRowStyle = $this->styleMerger->merge($cell->getStyle(), $rowStyle);
190 32
        $cell->setStyle($mergedCellAndRowStyle);
191 32
        $newCellStyle = $this->styleManager->applyExtraStylesIfNeeded($cell);
192
193 32
        $registeredStyle = $this->styleManager->registerStyle($newCellStyle);
194
195 32
        return $this->getCellXML($rowIndexOneBased, $columnIndexZeroBased, $cell, $registeredStyle->getId());
196
    }
197
198
    /**
199
     * Builds and returns xml for a single cell.
200
     *
201
     * @param int  $rowIndexOneBased
202
     * @param int  $columnIndexZeroBased
203
     * @param Cell $cell
204
     * @param int  $styleId
205
     *
206
     * @throws InvalidArgumentException If the given value cannot be processed
207
     * @return string
208
     */
209 32
    private function getCellXML($rowIndexOneBased, $columnIndexZeroBased, Cell $cell, $styleId)
210
    {
211 32
        $columnLetters = CellHelper::getColumnLettersFromColumnIndex($columnIndexZeroBased);
212 32
        $cellXML = '<c r="' . $columnLetters . $rowIndexOneBased . '"';
213 32
        $cellXML .= ' s="' . $styleId . '"';
214
215 32
        if ($cell->isString()) {
216 31
            $cellXML .= $this->getCellXMLFragmentForNonEmptyString($cell->getValue());
217 4
        } elseif ($cell->isBoolean()) {
218 1
            $cellXML .= ' t="b"><v>' . (int) ($cell->getValue()) . '</v></c>';
219 4
        } elseif ($cell->isNumeric()) {
220 1
            $cellXML .= '><v>' . $cell->getValue() . '</v></c>';
221 4
        } elseif ($cell->isEmpty()) {
222 2
            if ($this->styleManager->shouldApplyStyleOnEmptyCell($styleId)) {
223 1
                $cellXML .= '/>';
224
            } else {
225
                // don't write empty cells that do no need styling
226
                // NOTE: not appending to $cellXML is the right behavior!!
227 2
                $cellXML = '';
228
            }
229
        } else {
230 2
            throw new InvalidArgumentException('Trying to add a value with an unsupported type: ' . gettype($cell->getValue()));
231
        }
232
233 31
        return $cellXML;
234
    }
235
236
    /**
237
     * Returns the XML fragment for a cell containing a non empty string
238
     *
239
     * @param string $cellValue The cell value
240
     * @throws InvalidArgumentException If the string exceeds the maximum number of characters allowed per cell
241
     * @return string The XML fragment representing the cell
242
     */
243 31
    private function getCellXMLFragmentForNonEmptyString($cellValue)
244
    {
245 31
        if ($this->stringHelper->getStringLength($cellValue) > self::MAX_CHARACTERS_PER_CELL) {
246
            throw new InvalidArgumentException('Trying to add a value that exceeds the maximum number of characters allowed in a cell (32,767)');
247
        }
248
249 31
        if ($this->shouldUseInlineStrings) {
250 28
            $cellXMLFragment = ' t="inlineStr"><is><t>' . $this->stringsEscaper->escape($cellValue) . '</t></is></c>';
251
        } else {
252 3
            $sharedStringId = $this->sharedStringsManager->writeString($cellValue);
253 3
            $cellXMLFragment = ' t="s"><v>' . $sharedStringId . '</v></c>';
254
        }
255
256 31
        return $cellXMLFragment;
257
    }
258
259
    /**
260
     * {@inheritdoc}
261
     */
262 35
    public function close(Worksheet $worksheet)
263
    {
264 35
        $worksheetFilePointer = $worksheet->getFilePointer();
265
266 35
        if (!is_resource($worksheetFilePointer)) {
267
            return;
268
        }
269
270 35
        fwrite($worksheetFilePointer, '</sheetData>');
271 35
        fwrite($worksheetFilePointer, '</worksheet>');
272 35
        fclose($worksheetFilePointer);
273 35
    }
274
}
275