mirror of
https://github.com/joshuadavidthomas/django-language-server.git
synced 2025-09-13 13:56:25 +00:00
84 lines
2 KiB
Python
84 lines
2 KiB
Python
from __future__ import annotations
|
|
|
|
import argparse
|
|
import asyncio
|
|
import json
|
|
import logging
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Agent:
|
|
def __init__(self, ipc_path: Path):
|
|
self.ipc_path = ipc_path
|
|
|
|
async def start(self):
|
|
if sys.platform == "win32":
|
|
# Windows named pipe
|
|
pipe_path = rf"\\.\pipe\{self.ipc_path.name}"
|
|
server = await asyncio.start_server(
|
|
self.handle_client,
|
|
pipe=pipe_path,
|
|
)
|
|
else:
|
|
# Unix domain socket
|
|
try:
|
|
self.ipc_path.unlink()
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
server = await asyncio.start_unix_server(
|
|
self.handle_client,
|
|
path=str(self.ipc_path),
|
|
)
|
|
|
|
async with server:
|
|
await server.serve_forever()
|
|
|
|
async def handle_client(
|
|
self,
|
|
reader: asyncio.StreamReader,
|
|
writer: asyncio.StreamWriter,
|
|
):
|
|
try:
|
|
while True:
|
|
data = await reader.readline()
|
|
if not data:
|
|
break
|
|
|
|
request = json.loads(data)
|
|
response = await self.handle_request(request)
|
|
|
|
writer.write(json.dumps(response).encode() + b"\n")
|
|
await writer.drain()
|
|
except Exception:
|
|
logger.exception("Error handling client request")
|
|
finally:
|
|
writer.close()
|
|
await writer.wait_closed()
|
|
|
|
async def handle_request(self, request): ...
|
|
|
|
|
|
def main():
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--settings", required=True)
|
|
parser.add_argument("--ipc-path", required=True)
|
|
args = parser.parse_args()
|
|
|
|
import django
|
|
|
|
os.environ.setdefault("DJANGO_SETTINGS_MODULE", args.settings)
|
|
django.setup()
|
|
|
|
agent = Agent(Path(args.ipc_path))
|
|
asyncio.run(agent.start())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|