diff --git a/js/youtube.js b/js/youtube.js
index 9fe81b60..2eca99ae 100644
--- a/js/youtube.js
+++ b/js/youtube.js
@@ -24,22 +24,34 @@
onready(function(){
- var do_embed_yt = function(tag) {
- $('div.video-container a', tag).click(function() {
- var videoID = $(this.parentNode).data('video');
-
- $(this.parentNode).html('');
+ const ON = "[Remove]";
+ const OFF = "[Embed]";
- return false;
+ function addEmbedButton(index, videoNode) {
+ videoNode = $(videoNode);
+ var videoId = videoNode.data('video');
+ var span = $("[Embed]");
+ var embedNode = $('');
+ span.click(function() {
+ if (span.text() == ON){
+ embedNode.remove();
+ span.text(OFF);
+ } else{
+ videoNode.append(embedNode);
+ span.text(ON);
+ }
});
- };
- do_embed_yt(document);
- // allow to work with auto-reload.js, etc.
- $(document).on('new_post', function(e, post) {
- do_embed_yt(post);
- });
-});
+ videoNode.append(span);
+ }
+ $('div.video-container', document).each(addEmbedButton);
+
+
+ // allow to work with auto-reload.js, etc.
+ $(document).on('new_post', function(e, post) {
+ $('div.video-container', post).each(addEmbedButton);
+ });
+});