nim-lang - 如何模块化/抽象掉 os.walkDir 和 os.walkDirRec - Nim

Question

我想知道是否有可能使用函数指针或宏等，使下面的代码更加模块化，例如，不重复整个 for 循环，唯一的区别是 walkDirRec与 walkDir 相比。

我尝试使用函数指针和宏，但都无法正常工作。 (-> 尼姆菜鸟)

import os

proc containsFilesWithSuffix(dir: string, suffix: string, recursive: bool = true) : bool =
  if recursive:
    for file in os.walkDirRec(dir):
      if file.toLower().endsWith(suffix):
        return true
  else:
    for file in os.walkDir(dir):
      if file.toLower().endsWith(suffix):
        return true
  return false

Answer 1

import std/[os, strutils]

template anyOfIt*(sequence, predicate: untyped): bool =
  var result = false
  for it {.inject.} in sequence:
    if predicate:
      result = true
      break

  result


proc containsFilesWithSuffix(dir: string, suffix: string, recursive: bool = true) : bool =
  if recursive: anyOfIt(walkDirRec(dir), it.toLower.endsWith(suffix))
  else:         anyOfIt(walkDir(dir),    it.path.toLower.endsWith(suffix))

类似的东西可以正常工作，但在这种情况下，我还是会编写一个 for 循环，因为在我需要时添加更多条件/逻辑更容易。