From 5e01288754427831c121e9712e4293ea666a5dbc Mon Sep 17 00:00:00 2001
From: Michael Overmeyer <m.overmeyer@yahoo.ca>
Date: Mon, 4 Jun 2018 18:26:36 +0000
Subject: [PATCH] Regex matches only comments that start with TOC

Fixes #50
---
 src/extension.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/extension.ts b/src/extension.ts
index 03a12bc..676f178 100644
--- a/src/extension.ts
+++ b/src/extension.ts
@@ -15,8 +15,8 @@ import {
     TextEditorEdit
 } from 'vscode';
 
-const REGEXP_TOC_START          = /\s*<!--(.*)TOC(.*)-->/gi;
-const REGEXP_TOC_STOP           = /\s*<!--(.*)\/TOC(.*)-->/gi;
+const REGEXP_TOC_START          = /\s*<!--\s*TOC(\s(\.*))?-->/gi;
+const REGEXP_TOC_STOP           = /\s*<!--\s\/TOC(\s(\.*))?-->/gi;
 const REGEXP_TOC_CONFIG         = /\w+[:=][\w.]+/gi;
 const REGEXP_TOC_CONFIG_ITEM    = /(\w+)[:=]([\w.]+)/;
 const REGEXP_MARKDOWN_ANCHOR    = /^<a id="markdown-.+" name=".+"><\/a\>/;