mirror of
https://github.com/yeicor-3d/yet-another-cad-viewer.git
synced 2025-12-19 22:24:17 +01:00
118 lines
4.2 KiB
Python
118 lines
4.2 KiB
Python
import asyncio
|
|
import atexit
|
|
import os
|
|
import signal
|
|
import sys
|
|
from threading import Thread
|
|
from typing import Optional, Tuple
|
|
|
|
from OCP.TopoDS import TopoDS_Shape
|
|
from aiohttp import web
|
|
|
|
from pubsub import BufferedPubSub
|
|
from tessellate import _hashcode
|
|
|
|
FRONTEND_BASE_PATH = os.getenv('FRONTEND_BASE_PATH', '../dist')
|
|
UPDATES_API_PATH = '/api/updates'
|
|
OBJECTS_API_PATH = '/api/objects' # /{name}
|
|
|
|
|
|
# noinspection PyUnusedLocal
|
|
async def _index_handler(request: web.Request) -> web.Response:
|
|
return web.HTTPTemporaryRedirect(location='index.html')
|
|
|
|
|
|
class Server:
|
|
app = web.Application()
|
|
runner: web.AppRunner
|
|
thread: Optional[Thread] = None
|
|
do_shutdown = asyncio.Event()
|
|
show_events = BufferedPubSub[Tuple[TopoDS_Shape, str]]()
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
# --- Routes ---
|
|
# - APIs
|
|
self.app.router.add_route('GET', f'{UPDATES_API_PATH}', self.api_updates)
|
|
self.app.router.add_route('GET', f'{OBJECTS_API_PATH}/{{name}}', self.api_objects)
|
|
# - Static files from the frontend
|
|
self.app.router.add_get('/{path:(.*/|)}', _index_handler) # Any folder -> index.html
|
|
self.app.router.add_static('/', path=FRONTEND_BASE_PATH, name='static_frontend')
|
|
# --- Misc ---
|
|
self.loop = asyncio.new_event_loop()
|
|
|
|
def start(self):
|
|
"""Starts the web server in the background"""
|
|
assert self.thread is None, "Server already started"
|
|
# Start the server in a separate daemon thread
|
|
self.thread = Thread(target=self.run_server, name='yacv_server', daemon=True)
|
|
signal.signal(signal.SIGINT | signal.SIGTERM, self.stop)
|
|
atexit.register(self.stop)
|
|
self.thread.start()
|
|
|
|
# noinspection PyUnusedLocal
|
|
def stop(self, *args):
|
|
"""Stops the web server"""
|
|
if self.thread is None:
|
|
print('Cannot stop server because it is not running')
|
|
return
|
|
# FIXME: Wait for at least one client to confirm ready before stopping in case we are too fast?
|
|
self.loop.call_soon_threadsafe(lambda *a: self.do_shutdown.set())
|
|
self.thread.join(timeout=12)
|
|
self.thread = None
|
|
if len(args) >= 1 and args[0] in (signal.SIGINT, signal.SIGTERM):
|
|
sys.exit(0) # Exit with success
|
|
|
|
def run_server(self):
|
|
"""Runs the web server"""
|
|
asyncio.set_event_loop(self.loop)
|
|
self.loop.run_until_complete(self.run_server_async())
|
|
self.loop.stop()
|
|
self.loop.close()
|
|
|
|
async def run_server_async(self):
|
|
"""Runs the web server (async)"""
|
|
runner = web.AppRunner(self.app)
|
|
await runner.setup()
|
|
site = web.TCPSite(runner, os.getenv('YACV_HOST', 'localhost'), int(os.getenv('YACV_PORT', 32323)))
|
|
await site.start()
|
|
# print(f'Server started at {site.name}')
|
|
# Wait for a signal to stop the server while running
|
|
await self.do_shutdown.wait()
|
|
# print('Shutting down server...')
|
|
await runner.cleanup()
|
|
|
|
async def api_updates(self, request: web.Request) -> web.WebSocketResponse:
|
|
"""Handles a publish-only websocket connection that send show_object events along with their hashes and URLs"""
|
|
ws = web.WebSocketResponse()
|
|
await ws.prepare(request)
|
|
|
|
print('New client connected')
|
|
async for (obj, name) in self.show_events.subscribe():
|
|
hash_code = _hashcode(obj)
|
|
url = f'{UPDATES_API_PATH}/{name}'
|
|
print('New object:', name, hash_code, url)
|
|
await ws.send_json({'name': name, 'hash': hash_code, 'url': url})
|
|
|
|
# TODO: Start previous loop in a separate task and detect connection close to stop it
|
|
|
|
return ws
|
|
|
|
obj_counter = 0
|
|
|
|
def show_object(self, obj: TopoDS_Shape, name: Optional[str] = None):
|
|
"""Publishes a CAD object to the server"""
|
|
name = name or f'object_{self.obj_counter}'
|
|
self.obj_counter += 1
|
|
self.show_events.publish_nowait((obj, name))
|
|
|
|
async def api_objects(self, request: web.Request) -> web.Response:
|
|
return web.Response(body='TODO: Serve the object file here')
|
|
|
|
|
|
def get_app() -> web.Application:
|
|
"""Required by aiohttp-devtools"""
|
|
from logo.logo import build_logo
|
|
server = Server()
|
|
server.show_object(build_logo())
|
|
return server.app
|