64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
import logging
|
|
import typing
|
|
|
|
from . import data_dir, page
|
|
|
|
if typing.TYPE_CHECKING:
|
|
from . import params
|
|
|
|
|
|
class Registry:
|
|
HOST_FILE = ".host"
|
|
TOKEN_FILE = ".token" # noqa: S105
|
|
|
|
def __init__(self, params: params.Parameters) -> None:
|
|
self.logger = logging.getLogger(self.__class__.__name__)
|
|
self.pages: dict[str, page.Page] = {}
|
|
self.data_dir = data_dir.DataDir(params.data_dir)
|
|
|
|
def load_pages(self) -> None:
|
|
self.pages = {}
|
|
for path in self.data_dir.list_paths():
|
|
self.add(path)
|
|
|
|
def get_hosts(self) -> list[str]:
|
|
return [p.host for p in self.pages.values() if p.host is not None]
|
|
|
|
def add(self, path: str) -> None:
|
|
self.pages[path] = page.Page(
|
|
path,
|
|
self.data_dir.has_index(path),
|
|
self.data_dir.get_file(path, self.HOST_FILE),
|
|
self.data_dir.get_file(path, self.TOKEN_FILE),
|
|
)
|
|
self.logger.info("Updated %s", self.pages[path])
|
|
|
|
def set_host(self, path: str, host: str) -> None:
|
|
if path in self.pages and self.pages[path].host != host:
|
|
self.data_dir.set_file(path, self.HOST_FILE, host)
|
|
self.pages[path].host = host
|
|
self.logger.debug("Updated %s", self.pages[path])
|
|
|
|
def set_token_hash(self, path: str, token_hash: str) -> None:
|
|
if path in self.pages and self.pages[path].token_hash != token_hash:
|
|
self.data_dir.set_file(path, self.TOKEN_FILE, token_hash, 0o600)
|
|
self.pages[path].token_hash = token_hash
|
|
self.logger.debug("Updated %s", self.pages[path])
|
|
|
|
def remove(self, path: str) -> None:
|
|
if path in self.pages:
|
|
page = self.pages[path]
|
|
del self.pages[path]
|
|
self.logger.info("Removed %s", page)
|
|
|
|
def get_from_path(self, path: str) -> page.Page | None:
|
|
if path in self.pages:
|
|
return self.pages[path]
|
|
return None
|
|
|
|
def get_from_host(self, host: str) -> page.Page | None:
|
|
for p in self.pages.values():
|
|
if p.host == host:
|
|
return p
|
|
return None
|