# pylint: disable=C0111,E0611,E1101,E0001,C0301
from argparse import ArgumentParser
import sh


RELEASE = sh.rpm('-E', '%fedora').strip()


def install(*packages):
    sh.dnf.install(*packages, _fg=True)


def upgrade():
    sh.dnf.upgrade(refresh=True)


def fusion():
    install(
        '--nogpgcheck',
        'https://download1.rpmfusion.org/free/fedora/rpmfusion-free-release-%s.noarch.rpm' % RELEASE,
        'https://download1.rpmfusion.org/nonfree/fedora/rpmfusion-nonfree-release-%s.noarch.rpm' % RELEASE,
    )


def rfr():
    install(
        '--nogpgcheck',
        'http://mirror.yandex.ru/fedora/russianfedora/russianfedora/free/fedora/russianfedora-free-release-stable.noarch.rpm',
        'http://mirror.yandex.ru/fedora/russianfedora/russianfedora/nonfree/fedora/russianfedora-nonfree-release-stable.noarch.rpm',
        'http://mirror.yandex.ru/fedora/russianfedora/russianfedora/fixes/fedora/russianfedora-fixes-release-stable.noarch.rpm'
    )


def repos():
    fusion()
    rfr()


def video():
    install('xorg-x11-xinit',
            'xorg-x11-xauth',
            'xorg-x11-utils',
            'xorg-x11-server-Xorg',
            'xorg-x11-drv-libinput',
            'plymouth-system-theme',
            'mesa-dri-drivers',
            'glx-utils')

    devices = sh.lspci('-mm')
    if 'AMD/ATI' in devices:
        install('xorg-x11-drv-ati')
    if 'VMware' in devices:
        install('xorg-x11-drv-vmware')
    else:
        print("No video devices found.")


def desktop():
    install(
        'i3',
        'i3status',
        'i3lock'
        'dmenu',
        'feh',
        'lightdm'
    )
    install(
        'telegram-desktop',
        'chromium-browser'
    )


P = ArgumentParser()
P.add_argument('upgrade',
               action=upgrade,
               help="Advanced Upgrade")
P.add_argument('fusion',
               action=fusion,
               help="Install RPM Fusion repos")
P.add_argument('rfr',
               action=rfr,
               help="Install Russian Fedora repos")
P.add_argument('repos',
               action=repos,
               help="Install Russian Fedora and RPM Fusion repos")
P.add_argument('video',
               action=video,
               help="Install video packages")
P.add_argument('desktop',
               action=desktop,
               help="Install i3 desktop")