Completed
Pull Request — master (#1)
by Adrien
03:52
created

Rst::__construct()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 4
Code Lines 2

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 1
eloc 2
nc 1
nop 0
dl 0
loc 4
rs 10
c 0
b 0
f 0
1
<?php
2
3
declare(strict_types=1);
4
5
namespace Rusty\Extractor;
6
7
use Gregwar\RST\ErrorManager;
8
use Gregwar\RST\HTML\Nodes\CodeNode;
9
use Gregwar\RST\Parser;
10
use League\CommonMark\Block\Element;
11
use Rusty\CodeSample;
12
use Rusty\PragmaParser;
13
14
class Rst implements SampleExtractor
15
{
16
    protected $pragmaParser;
17
18
    public function __construct()
19
    {
20
        $this->pragmaParser = new PragmaParser();
21
    }
22
23
    public static function supportedExtensions(): array
24
    {
25
        return ['rst'];
26
    }
27
28
    public function extractSamples(\SplFileInfo $file): \Traversable
29
    {
30
        $parser = new Parser();
31
32
        $parser->getEnvironment()->setErrorManager(new class() extends ErrorManager {
33
            public function error($message)
34
            {
35
36
            }
37
        });
38
39
        $fileContent = file_get_contents($file->getRealPath());
40
41
        /** @var \Gregwar\RST\HTML\Document $document */
42
        $document = @$parser->parse($fileContent);
43
        $nodes = $document->getNodes(function($node) {
44
            return $node instanceof CodeNode;
45
        });
46
47
        foreach ($nodes as $node) {
48
            if ($node->getLanguage() != 'php') {
49
                continue;
50
            }
51
            $content = $node->getValue();
52
            $pragmaDirectives = $this->pragmaParser->getPragmaDirectives($content);
53
            yield new CodeSample($file, 0, $content, $pragmaDirectives);
54
        }
55
    }
56
}
57