Completed
Push — master ( 91f756...eb84ec )
by Adrien
13:07
created

WorksheetManager::applyStyleAndRegister()   A

Complexity

Conditions 4
Paths 4

Size

Total Lines 30

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 12
CRAP Score 4

Importance

Changes 0
Metric Value
dl 0
loc 30
ccs 12
cts 12
cp 1
rs 9.44
c 0
b 0
f 0
cc 4
nc 4
nop 2
crap 4
1
<?php
2
3
namespace Box\Spout\Writer\ODS\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\ODS as ODSEscaper;
11
use Box\Spout\Common\Helper\StringHelper;
12
use Box\Spout\Writer\Common\Entity\Worksheet;
13
use Box\Spout\Writer\Common\Manager\RegisteredStyle;
14
use Box\Spout\Writer\Common\Manager\Style\StyleMerger;
15
use Box\Spout\Writer\Common\Manager\WorksheetManagerInterface;
16
use Box\Spout\Writer\ODS\Manager\Style\StyleManager;
17
18
/**
19
 * Class WorksheetManager
20
 * ODS worksheet manager, providing the interfaces to work with ODS worksheets.
21
 */
22
class WorksheetManager implements WorksheetManagerInterface
23
{
24
    /** @var \Box\Spout\Common\Helper\Escaper\ODS Strings escaper */
25
    private $stringsEscaper;
26
27
    /** @var StringHelper String helper */
28
    private $stringHelper;
29
30
    /** @var StyleManager Manages styles */
31
    private $styleManager;
32
33
    /** @var StyleMerger Helper to merge styles together */
34
    private $styleMerger;
35
36
    /**
37
     * WorksheetManager constructor.
38
     *
39
     * @param StyleManager $styleManager
40
     * @param StyleMerger $styleMerger
41
     * @param ODSEscaper $stringsEscaper
42
     * @param StringHelper $stringHelper
43 38
     */
44
    public function __construct(
45
        StyleManager $styleManager,
46
        StyleMerger $styleMerger,
47
        ODSEscaper $stringsEscaper,
48
        StringHelper $stringHelper
49 38
    ) {
50 38
        $this->styleManager = $styleManager;
51 38
        $this->styleMerger = $styleMerger;
52 38
        $this->stringsEscaper = $stringsEscaper;
53 38
        $this->stringHelper = $stringHelper;
54
    }
55
56
    /**
57
     * Prepares the worksheet to accept data
58
     *
59
     * @param Worksheet $worksheet The worksheet to start
60
     * @throws \Box\Spout\Common\Exception\IOException If the sheet data file cannot be opened for writing
61
     * @return void
62 38
     */
63
    public function startSheet(Worksheet $worksheet)
64 38
    {
65 38
        $sheetFilePointer = \fopen($worksheet->getFilePath(), 'w');
66
        $this->throwIfSheetFilePointerIsNotAvailable($sheetFilePointer);
67 38
68 38
        $worksheet->setFilePointer($sheetFilePointer);
69
    }
70
71
    /**
72
     * Checks if the sheet has been sucessfully created. Throws an exception if not.
73
     *
74
     * @param bool|resource $sheetFilePointer Pointer to the sheet data file or FALSE if unable to open the file
75
     * @throws IOException If the sheet data file cannot be opened for writing
76
     * @return void
77 38
     */
78
    private function throwIfSheetFilePointerIsNotAvailable($sheetFilePointer)
79 38
    {
80
        if (!$sheetFilePointer) {
81
            throw new IOException('Unable to open sheet for writing.');
82 38
        }
83
    }
84
85
    /**
86
     * Returns the table XML root node as string.
87
     *
88
     * @param Worksheet $worksheet
89
     * @return string <table> node as string
90 35
     */
91
    public function getTableElementStartAsString(Worksheet $worksheet)
92 35
    {
93 35
        $externalSheet = $worksheet->getExternalSheet();
94 35
        $escapedSheetName = $this->stringsEscaper->escape($externalSheet->getName());
95
        $tableStyleName = 'ta' . ($externalSheet->getIndex() + 1);
96 35
97 35
        $tableElement = '<table:table table:style-name="' . $tableStyleName . '" table:name="' . $escapedSheetName . '">';
98
        $tableElement .= '<table:table-column table:default-cell-style-name="ce1" table:style-name="co1" table:number-columns-repeated="' . $worksheet->getMaxNumColumns() . '"/>';
99 35
100
        return $tableElement;
101
    }
102
103
    /**
104
     * Adds a row to the given worksheet.
105
     *
106
     * @param Worksheet $worksheet The worksheet to add the row to
107
     * @param Row $row The row to be added
108
     * @throws InvalidArgumentException If a cell value's type is not supported
109
     * @throws IOException If the data cannot be written
110
     * @return void
111 33
     */
112
    public function addRow(Worksheet $worksheet, Row $row)
113 33
    {
114 33
        $cells = $row->getCells();
115
        $rowStyle = $row->getStyle();
116 33
117
        $data = '<table:table-row table:style-name="ro1">';
118 33
119 33
        $currentCellIndex = 0;
120
        $nextCellIndex = 1;
121 33
122
        for ($i = 0; $i < $row->getNumCells(); $i++) {
123 33
            /** @var Cell $cell */
124
            $cell = $cells[$currentCellIndex];
125 33
            /** @var Cell|null $nextCell */
126
            $nextCell = isset($cells[$nextCellIndex]) ? $cells[$nextCellIndex] : null;
127 33
128 33
            if ($nextCell === null || $cell->getValue() !== $nextCell->getValue()) {
129 32
                $registeredStyle = $this->applyStyleAndRegister($cell, $rowStyle);
130
                $cellStyle = $registeredStyle->getStyle();
131
                if ($registeredStyle->isMatchingRowStyle()) {
132 32
                    $rowStyle = $cellStyle; // Replace actual rowStyle (possibly with null id) by registered style (with id)
133
                }
134
135 32
                $data .= $this->getCellXMLWithStyle($cell, $cellStyle, $currentCellIndex, $nextCellIndex);
136
                $currentCellIndex = $nextCellIndex;
137 32
            }
138 32
139
            $nextCellIndex++;
140
        }
141
142
        $data .= '</table:table-row>';
143 32
144 32
        $wasWriteSuccessful = \fwrite($worksheet->getFilePointer(), $data);
145 32
        if ($wasWriteSuccessful === false) {
146
            throw new IOException("Unable to write data in {$worksheet->getFilePath()}");
147
        }
148
149
        // only update the count if the write worked
150
        $lastWrittenRowIndex = $worksheet->getLastWrittenRowIndex();
151
        $worksheet->setLastWrittenRowIndex($lastWrittenRowIndex + 1);
152
    }
153
154
    /**
155
     * Applies styles to the given style, merging the cell's style with its row's style
156
     *
157
     * @param Cell $cell
158 33
     * @param Style $rowStyle
159
     * @throws InvalidArgumentException If a cell value's type is not supported
160
     * @return RegisteredStyle
161 33
     */
162 33
    private function applyStyleAndRegister(Cell $cell, Style $rowStyle) : RegisteredStyle
163 33
    {
164
        $isMatchingRowStyle = false;
165 33
        if ($cell->getStyle()->isEmpty()) {
166 33
            $cell->setStyle($rowStyle);
167
168 33
            $possiblyUpdatedStyle = $this->styleManager->applyExtraStylesIfNeeded($cell);
169
170 33
            if ($possiblyUpdatedStyle->isUpdated()) {
171
                $registeredStyle = $this->styleManager->registerStyle($possiblyUpdatedStyle->getStyle());
172
            } else {
173
                $registeredStyle = $this->styleManager->registerStyle($rowStyle);
174
                $isMatchingRowStyle = true;
175
            }
176
        } else {
177
            $mergedCellAndRowStyle = $this->styleMerger->merge($cell->getStyle(), $rowStyle);
178
            $cell->setStyle($mergedCellAndRowStyle);
179
180
            $possiblyUpdatedStyle = $this->styleManager->applyExtraStylesIfNeeded($cell);
181
            if ($possiblyUpdatedStyle->isUpdated()) {
182 33
                $newCellStyle = $possiblyUpdatedStyle->getStyle();
183
            } else {
184 33
                $newCellStyle = $mergedCellAndRowStyle;
185
            }
186 33
187 4
            $registeredStyle = $this->styleManager->registerStyle($newCellStyle);
188
        }
189
190 33
        return new RegisteredStyle($registeredStyle, $isMatchingRowStyle);
191 28
    }
192
193 28
    private function getCellXMLWithStyle(Cell $cell, Style $style, int $currentCellIndex, int $nextCellIndex) : string
194 28
    {
195 28
        $styleIndex = $style->getId() + 1; // 1-based
196
197
        $numTimesValueRepeated = ($nextCellIndex - $currentCellIndex);
198 28
199 7
        return $this->getCellXML($cell, $styleIndex, $numTimesValueRepeated);
200 2
    }
201 2
202 2
    /**
203 2
     * Returns the cell XML content, given its value.
204 6
     *
205 2
     * @param Cell $cell The cell to be written
206 2
     * @param int $styleIndex Index of the used style
207 2
     * @param int $numTimesValueRepeated Number of times the value is consecutively repeated
208 5
     * @throws InvalidArgumentException If a cell value's type is not supported
209
     * @return string The cell XML content
210 1
     */
211 1
    private function getCellXML(Cell $cell, $styleIndex, $numTimesValueRepeated)
212 1
    {
213 4
        $data = '<table:table-cell table:style-name="ce' . $styleIndex . '"';
214 2
215
        if ($numTimesValueRepeated !== 1) {
216 2
            $data .= ' table:number-columns-repeated="' . $numTimesValueRepeated . '"';
217
        }
218
219 32
        if ($cell->isString()) {
220
            $data .= ' office:value-type="string" calcext:value-type="string">';
221
222
            $cellValueLines = \explode("\n", $cell->getValue());
223
            foreach ($cellValueLines as $cellValueLine) {
224
                $data .= '<text:p>' . $this->stringsEscaper->escape($cellValueLine) . '</text:p>';
225
            }
226
227
            $data .= '</table:table-cell>';
228 35
        } elseif ($cell->isBoolean()) {
229
            $value = $cell->getValue() ? 'true' : 'false'; // boolean-value spec: http://docs.oasis-open.org/office/v1.2/os/OpenDocument-v1.2-os-part1.html#datatype-boolean
230 35
            $data .= ' office:value-type="boolean" calcext:value-type="boolean" office:boolean-value="' . $value . '">';
231
            $data .= '<text:p>' . $cell->getValue() . '</text:p>';
232 35
            $data .= '</table:table-cell>';
233
        } elseif ($cell->isNumeric()) {
234
            $data .= ' office:value-type="float" calcext:value-type="float" office:value="' . $cell->getValue() . '">';
235
            $data .= '<text:p>' . $cell->getValue() . '</text:p>';
236 35
            $data .= '</table:table-cell>';
237 35
        } elseif ($cell->isError() && is_string($cell->getValueEvenIfError())) {
238
            // only writes the error value if it's a string
239
            $data .= ' office:value-type="string" calcext:value-type="error" office:value="">';
240
            $data .= '<text:p>' . $cell->getValueEvenIfError() . '</text:p>';
241
            $data .= '</table:table-cell>';
242
        } elseif ($cell->isEmpty()) {
243
            $data .= '/>';
244
        } else {
245
            throw new InvalidArgumentException('Trying to add a value with an unsupported type: ' . \gettype($cell->getValue()));
246
        }
247
248
        return $data;
249
    }
250
251
    /**
252
     * Closes the worksheet
253
     *
254
     * @param Worksheet $worksheet
255
     * @return void
256
     */
257
    public function close(Worksheet $worksheet)
258
    {
259
        $worksheetFilePointer = $worksheet->getFilePointer();
260
261
        if (!\is_resource($worksheetFilePointer)) {
262
            return;
263
        }
264
265
        \fclose($worksheetFilePointer);
266
    }
267
}
268