shell-lite 0.3.3__py3-none-any.whl → 0.3.5__py3-none-any.whl

shell_lite/main.py

@@ -8,7 +8,8 @@ import subprocess
 from .lexer import Lexer
 from .parser import Parser
 from .interpreter import Interpreter
-
+from .ast_nodes import *
+import json
 def execute_source(source: str, interpreter: Interpreter):
     lines = source.split('\n')
     try:
@@ -16,7 +17,6 @@ def execute_source(source: str, interpreter: Interpreter):
         tokens = lexer.tokenize()
         parser = Parser(tokens)
         statements = parser.parse()
-        
         for stmt in statements:
             interpreter.visit(stmt)
     except Exception as e:
@@ -25,7 +25,6 @@ def execute_source(source: str, interpreter: Interpreter):
             if 0 <= e.line-1 < len(lines):
                 print(f"  {lines[e.line-1].strip()}")
         print(f"Exception: {e}")
-
 def run_file(filename: str):
     if not os.path.exists(filename):
         print(f"Error: File '{filename}' not found.")
@@ -34,7 +33,6 @@ def run_file(filename: str):
         source = f.read()
     interpreter = Interpreter()
     execute_source(source, interpreter)
-
 def run_repl():
     interpreter = Interpreter()
     print("\n" + "="*40)
@@ -43,19 +41,14 @@ def run_repl():
     print("Version: v0.03.3 | Made by Shrey Naithani")
     print("Commands: Type 'exit' to quit, 'help' for examples.")
     print("Note: Terminal commands (like 'shl install') must be run in CMD/PowerShell, not here.")
-    
     buffer = []
     indent_level = 0
-    
     while True:
         try:
             prompt = "... " if indent_level > 0 else ">>> "
             line = input(prompt)
-            
             if line.strip() == "exit":
                 break
-            
-            # Support line continuation with \
             if line.endswith("\\"):
                 buffer.append(line[:-1])
                 indent_level = 1
@@ -68,12 +61,10 @@ def run_repl():
                 print('  display(tasks)             # View the list')
                 print('  \\                          # Tip: Use \\ at the end of a line for multi-line typing')
                 continue
-
             if line.strip().startswith("shl"):
                 print("! Hint: You are already INSIDE ShellLite. You don't need to type 'shl' here.")
                 print("! To run terminal commands, exit this REPL first.")
                 continue
-
             if not line:
                 if indent_level > 0:
                     source = "\n".join(buffer)
@@ -81,7 +72,6 @@ def run_repl():
                     buffer = []
                     indent_level = 0
                 continue
-
             if line.strip().endswith(":"):
                 indent_level = 1
                 buffer.append(line)
@@ -93,10 +83,8 @@ def run_repl():
                      execute_source(source, interpreter)
                      buffer = []
                      indent_level = 0
-                
                 if line.strip():
                     execute_source(line, interpreter)
-            
         except KeyboardInterrupt:
             print("\nExiting...")
             break
@@ -104,34 +92,24 @@ def run_repl():
             print(f"Error: {e}")
             buffer = []
             indent_level = 0
-
 def install_globally():
-    """Performs the global PATH installation."""
     print("\n" + "="*50)
     print("  ShellLite Global Installer")
     print("="*50)
-    
     install_dir = os.path.join(os.environ['LOCALAPPDATA'], 'ShellLite')
     if not os.path.exists(install_dir):
         os.makedirs(install_dir)
-    
     target_exe = os.path.join(install_dir, 'shl.exe')
     current_path = sys.executable
-    
-    # If not running as EXE (e.g. py script), we can't 'install' the script easily as shl.exe
     is_frozen = getattr(sys, 'frozen', False)
-    
     try:
         if is_frozen:
             shutil.copy2(current_path, target_exe)
         else:
             print("Error: Installation requires the shl.exe file.")
             return
-
-        # Add to PATH (User level)
         ps_cmd = f'$oldPath = [Environment]::GetEnvironmentVariable("Path", "User"); if ($oldPath -notlike "*ShellLite*") {{ [Environment]::SetEnvironmentVariable("Path", "$oldPath;{install_dir}", "User") }}'
         subprocess.run(["powershell", "-Command", ps_cmd], capture_output=True)
-        
         print(f"\n[SUCCESS] ShellLite is now installed!")
         print(f"Location: {install_dir}")
         print("\nACTION REQUIRED:")
@@ -142,70 +120,45 @@ def install_globally():
         input("Press Enter to continue...")
     except Exception as e:
         print(f"Installation failed: {e}")
-
 def install_package(package_name: str):
-    """
-    Downloads a package from GitHub.
-    Format: shl get user/repo
-    Target: ~/.shell_lite/modules/<repo>/ (containing main.shl or similar)
-    """
     if '/' not in package_name:
         print("Error: Package must be in format 'user/repo'")
         return
-
     user, repo = package_name.split('/')
     print(f"Fetching '{package_name}' from GitHub...")
-    
-    # Define modules dir
     home = os.path.expanduser("~")
     modules_dir = os.path.join(home, ".shell_lite", "modules")
     if not os.path.exists(modules_dir):
         os.makedirs(modules_dir)
-        
-    # Clean up existing
     target_dir = os.path.join(modules_dir, repo)
     if os.path.exists(target_dir):
         print(f"Removing existing '{repo}'...")
         shutil.rmtree(target_dir)
-    
-    # Strategy 1: Download Main Branch ZIP
     zip_url = f"https://github.com/{user}/{repo}/archive/refs/heads/main.zip"
-    
     try:
         print(f"Downloading {zip_url}...")
         with urllib.request.urlopen(zip_url) as response:
             zip_data = response.read()
-            
         with zipfile.ZipFile(io.BytesIO(zip_data)) as z:
             z.extractall(modules_dir)
-            
-        # Rename extracted folder (repo-main -> repo)
         extracted_name = f"{repo}-main"
         extracted_path = os.path.join(modules_dir, extracted_name)
-        
         if os.path.exists(extracted_path):
             os.rename(extracted_path, target_dir)
             print(f"[SUCCESS] Installed '{repo}' to {target_dir}")
         else:
              print(f"Error: Could not find extracted folder '{extracted_name}'.")
-             
         return
-
     except urllib.error.HTTPError:
-        # Strategy 2: Try Master Branch
          zip_url = f"https://github.com/{user}/{repo}/archive/refs/heads/master.zip"
          try:
             print(f"Downloading {zip_url}...")
             with urllib.request.urlopen(zip_url) as response:
                 zip_data = response.read()
-                
             with zipfile.ZipFile(io.BytesIO(zip_data)) as z:
                 z.extractall(modules_dir)
-                
-            # Rename extracted folder (repo-master -> repo)
             extracted_name = f"{repo}-master"
             extracted_path = os.path.join(modules_dir, extracted_name)
-            
             if os.path.exists(extracted_path):
                 os.rename(extracted_path, target_dir)
                 print(f"[SUCCESS] Installed '{repo}' to {target_dir}")
@@ -213,30 +166,22 @@ def install_package(package_name: str):
                  print(f"Error: Could not find extracted folder '{extracted_name}'.")
          except Exception as e:
               print(f"Installation failed: {e}")
-              
     except Exception as e:
         print(f"Installation failed: {e}")
-
-
 def compile_file(filename: str, target: str = 'python'):
     if not os.path.exists(filename):
         print(f"Error: File '{filename}' not found.")
         return
-
     print(f"Compiling {filename} to {target.upper()}...")
-    
     with open(filename, 'r', encoding='utf-8') as f:
         source = f.read()
-        
     try:
         from .parser import Parser
         from .lexer import Lexer
-        
         lexer = Lexer(source)
         tokens = lexer.tokenize()
         parser = Parser(tokens)
         statements = parser.parse()
-        
         if target.lower() == 'js':
             from .js_compiler import JSCompiler
             compiler = JSCompiler()
@@ -247,17 +192,12 @@ def compile_file(filename: str, target: str = 'python'):
             compiler = Compiler()
             code = compiler.compile(statements)
             ext = '.py'
-        
         output_file = filename.replace('.shl', ext)
         if output_file == filename: output_file += ext
-        
         with open(output_file, 'w') as f:
             f.write(code)
-            
         print(f"[SUCCESS] Transpiled to {output_file}")
-        
         if target.lower() == 'python':
-            # Optional: Compile to EXE using PyInstaller
             try:
                 import PyInstaller.__main__
                 print("Building Executable with PyInstaller...")
@@ -269,23 +209,114 @@ def compile_file(filename: str, target: str = 'python'):
                 ])
                 print(f"[SUCCESS] Built {os.path.splitext(os.path.basename(filename))[0]}.exe")
             except ImportError:
-                 pass # Silent fail regarding exe if just transpiling
-
+                 pass 
     except Exception as e:
         print(f"Compilation Failed: {e}")
-
-
+def lint_file(filename: str):
+    if not os.path.exists(filename):
+        print(json.dumps([{"line": 0, "message": f"File {filename} not found"}]))
+        return
+    try:
+        with open(filename, 'r', encoding='utf-8') as f:
+            source = f.read()
+        lexer = Lexer(source)
+        tokens = lexer.tokenize()
+        parser = Parser(tokens)
+        parser.parse()
+        print(json.dumps([]))
+    except Exception as e:
+        line = getattr(e, 'line', 1)
+        print(json.dumps([{
+            "line": line, 
+            "message": str(e)
+        }]))
+def resolve_cursor(filename: str, line: int, col: int):
+    try:
+        with open(filename, 'r', encoding='utf-8') as f:
+            source = f.read()
+        lexer = Lexer(source)
+        tokens = lexer.tokenize()
+        parser = Parser(tokens)
+        nodes = parser.parse()
+        target_token = None
+        for t in tokens:
+            if t.line == line:
+                if t.column <= col <= t.column + len(t.value):
+                    target_token = t
+                    break
+        if not target_token or target_token.type != 'ID':
+             print(json.dumps({"found": False}))
+             return
+        word = target_token.value
+        def find_def(n_list, name):
+            for node in n_list:
+                if isinstance(node, FunctionDef) and node.name == name:
+                    return node, "Function"
+                if isinstance(node, ClassDef) and node.name == name:
+                    return node, "Class"
+                if isinstance(node, Assign) and node.name == name:
+                    return node, "Variable"
+                if isinstance(node, If):
+                     res = find_def(node.body, name)
+                     if res: return res
+            return None, None
+        found_node = None
+        found_type = None
+        queue = nodes[:]
+        while queue:
+            n = queue.pop(0)
+            if isinstance(n, FunctionDef):
+                if n.name == word: 
+                    found_node = n
+                    found_type = "Function"
+                    break
+                queue.extend(n.body)
+            elif isinstance(n, ClassDef):
+                if n.name == word:
+                    found_node = n
+                    found_type = "Class"
+                    break
+                queue.extend(n.methods)
+            elif isinstance(n, Assign) and n.name == word:
+                found_node = n
+                found_type = "Variable"
+                break
+            elif isinstance(n, If): queue.extend(n.body)
+            elif isinstance(n, While): queue.extend(n.body)
+            elif isinstance(n, For): queue.extend(n.body)
+        if found_node:
+            print(json.dumps({
+                "found": True,
+                "file": filename,
+                "line": found_node.line,
+                "hover": f"**{found_type}** `{word}`"
+            }))
+        else:
+             print(json.dumps({"found": False}))
+    except Exception:
+        print(json.dumps({"found": False}))
+def format_file(filename: str):
+    if not os.path.exists(filename):
+        print(f"Error: File '{filename}' not found.")
+        return
+    try:
+        with open(filename, 'r', encoding='utf-8') as f:
+            source = f.read()
+        from .formatter import Formatter
+        formatter = Formatter(source)
+        formatted_code = formatter.format()
+        with open(filename, 'w', encoding='utf-8') as f:
+            f.write(formatted_code)
+        print(f"[SUCCESS] Formatted {filename}")
+    except Exception as e:
+        print(f"Formatting failed: {e}")
 def self_install_check():
-    """Checks if shl is in PATH, if not, offer to install it."""
-    # Simple check: is shl.exe in a known Global path?
-    # Or just check if 'shl' works in shell
     res = subprocess.run(["where", "shl"], capture_output=True, text=True)
     if "ShellLite" not in res.stdout:
         print("\nShellLite is not installed globally.")
         choice = input("Would you like to install it so 'shl' works everywhere? (y/n): ").lower()
         if choice == 'y':
             install_globally()
-
 def show_help():
     print("""
 ShellLite - The English-Like Programming Language
@@ -294,11 +325,12 @@ Usage:
   shl                   Start the interactive REPL
   shl help              Show this help message
   shl compile <file>    Compile a script (Options: --target js)
+  shl fmt <file>        Format a script
+  shl check <file>      Lint a file (JSON output)
+  shl resolve <file> <line> <col>  Resolve symbol (JSON output)
   shl install           Install ShellLite globally to your system PATH
-
 For documentation, visit: https://github.com/Shrey-N/ShellDesk
 """)
-
 def main():
     if len(sys.argv) > 1:
         cmd = sys.argv[1]
@@ -325,14 +357,27 @@ def main():
             else:
                  print("Usage: shl get <user/repo>")
         elif cmd == "install":
-            # Install ShellLite itself
             install_globally()
+        elif cmd == "fmt" or cmd == "format":
+            if len(sys.argv) > 2:
+                filename = sys.argv[2]
+                format_file(filename)
+            else:
+                print("Usage: shl fmt <filename>")
+        elif cmd == "check":
+            if len(sys.argv) > 2:
+                filename = sys.argv[2]
+                lint_file(filename)
+        elif cmd == "resolve":
+            if len(sys.argv) > 4:
+                filename = sys.argv[2]
+                line = int(sys.argv[3])
+                col = int(sys.argv[4])
+                resolve_cursor(filename, line, col)
         else:
             run_file(sys.argv[1])
     else:
-        # No args - trigger install check, then REPL
         self_install_check()
         run_repl()
-
 if __name__ == "__main__":
     main()