Merge pull request #123 from weaverryan/add-code-node-options

Finish add code node options
doctrine · Dec 5, 2020 · 10a5e33 · 10a5e33
2 parents 596ace1 + ab10cf1
commit 10a5e33
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 0 deletions.
diff --git a/lib/Directives/CodeBlock.php b/lib/Directives/CodeBlock.php
@@ -44,6 +44,7 @@ public function process(
 
         if ($node instanceof CodeNode) {
             $node->setLanguage(trim($data));
+            $node->setOptions($options);
         }
 
         if ($variable !== '') {

diff --git a/lib/Nodes/CodeNode.php b/lib/Nodes/CodeNode.php
@@ -15,6 +15,9 @@ class CodeNode extends Node
     /** @var string|null */
     protected $language = null;
 
+    /** @var string[] */
+    private $options;
+
     /**
      * @param string[] $lines
      */
@@ -47,4 +50,20 @@ public function isRaw(): bool
     {
         return $this->raw;
     }
+
+    /**
+     * @param string[] $options
+     */
+    public function setOptions(array $options = []): void
+    {
+        $this->options = $options;
+    }
+
+    /**
+     * @return string[]
+     */
+    public function getOptions(): array
+    {
+        return $this->options;
+    }
 }
diff --git a/tests/Parser/ParserTest.php b/tests/Parser/ParserTest.php
@@ -71,6 +71,24 @@ public function testCodeNode(): void
         self::assertSame("A\nB\n C", trim($nodes[0]->getValueString()));
     }
 
+    /**
+     * Testing that code node options are parsed
+     */
+    public function testCodeNodeWithOptions(): void
+    {
+        $document = $this->parse('code-block-with-options.rst');
+
+        $nodes = $document->getNodes(static function (Node $node): bool {
+            return $node instanceof CodeNode;
+        });
+
+        self::assertSame(1, count($nodes));
+        $codeNode = $nodes[0];
+        assert($codeNode instanceof CodeNode);
+        self::assertSame("A\nB\nC", trim($codeNode->getValueString()));
+        self::assertSame(['name' => 'My Very Best Code'], $codeNode->getOptions());
+    }
+
     /**
      * Testing paragraph nodes
      */

diff --git a/tests/Parser/files/code-block-with-options.rst b/tests/Parser/files/code-block-with-options.rst
@@ -0,0 +1,7 @@
+
+.. code-block::
+    :name: My Very Best Code
+
+    A
+    B
+    C