<?php

namespace Revolution\Google\Sheets\Concerns;

use Google\Service\Sheets\BatchUpdateValuesRequest;
use Google\Service\Sheets\ClearValuesRequest;
use Google\Service\Sheets\ClearValuesResponse;
use Google\Service\Sheets\ValueRange;
use Illuminate\Support\Arr;
use Illuminate\Support\Str;

trait SheetsValues
{
    /**
     * @var string
     */
    protected $range;

    /**
     * @var string
     */
    protected $majorDimension;

    /**
     * @var string
     */
    protected $valueRenderOption;

    /**
     * @var string
     */
    protected $dateTimeRenderOption;

    /**
     * @return array
     */
    public function all()
    {
        $query = $this->query();

        $sheets = $this->serviceValues()->batchGet($this->spreadsheetId, $query);

        $values = $sheets->getValueRanges()[0]->getValues();

        return $values ?? [];
    }

    /**
     * @return array
     */
    public function first()
    {
        $values = $this->all();

        $first = head($values);

        return $first ?: [];
    }

    /**
     * @param  array  $value
     * @param  string  $valueInputOption
     * @return mixed|\Google_Service_Sheets_UpdateValuesResponse
     */
    public function update(array $value, string $valueInputOption = 'RAW')
    {
        $range = $this->ranges();

        $batch = new BatchUpdateValuesRequest();
        $batch->setValueInputOption($valueInputOption);

        $valueRange = new ValueRange();
        $valueRange->setValues($value);
        $valueRange->setRange($range);

        $batch->setData($valueRange);

        return $this->serviceValues()->batchUpdate($this->spreadsheetId, $batch);
    }

    /**
     * @return mixed|ClearValuesResponse
     */
    public function clear()
    {
        $range = $this->ranges();

        $clear = new ClearValuesRequest();

        return $this->serviceValues()->clear($this->spreadsheetId, $range, $clear);
    }

    /**
     * @param  array  $values
     * @param  string  $valueInputOption
     * @param  string  $insertDataOption
     * @return mixed|\Google_Service_Sheets_AppendValuesResponse
     */
    public function append(array $values, string $valueInputOption = 'RAW', string $insertDataOption = 'OVERWRITE')
    {
        $range = $this->ranges();
        $orderedValues = $this->orderAppendables($values);

        $valueRange = new ValueRange();
        $valueRange->setValues($orderedValues);
        $valueRange->setRange($range);

        $optParams = [
            'valueInputOption' => $valueInputOption,
            'insertDataOption' => $insertDataOption,
        ];

        return $this->serviceValues()->append($this->spreadsheetId, $range, $valueRange, $optParams);
    }

    /**
     * @param  array  $values
     * @return array
     */
    public function orderAppendables(array $values)
    {
        // The array has integer keys, so just append
        if (! Arr::isAssoc(head($values) ?: [])) {
            return $values;
        }

        // The array has keys, which we want to map to headers and order
        $header = $this->first();

        $ordered = [];
        // Gets just the values of an array that has been re-ordered to match the header order
        foreach ($values as $value) {
            array_push(
                $ordered,
                array_values(array_replace(array_flip($header), $value))
            );
        }

        // Replaces null values with empty strings to work with Google's API
        return array_map(function ($row) {
            $notNull = [];
            foreach ($row as $key => $value) {
                // If key is the same as value, that's because the user
                // didn't specify a header that exists in the sheet.
                if (is_null($value) || $key === $value) {
                    array_push($notNull, '');
                } else {
                    array_push($notNull, $value);
                }
            }

            return $notNull;
        }, $ordered);
    }

    /**
     * @return string
     */
    public function ranges()
    {
        // If no range is provided, we get the sheet automatically
        if (blank($this->range)) {
            return $this->sheet;
        }

        // If we only provide part of the range, we get the full proper range
        if (! Str::contains($this->range, '!')) {
            return $this->sheet.'!'.$this->range;
        }

        // If we provide the full range, it returns accurately
        return $this->range;
    }

    /**
     * @param  string  $range
     * @return $this
     */
    public function range(string $range)
    {
        $this->range = $range;

        return $this;
    }

    /**
     * @param  string  $majorDimension
     * @return $this
     */
    public function majorDimension(string $majorDimension)
    {
        $this->majorDimension = $majorDimension;

        return $this;
    }

    /**
     * @param  string  $valueRenderOption
     * @return $this
     */
    public function valueRenderOption(string $valueRenderOption)
    {
        $this->valueRenderOption = $valueRenderOption;

        return $this;
    }

    /**
     * @param  string  $dateTimeRenderOption
     * @return $this
     */
    public function dateTimeRenderOption(string $dateTimeRenderOption)
    {
        $this->dateTimeRenderOption = $dateTimeRenderOption;

        return $this;
    }

    /**
     * @return \Google_Service_Sheets_Resource_SpreadsheetsValues
     */
    protected function serviceValues()
    {
        return $this->getService()->spreadsheets_values;
    }

    /**
     * @return array
     */
    protected function query(): array
    {
        $query = [];

        $ranges = $this->ranges();

        if (! empty($ranges)) {
            $query['ranges'] = $ranges;
        }

        if (! empty($this->majorDimension)) {
            $query['majorDimension'] = $this->majorDimension;
        }

        if (! empty($this->valueRenderOption)) {
            $query['valueRenderOption'] = $this->valueRenderOption;
        }

        if (! empty($this->dateTimeRenderOption)) {
            $query['dateTimeRenderOption'] = $this->dateTimeRenderOption;
        }

        return $query;
    }
}