fix count columns in multiple headers

keboola · Dec 3, 2020 · 74052bb · 74052bb
1 parent 9be4ceb
commit 74052bb
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/Keboola/GoogleDriveExtractor/Extractor/Output.php b/src/Keboola/GoogleDriveExtractor/Extractor/Output.php
@@ -56,10 +56,11 @@ public function write(array $data, int $offset): void
         if ($this->header === null) {
             $headerRowNum = $this->sheetCfg['header']['rows'] - 1;
             $this->header = $data[$headerRowNum];
+            $headerLength = $this->getHeaderLength($data, (int) $headerRowNum);
+        } else {
+            $headerLength = count($this->header);
         }
 
-        $headerLength = count($this->header);
-
         foreach ($data as $k => $row) {
             // backward compatibility fix
             if ($this->sheetCfg['header']['rows'] === 1 && $k === 0 && $offset === 1) {
@@ -94,4 +95,13 @@ protected function normalizeCsvHeader(array $header): array
         }
         return $header;
     }
+
+    private function getHeaderLength(array $data, int $headerRowNum): int
+    {
+        $headerLength = 0;
+        for ($i = 0; $i <= $headerRowNum; $i++) {
+            $headerLength = max($headerLength, count($data[$i]));
+        }
+        return $headerLength;
+    }
 }