syncthing-monitor/syncthing_monitor/__main__.py

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


def main():
    syncthing_mon = SyncthingMonitor(os.getenv('SYNCTHING_HOSTNAME'),
                                     os.getenv('SYNCTHING_ADDRESS'),
                                     os.getenv('SYNCTHING_DATA_PATH'),
                                     SYNCTHING_CONFIG_XML_PATH, "sync", SYNCTHING_GUI_PORT,
                                     "etcd", ETCD_PORT)
    syncthing_mon.start()


class SyncthingMonitor:
    def __init__(self, syncthing_public_host, syncthing_address, syncthing_data_path,
                 syncthing_config_xml_path, syncthing_hostname, syncthing_gui_port,
                 etcd_hostname, etcd_port):
        self.syncthing_hostname = syncthing_public_host
        self.syncthing_address = syncthing_address
        self.syncthing_data_path = syncthing_data_path
        self.my_device_id = None
        self.syncthing = None
        self.etcd = EtcdClient(etcd_hostname, etcd_port)
        self.syncthing_gui_port = syncthing_gui_port
        self.syncthing_hostname = syncthing_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_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_hostname, self.syncthing_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()