Merge pull request #131 from spicecode-cli/structural-changes

Structural changes

Author: ManfredHair

cli/commands/analyze.py

@@ -1,6 +1,6 @@
 from InquirerPy import inquirer
 
-from cli.utils.get_translation import get_translation
+from utils.get_translation import get_translation
 from spice.analyze import analyze_file
 
 

cli/commands/export/export.py

@@ -5,7 +5,7 @@
 from rich.console import Console
 from rich.table import Table
 
-from cli.utils.get_translation import get_translation
+from utils.get_translation import get_translation
 
 def export_results(results, format_type, output_file, messages):
     """

cli/commands/hello.py

@@ -1,6 +1,6 @@
 from rich import print # this add colors to the printed text
 
-from cli.utils.get_translation import get_translation
+from utils.get_translation import get_translation
 
 
 def hello_command(LANG_FILE):

cli/commands/version.py

@@ -1,7 +1,7 @@
 import os
 from rich import print # this add colors to the printed text
 
-from cli.utils.get_translation import get_translation
+from utils.get_translation import get_translation
 
 
 def version_command(LANG_FILE, CURRENT_DIR):

spice/analyze.py

@@ -46,7 +46,7 @@ def analyze_file(file_path: str, selected_stats=None):
     # only put the code through the lexer and proceed with tokenization if needed
     if any(stat in selected_stats for stat in ["function_count"]):
         # get the lexer for the code's language
-        from spice.utils.get_lexer import get_lexer_for_file
+        from utils.get_lexer import get_lexer_for_file
         LexerClass = get_lexer_for_file(file_path)
 
         # tokenize the code via lexer

spice/utils/__init__.py

@@ -0,0 +1,22 @@
+import os
+
+
+# this will read the file extension and return the correct langague
+def get_lexer_for_file(file_path):
+    _, ext = os.path.splitext(file_path)
+
+    if ext == ".rb":
+        return "ruby"
+    
+    elif ext == ".py":
+        from lexers.python.pythonlexer import PythonLexer
+        return "python"
+    
+    elif ext == ".js":
+        return "javascript"
+    
+    elif ext == ".go":
+        return "go"
+    
+    else:
+        raise ValueError(f"Unsupported file extension: {ext}")