feat: ✨ Add C# support (#75)

getcodelimit · Jan 17, 2025 · 765ce4c · 765ce4c
1 parent b3a63f6
commit 765ce4c
Show file tree

Hide file tree

Showing 18 changed files with 500 additions and 88 deletions.
diff --git a/codelimit/commands/check.py b/codelimit/commands/check.py
@@ -30,9 +30,9 @@ def check_command(paths: list[Path], quiet: bool):
                     check_file(abs_path, check_result)
     exit_code = 1 if check_result.unmaintainable > 0 else 0
     if (
-        not quiet
-        or check_result.hard_to_maintain > 0
-        or check_result.unmaintainable > 0
+            not quiet
+            or check_result.hard_to_maintain > 0
+            or check_result.unmaintainable > 0
     ):
         check_result.report()
     raise typer.Exit(code=exit_code)

diff --git a/codelimit/common/Language.py b/codelimit/common/Language.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from abc import ABC, abstractmethod
 
 from codelimit.common.Token import Token
@@ -17,6 +19,6 @@ def extract_headers(self, tokens: list[Token]) -> list[Header]:
 
     @abstractmethod
     def extract_blocks(
-        self, tokens: list[Token], headers: list[Header]
+            self, tokens: list[Token], headers: list[Header]
     ) -> list[TokenRange]:
         pass
diff --git a/codelimit/common/report/format_markdown.py b/codelimit/common/report/format_markdown.py
@@ -126,5 +126,6 @@ def _print_findings_with_repository(report_units: list[ReportUnit],
         link = (f'https://github.com/{owner}/{name}/blob/{branch}/{unit.file}#L{unit.measurement.start.line}-L'
                 f'{unit.measurement.end.line}')
         console.print(
-            f"| {violation_type} \[{unit.measurement.unit_name}]({link}) | {unit.measurement.value} | {unit.file} |"
+            f"| {violation_type} " + '\[' + unit.measurement.unit_name + ']' + f"({link}) | {unit.measurement.value} "
+                                                                               f"| {unit.file} |"
         )
diff --git a/codelimit/common/token_matching/predicate/TokenValue.py b/codelimit/common/token_matching/predicate/TokenValue.py
@@ -22,4 +22,4 @@ def __hash__(self):
         return hash(self.value)
 
     def __str__(self):
-        return f"<TokenValue {self.value}>"
+        return self.value
diff --git a/codelimit/languages/CSharp.py b/codelimit/languages/CSharp.py
@@ -0,0 +1,20 @@
+from codelimit.common.Language import Language
+from codelimit.common.gsm.operator.OneOrMore import OneOrMore
+from codelimit.common.scope.scope_utils import (
+    get_blocks,
+    get_headers,
+)
+from codelimit.common.token_matching.predicate.Balanced import Balanced
+from codelimit.common.token_matching.predicate.Name import Name
+from codelimit.common.token_matching.predicate.Symbol import Symbol
+
+
+class CSharp(Language):
+    def __init__(self):
+        super().__init__('C#')
+
+    def extract_headers(self, tokens: list) -> list:
+        return get_headers(tokens, [Name(), OneOrMore(Balanced('(', ')'))], Symbol('{'))
+
+    def extract_blocks(self, tokens: list, headers: list) -> list:
+        return get_blocks(tokens, "{", "}")
diff --git a/codelimit/languages/Cpp.py b/codelimit/languages/Cpp.py
@@ -17,6 +17,6 @@ def extract_headers(self, tokens: list[Token]) -> list[Header]:
         return get_headers(tokens, [Name(), OneOrMore(Balanced("(", ")"))], Symbol("{"))
 
     def extract_blocks(
-        self, tokens: list[Token], headers: list[Header]
+            self, tokens: list[Token], headers: list[Header]
     ) -> list[TokenRange]:
         return get_blocks(tokens, "{", "}")
diff --git a/codelimit/languages/TypeScript.py b/codelimit/languages/TypeScript.py
@@ -6,10 +6,10 @@
 from codelimit.common.scope.Header import Header
 from codelimit.common.scope.scope_utils import get_blocks, get_headers
 from codelimit.common.token_matching.predicate.Balanced import Balanced
-from codelimit.common.token_matching.predicate.Or import Or
 from codelimit.common.token_matching.predicate.Keyword import Keyword
 from codelimit.common.token_matching.predicate.Name import Name
 from codelimit.common.token_matching.predicate.Operator import Operator
+from codelimit.common.token_matching.predicate.Or import Or
 from codelimit.common.token_matching.predicate.Symbol import Symbol
 
 
@@ -38,6 +38,6 @@ def extract_headers(self, tokens: list[Token]) -> list[Header]:
         return functions + arrow_functions
 
     def extract_blocks(
-        self, tokens: list[Token], headers: list[Header]
+            self, tokens: list[Token], headers: list[Header]
     ) -> list[TokenRange]:
         return get_blocks(tokens, "{", "}")
diff --git a/codelimit/languages/__init__.py b/codelimit/languages/__init__.py
@@ -1,4 +1,6 @@
+from codelimit.common.Language import Language
 from codelimit.languages.C import C
+from codelimit.languages.CSharp import CSharp
 from codelimit.languages.Cpp import Cpp
 from codelimit.languages.Java import Java
 from codelimit.languages.JavaScript import JavaScript
@@ -9,16 +11,14 @@
 class Languages:
     C = C()
     Cpp = Cpp()
+    CSharp = CSharp()
     Java = Java()
     JavaScript = JavaScript()
     Python = Python()
     TypeScript = TypeScript()
 
-    by_name = {
-        C.name: C,
-        Cpp.name: Cpp,
-        Java.name: Java,
-        JavaScript.name: JavaScript,
-        Python.name: Python,
-        TypeScript.name: TypeScript,
-    }
+    by_name: dict[str, Language] = {}
+    for subclass in Language.__subclasses__():
+        language = subclass() # type: ignore
+        by_name[language.name] = language
+