Added caption for Embed block

Author: Edward

src/Parser.php

@@ -240,9 +240,8 @@ private function parseLink($block)
 
     private function parseEmbed($block)
     {
-        $wrapper = $this->dom->createElement('div');
-
-        $wrapper->setAttribute('class', "{$this->prefix}-embed");
+        $figure = $this->dom->createElement('figure');
+        $figure->setAttribute('class', $block->type);
 
         switch ($block->data->service) {
             case 'youtube':
@@ -254,22 +253,25 @@ private function parseEmbed($block)
                     'allowfullscreen' => true
                 ];
 
-                $wrapper->appendChild($this->createIframe($attrs));
-
                 break;
-            case 'codepen' || 'gfycat':
+            // case 'codepen' || 'gfycat':
+            default:
 
                 $attrs = [
                     'height' => $block->data->height,
                     'src' => $block->data->embed,
                 ];
+        }
 
-                $wrapper->appendChild($this->createIframe($attrs));
+        $figure->appendChild($this->createIframe($attrs));
 
-                break;
+        if ($block->data->caption) {
+            $figCaption = $this->dom->createElement('figcaption');
+            $figCaption->appendChild($this->html5->loadHTMLFragment($block->data->caption));
+            $figure->appendChild($figCaption);
         }
 
-        $this->dom->appendChild($wrapper);
+        $this->dom->appendChild($figure);
     }
 
     private function createIframe(array $attrs)