From ab71cba94902a6e75f010571dd90f2a68f734a98 Mon Sep 17 00:00:00 2001
From: Alex Gleason <alex@alexgleason.me>
Date: Thu, 17 Sep 2015 09:08:37 -0500
Subject: [PATCH] Refactored StreamBlockMenu js

---
 .../static_src/wagtailadmin/js/blocks/stream.js            | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/wagtail/wagtailadmin/static_src/wagtailadmin/js/blocks/stream.js b/wagtail/wagtailadmin/static_src/wagtailadmin/js/blocks/stream.js
index 110878921..97a1ba5ca 100644
--- a/wagtail/wagtailadmin/static_src/wagtailadmin/js/blocks/stream.js
+++ b/wagtail/wagtailadmin/static_src/wagtailadmin/js/blocks/stream.js
@@ -30,10 +30,15 @@
             self.container.addClass('stream-menu-closed');
         };
 
+        self.addFirstBlock = function() {
+            if (opts.onChooseBlock) opts.onChooseBlock(opts.childBlocks[0]);
+        };
+
         self.toggle = function() {
             if (self.container.hasClass('stream-menu-closed')) {
                 if (opts.childBlocks.length == 1) {
-                    if (opts.onChooseBlock) opts.onChooseBlock(opts.childBlocks[0]);
+                    /* If there's only one block type, add it automatically */
+                    self.addFirstBlock();
                 } else {
                     self.show();
                 }