import os import sys import time from retrying import retry import syncthing_monitor.config_xml as xml from .etcd_client import EtcdClient from .syncthing_client import SyncthingClient SHARED_FOLDER_ID = "data" SHARED_FOLDER_LABEL = "syncthing_monitor_data" SYNCTHING_CONFIG_XML_PATH = '/config/config.xml' SYNCTHING_GUI_PORT = 8384 ETCD_PORT = 2379 ETCD_KEY = '/syncthing_monitor/cluster_info' def main(): syncthing_mon = SyncthingMonitor(os.getenv('SYNCTHING_NODE_NAME', 'sync'), os.getenv('SYNCTHING_PUBLISH_ADDRESS', 'tcp4://sync'), os.getenv('SYNCTHING_DATA_PATH', '/data'), os.getenv('SYNCTHING_CONFIG_XML_PATH', SYNCTHING_CONFIG_XML_PATH), os.getenv('SYNCTHING_LOCAL_HOSTNAME', 'sync'), os.getenv('SYNCTHING_GUI_PORT', SYNCTHING_GUI_PORT), os.getenv('ETCD_HOSTNAME', 'etcd'), os.getenv('ETCD_PORT', ETCD_PORT), os.getenv('ETCD_KEY', ETCD_KEY)) syncthing_mon.start() class SyncthingMonitor: def __init__(self, syncthing_node_name, syncthing_publish_address, syncthing_data_path, syncthing_config_xml_path, syncthing_local_hostname, syncthing_gui_port, etcd_hostname, etcd_port, etcd_key): self.syncthing_node_name = syncthing_node_name self.syncthing_publish_address = syncthing_publish_address self.syncthing_data_path = syncthing_data_path self.my_device_id = None self.syncthing = None self.etcd = EtcdClient(etcd_hostname, etcd_port, etcd_key) self.syncthing_gui_port = syncthing_gui_port self.syncthing_local_hostname = syncthing_local_hostname self.syncthing_config_xml_path = syncthing_config_xml_path @retry def initialize_syncthing(self): api_key = xml.parse_api_key(self.syncthing_config_xml_path) print("Found API Key: {0}".format(api_key)) print("Configuring Syncthing to listen on 0.0.0.0...") xml.set_listen_ip_to_any(self.syncthing_config_xml_path, self.syncthing_config_xml_path, self.syncthing_gui_port) self.syncthing = SyncthingClient(api_key, self.syncthing_local_hostname, self.syncthing_gui_port) @retry def start(self): self.initialize_syncthing() self.my_device_id = self.syncthing.get_my_device_id() print("Found My Device ID: {0}".format(self.my_device_id)) self.syncthing.disable_announce_discovery_and_relay() self.syncthing.create_shared_folder(SHARED_FOLDER_ID, SHARED_FOLDER_LABEL, self.syncthing_data_path, [{'id': self.my_device_id}]) self.syncthing.sync_config() self.etcd.register_device_update_handler(self.update_devices) self.etcd.add_device_to_cluster(self.my_device_id, self.syncthing_node_name, self.syncthing_publish_address) self.loop() @staticmethod def loop(): while True: sys.stdout.flush() time.sleep(1) # noinspection PyUnusedLocal def update_devices(self, event): device_list = self.etcd.get_device_list() self.syncthing.add_devices(device_list) self.syncthing.create_shared_folder(SHARED_FOLDER_ID, SHARED_FOLDER_LABEL, self.syncthing_data_path, device_list) self.syncthing.sync_config() self.syncthing.print_config() if __name__ == "__main__": main()