crocodilestick
112 lines
4.2 KiB
Python
Executable File
112 lines
4.2 KiB
Python
Executable File
# -*- coding: utf-8 -*-
|
||
# Calibre-Web Automated – fork of Calibre-Web
|
||
# Copyright (C) 2018-2025 Calibre-Web contributors
|
||
# Copyright (C) 2024-2025 Calibre-Web Automated contributors
|
||
# SPDX-License-Identifier: GPL-3.0-or-later
|
||
# See CONTRIBUTORS for full list of authors.
|
||
|
||
from tornado.wsgi import WSGIContainer
|
||
import tornado
|
||
|
||
from tornado import escape
|
||
from tornado import httputil
|
||
from tornado.ioloop import IOLoop
|
||
from tornado.log import access_log
|
||
|
||
from typing import List, Tuple, Optional, Callable, Any, Dict, Text
|
||
from types import TracebackType
|
||
import typing
|
||
|
||
if typing.TYPE_CHECKING:
|
||
from typing import Type # noqa: F401
|
||
from wsgiref.types import WSGIApplication as WSGIAppType # noqa: F4
|
||
|
||
class MyWSGIContainer(WSGIContainer):
|
||
|
||
def __call__(self, request: httputil.HTTPServerRequest) -> None:
|
||
if tornado.version_info < (6, 3, 0, -99):
|
||
data = {} # type: Dict[str, Any]
|
||
response = [] # type: List[bytes]
|
||
|
||
def start_response(
|
||
status: str,
|
||
headers: List[Tuple[str, str]],
|
||
exc_info: Optional[
|
||
Tuple[
|
||
"Optional[Type[BaseException]]",
|
||
Optional[BaseException],
|
||
Optional[TracebackType],
|
||
]
|
||
] = None,
|
||
) -> Callable[[bytes], Any]:
|
||
data["status"] = status
|
||
data["headers"] = headers
|
||
return response.append
|
||
|
||
app_response = self.wsgi_application(
|
||
MyWSGIContainer.environ(self, request), start_response
|
||
)
|
||
try:
|
||
response.extend(app_response)
|
||
body = b"".join(response)
|
||
finally:
|
||
if hasattr(app_response, "close"):
|
||
app_response.close() # type: ignore
|
||
if not data:
|
||
raise Exception("WSGI app did not call start_response")
|
||
|
||
status_code_str, reason = data["status"].split(" ", 1)
|
||
status_code = int(status_code_str)
|
||
headers = data["headers"] # type: List[Tuple[str, str]]
|
||
header_set = set(k.lower() for (k, v) in headers)
|
||
body = escape.utf8(body)
|
||
if status_code != 304:
|
||
if "content-length" not in header_set:
|
||
headers.append(("Content-Length", str(len(body))))
|
||
if "content-type" not in header_set:
|
||
headers.append(("Content-Type", "text/html; charset=UTF-8"))
|
||
if "server" not in header_set:
|
||
headers.append(("Server", "TornadoServer/%s" % tornado.version))
|
||
|
||
start_line = httputil.ResponseStartLine("HTTP/1.1", status_code, reason)
|
||
header_obj = httputil.HTTPHeaders()
|
||
for key, value in headers:
|
||
header_obj.add(key, value)
|
||
assert request.connection is not None
|
||
request.connection.write_headers(start_line, header_obj, chunk=body)
|
||
request.connection.finish()
|
||
self._log(status_code, request)
|
||
else:
|
||
IOLoop.current().spawn_callback(self.handle_request, request)
|
||
|
||
|
||
def environ(self, request: httputil.HTTPServerRequest) -> Dict[Text, Any]:
|
||
try:
|
||
environ = WSGIContainer.environ(self, request)
|
||
except TypeError as e:
|
||
environ = WSGIContainer.environ(request)
|
||
environ['RAW_URI'] = request.path
|
||
self.env = environ
|
||
return environ
|
||
|
||
def _log(self, status_code: int, request: httputil.HTTPServerRequest) -> None:
|
||
if status_code < 400:
|
||
log_method = access_log.info
|
||
elif status_code < 500:
|
||
log_method = access_log.warning
|
||
else:
|
||
log_method = access_log.error
|
||
request_time = 1000.0 * request.request_time()
|
||
assert request.method is not None
|
||
assert request.uri is not None
|
||
ip = self.env.get("HTTP_FORWARD_FOR", None) or request.remote_ip
|
||
summary = (
|
||
request.method # type: ignore[operator]
|
||
+ " "
|
||
+ request.uri
|
||
+ " ("
|
||
+ ip
|
||
+ ")"
|
||
)
|
||
log_method("%d %s %.2fms", status_code, summary, request_time)
|