mirror of
https://github.com/joshuadavidthomas/django-language-server.git
synced 2025-07-19 02:15:09 +00:00
Get rid of all transport types and settle on Protobuf (#25)
* Get rid of all transport types and settle on Protobuf hope i don't regret this * Update Cargo.toml * Update agent.py
This commit is contained in:
parent
643a47953e
commit
0a6e975ca5
38 changed files with 1484 additions and 685 deletions
131
python/djls/agent.py
Normal file
131
python/djls/agent.py
Normal file
|
@ -0,0 +1,131 @@
|
|||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
import struct
|
||||
import sys
|
||||
|
||||
from google.protobuf.message import Message
|
||||
|
||||
from .commands import COMMANDS
|
||||
from .commands import Command
|
||||
from .proto.v1 import messages_pb2
|
||||
|
||||
logger = logging.getLogger("djls")
|
||||
logger.setLevel(logging.DEBUG)
|
||||
|
||||
fh = logging.FileHandler("/tmp/djls_debug.log")
|
||||
fh.setLevel(logging.DEBUG)
|
||||
|
||||
ch = logging.StreamHandler(sys.stderr)
|
||||
ch.setLevel(logging.DEBUG)
|
||||
|
||||
formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
|
||||
fh.setFormatter(formatter)
|
||||
ch.setFormatter(formatter)
|
||||
|
||||
logger.addHandler(fh)
|
||||
logger.addHandler(ch)
|
||||
|
||||
|
||||
class LSPAgent:
|
||||
def __init__(self):
|
||||
self._commands: dict[str, Command] = {cmd.name: cmd() for cmd in COMMANDS}
|
||||
logger.debug(
|
||||
"LSPAgent initialized with commands: %s", list(self._commands.keys())
|
||||
)
|
||||
|
||||
def serve(self):
|
||||
print("ready", flush=True)
|
||||
|
||||
try:
|
||||
import django
|
||||
|
||||
django.setup()
|
||||
|
||||
except Exception as e:
|
||||
error_response = self.create_error(messages_pb2.Error.DJANGO_ERROR, str(e))
|
||||
self.write_message(error_response)
|
||||
|
||||
while True:
|
||||
try:
|
||||
data = self.read_message()
|
||||
if not data:
|
||||
break
|
||||
|
||||
response = self.handle_request(data)
|
||||
self.write_message(response)
|
||||
|
||||
except Exception as e:
|
||||
error_response = self.create_error(messages_pb2.Error.UNKNOWN, str(e))
|
||||
self.write_message(error_response)
|
||||
|
||||
def read_message(self) -> bytes | None:
|
||||
length_bytes = sys.stdin.buffer.read(4)
|
||||
logger.debug("Read length bytes: %r", length_bytes)
|
||||
if not length_bytes:
|
||||
return None
|
||||
|
||||
length = struct.unpack(">I", length_bytes)[0]
|
||||
logger.debug("Unpacked length: %d", length)
|
||||
data = sys.stdin.buffer.read(length)
|
||||
logger.debug("Read data bytes: %r", data)
|
||||
return data
|
||||
|
||||
def handle_request(self, request_data: bytes) -> Message:
|
||||
request = messages_pb2.Request()
|
||||
request.ParseFromString(request_data)
|
||||
|
||||
command_name = request.WhichOneof("command")
|
||||
logger.debug("Command name: %s", command_name)
|
||||
command = self._commands.get(command_name)
|
||||
|
||||
if not command:
|
||||
logger.error("Unknown command: %s", command_name)
|
||||
return self.create_error(
|
||||
messages_pb2.Error.INVALID_REQUEST, f"Unknown command: {command_name}"
|
||||
)
|
||||
|
||||
try:
|
||||
result = command.execute(getattr(request, command_name))
|
||||
return messages_pb2.Response(**{command_name: result})
|
||||
except Exception as e:
|
||||
logger.exception("Error executing command")
|
||||
return self.create_error(messages_pb2.Error.UNKNOWN, str(e))
|
||||
|
||||
def write_message(self, message: Message) -> None:
|
||||
data = message.SerializeToString()
|
||||
logger.debug(f"Sending response, length: {len(data)}, data: {data!r}")
|
||||
length = struct.pack(">I", len(data))
|
||||
logger.debug(f"Length bytes: {length!r}")
|
||||
sys.stdout.buffer.write(length)
|
||||
sys.stdout.buffer.write(data)
|
||||
sys.stdout.buffer.flush()
|
||||
|
||||
def create_error(
|
||||
self, code: messages_pb2.Error.Code, message: str
|
||||
) -> messages_pb2.Response:
|
||||
response = messages_pb2.Response()
|
||||
response.error.code = code
|
||||
response.error.message = message
|
||||
return response
|
||||
|
||||
|
||||
def main() -> None:
|
||||
logger.debug("Starting DJLS...")
|
||||
|
||||
try:
|
||||
logger.debug("Initializing LSPAgent...")
|
||||
agent = LSPAgent()
|
||||
logger.debug("Starting LSPAgent serve...")
|
||||
agent.serve()
|
||||
except KeyboardInterrupt:
|
||||
logger.debug("Received KeyboardInterrupt")
|
||||
sys.exit(0)
|
||||
except Exception as e:
|
||||
logger.exception("Fatal error")
|
||||
print(f"error: {e}", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
Loading…
Add table
Add a link
Reference in a new issue