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

from typing import Iterable

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

from .Package import Package

async def run_rpm(args: list[str], sudo: bool=False): # export
    cmd = ['/usr/bin/rpm']
    cmd.extend(args)
    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()