From 371c415085181d4e11ceacab8aca1ba072edb98a Mon Sep 17 00:00:00 2001
From: Geraldo Ribeiro <geraldo@stefaninirafael.com>
Date: Thu, 3 Feb 2022 20:29:45 -0300
Subject: [PATCH] Wraps the iframe in a div

---
 mkdocs_video/plugin.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/mkdocs_video/plugin.py b/mkdocs_video/plugin.py
index 7116e7e..bc7134b 100644
--- a/mkdocs_video/plugin.py
+++ b/mkdocs_video/plugin.py
@@ -59,12 +59,14 @@ def create_repl_tag(self, src):
             ["{}: {}".format(str(atr), str(style[atr])) for atr in style]
         )
 
-        return "<iframe "\
+        return "<div class=\"video-container\">"\
+            "<iframe "\
             "src=\"{}\" "\
             "style=\"{}\" "\
             "frameborder=\"0\" "\
             "allowfullscreen>"\
-            "</iframe>".format(src, style)
+            "</iframe>"\
+            "</div>".format(src, style)
 
 
     def find_marked_tags(self, content):