--- /dev/null
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+# Copyright: (c) 2019, Marcin Krol <hawk@tld-linux.org>
+# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
+
+from __future__ import absolute_import, division, print_function
+__metaclass__ = type
+
+ANSIBLE_METADATA = {'metadata_version': '1.0',
+ 'status': ['preview'],
+ 'supported_by': 'community'}
+
+DOCUMENTATION = '''
+---
+module: poldek
+short_description: Manage packages with I(poldek)
+description:
+ - Manage packages with the I(poldek) package manager, which is used by TLD Linux and PLD Linux
+version_added: "1.0"
+author:
+ - Marcin Krol <hawk@tld-linux.org>
+options:
+ name:
+ description:
+ - Name or list of names of the package(s) or file(s) to install, upgrade, or remove.
+ Can't be used in combination with C(upgrade).
+ aliases: [ package, pkg ]
+
+ state:
+ description:
+ - Desired state of the package.
+ default: present
+ choices: [ absent, latest, present ]
+
+ extra_args:
+ description:
+ - Additional option to pass to poldek when enforcing C(state).
+ default:
+
+ update_cache:
+ description:
+ - Whether or not to refresh the master package lists.
+ - This can be run as part of a package installation or as a separate step.
+ default: no
+ type: bool
+ aliases: [ update-cache ]
+
+ update_cache_extra_args:
+ description:
+ - Additional option to pass to poldek when enforcing C(update_cache).
+ default:
+
+ upgrade:
+ description:
+ - Whether or not to upgrade the whole system.
+ Can't be used in combination with C(name).
+ default: no
+ type: bool
+
+ upgrade_extra_args:
+ description:
+ - Additional option to pass to poldek when enforcing C(upgrade).
+ default:
+'''
+
+RETURN = '''
+packages:
+ description: a list of packages that have been changed
+ returned: when upgrade is set to yes
+ type: list
+ sample: [ package, other-package ]
+'''
+
+EXAMPLES = '''
+- name: Install packages foo and bar
+ poldek:
+ name:
+ - foo
+ - bar
+ state: present
+
+- name: Update package cache and upgrade package foo
+ poldek:
+ name: foo
+ state: latest
+ update_cache: yes
+
+- name: Remove packages foo and bar
+ poldek:
+ name:
+ - foo
+ - bar
+ state: absent
+
+- name: Update package cache only
+ poldek:
+ update_cache: yes
+
+- name: Update package cache and upgrade all packages
+ poldek:
+ upgrade: yes
+ update_cache: yes
+'''
+
+import re
+
+from ansible.module_utils.basic import AnsibleModule
+
+
+def query_package(module, poldek_path, name):
+ """Query the package status in both the local system and the repository. Returns a boolean to indicate if the package is installed, a second
+ boolean to indicate if the package is up-to-date. Note: indexes must be up to date to ensure that poldek knows latest version of package.
+ """
+ lcmd = "%s -q --shcmd='ls -n --installed %s'" % (poldek_path, name)
+ lrc, lstdout, lstderr = module.run_command(lcmd, check_rc=False)
+ rcmd = "%s -q --shcmd='ls -n --installed --upgradeable %s'" % (poldek_path, name)
+ rrc, rstdout, rstderr = module.run_command(rcmd, check_rc=False)
+ if lrc != 0 or rrc != 0:
+ return False, False
+
+ pkg_latest = False
+ if rstdout == "":
+ return True, True
+ else:
+ return True, False
+
+
+def update_package_db(module, poldek_path):
+ if module.params['force']:
+ module.params["update_cache_extra_args"] += " --upa"
+
+ cmd = "%s --noask --up %s" % (poldek_path, module.params["update_cache_extra_args"])
+ rc, stdout, stderr = module.run_command(cmd, check_rc=False)
+
+ if rc == 0:
+ return True
+ else:
+ module.fail_json(msg="error updating package database")
+
+
+def upgrade(module, poldek_path):
+ cmdupgrade = "%s -v --noask --upgrade-dist %s" % (poldek_path, module.params["upgrade_extra_args"])
+ cmdupgradeable = "%s -q --shcmd='ls -n --installed --upgradeable' %s" % (poldek_path, module.params["upgrade_extra_args"])
+ rc, stdout, stderr = module.run_command(cmdupgradeable, check_rc=False)
+
+ packages = stdout.split('\n')[:-1]
+ if stdout == "":
+ module.exit_json(changed=False, msg='Nothing to upgrade', packages=packages, stdout=stdout, stderr=stderr, rc=rc)
+ else:
+ if module.check_mode:
+ module.exit_json(changed=True, msg="%s package(s) would be upgraded" % (len(packages)), packages=packages, stdout=stdout, stderr=stderr, rc=rc)
+ rc, stdout, stderr = module.run_command(cmdupgrade, check_rc=False)
+ if rc == 0:
+ module.exit_json(changed=True, msg="%s package(s) upgraded" % (len(packages)), packages=packages, stdout=stdout, stderr=stderr, rc=rc)
+ else:
+ module.fail_json(msg="Error while upgrading packages", stdout=stdout, stderr=stderr, rc=rc)
+
+
+def remove_packages(module, poldek_path, packages):
+ if module.params["force"]:
+ module.params["extra_args"] += " --force --nodeps"
+
+ remove_c = 0
+ out = ""
+ err = ""
+
+ for package in packages:
+ installed, updated = query_package(module, poldek_path, package)
+ if not installed:
+ continue
+
+ cmd = "%s -v --noask --erase %s %s" % (poldek_path, module.params["extra_args"], package)
+ rc, stdout, stderr = module.run_command(cmd, check_rc=False)
+
+ out = out + stdout
+ err = err + stderr
+
+ if rc != 0:
+ module.fail_json(msg="failed to remove %s" % (package), stdout=out, stderr=err, rc=rc)
+
+ remove_c += 1
+
+ if remove_c > 0:
+ module.exit_json(changed=True, msg="removed %s package(s)" % (remove_c), packages=packages, stdout=out, stderr=err, rc=0)
+
+ module.exit_json(changed=False, msg="package(s) already absent", packages=packages, stdout=out, stderr=err, rc=0)
+
+
+def install_packages(module, poldek_path, state, packages):
+ if module.params["force"]:
+ module.params["extra_args"] += " --force --nodeps"
+
+ install_c = 0
+ out = ""
+ err = ""
+
+ for package in packages:
+ installed, updated = query_package(module, poldek_path, package)
+ if installed and (state == 'present' or (state == 'latest' and updated)):
+ continue
+
+ cmd = "%s -v --noask --upgrade %s %s" % (poldek_path, module.params["extra_args"], package)
+ rc, stdout, stderr = module.run_command(cmd, check_rc=False)
+
+ out = out + stdout
+ err = err + stderr
+
+ if rc != 0:
+ module.fail_json(msg="failed to install %s" % (package), stdout=out, stderr=err, rc=rc)
+
+ install_c += 1
+
+ if install_c > 0:
+ module.exit_json(changed=True, msg="installed %s package(s)" % (install_c), packages=packages, stdout=out, stderr=err, rc=0)
+
+ module.exit_json(changed=False, msg="package(s) already installed", packages=packages, stdout=out, stderr=err, rc=0)
+
+
+def check_packages(module, poldek_path, packages, state):
+ would_be_changed = []
+
+ for package in packages:
+ installed, updated = query_package(module, poldek_path, package)
+ if ((state in ["present", "latest"] and not installed) or
+ (state == "absent" and installed) or
+ (state == "latest" and not updated)):
+ would_be_changed.append(package)
+
+ if state == "absent":
+ state = "removed"
+ if state == "present" or state == "latest":
+ state = "installed"
+
+ if would_be_changed:
+ module.exit_json(changed=True, msg="%s package(s) would be %s" % (
+ len(would_be_changed), state))
+ else:
+ module.exit_json(changed=False, msg="package(s) already %s" % state)
+
+
+def main():
+ module = AnsibleModule(
+ argument_spec=dict(
+ name=dict(type='list', aliases=['pkg', 'package']),
+ state=dict(type='str', default='present', choices=['present', 'installed', 'latest', 'absent', 'removed']),
+ extra_args=dict(type='str', default=''),
+ upgrade=dict(type='bool', default=False),
+ upgrade_extra_args=dict(type='str', default=''),
+ update_cache=dict(type='bool', default=False, aliases=['update-cache']),
+ update_cache_extra_args=dict(type='str', default=''),
+ force=dict(type='bool', default=False),
+ ),
+ required_one_of=[['name', 'update_cache', 'upgrade']],
+ mutually_exclusive=[['name', 'upgrade']],
+ supports_check_mode=True,
+ )
+
+ poldek_path = module.get_bin_path('poldek', True)
+
+ p = module.params
+
+ if p['state'] in ['present', 'installed']:
+ p['state'] = 'present'
+ elif p['state'] in ['absent', 'removed']:
+ p['state'] = 'absent'
+
+ if p["update_cache"] and not module.check_mode:
+ update_package_db(module, poldek_path)
+ if not (p['name'] or p['upgrade']):
+ module.exit_json(changed=True, msg='Updated package database')
+
+ if p['update_cache'] and module.check_mode and not (p['name'] or p['upgrade']):
+ module.exit_json(changed=True, msg='Would have updated package database')
+
+ if p['upgrade']:
+ upgrade(module, poldek_path)
+
+ if p['name']:
+ pkgs = p['name']
+
+ if module.check_mode:
+ check_packages(module, poldek_path, pkgs, p['state'])
+
+ if p['state'] in ['present', 'latest']:
+ install_packages(module, poldek_path, p['state'], pkgs)
+ elif p['state'] == 'absent':
+ remove_packages(module, poldek_path, pkgs)
+ else:
+ module.exit_json(changed=False, msg="No package specified to work on")
+
+
+if __name__ == "__main__":
+ main()