LaravelFreelancerNL\Aranguent\Schema\Grammar

Complexity

Total Complexity

8

Size/Duplication

Total Lines	118
Duplicated Lines	0 %

Test Coverage

Coverage

97.92%

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	wrapBindVar()	0	8	2
A	compileDropAttribute()	0	19	2
A	compileHasColumn()	0	21	3
A	compileRenameAttribute()	0	25	1

<?php

namespace LaravelFreelancerNL\Aranguent\Schema;

use Illuminate\Database\Schema\Grammars\Grammar as IlluminateGrammar;
use Illuminate\Support\Fluent;
use LaravelFreelancerNL\FluentAQL\QueryBuilder;

class Grammar extends IlluminateGrammar
{
    /**
     * ArangoDB handles transactions itself. However most schema actions
     * are not done through queries but rather through commands.
     * So we just run schema actions sequentially.
     *
     * @var bool
     */
    protected $transactions = false;

    /**
     * Compile AQL to check if an attribute is in use within a document in the collection.
     * If multiple attributes are set then all must be set in one document.
     *
     * @param string $collection
     * @param Fluent $command
     *
     * @return Fluent
     */
    public function compileHasColumn($collection, Fluent $command)
    {
        $attributes = $command->getAttributes();
        if (!isset($attributes['columns'])) {
            return $command;
        }

        $filter = [];
        foreach ($attributes['columns'] as $column) {
            $filter[] = ['doc.' . $column, '!=', null];
        }

        $aqb = (new QueryBuilder())->for('doc', $collection)
            ->filter($filter)
            ->limit(1)
            ->return('true')
            ->get();

        $command->aqb = $aqb;

        return $command;
    }

    /**
     * Compile AQL to rename an attribute, if the new name isn't already in use.
     *
     * @param string $collection
     * @param Fluent $command
     *
     * @return Fluent
     */
    public function compileRenameAttribute($collection, Fluent $command)
    {
        $attributes = $command->getAttributes();

        $filter = [
            ['doc.' . $attributes['from'], '!=', null],
            ['doc.' . $attributes['to'], '==', null],
        ];

        $aqb = (new QueryBuilder())->for('doc', $collection)
            ->filter($filter)
            ->update(
                'doc',
                [
                    $attributes['from'] => null,
                    $attributes['to']   => 'doc.' . $command->from,
                ],
                $collection
            )
            ->options(['keepNull' => true])
            ->get();

        $command->aqb = $aqb;

        return $command;
    }

    /**
     * Compile AQL to drop one or more attributes.
     *
     * @param string $collection
     * @param Fluent $command
     *
     * @return Fluent
     */
    public function compileDropAttribute($collection, Fluent $command)
    {
        $filter = [];
        $attributes = $command->getAttributes();

        $data = [];
        foreach ($attributes['attributes'] as $attribute) {
            $filter[] = ['doc.' . $attribute, '!=', null, 'OR'];
            $data[$attribute] = null;
        }
        $aqb = (new QueryBuilder())->for('doc', $collection)
            ->filter($filter)
            ->update('doc', $data, $collection)
            ->options(['keepNull' => false])
            ->get();

        $command->aqb = $aqb;

        return $command;
    }

    /**
     * Prepare a bindVar for inclusion in an AQL query.
     */
    public function wrapBindVar(string $attribute): string
    {
        $attribute = trim($attribute);
        if (strpos($attribute, '@') === 0) {
            $attribute = "`$attribute`";
        }

        return $attribute;
    }
}