feat: add extraInfo parameter to codeGenerator

fjc0k · Mar 18, 2019 · f698baf · f698baf
1 parent 049b970
commit f698baf
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -81,6 +81,7 @@ function index(
   type CodeGenerator = (
     parsedPath: ParsedPath,
     changeCase: ChangeCase,
+    extraInfo: ExtraInfo,
   ) => string
 
   interface ParsedPath {
@@ -95,6 +96,17 @@ function index(
   interface ChangeCase {
     // See https://github.com/blakeembrey/change-case#usage
   }
+
+  interface ExtraInfo {
+    /** total number of items */
+    total: number,
+    /** index of current item */
+    index: number,
+    /** if current item is the first */
+    first: boolean,
+    /** if current item is the last */
+    last: boolean,
+  }
   ```
 
   See [all changeCase methods](https://github.com/blakeembrey/change-case#usage).

diff --git a/src/Generator.ts b/src/Generator.ts
@@ -20,6 +20,12 @@ type Pattern = string
 type CodeGenerator = (
   parsedPath: ParsedPath,
   changeCase: ChangeCase,
+  extraInfo: {
+    total: number,
+    index: number,
+    first: boolean,
+    last: boolean,
+  },
 ) => string
 
 interface Marker {
@@ -69,14 +75,23 @@ export default class Generator {
       )
       const codes = paths
         .filter(path => path !== currentFile)
-        .map(path => {
+        .map((path, index, paths) => {
           const pp = p.parse(path)
           const parsedPath: ParsedPath = {
             path: getRelativePath(currentDir, p.join(pp.dir, pp.name)),
             name: pp.name,
             ext: pp.ext,
           }
-          const code = marker.codeGenerator(parsedPath, changeCase)
+          const code = marker.codeGenerator(
+            parsedPath,
+            changeCase,
+            {
+              total: paths.length,
+              index: index,
+              first: index === 0,
+              last: index === paths.length - 1,
+            },
+          )
           return marker.indent + code
         })
       edit.replace(