import typing as t
import yaml

from src.util.package import Package, PackageAlreadyInstalled, InvalidPackage
from src.util.install import Install
from src.util import log


def obtain_packages() -> t.List[Package]:
    with open("packages.yaml") as f:
        yaml_file = yaml.safe_load(f)

    pacman_packages = yaml_file["pacman"]
    aur_packages = yaml_file["aur"]
    git_packages = yaml_file["git"]

    packages = []
    for package in git_packages:
        packages.append(Package(package, git=True))
    for package in pacman_packages:
        packages.append(Package(package))
    for package in aur_packages:
        packages.append(Package(package, aur=True))

    return packages


def install_packages() -> None:
    packages = obtain_packages()
    Install.upgrade_pacman()
    for package in packages:
        try:
            log.action(f"Installing {package}")
            package.install()
        except PackageAlreadyInstalled:
            log.cancel(f"Package {package} is already installed.")
        except InvalidPackage as e:
            log.warning(e)