From 5dc3d4eb6669305547971326ee53ea2d79e06773 Mon Sep 17 00:00:00 2001
From: Toby <mail@tobias-schneider.io>
Date: Wed, 20 Sep 2023 01:20:08 +0200
Subject: [PATCH] update

---
 sync.py | 123 +++++++++++++++++++++++++++++++++++++++++++++-----------
 1 file changed, 100 insertions(+), 23 deletions(-)

diff --git a/sync.py b/sync.py
index 56d1986..f85fbe4 100755
--- a/sync.py
+++ b/sync.py
@@ -5,48 +5,100 @@ import paramiko
 import threading
 import yaml
 
-class task_thread(threading.Thread):
-    def __init__(self, host, port, user, keys):
+def string_to_float(data):
+    return float(data.strip().split(' ', 2)[0].strip())
+
+def parse_top_string(data):
+    rows = data.splitlines()
+
+    load = rows[0].split('load average: ', 2)[1]
+
+    cpu_data = rows[1].split(':', 2)[1].strip().split(',')
+    cpu_percent = 100.0 - float(cpu_data[3].strip().split(' ', 2)[0].strip()) - float(cpu_data[7].strip().split(' ', 2)[0].strip())
+    if (cpu_percent < 0):
+        cpu_percent = 0
+
+    ram_unit, ram_data = rows[2].split(':', 2)
+    ram_factor = 1
+    if ram_unit[0] == 'K':
+        ram_factor = 1024 * 1024
+    elif ram_unit[0] == 'M':
+        ram_factor = 1024
+    elif ram_unit[0] == 'G':
+        ram_factor = 1
+    ram_data = ram_data.strip().split(',')
+    ram_total = float(ram_data[0].strip().split(' ', 2)[0].strip()) / ram_factor
+    ram_free = float(ram_data[1].strip().split(' ', 2)[0].strip()) / ram_factor
+
+    return load, cpu_percent, ram_total, ram_free
+
+class Host(threading.Thread):
+    def __init__(self, host, port, user, keys, host_length, idx):
         threading.Thread.__init__(self)
         self.host = host
         self.port = port
         self.user = user
         self.keys = keys
+        self.host_length = host_length
+        self.idx = idx
+        self.status = 'none'
+        self.row = ''
+
+    def render_status(self):
+        if self.status == 'none':
+            return '[ .. ] '
+        elif self.status == 'ok':
+            return '[ \033[92mok\033[0m ] '
+        elif self.status == 'fail':
+            return '[\033[91mfail\033[0m] '
+
+    def render_row(self):
+        self.row = self.render_status() + self.host.ljust(self.host_length + 3) + self.user
+
     def run(self):
-        update_keys(self.host, self.port, self.user, self.keys)
+        try:
+            self.status = 'none'
+            self.render_row()
+            self.client = paramiko.SSHClient()
+            self.client.set_missing_host_key_policy(paramiko.MissingHostKeyPolicy())
+            self.client.connect(self.host, port = self.port, username = self.user, timeout = 1)
+            self.client.exec_command('echo "###\n# Warning this file has been generated and will be overwritten!\n###\n' + '\n'.join(self.keys) + '" > ~/.ssh/authorized_keys2')
+            self.status = 'ok'
+        except Exception as e:
+            #print(e)
+            self.status = 'fail'
+        self.render_row()
 
 def read_config():
     with open('config.yaml', 'r') as stream:
         return yaml.safe_load(stream)
 
-def update_keys(host, port, user, keys):
-    try:
-        client = paramiko.SSHClient()
-        client.set_missing_host_key_policy(paramiko.MissingHostKeyPolicy())
-        client.connect(host, username = user, port = port, timeout = 1)
-        client.exec_command('echo "###\n# Warning this file has been generated and will be overwritten!\n###\n' + '\n'.join(keys) + '" > ~/.ssh/authorized_keys2')
-        client.close()
-        print('✅ ' + user + '@' + host)
-    except Exception:
-        time.sleep(1)
-        print('❌ ' + user + '@' + host)
-
 def main():
     config = read_config()
+    hosts = []
+
+    host_length = 0
     for host in config['hosts']:
+        if len(host['host']) > host_length:
+            host_length = len(host['host'])
+
+    #print('Host'.center(7 + host_length + 3) + 'User')
+
+    config_hosts = config['hosts']
+    config_hosts.sort(key=lambda x: x['host'])
+
+    for i, host in enumerate(config_hosts):
         for user_name, user_data in host['users'].items():
             host_keys = []
             if 'groups' in user_data.keys():
                 for group in user_data['groups']:
                     if group not in config['groups'].keys():
-                        print('WARNING: Key-group "' + group + '" not found!')
                         continue
                     for key_name in config['groups'][group]:
                         host_keys.append(config['keys'][key_name])
             if 'keys' in user_data.keys():
                 for key_name in user_data['keys']:
                     if key_name not in config['keys'].keys():
-                        print('WARNING: Key "' + key_name + '" not found!')
                         continue
                     host_keys.append(config['keys'][key_name])
             host_keys = list(set(host_keys)) # Filter duplicates
@@ -54,12 +106,37 @@ def main():
                 continue
             if not 'port' in host:
                 host['port'] = 22
-            try:
-                thread = task_thread(host['host'], host['port'], user_name, host_keys)
-                thread.start()
-            except:
-                time.sleep(1)
-                print('❌ ' + user_name + '@' + host['host'])
+            hosts.append(Host(host['host'], host['port'] or 22, user_name, host_keys, host_length, i))
+
+    for host in hosts:
+        host.start()
+
+    end = False
+    first = True
+
+    while True:
+        screen = []
+        for host in hosts:
+            screen.append(host.row)
+
+        if first:
+            first = False
+        else:
+            print('\033[' + str(len(screen)) + 'F', end = '')
+
+        print('\n'.join(screen))
+
+        if end:
+            break
+
+        states = []
+        for host in hosts:
+            states.append(not host.is_alive())
+
+        if all(states):
+            end = True
+
+        time.sleep(0.1)
 
 if __name__ == '__main__':
     main()