diff --git a/apt b/apt new file mode 100755 index 00000000000..e3429875637 --- /dev/null +++ b/apt @@ -0,0 +1,152 @@ +#!/usr/bin/python -tt +# (c) 2012, Flowroute LLC +# Written by Matthew Williams +# Based on yum module written by Seth Vidal +# +# This module is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This software is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this software. If not, see . +# + + +import os +import sys +import apt +import shlex +import subprocess + + +try: + import json +except ImportError: + import simplejson as json + +def run_apt(command): + try: + cmd = subprocess.Popen(command, shell=True, + stdout=subprocess.PIPE, stderr=subprocess.PIPE) + out, err = cmd.communicate() + except (OSError, IOError), e: + rc = 1 + err = str(e) + out = '' + except: + rc = 1 + err = traceback.format_exc() + out = '' + + if out is None: + out = '' + if err is None: + err = '' + else: + rc = cmd.returncode + + return rc, out, err + +def ensure(state, pkgspec): + cache = apt.Cache() + cache.update() + cache.open(None) + try: + pkg = cache[pkgspec] + except: + msg = "No package matching '%s' is available" % pkgsepc + return {'changed': False, 'failed': True, 'msg': msg} + if state == 'installed': + #check if package is installed + if pkg.is_installed: + return {'changed': False, 'failed': False} + cmd = "apt-get -q -y install '%s'" % pkgspec + rc, out, err = run_apt(cmd) + #pkg.mark_install() + #cache.commit(apt.progress.base.OpProgress()) + + elif state == 'removed': + #check if package is installed + if not pkg.is_installed: + return {'changed': False, 'failed': False} + cmd = "apt-get -q -y remove '%s'" % pkgspec + rc, out, err = run_apt(cmd) + #pkg.mark_delete() + #cache.commit(apt.progress.base.OpProgress()) + else: + return {'failed': True, 'msg': "Unknown state: %s" % state } + + return {'changed': True, 'failed': False} + +def update(args): + #generic update routine + pass + +def remove_only(pkgspec): + # remove this pkg and only this pkg - fail if it will require more to remove + pass + +def main(): + # state=installed pkg=pkgspec + # state=removed pkg=pkgspec + + if len(sys.argv) == 1: + msg = "the apt module requires arguments (-a)" + return 1, msg + + argfile = sys.argv[1] + if not os.path.exists(argfile): + msg = "Argument file not found" + return 1, msg + + args = open(argfile, 'r').read() + items = shlex.split(args) + + if not len(items): + msg = "the apt module requires arguments (-a)" + return 1, msg + + # if nothing else changes - it fails + results = { 'changed':False, + 'failed':True, + 'results':'', + 'errors':'', + 'msg':"; ".join(items) } + params = {} + for x in items: + try: + (k, v) = x.split("=", 1) + except ValueError: + msg = "invalid arguments: %s" % args + return 1, msg + + params[k] = v + + if 'state' in params: + if 'pkg' not in params: + results['msg'] = "No pkg specified" + else: + state = params['state'] + pkgspec = params['pkg'] + + devnull = open(os.devnull, 'w') + results = ensure(state, pkgspec) + print json.dumps(results) + return 0, None + +if __name__ == "__main__": + rc, msg = main() + if rc != 0: # something went wrong emit the msg + print json.dumps({ + "failed" : bool(rc), + "msg" : msg + }) + sys.exit(rc) + +