Passed
Pull Request — master (#40)
by Def
01:57
created

extractParametersFromTokens()   A

Complexity

Conditions 3
Paths 3

Size

Total Lines 17
Code Lines 8

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 9
CRAP Score 3

Importance

Changes 3
Bugs 1 Features 0
Metric Value
eloc 8
c 3
b 1
f 0
dl 0
loc 17
ccs 9
cts 9
cp 1
rs 10
cc 3
nc 3
nop 1
crap 3
1
<?php
2
3
declare(strict_types=1);
4
5
namespace Yiisoft\Translator;
6
7
use Yiisoft\Files\FileHelper;
8
use Yiisoft\Files\PathMatcher\PathMatcher;
9
10
/**
11
 * Extractor messages
12
 */
13
final class TranslationExtractor
14
{
15
    private string $translator = '->translate';
16
17
    /**
18
     * @var array<string|array{0: int, 1: string, 2: int}>
19
     */
20
    private array $translatorTokens = [];
21
    private int $sizeOfTranslator = 0;
22
23
    private array $skippedLines = [];
24
    private array $skippedLinesOfFile = [];
25
26
    private string $defaultCategory = '';
27
28
    /**
29
     * @var string[]
30
     */
31
    private array $only = ['**.php'];
32
33
    /**
34
     * @var string[]
35
     */
36
    private array $except = [
37
        '.svn',
38
        '.git',
39
        '.gitignore',
40
        '.gitkeep',
41
        '.hgignore',
42
        '.hgkeep',
43
        '/messages',
44
    ];
45
46
    private static array $commaSpare = [
47
        ')' => '(',
48
        ']' => '[',
49
        '}' => '{',
50
    ];
51
52
    /**
53
     * @param string $path
54
     * @param string[]|null $only
55
     * @param string[]|null $except
56
     *
57
     * @return array
58
     */
59 9
    public function extract(string $path, ?array $only = null, ?array $except = null): array
60
    {
61 9
        if (!is_dir($path)) {
62 1
            throw new \RuntimeException(sprintf('Directory "%s" does not exist.', $path));
63
        }
64
65 8
        $translatorTokens = token_get_all('<?php ' . $this->translator);
66 8
        array_shift($translatorTokens);
67 8
        $this->translatorTokens = $translatorTokens;
68 8
        $this->sizeOfTranslator = count($this->translatorTokens);
69
70
71 8
        if ($this->sizeOfTranslator < 2) {
72 1
            throw new \RuntimeException('Translator tokens cannot be shorttest 2 tokens.');
73
        }
74
75 7
        return $this->getMessageFromPath($path, $only === null ? $this->only : $only, $except === null ? $this->except : $except);
76
    }
77
78
    /**
79
     * @param string $path
80
     * @param string[] $only
81
     * @param string[] $except
82
     *
83
     * @psalm-return array<array-key|string, mixed|non-empty-list<string>>
84
     */
85 7
    private function getMessageFromPath(string $path, array $only, array $except): array
86
    {
87 7
        $messages = [];
88
89 7
        $files = FileHelper::findFiles($path, [
90 7
            'filter' => (new pathMatcher())->only(...$only)->except(...$except),
91
            'recursive' => true,
92
        ]);
93
94
        /**
95
         * @var string[] $files
96
         */
97 7
        foreach ($files as $file) {
98 7
            $messages = array_merge_recursive($messages, $this->extractMessagesFromFile($file));
99
        }
100
101 7
        return $messages;
102
    }
103
104
    /**
105
     * @param string $fileName
106
     *
107
     * @psalm-return array<array-key|string, mixed|non-empty-list<string>>
108
     */
109 7
    private function extractMessagesFromFile(string $fileName): array
110
    {
111 7
        $fileContent = file_get_contents($fileName);
112 7
        $tokens = token_get_all($fileContent);
113
114 7
        $this->skippedLinesOfFile = [];
115 7
        $messages = $this->extractMessagesFromTokens($tokens);
116
117 7
        if (!empty($this->skippedLinesOfFile)) {
118 4
            $this->skippedLines[$fileName] = $this->skippedLinesOfFile;
119
        }
120
121 7
        return $messages;
122
    }
123
124
    /**
125
     * @psalm-param array<string|array{0: int, 1: string, 2: int}> $tokens
126
     *
127
     * @psalm-return array<array-key|string, mixed|non-empty-list<string>>
128
     */
129 7
    private function extractMessagesFromTokens(array $tokens): array
130
    {
131 7
        $messages = $buffer = [];
132 7
        $matchedTokensCount = $pendingParenthesisCount = 0;
133 7
        $isStartedTranslator = false;
134
135 7
        foreach ($tokens as $tokenIndex => $token) {
136 7
            if (is_array($token) && in_array($token[0], [T_WHITESPACE, T_COMMENT])) {
137 7
                continue;
138
            }
139
140 7
            if ($isStartedTranslator && $pendingParenthesisCount === 0 && $this->tokensEqual($token, ')')) {
141 7
                $messages = array_merge_recursive($messages, $this->extractParametersFromTokens($buffer));
142 7
                $isStartedTranslator = false;
143 7
                $pendingParenthesisCount = 0;
144 7
                $buffer = [];
145 7
            } elseif ($isStartedTranslator) {
146 7
                if ($this->tokensEqual($token, '(')) {
147 7
                    $pendingParenthesisCount++;
148 7
                } elseif ($this->tokensEqual($token, ')')) {
149 7
                    $pendingParenthesisCount--;
150
                }
151 7
                $buffer[] = $token;
152
            } else {
153 7
                if ($matchedTokensCount === $this->sizeOfTranslator) {
154 7
                    if ($this->tokensEqual($token, '(')) {
155 7
                        $isStartedTranslator = true;
156
                    }
157 7
                    $matchedTokensCount = 0;
158
                }
159
160 7
                if ($this->tokensEqual($token, $this->translatorTokens[$matchedTokensCount])) {
161 7
                    $matchedTokensCount++;
162
                } else {
163 7
                    $matchedTokensCount = 0;
164
                }
165
            }
166
        }
167
168 7
        return $messages;
169
    }
170
171
    /**
172
     * @param array $tokens
173
     * @psalm-param array<string|array{0: int, 1: string, 2: int}> $tokens
174
     *
175
     * @return array
176
     * @psalm-return array<array-key|string, mixed|non-empty-list<string>>
177
     */
178 7
    private function extractParametersFromTokens(array $tokens): array
179
    {
180 7
        $parameters = $this->splitTokensAsParams($tokens);
181
182 7
        if (!isset($parameters['id'])) {
183 4
            $this->skippedLinesOfFile[] = $tokens;
184 4
            return [];
185
        }
186
187 5
        $messages = [$parameters['category'] ?? $this->defaultCategory => [$parameters['id']]];
188
189
        // Get translation messages from parameters
190 5
        if (isset($parameters['parameters'])) {
191 4
            $messages = array_merge_recursive($messages, $this->extractMessagesFromTokens($parameters['parameters']));
192
        }
193
194 5
        return $messages;
195
    }
196
197
    /**
198
     * @psalm-param array<string|array{0: int, 1: string, 2: int}> $tokens
199
     *
200
     * @psalm-return array{category?: null|string, id?: null|string, parameters?: null|list<array{0: int, 1: string, 2: int}|string>}
201
     */
202 7
    private function splitTokensAsParams(array $tokens): array
203
    {
204 7
        $parameters = [];
205 7
        $parameterIndex = 0;
206 7
        $commaStack = [];
207
208 7
        foreach ($tokens as $token) {
209 7
            if (empty($commaStack) && $token === ',') {
210 6
                $parameterIndex++;
211 6
                continue;
212
            }
213 7
            if (is_string($token)) {
214 6
                if (in_array($token, self::$commaSpare)) {
215 6
                    $commaStack[] = $token;
216 6
                } elseif (isset(self::$commaSpare[$token])&& array_pop($commaStack) !== self::$commaSpare[$token]) {
217
                    return [];
218
                }
219
            }
220 7
            $parameters[$parameterIndex][] = $token;
221
        }
222
223
        return [
224 7
            'id' => $this->getMessageStringFromTokens($parameters[0] ?? []),
225 7
            'parameters' => $parameters[1] ?? null,
226 7
            'category' => $this->getMessageStringFromTokens($parameters[2] ?? []),
227
        ];
228
    }
229
230
    /**
231
     * @psalm-param array<string|array{0: int, 1: string, 2: int}> $tokens
232
     *
233
     * @return string|null
234
     */
235 7
    private function getMessageStringFromTokens(array $tokens): ?string
236
    {
237 7
        if (empty($tokens) || $tokens[0][0] !== T_CONSTANT_ENCAPSED_STRING) {
238 6
            return null;
239
        }
240
241 7
        $fullMessage = substr($tokens[0][1], 1, -1);
242
243 7
        $i = 1;
244 7
        $countTokens = count($tokens);
245 7
        while ($i < $countTokens && $tokens[$i] === '.') {
246 6
            if ($tokens[$i + 1][0] === T_CONSTANT_ENCAPSED_STRING) {
247 6
                $fullMessage .= substr($tokens[$i + 1][1], 1, -1);
248 4
            } elseif (in_array($tokens[$i + 1][0], [T_LNUMBER, T_DNUMBER])) {
249 4
                $fullMessage .= $tokens[$i + 1][1];
250
            } else {
251 3
                return null;
252
            }
253
254 6
            $i += 2;
255
        }
256
257 5
        return stripcslashes($fullMessage);
258
    }
259
260
    /**
261
     * Finds out if two PHP tokens are equal.
262
     *
263
     * @param array{0: int, 1: string, 2: int}|string $a
264
     * @param array{0: int, 1: string, 2: int}|string $b
265
     *
266
     * @return bool
267
     */
268 7
    private function tokensEqual($a, $b): bool
269
    {
270 7
        if (is_string($a)) {
271 7
            return $a === $b;
272
        }
273
274 7
        return $a[0] === $b[0] && $a[1] == $b[1];
275
    }
276
277 7
    public function hasSkippedLines(): bool
278
    {
279 7
        return !empty($this->skippedLines);
280
    }
281
282 4
    public function getSkippedLines(): array
283
    {
284 4
        return $this->skippedLines;
285
    }
286
287 2
    public function getDefaultCategory(): string
288
    {
289 2
        return $this->defaultCategory;
290
    }
291
292 6
    public function setDefaultCategory(string $defaultCategory): void
293
    {
294 6
        $this->defaultCategory = $defaultCategory;
295 6
    }
296
297 2
    public function withTranslator(string $translator): self
298
    {
299 2
        $new = clone $this;
300 2
        $new->translator = $translator;
301 2
        return $new;
302
    }
303
}
304