Strict check for open/close

Author: leaysgur

lib/tokenizer.js

@@ -34,25 +34,33 @@ function tokenize(str) {
   // 空文字は無視します
   if (str.length === 0) { return tokens; }
 
-  var isCGMD = false;
+  var isCGMD = null; // or note, column, imgbox, ...
   // 全ての行をなめる
   while ((line = lines.shift()) !== undefined) {
     // CGMDのスタート / エンドでフラグを打つ
     if (CGMD_START_RE.exec(line)) {
-      isCGMD = true;
+      // ネストしようとした
+      if (isCGMD !== null)
+        throw new Error(`[${isCGMD}] is still opened, nesting is not allowed!`);
+
+      isCGMD = line.slice(1, -1);
       tokens.push(new CGMD_Token());
       tokens[tokens.length - 1].addBody(line);
       continue;
     }
     if (CGMD_END_RE.exec(line)) {
-      isCGMD = false;
+      // 異なる種類で閉じようとした
+      if (isCGMD !== line.slice(2, -1))
+        throw new Error(`[${isCGMD}] should not be closed by ${line}!`);
+
+      isCGMD = null;
       tokens[tokens.length - 1].addBody(line);
       continue;
     }
 
     if (
       // ふつうのMDでありつつ
-      isCGMD === false &&
+      isCGMD === null &&
       (
         // 最初か
         tokens.length === 0 ||
@@ -69,8 +77,8 @@ function tokenize(str) {
   }
 
   // 全行見終わってまだこのフラグが立ってるなら閉じ忘れ
-  if (isCGMD) {
-    throw new Error('Missing close tag for CGMD!');
+  if (isCGMD !== null) {
+    throw new Error(`Missing close tag for [${isCGMD}]!`);
   }
 
   return tokens;

test/cgmd/renderer/md/code.js

@@ -22,6 +22,19 @@ describe('#code', function() {
     var expect = '<pre><code class="lang-html">hoge</code></pre>\n';
     assert.equal(html, expect);
   });
+  it('言語とタイトル指定が正しい場合は拡張したやつ', function() {
+    var html1 = renderer.render('```#title\nhoge\n```');
+    var expect = '' +
+      '<section class="CG2-livecode">\n' +
+        '<header class="CG2-livecode__header">\n' +
+          '<div class="CG2-livecode__label">title</div>\n' +
+        '</header>\n' +
+        '<div class="CG2-livecode__body">' +
+          '<pre><code>hoge</code></pre>\n' +
+        '</div>\n' +
+      '</section>\n';
+    assert.equal(html1, expect);
+  });
   it('言語とタイトル指定が正しい場合は拡張したやつ', function() {
     var html1 = renderer.render('```html#title\nhoge\n```');
     // スペースがあっても気にしない

test/cgmd/tokenizer.js

@@ -43,6 +43,18 @@ describe('#tokenize', function() {
     });
   });
 
+  it('cgmdネストもエラー', function() {
+    assert.throws(function() {
+      Tokenizer.tokenize('[demo]\n# Nesting not allowed!\n[imgbox]');
+    });
+  });
+
+  it('cgmd違うので閉じるとエラー', function() {
+    assert.throws(function() {
+      Tokenizer.tokenize('[imgbox]\n# Wrong closing!\n[/column]');
+    });
+  });
+
   it('cgmdの規定のタイプ以外は無視', function() {
     var tokens = Tokenizer.tokenize([
       '```toml',