58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
|
#!/usr/bin/python
|
||
|
|
|
||
|
|
from ansible.module_utils.basic import *
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
class GitlabRunner:
|
||
|
|
def __init__(self, name):
|
||
|
|
self.name = name
|
||
|
|
self.exist = self.name in open('/etc/gitlab-runner/config.toml').read()
|
||
|
|
|
||
|
|
def register(self, url, token, executor, options):
|
||
|
|
opts = []
|
||
|
|
for k, v in options.items():
|
||
|
|
if v is False:
|
||
|
|
opts.append('--%s=false' % k)
|
||
|
|
else:
|
||
|
|
opts.append('--%s' % k)
|
||
|
|
if v is not True:
|
||
|
|
opts.append(str(v))
|
||
|
|
|
||
|
|
subprocess.check_call(['gitlab-runner', 'register',
|
||
|
|
'--non-interactive',
|
||
|
|
'--name', self.name,
|
||
|
|
'--url', url,
|
||
|
|
'--registration-token', token,
|
||
|
|
'--executor', executor,
|
||
|
|
] + opts )
|
||
|
|
|
||
|
|
def unregister(self):
|
||
|
|
subprocess.check_call(['gitlab-runner', 'unregister', '--name', self.name])
|
||
|
|
|
||
|
|
def main():
|
||
|
|
changed = False
|
||
|
|
fields = {
|
||
|
|
'name': { 'type': 'str', 'required': True },
|
||
|
|
'url': { 'type': 'str', 'required': True },
|
||
|
|
'token': { 'type': 'str', 'required': True, 'no_log': True },
|
||
|
|
'executor': { 'type': 'str', 'required': True },
|
||
|
|
'options': { 'type': 'dict', 'default': {} },
|
||
|
|
'state': { 'type': 'str', 'default': 'present', 'choices': ['present', 'absent'] }
|
||
|
|
}
|
||
|
|
module = AnsibleModule(argument_spec=fields)
|
||
|
|
|
||
|
|
runner = GitlabRunner(module.params['name'])
|
||
|
|
if module.params['state'] == 'present':
|
||
|
|
if not runner.exist:
|
||
|
|
runner.register(module.params['url'], module.params['token'],
|
||
|
|
module.params['executor'], module.params['options'])
|
||
|
|
changed = True
|
||
|
|
elif runner.exist:
|
||
|
|
runner.unregister()
|
||
|
|
changed = True
|
||
|
|
|
||
|
|
module.exit_json(changed=changed)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|