mirror of
https://github.com/release-engineering/repo-autoindex.git
synced 2025-02-24 14:12:54 +00:00
137388d475
startElement signature changed in the .pyi stubs for XML classes, triggering a mypy complaint here. Suppress it as there is no actual error here.
256 lines
8.1 KiB
Python
256 lines
8.1 KiB
Python
import datetime
|
|
import logging
|
|
import os
|
|
from collections.abc import AsyncGenerator, Generator, Iterable, Mapping
|
|
from dataclasses import dataclass
|
|
from typing import BinaryIO, Optional, Type, Any
|
|
from xml.dom.minidom import Element
|
|
from xml.dom.pulldom import END_ELEMENT, START_ELEMENT
|
|
from xml.sax.handler import ContentHandler
|
|
|
|
from defusedxml import pulldom, sax # type: ignore
|
|
|
|
from .base import (
|
|
ICON_PACKAGE,
|
|
GeneratedIndex,
|
|
IOFetcher,
|
|
IndexEntry,
|
|
Repo,
|
|
ContentError,
|
|
)
|
|
from .template import TemplateContext
|
|
from .tree import treeify
|
|
|
|
LOG = logging.getLogger("autoindex")
|
|
|
|
|
|
def assert_repodata_ok(condition: Any, msg: str):
|
|
if not condition:
|
|
raise ContentError(msg)
|
|
|
|
|
|
def get_tag(elem: Element, name: str) -> Element:
|
|
elems: list[Element] = elem.getElementsByTagName(name)
|
|
assert_repodata_ok(len(elems) == 1, f"expected exactly one {name} tag")
|
|
return elems[0]
|
|
|
|
|
|
def get_text_tag(elem: Element, name: str) -> str:
|
|
tagnode = get_tag(elem, name)
|
|
child = tagnode.firstChild
|
|
assert_repodata_ok(child, f"missing text {name} tag")
|
|
return str(child.toxml()) # type: ignore
|
|
|
|
|
|
@dataclass
|
|
class Package:
|
|
href: str
|
|
time: str
|
|
size: int
|
|
|
|
@property
|
|
def index_entry(self) -> IndexEntry:
|
|
return IndexEntry(
|
|
icon=ICON_PACKAGE,
|
|
href=self.href,
|
|
text=os.path.basename(self.href),
|
|
time=datetime.datetime.utcfromtimestamp(float(self.time)).isoformat(),
|
|
size=str(self.size),
|
|
)
|
|
|
|
|
|
def pulldom_elements(
|
|
xml_str: str, path_matcher, attr_matcher=lambda _: True
|
|
) -> Generator[Element, None, None]:
|
|
stream = pulldom.parseString(xml_str)
|
|
current_path = []
|
|
for event, node in stream:
|
|
if event == START_ELEMENT:
|
|
current_path.append(node.tagName)
|
|
|
|
if path_matcher(current_path) and attr_matcher(node.attributes):
|
|
stream.expandNode(node)
|
|
yield node
|
|
|
|
# expandNode makes it so that we don't get END_ELEMENT any more
|
|
# for this node, so pop here.
|
|
current_path.pop()
|
|
|
|
elif event == END_ELEMENT:
|
|
LOG.debug("leaving element %s", current_path)
|
|
current_path.pop()
|
|
|
|
|
|
class PackagesParser(ContentHandler):
|
|
# SAX-integrated parser to load Package instances from a primary XML.
|
|
#
|
|
# We use this rather than pulldom because the pulldom memory usage while
|
|
# parsing a large primary XML seems unreasonably high.
|
|
#
|
|
def __init__(self) -> None:
|
|
self.current_path: list[str] = []
|
|
self.current_package: Optional[Package] = None
|
|
self.packages: list[Package] = []
|
|
|
|
def parse(self, xml: BinaryIO) -> Iterable[Package]:
|
|
self.packages = []
|
|
|
|
# Parse the XML document; this will invoke our start/end element handlers
|
|
# which in turn will populate self.packages
|
|
sax.parse(xml, self)
|
|
|
|
return self.packages
|
|
|
|
def startElement(self, name: str, attrs: Mapping[str, Any]): # type: ignore
|
|
self.current_path.append(name)
|
|
LOG.debug("entering element %s", self.current_path)
|
|
|
|
if self.current_path == ["metadata", "package"] and attrs.get("type") == "rpm":
|
|
self.current_package = Package("<unknown package>", "", 0)
|
|
elif self.current_path == ["metadata", "package", "location"]:
|
|
assert self.current_package
|
|
self.current_package.href = attrs["href"]
|
|
elif self.current_path == ["metadata", "package", "time"]:
|
|
assert self.current_package
|
|
self.current_package.time = attrs["file"]
|
|
elif self.current_path == ["metadata", "package", "size"]:
|
|
assert self.current_package
|
|
self.current_package.size = attrs["package"]
|
|
|
|
def endElement(self, _name: str):
|
|
LOG.debug("leaving element %s", self.current_path)
|
|
|
|
if self.current_path == ["metadata", "package"]:
|
|
assert self.current_package
|
|
self.packages.append(self.current_package)
|
|
self.current_package = None
|
|
|
|
self.current_path.pop()
|
|
|
|
|
|
class YumRepo(Repo):
|
|
async def render_index(
|
|
self, index_href_suffix: str
|
|
) -> AsyncGenerator[GeneratedIndex, None]:
|
|
LOG.debug("repomd.xml: %s", self.entry_point_content)
|
|
|
|
entries = []
|
|
entries.extend(await self._repodata_entries())
|
|
entries.extend(await self._package_entries())
|
|
|
|
for page in self.__render_entries(entries, index_href_suffix):
|
|
yield page
|
|
|
|
async def _repodata_entries(self) -> list[IndexEntry]:
|
|
out = []
|
|
|
|
# There's always an entry for repomd.xml itself...
|
|
size = len(self.entry_point_content)
|
|
time = "-"
|
|
revision_nodes = list(
|
|
pulldom_elements(
|
|
self.entry_point_content,
|
|
path_matcher=lambda p: p == ["repomd", "revision"],
|
|
)
|
|
)
|
|
if len(revision_nodes) == 1:
|
|
timestamp_node = revision_nodes[0].firstChild
|
|
assert_repodata_ok(timestamp_node, "missing timestamp node")
|
|
time = datetime.datetime.utcfromtimestamp(
|
|
int(timestamp_node.toxml()) # type: ignore
|
|
).isoformat()
|
|
|
|
out.append(
|
|
IndexEntry(
|
|
href="repodata/repomd.xml",
|
|
text="repomd.xml",
|
|
time=time,
|
|
size=str(size),
|
|
)
|
|
)
|
|
|
|
data_nodes = list(
|
|
pulldom_elements(
|
|
self.entry_point_content,
|
|
path_matcher=lambda p: p == ["repomd", "data"],
|
|
attr_matcher=lambda attrs: attrs.get("type"),
|
|
)
|
|
)
|
|
data_nodes.sort(key=lambda node: str(node.attributes["type"].value))
|
|
|
|
for node in data_nodes:
|
|
href = get_tag(node, "location").attributes["href"].value
|
|
basename = os.path.basename(href)
|
|
timestamp = get_text_tag(node, "timestamp")
|
|
time = datetime.datetime.utcfromtimestamp(float(timestamp)).isoformat()
|
|
size = int(get_text_tag(node, "size"))
|
|
|
|
out.append(
|
|
IndexEntry(
|
|
href=href,
|
|
text=basename,
|
|
time=time,
|
|
size=str(size),
|
|
)
|
|
)
|
|
|
|
return out
|
|
|
|
async def _package_entries(self) -> list[IndexEntry]:
|
|
primary_nodes = list(
|
|
pulldom_elements(
|
|
self.entry_point_content,
|
|
path_matcher=lambda p: p == ["repomd", "data"],
|
|
attr_matcher=lambda attrs: attrs.get("type")
|
|
and attrs["type"].value == "primary",
|
|
)
|
|
)
|
|
assert len(primary_nodes) == 1
|
|
primary_node = primary_nodes[0]
|
|
|
|
location = get_tag(primary_node, "location")
|
|
href = location.attributes["href"].value
|
|
|
|
primary_url = "/".join([self.base_url, href])
|
|
primary_xml = await self.fetcher(primary_url)
|
|
|
|
assert_repodata_ok(primary_xml, f"missing primary XML at {primary_url}")
|
|
|
|
return sorted(
|
|
[p.index_entry for p in self.__packages_from_primary(primary_xml)], # type: ignore
|
|
key=lambda e: e.text,
|
|
)
|
|
|
|
def __packages_from_primary(self, primary_xml: BinaryIO) -> Iterable[Package]:
|
|
return PackagesParser().parse(primary_xml)
|
|
|
|
def __render_entries(
|
|
self,
|
|
entries: Iterable[IndexEntry],
|
|
index_href_suffix: str,
|
|
) -> Generator[GeneratedIndex, None, None]:
|
|
ctx = TemplateContext()
|
|
nodes = [treeify(entries, index_href_suffix=index_href_suffix)]
|
|
while nodes:
|
|
node = nodes.pop()
|
|
yield GeneratedIndex(
|
|
content=ctx.render_index(index_entries=node.entries),
|
|
relative_dir=node.relative_dir,
|
|
)
|
|
nodes.extend(node.children)
|
|
|
|
@classmethod
|
|
async def probe(
|
|
cls: Type["YumRepo"],
|
|
fetcher: IOFetcher,
|
|
url: str,
|
|
) -> Optional["YumRepo"]:
|
|
repomd_xml_url = f"{url}/repodata/repomd.xml"
|
|
repomd_xml = await fetcher(repomd_xml_url)
|
|
|
|
if repomd_xml is None:
|
|
# not yum repo
|
|
return None
|
|
|
|
# it is a yum repo
|
|
return cls(url, repomd_xml.read().decode(), fetcher)
|