Compare commits
2 Commits
d9b559d13d
...
04360b42d8
| Author | SHA1 | Date | |
|---|---|---|---|
| 04360b42d8 | |||
| 4edcc6acc7 |
+1
-1
@@ -1,6 +1,6 @@
|
|||||||
[project]
|
[project]
|
||||||
name = "stapler"
|
name = "stapler"
|
||||||
version = "1.3.0"
|
version = "1.3.1"
|
||||||
description = "Static pages as simple as a gzip file"
|
description = "Static pages as simple as a gzip file"
|
||||||
requires-python = ">=3.14"
|
requires-python = ">=3.14"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
|
|||||||
@@ -200,6 +200,7 @@ class CertManager:
|
|||||||
cert_file,
|
cert_file,
|
||||||
key_file,
|
key_file,
|
||||||
)
|
)
|
||||||
|
new_context.set_alpn_protocols(["http/1.1"])
|
||||||
socket.context = new_context
|
socket.context = new_context
|
||||||
except Exception:
|
except Exception:
|
||||||
self.logger.exception("Could not create HTTPS context for %s", host)
|
self.logger.exception("Could not create HTTPS context for %s", host)
|
||||||
|
|||||||
+12
-2
@@ -27,6 +27,7 @@ if typing.TYPE_CHECKING:
|
|||||||
|
|
||||||
class BaseHandler(abc.ABC, http.server.BaseHTTPRequestHandler):
|
class BaseHandler(abc.ABC, http.server.BaseHTTPRequestHandler):
|
||||||
timeout = 10
|
timeout = 10
|
||||||
|
protocol_version = "HTTP/1.1"
|
||||||
REQUEST_COUNT = 0
|
REQUEST_COUNT = 0
|
||||||
|
|
||||||
@typing.override
|
@typing.override
|
||||||
@@ -111,6 +112,8 @@ class BaseHandler(abc.ABC, http.server.BaseHTTPRequestHandler):
|
|||||||
@typing.override
|
@typing.override
|
||||||
def log_request(self, code: str = "?", size: str = "-") -> None: # ty:ignore[invalid-method-override] # pragma: no cover
|
def log_request(self, code: str = "?", size: str = "-") -> None: # ty:ignore[invalid-method-override] # pragma: no cover
|
||||||
if isinstance(code, http.HTTPStatus):
|
if isinstance(code, http.HTTPStatus):
|
||||||
|
code = code.value
|
||||||
|
if isinstance(code, int):
|
||||||
color = logs.TermColor.RED
|
color = logs.TermColor.RED
|
||||||
if 100 <= code < 200:
|
if 100 <= code < 200:
|
||||||
color = logs.TermColor.CYAN
|
color = logs.TermColor.CYAN
|
||||||
@@ -120,7 +123,7 @@ class BaseHandler(abc.ABC, http.server.BaseHTTPRequestHandler):
|
|||||||
color = logs.TermColor.BLUE
|
color = logs.TermColor.BLUE
|
||||||
elif 400 <= code < 500:
|
elif 400 <= code < 500:
|
||||||
color = logs.TermColor.YELLOW
|
color = logs.TermColor.YELLOW
|
||||||
code = color + str(code.value) + logs.TermColor.RESET
|
code = color + str(code) + logs.TermColor.RESET
|
||||||
if size == "" and self.out_size > 0:
|
if size == "" and self.out_size > 0:
|
||||||
size = str(self.out_size)
|
size = str(self.out_size)
|
||||||
args = (
|
args = (
|
||||||
@@ -193,6 +196,7 @@ class BaseHandler(abc.ABC, http.server.BaseHTTPRequestHandler):
|
|||||||
headers=headers,
|
headers=headers,
|
||||||
allow_redirects=False,
|
allow_redirects=False,
|
||||||
timeout=480,
|
timeout=480,
|
||||||
|
stream=False,
|
||||||
)
|
)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self.send_error(
|
self.send_error(
|
||||||
@@ -385,6 +389,7 @@ class RequestHandler(http.server.SimpleHTTPRequestHandler, BaseHandler):
|
|||||||
self._pre_log_request()
|
self._pre_log_request()
|
||||||
if not self._proxy_or_redirect():
|
if not self._proxy_or_redirect():
|
||||||
super().do_HEAD()
|
super().do_HEAD()
|
||||||
|
self.close_connection = True
|
||||||
|
|
||||||
@typing.override
|
@typing.override
|
||||||
def do_GET(self) -> None:
|
def do_GET(self) -> None:
|
||||||
@@ -394,7 +399,9 @@ class RequestHandler(http.server.SimpleHTTPRequestHandler, BaseHandler):
|
|||||||
return None
|
return None
|
||||||
if self.path == "/" and self.host == self.default_host:
|
if self.path == "/" and self.host == self.default_host:
|
||||||
return self.send_basic_body(self.server_signature())
|
return self.send_basic_body(self.server_signature())
|
||||||
return super().do_GET()
|
super().do_GET()
|
||||||
|
self.close_connection = True
|
||||||
|
return None
|
||||||
|
|
||||||
def do_PUT(self) -> None:
|
def do_PUT(self) -> None:
|
||||||
with self.handle_errors():
|
with self.handle_errors():
|
||||||
@@ -620,16 +627,19 @@ class RequestHandler(http.server.SimpleHTTPRequestHandler, BaseHandler):
|
|||||||
|
|
||||||
|
|
||||||
class UpgradeHandler(RequestHandler):
|
class UpgradeHandler(RequestHandler):
|
||||||
|
protocol_version = "HTTP/1.0"
|
||||||
server_version = "StaplerUpgradeServer/" + PKG_VERSION
|
server_version = "StaplerUpgradeServer/" + PKG_VERSION
|
||||||
|
|
||||||
def do_HEAD(self) -> None:
|
def do_HEAD(self) -> None:
|
||||||
with self.handle_errors():
|
with self.handle_errors():
|
||||||
self._pre_log_request()
|
self._pre_log_request()
|
||||||
self.send_redirect(f"https://{self.host}{self.path}")
|
self.send_redirect(f"https://{self.host}{self.path}")
|
||||||
|
self.close_connection = True
|
||||||
|
|
||||||
def do_GET(self) -> None:
|
def do_GET(self) -> None:
|
||||||
with self.handle_errors():
|
with self.handle_errors():
|
||||||
if self.path.startswith(self.CERTBOT_CHALLENGE_PATH):
|
if self.path.startswith(self.CERTBOT_CHALLENGE_PATH):
|
||||||
super().do_GET()
|
super().do_GET()
|
||||||
|
self.close_connection = True
|
||||||
else:
|
else:
|
||||||
self.do_HEAD()
|
self.do_HEAD()
|
||||||
|
|||||||
+2
-4
@@ -29,7 +29,6 @@ class StaplerServer:
|
|||||||
"logger",
|
"logger",
|
||||||
"params",
|
"params",
|
||||||
"registry",
|
"registry",
|
||||||
"server",
|
|
||||||
"token_manager",
|
"token_manager",
|
||||||
]
|
]
|
||||||
|
|
||||||
@@ -41,7 +40,6 @@ class StaplerServer:
|
|||||||
self.token_manager: TokenManager = TokenManager(params, self.registry)
|
self.token_manager: TokenManager = TokenManager(params, self.registry)
|
||||||
self.data_dir: DataDir = DataDir(params.data_dir)
|
self.data_dir: DataDir = DataDir(params.data_dir)
|
||||||
self.default_host: str = params.host.split(":", maxsplit=2)[0]
|
self.default_host: str = params.host.split(":", maxsplit=2)[0]
|
||||||
self.server: http.server.ThreadingHTTPServer | None = None
|
|
||||||
|
|
||||||
def __get_all_hosts(self) -> list[str]:
|
def __get_all_hosts(self) -> list[str]:
|
||||||
return [self.default_host, *self.registry.get_hosts()]
|
return [self.default_host, *self.registry.get_hosts()]
|
||||||
@@ -131,7 +129,7 @@ class StaplerServer:
|
|||||||
for line in STAPLER_ASCII.split("\n"):
|
for line in STAPLER_ASCII.split("\n"):
|
||||||
self.logger.debug(line.ljust(36))
|
self.logger.debug(line.ljust(36))
|
||||||
self.__startup()
|
self.__startup()
|
||||||
self.server = self.__create_base_server()
|
base_server = self.__create_base_server()
|
||||||
upgrade_server = self.__start_upgrade_server() if self.params.https else None
|
upgrade_server = self.__start_upgrade_server() if self.params.https else None
|
||||||
self.logger.info(
|
self.logger.info(
|
||||||
"Server up and ready on %s://%s",
|
"Server up and ready on %s://%s",
|
||||||
@@ -140,7 +138,7 @@ class StaplerServer:
|
|||||||
)
|
)
|
||||||
self.__start_background_tasks()
|
self.__start_background_tasks()
|
||||||
with contextlib.suppress(KeyboardInterrupt):
|
with contextlib.suppress(KeyboardInterrupt):
|
||||||
self.server.serve_forever()
|
base_server.serve_forever()
|
||||||
self.logger.info("Shutting down...")
|
self.logger.info("Shutting down...")
|
||||||
if upgrade_server is not None:
|
if upgrade_server is not None:
|
||||||
upgrade_server.shutdown()
|
upgrade_server.shutdown()
|
||||||
|
|||||||
@@ -859,6 +859,7 @@ class TestRequestHandler(BaseHandlerTestCase):
|
|||||||
},
|
},
|
||||||
"allow_redirects": False,
|
"allow_redirects": False,
|
||||||
"timeout": 480,
|
"timeout": 480,
|
||||||
|
"stream": False,
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
self.expects_status_only(handler, 200, "OK"),
|
self.expects_status_only(handler, 200, "OK"),
|
||||||
@@ -903,6 +904,7 @@ class TestRequestHandler(BaseHandlerTestCase):
|
|||||||
},
|
},
|
||||||
"allow_redirects": False,
|
"allow_redirects": False,
|
||||||
"timeout": 480,
|
"timeout": 480,
|
||||||
|
"stream": False,
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
self.expects_status_only(handler, 200, "OK"),
|
self.expects_status_only(handler, 200, "OK"),
|
||||||
@@ -945,6 +947,7 @@ class TestRequestHandler(BaseHandlerTestCase):
|
|||||||
},
|
},
|
||||||
"allow_redirects": False,
|
"allow_redirects": False,
|
||||||
"timeout": 480,
|
"timeout": 480,
|
||||||
|
"stream": False,
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
self.expects_basic_body(handler, "hello", message="OK"),
|
self.expects_basic_body(handler, "hello", message="OK"),
|
||||||
@@ -979,6 +982,7 @@ class TestRequestHandler(BaseHandlerTestCase):
|
|||||||
},
|
},
|
||||||
"allow_redirects": False,
|
"allow_redirects": False,
|
||||||
"timeout": 480,
|
"timeout": 480,
|
||||||
|
"stream": False,
|
||||||
},
|
},
|
||||||
) as request_mock,
|
) as request_mock,
|
||||||
self.expects_status_only(
|
self.expects_status_only(
|
||||||
@@ -1022,6 +1026,7 @@ class TestRequestHandler(BaseHandlerTestCase):
|
|||||||
},
|
},
|
||||||
"allow_redirects": False,
|
"allow_redirects": False,
|
||||||
"timeout": 480,
|
"timeout": 480,
|
||||||
|
"stream": False,
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
self.expects_status_only(handler, 200, "OK"),
|
self.expects_status_only(handler, 200, "OK"),
|
||||||
@@ -1062,6 +1067,7 @@ class TestRequestHandler(BaseHandlerTestCase):
|
|||||||
},
|
},
|
||||||
"allow_redirects": False,
|
"allow_redirects": False,
|
||||||
"timeout": 480,
|
"timeout": 480,
|
||||||
|
"stream": False,
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
self.expects_status_only(handler, 200, "OK"),
|
self.expects_status_only(handler, 200, "OK"),
|
||||||
|
|||||||
Reference in New Issue
Block a user