Provide method for filtering toolbox XML

package.json

@@ -33,6 +33,7 @@
     "gh-pages": "^0.12.0",
     "highlightjs": "^9.8.0",
     "htmlparser2": "3.9.2",
+    "jsdom": "^9.11.0",
     "json": "^9.0.4",
     "lodash.defaultsdeep": "4.6.0",
     "minilog": "3.1.0",

src/playground/playground.js

@@ -59,6 +59,11 @@ window.onload = function () {
     });
     window.workspace = workspace;
 
+    // Filter available blocks
+    var toolbox = vm.filterToolbox(workspace.options.languageTree);
+    // var toolbox = workspace.options.languageTree;
+    workspace.updateToolbox(toolbox);
+
     // Attach scratch-blocks events to VM.
     workspace.addChangeListener(vm.blockListener);
     var flyoutWorkspace = workspace.getFlyout().getWorkspace();

src/util/filter-toolbox.js

@@ -0,0 +1,49 @@
+/**
+ * Filter Blockly toolbox XML node containing blocks to only those with
+ * valid opcodes. Return a copy of the node with valid blocks.
+ * @param {HTMLElement} node Blockly toolbox XML node
+ * @param {Array.<string>} opcodes Valid opcodes. Blocks producing other opcodes
+ * will be filtered.
+ * @returns {HTMLElement} filtered toolbox XML node
+ */
+var filterToolboxNode = function (node, opcodes) {
+    var filteredCategory = node.cloneNode();
+    for (var block = node.firstElementChild; block; block = block.nextElementSibling) {
+        if (block.nodeName.toLowerCase() !== 'block') continue;
+        var opcode = block.getAttribute('type').toLowerCase();
+        if (opcodes.indexOf(opcode) !== -1) {
+            filteredCategory.appendChild(block.cloneNode(true));
+        }
+    }
+    return filteredCategory;
+};
+
+/**
+ * Filter Blockly toolbox XML and return a copy which only contains blocks with
+ * existent opcodes. Categories with no valid children will be removed.
+ * @param {HTMLElement} toolbox Blockly toolbox XML node
+ * @param {Array.<string>} opcodes Valid opcodes. Blocks producing other opcodes
+ * will be filtered.
+ * @returns {HTMLElement} filtered toolbox XML node
+ */
+var filterToolbox = function (toolbox, opcodes) {
+    if (!toolbox.hasChildNodes()) return toolbox;
+    var filteredToolbox;
+    if (toolbox.firstElementChild.nodeName.toLowerCase() === 'category') {
+        filteredToolbox = toolbox.cloneNode();
+        for (
+            var category = toolbox.firstElementChild;
+            category;
+            category = category.nextElementSibling
+        ) {
+            if (category.nodeName.toLowerCase() !== 'category') continue;
+            var filteredCategory = filterToolboxNode(category, opcodes);
+            if (filteredCategory.hasChildNodes()) filteredToolbox.appendChild(filteredCategory);
+        }
+    } else {
+        filteredToolbox = filterToolboxNode(toolbox, opcodes);
+    }
+    return filteredToolbox;
+};
+
+module.exports = filterToolbox;

src/virtual-machine.js

@@ -1,6 +1,7 @@
 var EventEmitter = require('events');
 var util = require('util');
 
+var filterToolbox = require('./util/filter-toolbox');
 var Runtime = require('./engine/runtime');
 var sb2import = require('./import/sb2import');
 
@@ -336,4 +337,17 @@ VirtualMachine.prototype.postSpriteInfo = function (data) {
     this.editingTarget.postSpriteInfo(data);
 };
 
+
+/**
+ * Filter Blockly toolbox XML and return a copy which only contains blocks with
+ * existent opcodes. Categories with no valid children will be removed.
+ * @param {HTMLElement} toolbox Blockly toolbox XML node
+ * @returns {HTMLElement} filtered toolbox XML node
+ */
+VirtualMachine.prototype.filterToolbox = function (toolbox) {
+    var opcodes = Object.keys(this.runtime._primitives)
+        .concat(Object.keys(this.runtime._hats));
+    return filterToolbox(toolbox, opcodes);
+};
+
 module.exports = VirtualMachine;

test/fixtures/.eslintrc.js

@@ -0,0 +1,5 @@
+module.exports = {
+    rules: {
+        'max-len': [0]
+    }
+};