57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
from ast import Assign, Constant, NodeTransformer, Subscript
|
|
from typing import TYPE_CHECKING, Any, List
|
|
|
|
from sanic.http.constants import HTTP
|
|
|
|
from .base import BaseScheme
|
|
|
|
|
|
if TYPE_CHECKING: # no cov
|
|
from sanic import Sanic
|
|
|
|
|
|
class AltSvcCheck(BaseScheme):
|
|
ident = "ALTSVC"
|
|
|
|
def visitors(self) -> List[NodeTransformer]:
|
|
return [RemoveAltSvc(self.app, self.app.state.verbosity)]
|
|
|
|
|
|
class RemoveAltSvc(NodeTransformer):
|
|
def __init__(self, app: Sanic, verbosity: int = 0) -> None:
|
|
self._app = app
|
|
self._verbosity = verbosity
|
|
self._versions = {
|
|
info.settings["version"] for info in app.state.server_info
|
|
}
|
|
|
|
def visit_Assign(self, node: Assign) -> Any:
|
|
if any(self._matches(target) for target in node.targets):
|
|
if self._should_remove():
|
|
return None
|
|
assert isinstance(node.value, Constant)
|
|
node.value.value = self.value()
|
|
return node
|
|
|
|
def _should_remove(self) -> bool:
|
|
return len(self._versions) == 1
|
|
|
|
@staticmethod
|
|
def _matches(node) -> bool:
|
|
return (
|
|
isinstance(node, Subscript)
|
|
and isinstance(node.slice, Constant)
|
|
and node.slice.value == "alt-svc"
|
|
)
|
|
|
|
def value(self):
|
|
values = []
|
|
for info in self._app.state.server_info:
|
|
port = info.settings["port"]
|
|
version = info.settings["version"]
|
|
if version is HTTP.VERSION_3:
|
|
values.append(f'h3=":{port}"')
|
|
return ", ".join(values)
|