To adjust other extension, Changd proc name #4

Yamato-Security · Sep 30, 2022 · be50402 · be50402
1 parent f8e14bf
commit be50402
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/takajo.nim b/src/takajo.nim
@@ -26,7 +26,7 @@ when isMainModule:
     var ymlLists: seq[string]
     if args["--check-undetected"]:
       let rulePath: string = $args["--check-undetected"]
-      ymlLists = getYMLLists(rulePath)
+      ymlLists = getTargetExtFileLLists(rulePath, "*.yml")
     if args["--target-column"]:
       let targetColumn = $args["--target-column"]
       var detectedRulePath: seq[string] = csvData[targetColumn]

diff --git a/src/takajopkg/submodule.nim b/src/takajopkg/submodule.nim
@@ -10,11 +10,11 @@ import std/strutils
 import std/sequtils
 from std/streams import newFileStream
 
-proc getYMLLists*(targetDirPath: string): seq[string] =
+proc getTargetExtFileLLists*(targetDirPath: string, targetExt: string): seq[string] =
   ## extract yml file name seq to specified directory path
   var r: seq[string] = @[]
   for f in walkDirRec(targetDirPath):
-    if f.endsWith(".yml"):
+    if f.endsWith(targetExt):
       var (_, file, ext) = splitFile(f)
       file &= ext
       r.insert(file)