Scrawler\Arca\QueryBuilder

Complexity

Total Complexity

6

Size/Duplication

Total Lines	44
Duplicated Lines	0 %

Importance

Changes	2
Bugs	0	Features	0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	get()	0	7	1
A	with()	0	4	1
A	from()	0	7	1
A	first()	0	6	2

<?php

namespace Scrawler\Arca;

/**
 * Extended implementation of \Doctrine\DBAL\Query\QueryBuilder
 */
class QueryBuilder extends \Doctrine\DBAL\Query\QueryBuilder
{
    private string $table;
    private array $relations= [];

    private Database $db;

    public function __construct(Database $db)
    {
        $this->db = $db;
        parent::__construct($this->db->connection);

    }

    public function with(string $relation): QueryBuilder
    {
        array_push($this->relations, $relation);
        return $this;
    }

    public function from($from, $alias = null): QueryBuilder
    {
        $this->table = $from;
        return $this->add('from', [
            'table' => $from,
            'alias' => $alias,
        ], true);
    }

    public function get(): Collection
    {
        $model = $this->db->create($this->table);
        $relations = $this->relations;
        $this->relations = [];
        return Collection::fromIterable($this->fetchAllAssociative())
            ->map(static fn($value): Model => ($model)->setProperties($value)->with($relations)->setLoaded());
    }

    public function first(): Model
    {
        $relations = $this->relations;
        $this->relations = [];
        $result = $this->fetchAssociative() ? $this->fetchAssociative() : [];
        return ($this->db->create($this->table))->setProperties($result)->with($relations)->setLoaded();
    }
}