add local dir

idleberg · Oct 21, 2018 · 682a943 · 682a943
1 parent a68482c
commit 682a943
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/generators/app/index.js b/generators/app/index.js
@@ -23,11 +23,17 @@ function getModuleDir() {
 
   const yarnDir = join(yarnModules(), 'node_modules', 'highlight.js');
 
-   if (existsSync(yarnDir)) {
+  if (existsSync(yarnDir)) {
     return yarnDir;
   }
 
-  throw 'highlight.js not found in global modules';
+  const localDir = join(__dirname, '..', '..', 'node_modules', 'highlight.js');
+
+  if (existsSync(localDir)) {
+    return yarnDir;
+  }
+
+  throw 'Error: highlight.js not found in node_modules';
 }
 
 function getLanguages() {