QB\PostgreSQL\Statement\Insert

Complexity

Total Complexity

16

Size/Duplication

Total Lines	130
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes	3
Bugs	0	Features	0

9 Methods

Rating	Name	Duplication	Size	Complexity
A	setDoNothing()	0	6	1
A	setDoUpdate()	0	6	1
A	setOnConflict()	0	5	1
A	setReturning()	0	5	1
A	getOnConflict()	0	20	6
A	getReturning()	0	7	2
A	__toString()	0	9	1
A	getRawValues()	0	7	2
A	isValid()	0	3	1

<?php

declare(strict_types=1);

namespace QB\PostgreSQL\Statement;

use QB\Generic\Statement\Insert as GenericInsert;

class Insert extends GenericInsert
{
    public const DEFAULT_VALUES = 'DEFAULT VALUES';

    public const CONFLICT_DO_NOTHING = 'DO NOTHING';
    public const CONFLICT_DO_UPDATE  = 'DO UPDATE';

    /** @var array<int,string> */
    protected array $onConflict = [];

    /** @var array<int,string> */
    protected array $doUpdate = [];

    /** @var bool */
    protected bool $doNothing = false;

    /** @var array<int,string> */
    protected array $returning = [];

    /**
     * @param string ...$columns
     *
     * @return $this
     */
    public function setOnConflict(string ...$columns): static
    {
        $this->onConflict = $columns;

        return $this;
    }

    /**
     * @param string ...$columns
     *
     * @return $this
     */
    public function setDoUpdate(string ...$columns): static
    {
        $this->doNothing = false;
        $this->doUpdate  = $columns;

        return $this;
    }

    /**
     * @return $this
     */
    public function setDoNothing(): static
    {
        $this->doNothing = true;
        $this->doUpdate  = [];

        return $this;
    }

    /**
     * @param string ...$columns
     *
     * @return $this
     */
    public function setReturning(string ...$columns): static
    {
        $this->returning = $columns;

        return $this;
    }

    /**
     * @return string
     */
    public function __toString(): string
    {
        $parts = array_merge(
            [parent::__toString()],
            $this->getOnConflict(),
            $this->getReturning(),
        );

        return implode(PHP_EOL, $parts);
    }

    protected function getOnConflict(): array
    {
        if (!$this->doNothing && count($this->doUpdate) == 0) {
            return [];
        }

        $parts = [];

        $action = $this->doNothing ? static::CONFLICT_DO_NOTHING : static::CONFLICT_DO_UPDATE;
        if (count($this->onConflict) > 0) {
            $parts[] = sprintf('ON CONFLICT (%s) %s', implode(', ', $this->onConflict), $action);
        } else {
            $parts[] = sprintf('ON CONFLICT %s', $action);
        }

        if (count($this->doUpdate) > 0) {
            $parts[] = sprintf('SET %s', implode(', ', $this->doUpdate));
        }

        return $parts;
    }

    protected function getReturning(): array
    {
        if (count($this->returning) == 0) {
            return [];
        }

        return [sprintf('RETURNING %s', implode(',', $this->returning))];
    }

    /**
     * @return bool
     */
    public function isValid(): bool
    {
        return count($this->tables) === 1;
    }

    /**
     * @return string[]
     */
    protected function getRawValues(): array
    {
        if (count($this->rawValues) == 0) {
            return [self::DEFAULT_VALUES];
        }

        return parent::getRawValues();
    }
}