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

from typing import Iterable

from ....lib.util import run_cmd, run_sudo

from .Package import Package

async def run_rpm(*args, sudo=False): # export
    cmd = ['/usr/bin/rpm']
    cmd.extend(args)
    # FIXME: usage of run_xxx() (unpacked vs regular list is highly inconsistent)
    if sudo:
        return await run_sudo(cmd)
    return await run_cmd(*cmd)

async def all_installed_packages() -> Iterable[Package]: # export
    ret: list[Package] = []
    opts: list[str] = []
    query_tags: list[str] = []
    query_tags.append('Name')
    query_tags.append('Vendor')
    query_tags.append('Packager')
    query_tags.append('URL')
    opts.append('--queryformat')
    tag_str = '|'.join([f'%{{{tag}}}' for tag in query_tags]) + r'\n'
    opts.append(tag_str)
    package_list_str, stderr = await run_rpm('-qa', *opts, sudo=False)
    for package_str in package_list_str.splitlines():
        tags = package_str.split('|')
        ret.append(Package(name=tags[0], vendor=tags[1], packager=tags[2], url=tags[3]))
    return ret

async def list_files(pkg: str) -> list[str]: # export
    opts: list[str] = []
    file_list_str, stderr = await run_rpm('-ql', pkg, *opts, sudo=False)
    return file_list_str.splitlines()