jw-pkg/src/python/jw/pkg/cmds/projects/BaseCmdPkgRelations.py

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

import re
from argparse import Namespace, ArgumentParser
from enum import Enum, auto

from ...lib.log import *
from ..Cmd import Cmd
from ..CmdProjects import CmdProjects

class BaseCmdPkgRelations(Cmd):

    class Syntax(Enum):
        semver = auto()
        debian = auto()
        names_only = auto()

    def pkg_relations_list(
            self,
            rel_type: str,
            flavours: list[str],
            modules: list[str],
            subsections: list[str]|None=None,
            delimiter: str=' ',
            no_subpackages: bool=False,
            dont_strip_revision: bool=False,
            expand_semver_revision_range: bool=False,
            syntax: Syntax=Syntax.semver,
            recursive: bool=False,
            dont_expand_version_macros: bool=False,
            ignore: set[str] = set(),
        ) -> list[str]:

        if subsections is None:
            subsecs = self.app.os_cascade()
            subsecs.append('jw')
        else:
            subsecs = subsections

        expand_semver_revision_range = expand_semver_revision_range
        if syntax == self.Syntax.debian:
            expand_semver_revision_range = True

        log(DEBUG, f'flavours="{", ".join(flavours)}", subsecs="{", ".join(subsecs)}", "ignore="{", ".join(ignore)}"')

        version_pattern = re.compile("[0-9-.]*")
        r: list[str] = []
        for flavour in flavours:
            for s in subsecs:
                section = 'pkg.' + rel_type + '.' + s
                visited = set()
                sec_modules = modules.copy()
                while len(sec_modules):
                    m = sec_modules.pop(0)
                    if m in visited or m in ignore:
                        continue
                    visited.add(m)
                    value = self.app.get_value(m, section, flavour)
                    if not value:
                        continue
                    deps = value.split(',')
                    for spec in deps:
                        dep = re.split('([=><]+)', spec)
                        if syntax == self.Syntax.names_only:
                            dep = dep[:1]
                        dep = list(map(str.strip, dep))
                        dep_name = re.sub('-dev$|-devel$|-run$', '', dep[0])
                        if dep_name in ignore or dep[0] in ignore:
                            continue
                        if no_subpackages:
                            dep[0] = dep_name
                        for i, item in enumerate(dep):
                            dep[i] = item.strip()
                        if s == 'jw':
                            if recursive and not dep_name in visited and not dep_name in sec_modules:
                                sec_modules.append(dep_name)
                            if len(dep) == 3:
                                if dont_expand_version_macros and dep_name in sec_modules:
                                    version = dep[2]
                                else:
                                    version = self.app.get_value(dep_name, 'version', '')
                                if dep[2] == 'VERSION':
                                    if dont_strip_revision:
                                        dep[2] = version
                                    else:
                                        dep[2] = version.split('-')[0]
                                elif dep[2] == 'VERSION-REVISION':
                                    dep[2] = version
                                elif version_pattern.match(dep[2]):
                                    # dep[2] = dep[2]
                                    pass
                                else:
                                    raise Exception("Unknown version specifier in " + spec)
                        if not expand_semver_revision_range:
                            expanded_deps = [dep]
                        else:
                            expanded_deps = []
                            if expand_semver_revision_range and len(dep) == 3:
                                semver = re.split(r'[.-]', version)
                                if len(semver) != 4:
                                    expanded_deps = [dep]
                                else:
                                    release = int(semver[2])
                                    major_minor = f'{semver[0]}.{semver[1]}'
                                    match dep[1]:
                                        case '>' | '>=':
                                            expanded_deps.append([dep[0], dep[1], dep[2]])
                                            expanded_deps.append([dep[0], '<', f'{major_minor}.{release + 1}'])
                                        case '<' | '<=':
                                            expanded_deps.append([dep[0], dep[1], dep[2]])
                                        case '=':
                                            expanded_deps.append([dep[0], '>=', f'{major_minor}.{release}'])
                                            expanded_deps.append([dep[0], '<', f'{major_minor}.{release + 1}'])
                                        case _:
                                            raise NotImplementedError(f'Expanding SemVer range "{dep[0]} {dep[1]} {dep[3]}" is not yet implemented')
                        for expanded_dep in expanded_deps:
                            match syntax:
                                case self.Syntax.semver:
                                    pass
                                case self.Syntax.debian:
                                    match expanded_dep[1]:
                                        case '<':
                                            expanded_dep[1] = '<<'
                                        case '>':
                                            expanded_dep[1] = '>>'
                                case '_':
                                   raise NotImplementedError(f'Unknown dependency syntax "{syntax}" for dependency "{dep[0]} {dep[1]} {dep[3]}"')
                            dep_str = ' '.join(expanded_dep)
                            if not dep_str in r:
                                log(DEBUG, f'Appending dependency "{dep_str}"')
                                r.append(dep_str)
        return r

    def pkg_relations(self, rel_type: str, args: Namespace) -> str:

        return args.delimiter.join(
            self.pkg_relations_list(
                rel_type=rel_type,
                flavours = args.flavours.split(','),
                modules = args.modules,
                subsections = None if args.subsections is None else args.subsections.split(','),
                no_subpackages = args.no_subpackages,
                dont_strip_revision = args.dont_strip_revision,
                expand_semver_revision_range = args.expand_semver_revision_range,
                syntax = self.Syntax[args.syntax.replace('-', '_')],
                recursive = args.recursive,
                dont_expand_version_macros = args.dont_expand_version_macros,
                ignore = set(args.ignore.split(',')),
            )
        )

    def print_pkg_relations(self, rel_type: str, args: Namespace) -> None:
        print(self.pkg_relations(rel_type, args))

    def __init__(self, parent: CmdProjects, relation: str, help: str) -> None:
        super().__init__(parent, 'pkg-' + relation, help=help)
        self.relation = relation

    def add_arguments(self, parser: ArgumentParser) -> None:
        super().add_arguments(parser)
        parser.add_argument('-S', '--subsections', nargs='?', default=None, help='Subsections to consider, comma-separated')
        parser.add_argument('-d', '--delimiter', nargs='?', default=', ', help='Output words delimiter')
        parser.add_argument('flavours', help='Dependency flavours (run, build, devel, release)')
        parser.add_argument('modules', nargs='*', help='Modules')
        parser.add_argument('-p', '--no-subpackages', action='store_true',
            default=False, help='Cut -run and -devel from package names')
        parser.add_argument('--dont-strip-revision', action='store_true',
            default=False, help='Always treat VERSION macro as VERSION-REVISION')
        parser.add_argument('--expand-semver-revision-range', action='store_true',
            default=False, help='Always treat =VERSION macro as >= VERSION-0 and < (VERSION+1)-0')
        parser.add_argument('--syntax', choices=['semver', 'debian', 'names-only'],
            default='semver', help='Output syntax')
        parser.add_argument('--recursive', action='store_true',
            default=False, help='Find dependencies recursively')
        parser.add_argument('--dont-expand-version-macros', action='store_true',
            default=False, help='Don\'t expand VERSION and REVISION macros')
        parser.add_argument('--ignore', nargs='?', default='', help='Packages that '
            'should be ignored together with their dependencies')

    async def _run(self, args: Namespace) -> None:
        return self.print_pkg_relations(self.relation, args)