mirror of
https://github.com/iritmaximus/compilers.git
synced 2026-10-12 19:13:35 +03:00
105 lines
3.3 KiB
Python
105 lines
3.3 KiB
Python
from base64 import b64encode
|
|
import json
|
|
import re
|
|
import sys
|
|
from socketserver import ForkingTCPServer, StreamRequestHandler
|
|
from traceback import format_exception
|
|
from typing import Any
|
|
|
|
|
|
def call_compiler(source_code: str) -> bytes:
|
|
# *** TODO ***
|
|
# Call your compiler here and return the compiled executable.
|
|
# Raise an exception on compilation error.
|
|
# *** TODO ***
|
|
raise NotImplementedError("Compiler not implemented")
|
|
|
|
|
|
def main() -> int:
|
|
# === Option parsing ===
|
|
command: str | None = None
|
|
input_file: str | None = None
|
|
output_file: str | None = None
|
|
host = "127.0.0.1"
|
|
port = 3000
|
|
for arg in sys.argv[1:]:
|
|
if (m := re.fullmatch(r'--output=(.+)', arg)) is not None:
|
|
output_file = m[1]
|
|
elif (m := re.fullmatch(r'--host=(.+)', arg)) is not None:
|
|
host = m[1]
|
|
elif (m := re.fullmatch(r'--port=(.+)', arg)) is not None:
|
|
port = int(m[1])
|
|
elif arg.startswith('-'):
|
|
raise Exception(f"Unknown argument: {arg}")
|
|
elif command is None:
|
|
command = arg
|
|
elif input_file is None:
|
|
input_file = arg
|
|
else:
|
|
raise Exception("Multiple input files not supported")
|
|
|
|
valid_commands = ['compile', 'serve']
|
|
if command is None:
|
|
print(f"Error: command argument missing. Valid commands: {', '.join(valid_commands)}", file=sys.stderr)
|
|
return 1
|
|
|
|
if command not in valid_commands:
|
|
print(f"Error: unknown command: {command}", file=sys.stderr)
|
|
return 1
|
|
|
|
def read_source_code() -> str:
|
|
if input_file is not None:
|
|
with open(input_file) as f:
|
|
return f.read()
|
|
else:
|
|
return sys.stdin.read()
|
|
|
|
# === Command implementations ===
|
|
|
|
if command == 'compile':
|
|
source_code = read_source_code()
|
|
if output_file is None:
|
|
raise Exception("Output file flag --output=... required")
|
|
executable = call_compiler(source_code)
|
|
with open(output_file, 'wb') as f:
|
|
f.write(executable)
|
|
elif command == 'serve':
|
|
try:
|
|
run_server(host, port)
|
|
except KeyboardInterrupt:
|
|
pass
|
|
return 0
|
|
|
|
|
|
def run_server(host: str, port: int) -> None:
|
|
class Server(ForkingTCPServer):
|
|
allow_reuse_address = True
|
|
request_queue_size = 32
|
|
|
|
class Handler(StreamRequestHandler):
|
|
def handle(self) -> None:
|
|
result: dict[str, Any] = {}
|
|
try:
|
|
input_str = self.rfile.read().decode()
|
|
input = json.loads(input_str)
|
|
if input["command"] == "compile":
|
|
source_code = input["code"]
|
|
executable = call_compiler(source_code)
|
|
result["program"] = b64encode(executable).decode()
|
|
elif input["command"] == "ping":
|
|
pass
|
|
else:
|
|
result["error"] = "Unknown command: " + input['command']
|
|
except Exception as e:
|
|
result["error"] = "".join(format_exception(e))
|
|
result_str = json.dumps(result)
|
|
self.request.sendall(str.encode(result_str))
|
|
|
|
print(f"Starting TCP server at {host}:{port}")
|
|
with Server((host, port), Handler) as server:
|
|
server.serve_forever()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|