mirror of
https://github.com/joshuadavidthomas/django-language-server.git
synced 2025-07-18 18:05:04 +00:00
Josh Thomas
0a6e975ca5
* Get rid of all transport types and settle on Protobuf hope i don't regret this * Update Cargo.toml * Update agent.py
68 lines
1.9 KiB
Python
68 lines
1.9 KiB
Python
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_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
|
|
|
|
data = json.loads(line)
|
|
response = handle_json_command(data)
|
|
print(json.dumps(response), flush=True)
|
|
|
|
except Exception as e:
|
|
print(json.dumps({"status": "error", "error": str(e)}), flush=True)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|