#!/usr/bin/env python

from ansible.module_utils.basic import *


helper_cmd = {
    'trizen': ['env', 'LC_ALL=C', 'trizen', '-S', '--noconfirm', '--noedit', '--needed', '--aur'],
    'pacaur': ['env', 'LC_ALL=C', 'pacaur', '-S', '--noconfirm', '--noedit', '--needed', '--aur'],
    'yaourt': ['env', 'LC_ALL=C', 'yaourt', '-S', '--noconfirm', '--needed'],
    'yay': ['env', 'LC_ALL=C', 'yay', '-S', '--noconfirm'],
}


def upgrade(module, use):
    assert use in helper_cmd

    cmd = helper_cmd[use] + ['-u']

    rc, out, err = module.run_command(cmd, check_rc=True)

    module.exit_json(
        changed=not (out == '' or 'there is nothing to do' in out or 'No AUR updates found' in out),
        msg='upgraded system',
    )


def install_packages(module, packages, use):
    assert use in helper_cmd

    changed_iter = False
    for package in packages:
        cmd = helper_cmd[use] + [package]

        rc, out, err = module.run_command(cmd, check_rc=True)
        changed_iter = changed_iter or not (out == '' or '-- skipping' in out)

    module.exit_json(
        changed=changed_iter,
        msg='installed package',
    )


def main():
    module = AnsibleModule(
        argument_spec={
            'name': {
                'type': 'list',
            },
            'upgrade': {
                'default': False,
                'type': 'bool',
            },
            'use': {
                'default': 'auto',
                'choices': ['auto', 'pacaur', 'trizen', 'yaourt', 'yay'],
            },
        },
        required_one_of=[['name', 'upgrade']],
    )

    params = module.params

    if params['use'] == 'auto':
        for k in helper_cmd:
            if module.get_bin_path(k, False):
                helper = k
                break
    else:
        helper = params['use']

    if params['upgrade'] and params['name']:
        module.fail_json(msg="Upgrade and install must be requested separately")
    if params['upgrade']:
        upgrade(module, helper)
    else:
        install_packages(module, params['name'], helper)


if __name__ == '__main__':
    main()