bindmanager/core/cluster_manager.py

import json
import threading
import urllib.request
import config
from core.file_manager import load_panel_settings, load_adblock_lists, load_custom_rules

def broadcast_to_cluster(action, domain):
    def task():
        settings = load_panel_settings()
        nodes = settings.get('cluster_nodes', [])
        for node in nodes:
            try:
                url = f"{node['url'].rstrip('/')}/api/v1/cluster/zone"
                headers = {"X-API-Key": node['api_key'], "Content-Type": "application/json"}
                payload = {"domain": domain}
                if action == 'add_slave':
                    req = urllib.request.Request(url, data=json.dumps(payload).encode('utf-8'), headers=headers, method='POST')
                elif action == 'delete_zone':
                    req = urllib.request.Request(f"{url}/{domain}", headers=headers, method='DELETE')
                urllib.request.urlopen(req, timeout=5, context=config.ssl_ctx)
                print(f"Cluster Sync OK: {node['name']} ({action} {domain})")
            except Exception as e:
                print(f"Cluster Sync FAIL: {node['name']} -> {e}")
    threading.Thread(target=task, daemon=True).start()

def broadcast_adblock_to_cluster(action):
    def task():
        s = load_panel_settings()
        lists = load_adblock_lists()
        rules = load_custom_rules()
        count = s.get('adblock_domains_count', 0)
        update_time = s.get('adblock_last_update', 'Never')
        
        for node in s.get('cluster_nodes', []):
            try:
                url = f"{node['url'].rstrip('/')}/api/v1/cluster/adblock"
                headers = {"X-API-Key": node['api_key'], "Content-Type": "application/json"}
                payload = {
                    "action": action, 
                    "lists": lists, 
                    "rules": rules, 
                    "count": count, 
                    "time": update_time
                }
                req = urllib.request.Request(url, data=json.dumps(payload).encode(), headers=headers, method='POST')
                urllib.request.urlopen(req, timeout=10, context=config.ssl_ctx)
                print(f"Cluster AdBlock Sync OK: {node['name']} ({action})")
            except Exception as e:
                print(f"Cluster AdBlock Sync FAIL: {node['name']} -> {e}")
    threading.Thread(target=task, daemon=True).start()