dedent and fix underslug

Co-authored-by: ee7 <[email protected]>

Author: ynfle

src/representer.nim

@@ -4,31 +4,31 @@ import representer/[mapping, types]
 import docopt
 
 const doc = """
-Exercism nim representation normalizer.
+  Exercism nim representation normalizer.
 
-Usage:
-  representer --slug=<slug> --input-dir=<in-dir> [--output-dir=<out-dir>] [--print]
+  Usage:
+    representer --slug=<slug> --input-dir=<in-dir> [--output-dir=<out-dir>] [--print]
 
-Options:
-  -h --help                             Show this help message.
-  -v, --version                         Display version.
-  -p, --print                           Print the results.
-  -s <slug>, --slug=<slug>              The exercise slug.
-  -i <in-dir>, --input-dir=<in-dir>     The directory of the submission and exercise files.
-  -o <out-dir>, --output-dir=<out-dir>  The directory to output to.
-                                        If omitted, output will be written to stdout.
-"""
+  Options:
+    -h --help                             Show this help message.
+    -v, --version                         Display version.
+    -p, --print                           Print the results.
+    -s <slug>, --slug=<slug>              The exercise slug.
+    -i <in-dir>, --input-dir=<in-dir>     The directory of the submission and exercise files.
+    -o <out-dir>, --output-dir=<out-dir>  The directory to output to.
+                                          If omitted, output will be written to stdout.
+  """.dedent
 
 proc getFileContents(fileName: string): string = readFile fileName
 
-func underSlug(s: string): string = s.replace('-', '_')
+func kebabToSnakeCase(s: string): string = s.replace('-', '_')
 
 proc main() =
   let args = docopt(doc)
   let intr = loadScript(NimScriptPath("src/representer/loader.nims"))
   let (tree, map) = intr.invoke(
     getTestableRepresentation,
-    getFileContents($args["--input-dir"] / underslug($args["--slug"]) & ".nim"), true,
+    getFileContents($args["--input-dir"] / kebabToSnakeCase($args["--slug"]) & ".nim"), true,
     returnType = SerializedRepresentation
   )
   if args["--output-dir"]:

src/representer/microsextensions.nim

@@ -0,0 +1,87 @@
+import micros
+# CaseStmt ConstDef ElIfBranch ElseBranch EnumDef EnumField ForLoop IdentDef IfStmt LetDef NimName  ObjectDef OfBranch PragmaVal RoutineNode RoutineSym StmtList TypeDefs = (EnumDef or ObjectDef) VarDef VarDefs = (VarDef or LetDef or ConstDef) WhenStmt WhileLoop
+
+type
+  Comment*      = distinct NimNode
+  Call*         = distinct NimNode
+  Infix*        = distinct NimNode
+  Command*      = distinct NimNode
+  Prefix*       = distinct NimNode
+  Calls*        = Call | Infix | Command | Prefix
+  ImportStmt*   = distinct NimNode
+  ImportFrom*   = distinct NimNode
+  ImportExcept* = distinct NimNode
+  Imports*      = ImportStmt | ImportFrom | ImportExcept
+  ExportStmt*   = distinct NimNode
+  DotExpr*      = distinct NimNode
+
+func isa*(n: NimNode, _: typedesc[Comment]): bool =
+  n.checkIt {nnkCommentStmt}
+
+func comment*(n: NimNode): Comment = n.checkConv Comment
+
+func isa*(n: NimNode, _: typedesc[Call]): bool =
+  n.checkIt {nnkCall}
+
+func call*(n: NimNode): Call = n.checkConv Call
+
+func isa*(n: NimNode, _: typedesc[Infix]): bool =
+  n.checkIt {nnkInfix}
+
+func infix*(n: NimNode): Infix = n.checkConv Infix
+
+func isa*(n: NimNode, _: typedesc[Command]): bool =
+  n.checkIt {nnkCommand}
+
+func command*(n: NimNode): Command = n.checkConv Command
+
+func isa*(n: NimNode, _: typedesc[Prefix]): bool =
+  n.checkIt {nnkPrefix}
+
+func prefix*(n: NimNode): Prefix = n.checkConv Prefix
+
+func isa*(n: NimNode, T: typedesc[Calls]): bool =
+  n.checkIt:
+    when T is Call:
+      {nnkCall}
+    elif T is Infix:
+      {nnkInfix}
+    elif T is Command:
+      {nnkCommand}
+    elif T is Prefix:
+      {nnkPrefix}
+
+func isa*(n: NimNode, _: typedesc[ImportStmt]): bool =
+  n.checkIt {nnkImportStmt}
+
+func importStmt*(n: NimNode): ImportStmt = n.checkConv ImportStmt
+
+func isa*(n: NimNode, _: typedesc[ImportFrom]): bool =
+  n.checkIt {nnkFromStmt}
+
+func importFrom*(n: NimNode): ImportFrom = n.checkConv ImportFrom
+
+func isa*(n: NimNode, _: typedesc[ImportExcept]): bool =
+  n.checkIt {nnkImportExceptStmt}
+
+func importExcept*(n: NimNode): ImportExcept = n.checkConv ImportExcept
+
+func isa*(n: NimNode, _: typedesc[Imports]): bool =
+  n.checkIt:
+    when T is ImportStmt:
+      {nnkImportStmt}
+    elif T is From:
+      {nnkFromStmt}
+    elif T is ImportExcept:
+      {nnkImportExceptStmt}
+
+func isa*(n: NimNode, _: typedesc[ExportStmt]): bool = n.checkIt {nnkExportStmt}
+
+func exportStmt*(n: NimNode): ExportStmt = n.checkConv ExportStmt
+
+func isa*(n: NimNode, _: typedesc[DotExpr]): bool = n.checkIt {nnkDotExpr}
+
+func dotExpr*(n: NimNode): DotExpr = n.checkConv DotExpr
+
+func quoted*(n: NimName): NimName =
+  NimName nnkAccQuoted.newTree(n.NimNode)