Automatically recurse folders (#11)

* Automatically recurse folders

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "rapidchecker"
-version = "0.1.3"
+version = "0.1.4"
 description = " Grammar and format checker for ABB RAPID code. "
 readme = "README.md"
 requires-python = ">=3.10"

rapidchecker/cli.py

@@ -1,5 +1,4 @@
 import sys
-from pathlib import Path
 
 import click
 from pyparsing import ParseBaseException
@@ -8,14 +7,10 @@
 from rapidchecker.whitespace_checks import WhiteSpaceError
 
 from .check import check_format
+from .io import get_sys_files, read_sys_file
 from .whitespace_checks import check_whitespace
 
 
-def read_sys_file(path: str | Path) -> str:
-    with Path(path).open() as f:
-        return f.read()
-
-
 def check_file(file_contents: str) -> list[ParseBaseException | WhiteSpaceError]:
     errors: list[ParseBaseException | WhiteSpaceError] = []
     errors.extend(check_format(file_contents))
@@ -25,11 +20,11 @@ def check_file(file_contents: str) -> list[ParseBaseException | WhiteSpaceError]
 
 
 @click.command()
-@click.argument("paths", nargs=-1, type=click.Path(exists=True, dir_okay=False))
+@click.argument("paths", nargs=-1, type=click.Path(exists=True, dir_okay=True))
 def cli(paths: list[str]) -> None:
     found_errors = False
 
-    for filepath in paths:
+    for filepath in get_sys_files(paths):
         errors = check_file(read_sys_file(filepath))
         if not errors:
             continue

rapidchecker/io.py

@@ -0,0 +1,16 @@
+from collections.abc import Iterable
+from pathlib import Path
+
+
+def get_sys_files(paths: list[str]) -> Iterable[Path]:
+    for path_str in paths:
+        path = Path(path_str)
+        if path.is_dir():
+            yield from path.glob("**/*.sys")
+        else:
+            yield path
+
+
+def read_sys_file(path: str | Path) -> str:
+    with Path(path).open() as f:
+        return f.read()