<?php
namespace catcher\library\excel;

trait MacroExcel
{
    /**
     * @var string
     */
    protected $start = 'A';

    /**
     * 开始行
     *
     * @var int
     */
    protected $row = 1;

    /**
     * @var array
     */
    protected $columns = [];

    /**
     * 设置开始的单元
     *
     * @time 2020年05月25日
     * @return string
     */
    protected function getStartSheet(): string
    {
        if (method_exists($this->excel, 'start')) {
            return $this->excel->start();
        }

        return $this->start;
    }

    /**
     * 设置单元格宽度
     *
     * @time 2020年05月25日
     * @return void
     */
    protected function setSheetWidth()
    {
        if (method_exists($this->excel, 'setWidth')) {
            $width = $this->excel->setWidth();

            foreach ($width as $sheet => $w) {
                $this->getWorksheet()->getColumnDimension($sheet)->setWidth($w);
            }
        }
    }

    /**
     * 设置 column 信息 ['A', 'B', 'C' ...]
     *
     * @time 2020年05月25日
     * @return array
     */
    protected function getSheetColumns()
    {
        if (empty($this->columns)) {
            if (method_exists($this->excel, 'sheetColumns')) {
                $this->columns = $this->excel->sheetColumns();
                return $this->columns;
            }

            $start = $this->getStartSheet();

            $columns = [];
            // 通过 headers 推断需要的 columns
            foreach ($this->excel->headers() as $k => $header) {
                $columns[] = chr(ord($start) + $k);
            }

            return $columns;
        }

        return $this->columns;
    }

    /**
     * set keys
     *
     * @time 2020年05月25日
     * @return array
     */
    protected function getKeys()
    {
        if (method_exists($this->excel, 'keys')) {
            return $this->excel->keys();
        }

        return [];
    }

    /**
     * set start row
     *
     * @time 2020年05月25日
     * @return int
     */
    protected function getStartRow()
    {
        if (method_exists($this->excel, 'setRow')) {
            $this->row = $this->excel->setRow();
        }

        return $this->row;
    }

    /**
     * 设置 title
     *
     * @time 2020年05月25日
     * @return void
     */
    protected function setTitle()
    {
        if (method_exists($this->excel, 'setTitle')) {

            [$cells, $title, $style] = $this->excel->setTitle();

            $this->getWorksheet()
                 ->mergeCells($cells) // 合并单元格
                 ->setCellValue(explode(':', $cells)[0], $title)
                 ->getStyle($cells) // 设置样式
                 ->getAlignment()
                 ->setHorizontal($style);

        }
    }

    /**
     * 设置其他信息
     *
     * @time 2020年05月25日
     * @return void
     */
    protected function setOther()
    {
        if (method_exists($this->excel, 'setOther')) {
             $this->excel->setOther($this->getWorksheet());
        }
    }

    /**
     * register worksheet for excel
     *
     * @time 2020年05月25日
     * @return void
     */
    protected function registerWorksheet()
    {
        if (method_exists($this->excel, 'getWorksheet')) {
            $this->excel->getWorksheet($this->getWorksheet());
        }
    }

    /**
     * 增加 start row
     *
     * @time 2020年05月25日
     * @return void
     */
    protected function incRow()
    {
        ++$this->row;
    }
}