80 lines
3.4 KiB
Python
80 lines
3.4 KiB
Python
|
from ansible.plugins.action import ActionBase
|
||
|
|
import re
|
||
|
|
import yaml
|
||
|
|
from yaml.loader import SafeLoader
|
||
|
|
|
||
|
|
class ActionModule(ActionBase):
|
||
|
|
def run(self, tmp=None, task_vars=None):
|
||
|
|
if task_vars is None:
|
||
|
|
task_vars = dict()
|
||
|
|
result = super(ActionModule, self).run(tmp, task_vars)
|
||
|
|
result['changed'] = False
|
||
|
|
result['failed'] = False
|
||
|
|
|
||
|
|
error_msgs = []
|
||
|
|
|
||
|
|
### Syncthing variables for non ansible hosts #########################
|
||
|
|
peers = {}
|
||
|
|
with open('syncthing_data.yaml') as f:
|
||
|
|
data = yaml.load(f, Loader=SafeLoader)
|
||
|
|
for peer in data:
|
||
|
|
peers[peer['name']] = peer
|
||
|
|
if not 'address' in peer.keys():
|
||
|
|
peer['address'] = 'dynamic'
|
||
|
|
|
||
|
|
### Syncthing host vars ###############################################
|
||
|
|
for hostname, hostvars in task_vars['hostvars'].items() :
|
||
|
|
if 'syncthing' in hostvars.keys():
|
||
|
|
syncthing = hostvars['syncthing']
|
||
|
|
peer = {
|
||
|
|
'address': 'dynamic',
|
||
|
|
'id': '0000000-0000000-0000000-0000000-0000000-0000000-0000000-0000000',
|
||
|
|
'shared': [],
|
||
|
|
}
|
||
|
|
if 'address' in syncthing.keys():
|
||
|
|
peer['address'] = syncthing['address']
|
||
|
|
for shared in syncthing['shared']:
|
||
|
|
peer['shared'].append({ 'name': shared['name'], 'path': shared['path'], 'peers': shared['peers']})
|
||
|
|
if 'syncthing' in hostvars['ansible_local']:
|
||
|
|
peer['id'] = hostvars['ansible_local']['syncthing']['id']
|
||
|
|
peers[hostname] = peer
|
||
|
|
|
||
|
|
### Compiling host configuration ######################################
|
||
|
|
config = {}
|
||
|
|
if task_vars['ansible_host'] in peers.keys():
|
||
|
|
myself = peers[task_vars['ansible_host']]
|
||
|
|
config = {
|
||
|
|
'config_path': "",
|
||
|
|
'folders_to_create': [],
|
||
|
|
'packages': [],
|
||
|
|
'peers': {},
|
||
|
|
'service': "syncthing",
|
||
|
|
'shared': myself['shared'],
|
||
|
|
'user_group': "syncthing",
|
||
|
|
}
|
||
|
|
if task_vars['ansible_distribution'] == 'FreeBSD':
|
||
|
|
config['config_path'] = "/usr/local/etc/syncthing/config.xml"
|
||
|
|
config['folders_to_create'] = ["/usr/local/etc/syncthing/", "/var/syncthing"]
|
||
|
|
config['packages'] = ["p5-libwww", "syncthing"]
|
||
|
|
elif task_vars['ansible_distribution'] == 'Gentoo':
|
||
|
|
config['config_path'] = "/var/lib/syncthing/.config/syncthing/config.xml"
|
||
|
|
config['folders_to_create'] = ["/var/lib/syncthing/.config/syncthing"]
|
||
|
|
config['packages'] = ["net-p2p/syncthing"]
|
||
|
|
else:
|
||
|
|
error_msgs.append(f"syncthing role does not support {task_vars['ansible_distribution']} hosts yet")
|
||
|
|
for shared in myself['shared']:
|
||
|
|
for peer in shared['peers']:
|
||
|
|
if not peer in config['peers'].keys():
|
||
|
|
config['peers'][peer] = { 'id': peers[peer]['id'], 'address': peers[peer]['address'] }
|
||
|
|
|
||
|
|
### Results compilation ##############################################
|
||
|
|
if error_msgs != []:
|
||
|
|
result['msg'] = ' ; '.join(error_msgs)
|
||
|
|
result['failed'] = True
|
||
|
|
|
||
|
|
result['ansible_facts'] = {
|
||
|
|
'syncthing_config': config,
|
||
|
|
}
|
||
|
|
|
||
|
|
return result
|