ansible-aur/aur.py

247 lines
6.9 KiB
Python
Raw Normal View History

2018-05-23 23:01:47 +03:00
#!/usr/bin/python
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
2017-12-23 20:27:42 +03:00
from ansible.module_utils.basic import *
from ansible.module_utils.urls import open_url
2017-12-31 20:12:28 +03:00
import json
import tarfile
import os
import os.path
import tempfile
2018-04-14 13:53:14 +03:00
2018-05-23 23:01:47 +03:00
DOCUMENTATION = '''
---
2018-05-23 23:25:05 +03:00
module: aur
short_description: Manage packages from the AUR
2018-05-23 23:01:47 +03:00
description:
2018-05-23 23:07:53 +03:00
- Manage packages from the Arch User Repository (AUR)
2018-05-23 23:01:47 +03:00
author:
- Kewl <xrjy@nygb.rh.bet(rot13)>
options:
name:
description:
- Name or list of names of the package(s) to install or upgrade.
upgrade:
description:
- Whether or not to upgrade whole system.
type: bool
default: no
use:
description:
- The helper to use, 'auto' uses the first known helper found and makepkg as a fallback.
default: auto
choices: [ auto, aurman, pacaur, trizen, pikaur, yay, makepkg ]
skip_installed:
description:
- Skip operations if the package is present.
type: bool
default: no
skip_pgp_check:
description:
- Skip verification of PGP signatures.
This is useful when installing packages on a host without GnuPG (properly) configured.
Only valid with makepkg.
type: bool
default: no
notes:
- When used with a `loop:` each package will be processed individually,
it is much more efficient to pass the list directly to the `name` option.
'''
RETURN = '''
msg:
description: action that has been taken
helper:
the helper that was actually used
'''
EXAMPLES = '''
- name: Install trizen using makepkg, skip if trizen is already installed
aur: name=trizen use=makepkg skip_installed=true
become: yes
become_user: aur_builder
'''
2018-03-17 11:15:06 +03:00
def_lang = ['env', 'LC_ALL=C']
2017-12-31 20:12:28 +03:00
use_cmd = {
2018-05-08 09:20:00 +03:00
'aurman': ['aurman', '-S', '--noconfirm', '--noedit', '--needed'],
'pacaur': ['pacaur', '-S', '--noconfirm', '--noedit', '--needed'],
'trizen': ['trizen', '-S', '--noconfirm', '--noedit', '--needed'],
2018-03-17 11:15:06 +03:00
'pikaur': ['pikaur', '-S', '--noconfirm', '--noedit', '--needed'],
'yay': ['yay', '-S', '--noconfirm'],
'makepkg': ['makepkg', '--syncdeps', '--install', '--noconfirm', '--needed']
2017-12-23 20:27:42 +03:00
}
2018-05-08 09:20:00 +03:00
# optional: aurman, pacaur, trizen have a --aur option, do things only for aur
2017-12-23 20:27:42 +03:00
2017-12-31 20:12:28 +03:00
def package_installed(module, package):
2018-05-16 22:30:58 +03:00
"""
Determine if the package is already installed
"""
2017-12-31 20:12:28 +03:00
rc, _, _ = module.run_command(['pacman', '-Q', package], check_rc=False)
return rc == 0
2018-05-16 21:39:57 +03:00
def check_packages(module, packages):
"""
Inform the user what would change if the module were run
"""
would_be_changed = []
for package in packages:
installed = package_installed(module, package)
if not installed:
would_be_changed.append(package)
if would_be_changed:
status = True
if (len(packages) > 1):
message = '%s package(s) would be installed' % str(len(would_be_changed))
else:
message = 'package would be installed'
else:
status = False
if (len(packages) > 1):
message = 'all packages are already installed'
else:
message = 'package is already installed'
module.exit_json(changed=status, msg=message)
def install_with_makepkg(module, package):
2018-05-16 22:30:58 +03:00
"""
Install the specified package with makepkg
"""
f = open_url('https://aur.archlinux.org/rpc/?v=5&type=info&arg={}'.format(package))
2017-12-31 20:12:28 +03:00
result = json.loads(f.read().decode('utf8'))
if result['resultcount'] != 1:
return (1, '', 'package not found')
result = result['results'][0]
f = open_url('https://aur.archlinux.org/{}'.format(result['URLPath']))
2017-12-31 20:12:28 +03:00
current_path = os.getcwd()
with tempfile.TemporaryDirectory() as tmpdir:
os.chdir(tmpdir)
tar_file = '{}.tar.gz'.format(result['Name'])
with open(tar_file, 'wb') as out:
out.write(f.read())
tar = tarfile.open(tar_file)
tar.extractall()
tar.close()
os.chdir(format(result['Name']))
if module.params['skip_pgp_check']:
use_cmd['makepkg'].append('--skippgpcheck')
rc, out, err = module.run_command(use_cmd['makepkg'], check_rc=True)
2017-12-31 20:12:28 +03:00
os.chdir(current_path)
return (rc, out, err)
2017-12-23 20:27:42 +03:00
2017-12-31 20:12:28 +03:00
def upgrade(module, use):
2018-05-16 22:30:58 +03:00
"""
Upgrade the whole system
"""
2017-12-31 20:12:28 +03:00
assert use in use_cmd
2018-03-17 11:15:06 +03:00
rc, out, err = module.run_command(def_lang + use_cmd[use] + ['-u'], check_rc=True)
2017-12-23 20:27:42 +03:00
module.exit_json(
2018-04-14 13:37:11 +03:00
changed=not (out == '' or 'nothing to do' in out or 'No AUR updates found' in out),
2017-12-23 20:27:42 +03:00
msg='upgraded system',
helper=use,
2017-12-23 20:27:42 +03:00
)
2017-12-31 20:12:28 +03:00
def install_packages(module, packages, use, skip_installed):
2018-05-16 22:30:58 +03:00
"""
Install the specified packages
"""
2017-12-31 20:12:28 +03:00
assert use in use_cmd
2017-12-23 20:27:42 +03:00
2017-12-23 23:05:07 +03:00
changed_iter = False
2017-12-23 20:27:42 +03:00
2017-12-31 20:12:28 +03:00
for package in packages:
if skip_installed:
if package_installed(module, package):
rc = 0
continue
if use == 'makepkg':
rc, out, err = install_with_makepkg(module, package)
2017-12-31 20:12:28 +03:00
else:
2018-03-17 11:15:06 +03:00
rc, out, err = module.run_command(def_lang + use_cmd[use] + [package], check_rc=True)
2018-04-14 13:37:11 +03:00
changed_iter = changed_iter or not (out == '' or '-- skipping' in out or 'nothing to do' in out)
2017-12-23 20:27:42 +03:00
if changed_iter:
message = 'installed package(s)'
else:
message = 'package(s) already installed'
2017-12-23 20:27:42 +03:00
module.exit_json(
2017-12-23 23:05:07 +03:00
changed=changed_iter,
msg=message if not rc else err,
helper=use,
2017-12-31 20:12:28 +03:00
rc=rc,
2017-12-23 20:27:42 +03:00
)
def main():
module = AnsibleModule(
argument_spec={
'name': {
2017-12-23 23:05:07 +03:00
'type': 'list',
2017-12-23 20:27:42 +03:00
},
'upgrade': {
'default': False,
'type': 'bool',
},
2017-12-24 10:21:11 +03:00
'use': {
'default': 'auto',
'choices': ['auto', 'aurman', 'pacaur', 'trizen', 'pikaur', 'yaourt', 'yay', 'makepkg'],
2017-12-31 20:12:28 +03:00
},
'skip_installed': {
2018-05-08 09:20:00 +03:00
'default': False,
2017-12-31 20:12:28 +03:00
'type': 'bool',
2017-12-23 20:27:42 +03:00
},
'skip_pgp_check': {
'default': False,
'type': 'bool',
},
2017-12-23 20:27:42 +03:00
},
required_one_of=[['name', 'upgrade']],
2018-05-16 21:39:57 +03:00
supports_check_mode=True
2017-12-23 20:27:42 +03:00
)
params = module.params
2018-05-16 21:39:57 +03:00
if module.check_mode:
check_packages(module, params['name'])
2017-12-24 10:21:11 +03:00
if params['use'] == 'auto':
use = 'makepkg'
2018-05-08 09:20:00 +03:00
# auto: select the first helper for which the bin is found
2017-12-31 20:12:28 +03:00
for k in use_cmd:
2017-12-24 10:21:11 +03:00
if module.get_bin_path(k, False):
2017-12-31 20:12:28 +03:00
use = k
2017-12-24 10:21:11 +03:00
break
else:
2017-12-31 20:12:28 +03:00
use = params['use']
2017-12-24 10:21:11 +03:00
if params['upgrade'] and (params['name'] or params['skip_installed'] or use == 'makepkg'):
2017-12-31 20:12:28 +03:00
module.fail_json(msg="Upgrade cannot be used with this option.")
2017-12-23 20:27:42 +03:00
else:
2017-12-31 20:12:28 +03:00
if params['upgrade']:
upgrade(module, use)
else:
install_packages(module, params['name'], use, params['skip_installed'])
2017-12-23 20:27:42 +03:00
if __name__ == '__main__':
main()