From 5fea642991cdc5770062be60fa086cf9c1b9560f Mon Sep 17 00:00:00 2001
From: laggingreflex <laggingreflex@gmail.com>
Date: Wed, 30 Nov 2016 00:13:39 +0530
Subject: [PATCH] fix: update stopped working with banner

With the addition of `--banner` option index files no longer start with '// @create-index'

This new regex looks for '// @create-index' that are either at the beginning of the file (old behavior) or if it is followed by a newline.
Basically it just checks whether it exists in a file or not.

Added a test for it now.
---
 src/utilities/validateTargetDirectory.js                     | 3 +--
 .../safe-index-with-banner/index.js                          | 3 +++
 test/validateTargetDirectory.js                              | 5 +++++
 3 files changed, 9 insertions(+), 2 deletions(-)
 create mode 100644 test/fixtures/validate-target-directory/safe-index-with-banner/index.js

diff --git a/src/utilities/validateTargetDirectory.js b/src/utilities/validateTargetDirectory.js
index 4f917fa..48a4c6e 100644
--- a/src/utilities/validateTargetDirectory.js
+++ b/src/utilities/validateTargetDirectory.js
@@ -1,6 +1,5 @@
 import fs from 'fs';
 import path from 'path';
-import _ from 'lodash';
 
 export default (targetDirectory) => {
   let stats;
@@ -25,7 +24,7 @@ export default (targetDirectory) => {
 
   const indexFile = fs.readFileSync(indexFilePath, 'utf8');
 
-  if (!_.startsWith(indexFile, '// @create-index\n')) {
+  if (!indexFile.match(/(?:^|[\n\r]+)\/\/ @create-index[\n\r]+/)) {
     throw new Error('"' + indexFilePath + '" unsafe index.');
   }
 
diff --git a/test/fixtures/validate-target-directory/safe-index-with-banner/index.js b/test/fixtures/validate-target-directory/safe-index-with-banner/index.js
new file mode 100644
index 0000000..69a1f5b
--- /dev/null
+++ b/test/fixtures/validate-target-directory/safe-index-with-banner/index.js
@@ -0,0 +1,3 @@
+/* eslint-disable */
+
+// @create-index
diff --git a/test/validateTargetDirectory.js b/test/validateTargetDirectory.js
index 9a27c8e..63429da 100644
--- a/test/validateTargetDirectory.js
+++ b/test/validateTargetDirectory.js
@@ -36,6 +36,11 @@ describe('validateTargetDirectory()', () => {
         expect(validateTargetDirectory(path.resolve(fixturesPath, 'safe-index'))).to.equal(true);
       });
     });
+    context('safe with banner', () => {
+      it('returns true', () => {
+        expect(validateTargetDirectory(path.resolve(fixturesPath, 'safe-index-with-banner'))).to.equal(true);
+      });
+    });
     context('unsafe', () => {
       it('throws an error', () => {
         expect(() => {