Merge pull request #9 from Yamato-Security/8-check-yml-files-list-in-…

…target-folder

Check yml files list in target folder

src/takajo.nim

@@ -20,5 +20,12 @@ Options:
 
 when isMainModule:
   let args = docopt(doc)
-  if args["c"] == true and not args["<CSV-FILE>"]:
+  if args["<CSV-FILE>"]:
     let csvData = getHayabusaCsvData($args["<CSV-FILE>"])
+    if args["<hayabusa-rulespath>"]:
+      let rulePath: string = args["<hayabusa-rulespath>"]
+      for f in walkDirRec(rulePath, "*.yml"):
+
+    if args["c"] == true and args["<column>"]:
+      let targetColumn = args["<column>"]
+      csvData[targetColumn]

src/takajopkg/submodule.nim

@@ -5,8 +5,17 @@
 
 import std/tables
 import std/parsecsv
+import std/os
+import std/strutils
 from std/streams import newFileStream
 
+proc getYMLLists*(targetDirPath: string): seq[string] =
+  var r: seq[string] = @[]
+  for f in walkDirRec(targetDirPath):
+    if f.endsWith(".yml"):
+      r.insert(f)
+  return r
+
 proc getHayabusaCsvData*(csvPath: string): Tableref[string, seq[string]] =
   ## procedure for Hayabusa output csv read data.
 

tests/testsubmodule.nim

@@ -34,3 +34,6 @@ test "csv file path import":
   writeFile("temp.csv", expect_content)
   check getHayabusaCsvData("./tests/data/1.csv") == expect_table
 
+test "check getYMLLists":
+  let expect = @["tests\\data\\1.yml"]
+  check getYMLLists("./tests") == expect