production init

front/ymsg/entry.py

@@ -0,0 +1,68 @@
+from typing import Optional, Callable
+import asyncio, settings
+
+from aiohttp import web
+from core.backend import Backend
+from util.misc import Logger
+
+from .ymsg_ctrl import YMSGCtrlBase
+
+def register(loop: asyncio.AbstractEventLoop, backend: Backend, http_app: web.Application, *, devmode: bool = False) -> None:
+	from util.misc import ProtocolRunner
+	from . import pager, http, videochat, voicechat
+	
+	backend.add_runner(ProtocolRunner('0.0.0.0', 5050, ListenerYMSG, args = ['Yahoo', backend, pager.YMSGCtrlPager], service = "YMSG Pager"))
+	# Funny that Yahoo! used the FTP transfer, Telnet, SMTP, and NNTP (Usenet) ports as the fallback ports.
+	backend.add_runner(ProtocolRunner('0.0.0.0', 20, ListenerYMSG, args = ['Yahoo', backend, pager.YMSGCtrlPager], service = "YMSG Pager"))
+	backend.add_runner(ProtocolRunner('0.0.0.0', 23, ListenerYMSG, args = ['Yahoo', backend, pager.YMSGCtrlPager], service = "YMSG Pager"))
+	#backend.add_runner(ProtocolRunner('0.0.0.0', 25, ListenerYMSG, args = ['Yahoo', backend, pager.YMSGCtrlPager], service = "YMSG Pager"))
+	#backend.add_runner(ProtocolRunner('0.0.0.0', 119, ListenerYMSG, args = ['Yahoo', backend, pager.YMSGCtrlPager], service = "YMSG Pager"))
+	# Yahoo! also utilized port 80 for YMSG communication via TCP, but that interferes with the port 80 binded to the HTTP
+	# services when the server is run in dev mode.
+	#backend.add_runner(ProtocolRunner('0.0.0.0', 80, ListenerYMSG, args = ['Yahoo', backend, pager.YMSGCtrlPager], service = "YMSG Pager"))
+	backend.add_runner(ProtocolRunner('0.0.0.0', 8001, ListenerYMSG, args = ['Yahoo', backend, pager.YMSGCtrlPager], service = "YMSG Pager"))
+	backend.add_runner(ProtocolRunner('0.0.0.0', 8002, ListenerYMSG, args = ['Yahoo', backend, pager.YMSGCtrlPager], service = "YMSG Pager"))
+	http.register(http_app, devmode = devmode)
+	voicechat.register(backend)
+	videochat.register(backend)
+
+class ListenerYMSG(asyncio.Protocol):
+	logger: Logger
+	backend: Backend
+	controller: YMSGCtrlBase
+	transport: Optional[asyncio.WriteTransport]
+	
+	def __init__(self, logger_prefix: str, backend: Backend, controller_factory: Callable[[Logger, str, Backend], YMSGCtrlBase]) -> None:
+		super().__init__()
+		self.logger = Logger(logger_prefix, self)
+		self.backend = backend
+		self.controller = controller_factory(self.logger, 'direct', backend)
+		self.controller.close_callback = self._on_close
+		self.transport = None
+	
+	def connection_made(self, transport: asyncio.BaseTransport) -> None:
+		assert isinstance(transport, asyncio.WriteTransport)
+		self.transport = transport
+		self.logger.log_connect()
+	
+	def connection_lost(self, exc: Optional[Exception]) -> None:
+		self.controller.close()
+		self.logger.log_disconnect()
+		self.transport = None
+	
+	def data_received(self, data: bytes) -> None:
+		transport = self.transport
+		assert transport is not None
+		if self.backend.maintenance_mode:
+			transport.close()
+			return
+		self.controller.transport = None
+		if self.controller.transport is None:
+			self.controller.transport = self.transport
+		self.controller.data_received(data)
+		transport.write(self.controller.flush())
+		self.controller.transport = transport
+	
+	def _on_close(self) -> None:
+		if self.transport is None: return
+		self.transport.close()