82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
# Copyright 2019 Red Hat, Inc
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import os
|
|
|
|
from ansible.module_utils.basic import AnsibleModule
|
|
from ansible.module_utils import remarshal
|
|
|
|
|
|
def get_location(prefix, location):
|
|
# To support usage with both docker and podman, the buildset
|
|
# registry keeps "docker.io" entries un-namespaced.
|
|
if prefix == 'docker.io':
|
|
return location
|
|
else:
|
|
return location + '/' + prefix
|
|
|
|
|
|
def ansible_main():
|
|
module = AnsibleModule(
|
|
argument_spec=dict(
|
|
path=dict(required=True, type='path'),
|
|
buildset_registry=dict(type='raw'),
|
|
buildset_registry_alias=dict(type='str'),
|
|
namespaces=dict(type='raw'),
|
|
)
|
|
)
|
|
p = module.params
|
|
location = '%s:%s' % (p['buildset_registry_alias'],
|
|
p['buildset_registry']['port'])
|
|
|
|
if os.path.exists(p['path']):
|
|
with open(p['path'], 'rb') as f:
|
|
input_data = f.read()
|
|
data = remarshal.decode('toml', input_data, True)
|
|
else:
|
|
input_data = None
|
|
data = {}
|
|
|
|
unseen = set(p['namespaces'])
|
|
if 'registry' not in data:
|
|
data['registry'] = []
|
|
for reg in data['registry']:
|
|
if reg['prefix'] in unseen:
|
|
unseen.remove(reg['prefix'])
|
|
else:
|
|
continue
|
|
mirrors = reg.setdefault('mirror', [])
|
|
new_loc = dict(location=get_location(reg['prefix'], location))
|
|
if not mirrors or new_loc != mirrors[0]:
|
|
mirrors.insert(0, new_loc)
|
|
for prefix in unseen:
|
|
mirrors = [{'location': get_location(prefix, location)},
|
|
{'location': prefix}]
|
|
reg = {'prefix': prefix,
|
|
'location': prefix,
|
|
'mirror': mirrors}
|
|
data['registry'].append(reg)
|
|
|
|
output_data = remarshal.encode_toml(data, True)
|
|
changed = input_data is None or input_data != output_data
|
|
if changed:
|
|
with open(p['path'], 'wb') as f:
|
|
f.write(output_data.encode('utf8'))
|
|
|
|
module.exit_json(changed=changed, data=data)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
ansible_main()
|