2023-01-22 00:16:43 +01:00
|
|
|
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',
|
2023-01-23 17:33:53 +01:00
|
|
|
'id': '',
|
2023-01-22 00:16:43 +01:00
|
|
|
'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']})
|
2023-01-23 17:33:53 +01:00
|
|
|
if 'ansible_local' in hostvars and 'syncthing' in hostvars['ansible_local']:
|
2023-01-22 00:16:43 +01:00
|
|
|
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 = {
|
|
|
|
'peers': {},
|
|
|
|
'shared': myself['shared'],
|
|
|
|
}
|
|
|
|
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
|