Completed
Push — master ( 94029f...e94e78 )
by Beniamin
02:32
created

ReferenceCompiler::convertScalarReference()   A

Complexity

Conditions 2
Paths 2

Size

Total Lines 8
Code Lines 4

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 0
CRAP Score 6

Importance

Changes 0
Metric Value
c 0
b 0
f 0
dl 0
loc 8
ccs 0
cts 4
cp 0
rs 9.4285
cc 2
eloc 4
nc 2
nop 1
crap 6
1
<?php
2
3
/**
4
 * This file is part of UnderQuery package.
5
 *
6
 * Copyright (c) 2016 Beniamin Jonatan Šimko
7
 *
8
 * For the full copyright and license information, please view the LICENSE
9
 * file that was distributed with this source code.
10
 */
11
12
namespace Phuria\UnderQuery\QueryCompiler;
13
14
use Phuria\UnderQuery\Language\Expression\RelativeClause;
15
use Phuria\UnderQuery\QueryBuilder\AbstractBuilder;
16
use Phuria\UnderQuery\QueryBuilder\BuilderInterface;
17
use Phuria\UnderQuery\Table\AbstractTable;
18
19
/**
20
 * @author Beniamin Jonatan Šimko <[email protected]>
21
 */
22
class ReferenceCompiler
23
{
24
    /**
25
     * @param CompilerPayload $payload
26
     *
27
     * @return CompilerPayload
28
     */
29
    public function compileReference(CompilerPayload $payload)
30
    {
31
        $builder = $payload->getBuilder();
32
        $references = [];
33
34
        if ($builder instanceof AbstractBuilder) {
35
            $references = $builder->getReferences()->toArray();
36
        }
37
38
        $actualSQL = $this->compile($payload->getActualSQL(), $references);
39
40
        return $payload->updateSQL($actualSQL);
41
    }
42
43
    /**
44
     * @param string $rawSQL
45
     * @param array  $references
46
     *
47
     * @return string
48
     */
49
    public function compile($rawSQL, array $references)
50
    {
51
        foreach ($references as &$value) {
52
            $value = $this->convertReferenceToValue($value);
53
        }
54
55
        return str_replace(array_keys($references), array_values($references), $rawSQL);
56
    }
57
58
    /**
59
     * @param $reference
60
     *
61
     * @return string
62
     */
63
    private function convertReferenceToValue($reference)
64
    {
65
        if (is_scalar($reference)) {
66
            return $this->convertScalarReference($reference);
67
        } else {
68
            return $this->convertObjectReference($reference);
69
        }
70
    }
71
72
    /**
73
     * @param $reference
74
     *
75
     * @return string
76
     */
77
    private function convertScalarReference($reference)
78
    {
79
        if (is_string($reference)) {
80
            return "\"" . $reference ."\"";
81
        }
82
83
        return $reference;
84
    }
85
86
    /**
87
     * @param mixed $reference
88
     *
89
     * @return string
90
     */
91
    private function convertObjectReference($reference)
92
    {
93
        if ($reference instanceof AbstractTable) {
94
            return $reference->getAliasOrName();
95
        } elseif ($reference instanceof BuilderInterface) {
96
            return $reference->buildSQL();
97
        } elseif ($reference instanceof RelativeClause) {
98
            $table = $this->convertObjectReference($reference->getRelatedTable());
99
            $clause = $this->convertObjectReference($reference->getClause());
100
101
            return str_replace($reference->getDirective(), $table . '.', $clause);
102
        }
103
104
        return $reference;
105
    }
106
}