#!/usr/bin/env python3
"""ACB Processor Service - Prefect Managed"""
import subprocess
import sys
import time
from prefect import flow, task, get_run_logger

SERVICE_SCRIPT = "/mnt/dolphinng5_predict/prod/acb_processor_service.py"

@task
def check_acb() -> bool:
    """Check if ACB data is fresh in Hz."""
    try:
        import hazelcast
        import json
        from datetime import datetime, timezone
        
        client = hazelcast.HazelcastClient(
            cluster_name="dolphin",
            cluster_members=["127.0.0.1:5701"],
        )
        features = client.get_map('DOLPHIN_FEATURES').blocking()
        data = features.get('acb_boost')
        client.shutdown()
        
        if data:
            print("ACB data present")
            return True
        return False
    except Exception as e:
        print(f"ACB check error: {e}")
        return False

@flow(name="acb-processor-service")
def acb_service_flow():
    """Manage ACB Processor Service."""
    logger = get_run_logger()
    logger.info("Starting ACB Processor Service...")
    
    # Start service
    proc = subprocess.Popen(
        [sys.executable, SERVICE_SCRIPT],
        stdout=subprocess.PIPE,
        stderr=subprocess.STDOUT,
        universal_newlines=True
    )
    
    logger.info(f"ACB Service started (PID: {proc.pid})")
    
    try:
        while True:
            time.sleep(60)  # Check every minute
            if proc.poll() is not None:
                logger.error("ACB Service died, restarting...")
                proc = subprocess.Popen(
                    [sys.executable, SERVICE_SCRIPT],
                    stdout=subprocess.PIPE,
                    stderr=subprocess.STDOUT,
                    universal_newlines=True
                )
            else:
                healthy = check_acb()
                if healthy:
                    logger.info("✅ ACB healthy")
                else:
                    logger.warning("⚠️ ACB data not in Hz yet")
    except KeyboardInterrupt:
        logger.info("Stopping ACB Service...")
        proc.terminate()
        proc.wait()

if __name__ == "__main__":
    acb_service_flow()