<?php

namespace PhpOffice\PhpSpreadsheetTests\Functional;

use PhpOffice\PhpSpreadsheet\Reader\IReadFilter;
use PhpOffice\PhpSpreadsheet\Spreadsheet;

class ColumnWidthTest extends AbstractFunctional
{
    public function providerFormats()
    {
        return [
            ['Xlsx'],
        ];
    }

    /**
     * @dataProvider providerFormats
     *
     * @param $format
     */
    public function testReadColumnWidth($format): void
    {
        // create new sheet with column width
        $spreadsheet = new Spreadsheet();
        $sheet = $spreadsheet->getActiveSheet();
        $sheet->setCellValue('A1', 'Hello World !');
        $sheet->getColumnDimension('A')->setWidth(20);
        $this->assertColumn($spreadsheet);

        $reloadedSpreadsheet = $this->writeAndReload($spreadsheet, $format);
        $this->assertColumn($reloadedSpreadsheet);
    }

    /**
     * @dataProvider providerFormats
     *
     * @param $format
     */
    public function testReadColumnWidthWithReadFilter($format): void
    {
        // Same test with a read filter removing a single row
        $spreadsheet = new Spreadsheet();
        $sheet = $spreadsheet->getActiveSheet();
        foreach (range(1, 5) as $row) {
            $sheet->setCellValue("A{$row}", 'Hello');
            $sheet->setCellValue("B{$row}", 'World !');
        }

        // isFilteredColumn iterates rowAttributes when calling the read filter
        $sheet->getRowDimension(5)->setRowHeight(10);

        $sheet->getColumnDimension('A')->setWidth(20);
        $sheet->getColumnDimension('B')->setWidth(30);
        $this->assertColumn($spreadsheet);
        $this->assertColumn($spreadsheet, 'B', 30);

        // A reader-customeiser closure and ReadFilter implementation that skips rows >4
        $readerCustomizer = function ($reader) {
            $readFilterStub = $this->createMock(IReadFilter::class);
            $readFilterStub->method('readCell')
                ->willReturnCallback(function ($column, $row, $worksheetName = '') {
                    if ($column == 'B') {
                        return false;
                    }
                    return $row <= 4;
                });
            $reader->setReadFilter($readFilterStub);
        };

        // Save and reload a filtered set and assert the same width
        $reloadedSpreadsheet = $this->writeAndReload($spreadsheet, $format, $readerCustomizer);
        $this->assertColumn($reloadedSpreadsheet);
        $this->assertNotColumn($reloadedSpreadsheet, 'B', 30);
    }

    private function assertColumn(Spreadsheet $spreadsheet, $column = 'A', $expected = 20): void
    {
        $sheet = $spreadsheet->getActiveSheet();
        $columnDimensions = $sheet->getColumnDimensions();

        self::assertArrayHasKey($column, $columnDimensions);
        self::assertEquals($expected, $columnDimensions[$column]->getWidth());
    }

    private function assertNotColumn(Spreadsheet $spreadsheet, $column = 'A', $expected = 20): void
    {
        $sheet = $spreadsheet->getActiveSheet();
        $columnDimensions = $sheet->getColumnDimensions();

        self::assertArrayNotHasKey($column, $columnDimensions);
    }
}