diff --git a/library/service b/library/service index e8ede0678f6..406dc7c4bf0 100755 --- a/library/service +++ b/library/service @@ -66,20 +66,29 @@ examples: ''' import platform +import os +import re SERVICE = None CHKCONFIG = None INITCTL = None +INITSCRIPT = None +RCCONF = None + PS_OPTIONS = 'auxww' -def _find_binaries(m): +def _find_binaries(m,name): # list of possible paths for service/chkconfig binaries # with the most probable first global SERVICE global CHKCONFIG global INITCTL + global INITSCRIPT + global RCCONF paths = ['/sbin', '/usr/sbin', '/bin', '/usr/bin'] binaries = [ 'service', 'chkconfig', 'update-rc.d', 'initctl', 'systemctl'] + initpaths = [ '/etc/init.d','/etc/rc.d','/usr/local/etc/rc.d' ] + rcpaths = [ '/etc/rc.conf','/usr/local/etc/rc.conf' ] location = dict() for binary in binaries: @@ -95,11 +104,20 @@ def _find_binaries(m): elif location.get('update-rc.d', None): CHKCONFIG = location['update-rc.d'] else: + for rcfile in rcpaths: + if os.path.isfile(rcfile): + RCCONF = rcfile + if not CHKCONFIG and not RCCONF: m.fail_json(msg='unable to find chkconfig or update-rc.d binary') if location.get('service', None): SERVICE = location['service'] else: - m.fail_json(msg='unable to find service binary') + for rcdir in initpaths: + initscript = "%s/%s" % (rcdir,name) + if os.path.isfile(initscript): + INITSCRIPT = initscript + if not SERVICE and not INITSCRIPT: + m.fail_json(msg='unable to find service binary nor initscript') if location.get('initctl', None): INITCTL = location['initctl'] else: @@ -198,19 +216,39 @@ def _do_enable(name, enable): if enable: on_off = "on" enable_disable = "enable" + rc = "YES" else: on_off = "off" enable_disable = "disable" - - if CHKCONFIG.endswith("update-rc.d"): - args = (CHKCONFIG, name, enable_disable) - elif CHKCONFIG.endswith("systemctl"): - args = (CHKCONFIG, enable_disable, name + ".service") + rc = "NO" + + if RCCONF: + entry = "%s_enable" % name + full_entry = '%s="%s"' % (entry,rc) + rc = open(RCCONF,"r+") + rctext = rc.read() + if re.search("^%s" % full_entry,rctext,re.M) is None: + if re.search("^%s" % entry,rctext,re.M) is None: + rctext += "\n%s" % full_entry + else: + rctext = re.sub("^%s.*" % entry,full_entry,rctext,1,re.M) + rc.truncate(0) + rc.seek(0) + rc.write(rctext) + rc.close() + + rc=0 + stderr=stdout='' else: - args = (CHKCONFIG, name, on_off) + if CHKCONFIG.endswith("update-rc.d"): + args = (CHKCONFIG, name, enable_disable) + elif CHKCONFIG.endswith("systemctl"): + args = (CHKCONFIG, enable_disable, name + ".service") + else: + args = (CHKCONFIG, name, on_off) - if enable is not None: - rc, stdout, stderr = _run("%s %s %s" % args) + if enable is not None: + rc, stdout, stderr = _run("%s %s %s" % args) return rc, stdout, stderr @@ -237,7 +275,7 @@ def main(): # =========================================== # find binaries locations on minion - _find_binaries(module) + _find_binaries(module,name) # =========================================== # get service status @@ -250,6 +288,12 @@ def main(): err = '' out = '' + # set command to run + if SERVICE: + svc_cmd = "%s %s" % (SERVICE, name) + elif INITSCRIPT: + svc_cmd = "%s" % INITSCRIPT + if module.params['enabled']: rc_enable, out_enable, err_enable = _do_enable(name, enable) rc += rc_enable @@ -275,15 +319,24 @@ def main(): # run change commands if we need to if changed: + if platform.system() == 'FreeBSD': + start = "onestart" + stop = "onestop" + reload = "onereload" + else: + start = "start" + stop = "stop" + reload = "reload" + if state in ['started', 'running']: - rc_state, stdout, stderr = _run("%s %s start" % (SERVICE, name)) + rc_state, stdout, stderr = _run("%s %s" % (svc_cmd,start)) elif state == 'stopped': - rc_state, stdout, stderr = _run("%s %s stop" % (SERVICE, name)) + rc_state, stdout, stderr = _run("%s %s" % (svc_cmd,stop)) elif state == 'reloaded': - rc_state, stdout, stderr = _run("%s %s reload" % (SERVICE, name)) + rc_state, stdout, stderr = _run("%s %s" % (svc_cmd,reload)) elif state == 'restarted': - rc1, stdout1, stderr1 = _run("%s %s stop" % (SERVICE, name)) - rc2, stdout2, stderr2 = _run("%s %s start" % (SERVICE, name)) + rc1, stdout1, stderr1 = _run("%s %s" % (svc_cmd,stop)) + rc2, stdout2, stderr2 = _run("%s %s" % (svc_cmd,start)) if rc1 != 0 and rc2 == 0: rc_state = rc + rc2 stdout = stdout2 @@ -305,7 +358,8 @@ def main(): result['enabled'] = module.params['enabled'] if state: result['state'] = state - rc, stdout, stderr = _run("%s %s status" % (SERVICE, name)) + + rc, stdout, stderr = _run("%s status" % (svc_cmd)) module.exit_json(**result) # this is magic, see lib/ansible/module_common.py