Merge pull request #85 from GitbookIO/pr/63

Pr/63

1 files changed, 49 insertions, 5 deletions

diff --git a/lib/parse/lex.js b/lib/parse/lex.js
index cec6047..813dd33 100644
--- a/lib/parse/lex.js
+++ b/lib/parse/lex.js
@@ -17,21 +17,65 @@ function splitSections(nodes) {
     }, []).concat([section]); // Add remaining nodes
 }
 
-// What is the type of this section
-function sectionType(nodes, idx) {
+function isQuizNode(node) {
+    return (/^[(\[][ x][)\]]/).test(node.text || node);
+}
+
+function isExercise(nodes) {
     var codeType = { type: 'code' };
 
     // Number of code nodes in section
     var len = _.filter(nodes, codeType).length;
 
-    if(
+    return (
         // Got 3 or 4 code blocks
         (len === 3 || len === 4) &&
         // Ensure all nodes are at the end
         _.all(_.last(nodes, len), codeType)
-    )
-    {
+    );
+}
+
+function isQuiz(nodes) {
+    if (nodes.length < 3) {
+        return false;
+    }
+
+    // Support having a first paragraph block
+    // before our series of questions
+    var quizNodes = nodes.slice(nodes[0].type === 'paragraph' ? 1 : 0);
+
+    // No questions
+    if (!_.some(quizNodes, { type: 'blockquote_start' })) {
+        return false;
+    }
+
+    // Check if section has list of questions
+    // or table of questions
+    var listIdx = _.findIndex(quizNodes, { type: 'list_item_start' });
+    var tableIdx = _.findIndex(quizNodes, { type: 'table' });
+
+    if(
+        // List of questions
+        listIdx !== -1 && isQuizNode(quizNodes[listIdx + 1]) &&
+
+        // Table of questions
+        (
+            tableIdx !== -1 &&
+            _.every(quizNodes[tableIdx].cells[0].slice(1), isQuizNode)
+        )
+    ) {
+        return true;
+    }
+
+    return false;
+}
+
+// What is the type of this section
+function sectionType(nodes, idx) {
+    if(isExercise(nodes)) {
         return 'exercise';
+    } else if(isQuiz(nodes)) {
+        return 'quiz';
     }
 
     return 'normal';