omdev 0.0.0.dev140__tar.gz → 0.0.0.dev141__tar.gz

{omdev-0.0.0.dev140/omdev.egg-info → omdev-0.0.0.dev141}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omdev
-Version: 0.0.0.dev140
+Version: 0.0.0.dev141
 Summary: omdev
 Author: wrmsr
 License: BSD-3-Clause
@@ -12,7 +12,7 @@ Classifier: Operating System :: OS Independent
 Classifier: Operating System :: POSIX
 Requires-Python: >=3.12
 License-File: LICENSE
-Requires-Dist: omlish==0.0.0.dev140
+Requires-Dist: omlish==0.0.0.dev141
 Provides-Extra: all
 Requires-Dist: black~=24.10; extra == "all"
 Requires-Dist: pycparser~=2.22; extra == "all"

{omdev-0.0.0.dev140 → omdev-0.0.0.dev141}/omdev/.manifests.json

@@ -230,6 +230,18 @@
       }
     }
   },
+  {
+    "module": ".tools.cloc",
+    "attr": "_CLI_MODULE",
+    "file": "omdev/tools/cloc.py",
+    "line": 144,
+    "value": {
+      "$.cli.types.CliModule": {
+        "cmd_name": "cloc",
+        "mod_name": "omdev.tools.cloc"
+      }
+    }
+  },
   {
     "module": ".tools.doc",
     "attr": "_CLI_MODULE",

omdev-0.0.0.dev141/omdev/tools/cloc.py

@@ -0,0 +1,149 @@
+import dataclasses as dc
+import os
+import re
+import typing as ta
+
+from ..cli import CliModule
+
+
+SUPPORTED_EXTENSIONS: ta.Mapping[str, str] = {
+    '.c': 'c',
+    '.cpp': 'c',
+    '.h': 'c',
+    '.hpp': 'c',
+    '.py': 'python',
+    '.js': 'javascript',
+}
+
+
+COMMENT_PATTERNS: ta.Mapping[str, tuple[str, str | None, str | None]] = {
+    'c': (r'//', r'/\*', r'\*/'),
+    'python': (r'#', None, None),
+    'javascript': (r'//', r'/\*', r'\*/'),
+}
+
+
+@dc.dataclass(frozen=True)
+class FileLineCount:
+    loc: int
+    blanks: int
+    comments: int
+
+
+def count_lines(file_path: str, language: str) -> FileLineCount:
+    single_line_comment, block_comment_start, block_comment_end = COMMENT_PATTERNS[language]
+
+    in_block_comment = False
+
+    loc = 0
+    blank_lines = 0
+    comment_lines = 0
+
+    with open(file_path, encoding='utf-8') as file:
+        for line in file:
+            stripped = line.strip()
+            if not stripped:
+                blank_lines += 1
+                continue
+
+            if in_block_comment:
+                comment_lines += 1
+                if block_comment_end and re.search(block_comment_end, stripped):
+                    in_block_comment = False
+                continue
+
+            if block_comment_start and re.search(block_comment_start, stripped):
+                comment_lines += 1
+                if block_comment_end and not re.search(block_comment_end, stripped):
+                    in_block_comment = True
+                continue
+
+            if single_line_comment and stripped.startswith(single_line_comment):
+                comment_lines += 1
+                continue
+
+            loc += 1
+
+    return FileLineCount(
+        loc=loc,
+        blanks=blank_lines,
+        comments=comment_lines,
+    )
+
+
+def count_lines_in_directory(directory: str) -> ta.Mapping[str, FileLineCount]:
+    results: dict[str, FileLineCount] = {}
+    for root, _, files in os.walk(directory):
+        for file in files:
+            ext = os.path.splitext(file)[1]
+            if ext in SUPPORTED_EXTENSIONS:
+                language = SUPPORTED_EXTENSIONS[ext]
+                file_path = os.path.join(root, file)
+                results[file_path] = count_lines(file_path, language)
+    return results
+
+
+def display_results(results: ta.Mapping[str, FileLineCount]) -> None:
+    total_loc = total_blanks = total_comments = 0
+    file_width = max(map(len, results))
+    dash_width = 41 + file_width
+
+    print(
+        f"{'File'.ljust(file_width)} "
+        f"{'LOC':<10} "
+        f"{'Blank Lines':<15} "
+        f"{'Comment Lines':<15}",
+    )
+
+    print('-' * dash_width)
+
+    for file, counts in sorted(results.items()):
+        loc, blanks, comments = counts.loc, counts.blanks, counts.comments
+        total_loc += loc
+        total_blanks += blanks
+        total_comments += comments
+        print(
+            f'{file.ljust(file_width)} '
+            f'{loc:<10} '
+            f'{blanks:<15} '
+            f'{comments:<15}',
+        )
+
+    print('-' * dash_width)
+
+    print(
+        f"{' ' * file_width} "
+        f"{'LOC':<10} "
+        f"{'Blank Lines':<15} "
+        f"{'Comment Lines':<15}",
+    )
+
+    print('-' * dash_width)
+
+    print(
+        f"{'Total'.ljust(file_width)} "
+        f"{total_loc:<10} "
+        f"{total_blanks:<15} "
+        f"{total_comments:<15}",
+    )
+
+
+def _main() -> None:
+    import argparse
+
+    parser = argparse.ArgumentParser(description='Count lines of code in source files.')
+    parser.add_argument('directory', help='The directory to analyze.', nargs='+')
+    args = parser.parse_args()
+
+    for directory in args.directory:
+        results = count_lines_in_directory(directory)
+        display_results(results)
+        print()
+
+
+# @omlish-manifest
+_CLI_MODULE = CliModule('cloc', __name__)
+
+
+if __name__ == '__main__':
+    _main()

{omdev-0.0.0.dev140 → omdev-0.0.0.dev141/omdev.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omdev
-Version: 0.0.0.dev140
+Version: 0.0.0.dev141
 Summary: omdev
 Author: wrmsr
 License: BSD-3-Clause
@@ -12,7 +12,7 @@ Classifier: Operating System :: OS Independent
 Classifier: Operating System :: POSIX
 Requires-Python: >=3.12
 License-File: LICENSE
-Requires-Dist: omlish==0.0.0.dev140
+Requires-Dist: omlish==0.0.0.dev141
 Provides-Extra: all
 Requires-Dist: black~=24.10; extra == "all"
 Requires-Dist: pycparser~=2.22; extra == "all"

{omdev-0.0.0.dev140 → omdev-0.0.0.dev141}/omdev.egg-info/SOURCES.txt

@@ -148,6 +148,7 @@ omdev/toml/__init__.py
 omdev/toml/parser.py
 omdev/toml/writer.py
 omdev/tools/__init__.py
+omdev/tools/cloc.py
 omdev/tools/doc.py
 omdev/tools/docker.py
 omdev/tools/git.py

{omdev-0.0.0.dev140 → omdev-0.0.0.dev141}/omdev.egg-info/requires.txt

@@ -1,4 +1,4 @@
-omlish==0.0.0.dev140
+omlish==0.0.0.dev141
 
 [all]
 black~=24.10

{omdev-0.0.0.dev140 → omdev-0.0.0.dev141}/pyproject.toml

@@ -12,7 +12,7 @@ authors = [
 urls = {source = 'https://github.com/wrmsr/omlish'}
 license = {text = 'BSD-3-Clause'}
 requires-python = '>=3.12'
-version = '0.0.0.dev140'
+version = '0.0.0.dev141'
 classifiers = [
     'License :: OSI Approved :: BSD License',
     'Development Status :: 2 - Pre-Alpha',
@@ -22,7 +22,7 @@ classifiers = [
 ]
 description = 'omdev'
 dependencies = [
-    'omlish == 0.0.0.dev140',
+    'omlish == 0.0.0.dev141',
 ]
 
 [project.optional-dependencies]