jw-pkg/src/python/jw/pkg/cmds/secrets/lib/DistroContext.py

# -*- coding: utf-8 -*-

from __future__ import annotations

import re

from pathlib import Path

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from typing import Iterable
    from ....lib.FileContext import FileContext

from ....lib.log import *
from ....lib.util import run_cmd

from .base import Attrs
from .FilesContext import FilesContext

class DistroContext(FilesContext):

    def __init__(self, distro: Distro) -> None:
        super().__init__(distro.ctx)
        self.__distro = distro

    async def match_files(self, pkg_names: Iterable[str], pattern: str) -> list[str]:
        ret: list[str] = []
        for pkg_name in pkg_names:
            for path in await self.__distro.pkg_files(pkg_name):
                if re.match(pattern, path):
                    ret.append(path)
        return ret

    async def list_template_files(self, pkg_names: Iterable[str]) -> list[str]:
        if not pkg_names:
            pkg_names = [p.name for p in await self.__distro.select()]
        return await self.match_files(pkg_names, pattern=r'.*\.jw-tmpl$')

    async def list_secret_paths(self, pkg_names: Iterable[str], ignore_missing: bool=False) -> list[str]:
        ret = []
        for tmpl in await self.list_template_files(pkg_names):
            path = str(Path(tmpl).with_suffix(".jw-secret"))
            if ignore_missing and not await self.ctx.file_exists(path):
                continue
            ret.append(path)
        return ret

    async def list_compilation_targets(self, pkg_names: Iterable[str], ignore_missing: bool=False) -> list[str]:
        ret = []
        for tmpl in await self.list_template_files(pkg_names):
            path = tmpl.removesuffix('.jw-tmpl')
            if ignore_missing and not await self.ctx.file_exists(path):
                continue
            ret.append(path)
        return ret

    async def remove_compilation_targets(self, pkg_names: Iterable[str]) -> list[str]:
        for path in await self.list_compilation_targets(pkg_names):
            try:
                self.ctx.stat(path)
                log(NOTICE, f'Removing {path}')
                await self.ctx.unlink(path)
            except FileNotFoundError as e:
                log(DEBUG, f'Compilation target {path} doesn\'t exist (ignored)')
                continue

    async def compile_template_files(self, pkg_names: Iterable[str], default_attrs: Attrs) -> list[str]:
        missing = 0
        for target in await self.list_compilation_targets(pkg_names):
            if not await self.compile_template_file(target, default_attrs):
                missing += 1
        if missing > 0:
            log(WARNING, f'{missing} missing secrets found. You might want to add them and run sudo {app.cmdline} again')