switch from runner to ipc and long-running sidecar process (#21)

python/djls/lsp.py

@@ -0,0 +1,104 @@
+from __future__ import annotations
+
+import json
+import sys
+from typing import Any
+
+from .scripts import django_setup
+from .scripts import has_import
+from .scripts import python_setup
+
+
+def handle_command(command: str) -> str:
+    parts = command.strip().split()
+    command = parts[0]
+    args = parts[1:] if len(parts) > 1 else []
+
+    if command == "django_setup":
+        return json.dumps(django_setup.get_django_setup_info())
+    if command == "has_import":
+        if not args:
+            return "error: Missing module name argument"
+        return json.dumps({"can_import": has_import.check_import(args[0])})
+    if command == "health":
+        return "ok"
+    if command == "installed_apps_check":
+        import django
+        from django.conf import settings
+
+        django.setup()
+        if not args:
+            return "error: Missing module name argument"
+        return json.dumps({"has_app": args[0] in settings.INSTALLED_APPS})
+    if command == "python_setup":
+        return json.dumps(python_setup.get_python_info())
+    if command == "version":
+        return "0.1.0"
+    return f"Unknown command: {command}"
+
+
+def handle_json_command(data: dict[str, Any]) -> dict[str, Any]:
+    command = data["command"]
+    args = data.get("args", [])  # Get args if they exist
+
+    if command == "django_setup":
+        import django
+
+        django.setup()
+        return {"status": "ok", "data": django_setup.get_django_setup_info()}
+    if command == "has_import":
+        if not args:
+            return {"status": "error", "error": "Missing module name argument"}
+        return {
+            "status": "ok",
+            "data": {"can_import": has_import.check_import(args[0])},
+        }
+    if command == "health":
+        return {"status": "ok"}
+    if command == "installed_apps_check":
+        import django
+        from django.conf import settings
+
+        django.setup()
+        if not args:
+            return {"status": "error", "error": "Missing module name argument"}
+        return {
+            "status": "ok",
+            "data": {"has_app": args[0] in settings.INSTALLED_APPS},
+        }
+    if command == "python_setup":
+        return {"status": "ok", "data": python_setup.get_python_info()}
+    if command == "version":
+        return {"status": "ok", "data": "0.1.0"}
+
+    return {"status": "error", "error": f"Unknown command: {command}"}
+
+
+def main():
+    transport_type = sys.stdin.readline().strip()
+    print("ready", flush=True)
+
+    while True:
+        try:
+            line = sys.stdin.readline()
+            if not line:
+                break
+
+            if transport_type == "json":
+                data = json.loads(line)
+                response = handle_json_command(data)
+                print(json.dumps(response), flush=True)
+            else:
+                command = line.strip()
+                response = handle_command(command)
+                print(response, flush=True)
+
+        except Exception as e:
+            if transport_type == "json":
+                print(json.dumps({"status": "error", "error": str(e)}), flush=True)
+            else:
+                print(f"error: {str(e)}", flush=True)
+
+
+if __name__ == "__main__":
+    main()