#!/usr/bin/env python3
"""
K1RL QUANT - Real-Time Monitoring Dashboard
HuggingFace Spaces Edition
Displays live metrics, logs, and system status

FIXES APPLIED:
  BUG-1: avn_accuracy regex 'Avg? Accuracy' never matched 'AVN Accuracy:'
          + operator precedence broke the not-yet-set guard
  BUG-2: get_service_status() checked 'RUNNING' in the whole output blob
          instead of per-service line → every service showed ACTIVE even
          when crashed
  BUG-3: /logs/<service> only validated against LOG_FILES keys ('quasar',
          'features', 'rewards') but dashboard fetched '/logs/quasar_engine'
          → always 400, silently fell back to /api/logs/quasar
"""

from flask import Flask, jsonify, Response, session, request, send_from_directory

# Accuracy tracking module (optional — won't crash if missing)
try:
    import accuracy_api
    ACCURACY_AVAILABLE = True
except ImportError:
    ACCURACY_AVAILABLE = False
    print("⚠️  accuracy_api not found — accuracy endpoints disabled")

from flask_cors import CORS
import subprocess
import json
import time
import os
import re
from datetime import datetime, timedelta
from collections import deque
import psutil
import uuid

app = Flask(__name__)
CORS(app)

# ⚠️ IMPORTANT: Secret key for sessions (required for visitor tracking)
app.secret_key = 'k1rl-quant-2026-' + str(uuid.uuid4())[:8]

# ✅ HuggingFace Spaces compatible paths
LOG_FILES = {
    'quasar':        '/home/user/app/logs/quasar_engine.log',
    'quasar_engine': '/home/user/app/logs/quasar_engine.log',  # FIX BUG-3: alias so /logs/quasar_engine resolves correctly
    'features':      '/home/user/app/logs/features.log',
    'rewards':       '/home/user/app/logs/rewards.log'
}

CHECKPOINT_DIR = '/home/user/app/checkpoints'

# In-memory storage
metrics_history = deque(maxlen=100)
log_buffer = deque(maxlen=500)

# ============================================================================
# VISITOR TRACKING SYSTEM (HF Spaces Optimized)
# ============================================================================

active_visitors = {}   # {session_id: {'last_seen': datetime, 'ip': str}}
visitor_stats = {
    'total_unique': 0,
    'total_pageviews': 0,
    'first_visit': datetime.now().isoformat()
}

VISITOR_DATA_FILE = '/home/user/app/logs/visitor_data.json'


def load_visitor_data():
    """Load visitor statistics from file on startup"""
    global visitor_stats
    if os.path.exists(VISITOR_DATA_FILE):
        try:
            with open(VISITOR_DATA_FILE, 'r') as f:
                visitor_stats = json.load(f)
            print(f"📊 Loaded visitor stats: {visitor_stats['total_unique']} total unique visitors")
        except Exception as e:
            print(f"⚠️  Could not load visitor data: {e}")


def save_visitor_data():
    """Save visitor statistics to file"""
    try:
        os.makedirs(os.path.dirname(VISITOR_DATA_FILE), exist_ok=True)
        with open(VISITOR_DATA_FILE, 'w') as f:
            json.dump(visitor_stats, f, indent=2)
    except Exception as e:
        print(f"⚠️  Could not save visitor data: {e}")


def cleanup_inactive_visitors():
    """Remove visitors inactive for more than 5 minutes"""
    global active_visitors
    now = datetime.now()
    inactive_threshold = timedelta(minutes=5)

    to_remove = [vid for vid, data in active_visitors.items()
                 if now - data['last_seen'] > inactive_threshold]

    for vid in to_remove:
        del active_visitors[vid]

    if to_remove:
        print(f"🧹 Cleaned up {len(to_remove)} inactive visitors")

    return len(to_remove)


def track_visitor():
    """Track visitor activity and return visitor ID"""
    global active_visitors, visitor_stats

    cleanup_inactive_visitors()

    if 'visitor_id' not in session:
        session['visitor_id'] = str(uuid.uuid4())
        visitor_stats['total_unique'] += 1
        save_visitor_data()
        visitor_ip = request.headers.get('X-Forwarded-For', request.remote_addr)
        print(f"👤 New visitor #{visitor_stats['total_unique']}: {visitor_ip}")

    visitor_id = session['visitor_id']
    active_visitors[visitor_id] = {
        'last_seen': datetime.now(),
        'ip': request.headers.get('X-Forwarded-For', request.remote_addr)
    }

    visitor_stats['total_pageviews'] += 1

    if visitor_stats['total_pageviews'] % 50 == 0:
        save_visitor_data()

    return visitor_id


@app.before_request
def before_request():
    """Track visitors on every request"""
    if not request.path.startswith('/api/') and not request.path.startswith('/static/'):
        try:
            track_visitor()
        except Exception as e:
            print(f"⚠️  Error tracking visitor: {e}")


# ============================================================================
# METRIC EXTRACTION
# ============================================================================

def extract_latest_metrics():
    """Extract latest metrics from quasar log"""
    try:
        if not os.path.exists(LOG_FILES['quasar']):
            print(f"⚠️  Log file not found: {LOG_FILES['quasar']}")
            return None

        result = subprocess.run(
            ['tail', '-600', LOG_FILES['quasar']],  # v3.0.1: raised from 200 — worst-case AVN gap is 511 lines
            capture_output=True,
            text=True,
            timeout=5
        )

        lines = result.stdout.split('\n')

        metrics = {
            'training_steps':   None,
            'actor_loss':       None,
            'critic_loss':      None,
            'avn_loss':         None,
            'avn_accuracy':     None,
            'buffer_size':      None,
            'matched_rewards':  None,
            'unmatched_rewards': None,
            'duplicates':       None,
            'timestamp':        datetime.now().isoformat()
        }

        for line in reversed(lines):

            # Training steps
            if 'avn_training_steps:' in line and not metrics['training_steps']:
                match = re.search(r'avn_training_steps:\s*(\d+)', line)
                if match:
                    metrics['training_steps'] = int(match.group(1))

            # Actor loss
            if 'Actor Loss:' in line and not metrics['actor_loss']:
                match = re.search(r'Actor Loss:\s*([-\d.]+)', line)
                if match:
                    metrics['actor_loss'] = float(match.group(1))

            # Critic loss
            # ── FIX BUG-4a (2026-04-19): regex must include '-' in the char class
            #     so negative critic_loss values parse correctly.  The Actor
            #     Loss regex on line 193 already had [-\d.]+; this one was left
            #     as [\d.]+ and would silently drop negative values.
            if 'Critic Loss:' in line and not metrics['critic_loss']:
                match = re.search(r'Critic Loss:\s*([-\d.]+)', line)
                if match:
                    metrics['critic_loss'] = float(match.group(1))

            # AVN loss
            # ── FIX BUG-4b (2026-04-19): same issue as BUG-4a.  Regex now
            #     matches negative values.  Aligns with log_metrics_reader.py.
            if 'Avg Loss:' in line and not metrics['avn_loss']:
                match = re.search(r'Avg Loss:\s*([-\d.]+)', line)
                if match:
                    metrics['avn_loss'] = float(match.group(1))

            # ── FIX BUG-1a: parenthesise the OR so the `and not` guard applies
            #                 to BOTH branches, not just the second one.
            # ── FIX BUG-1b: regex now matches both 'AVN Accuracy:' and
            #                 'Avg Accuracy:' using a non-capturing group.
            #                 Old pattern  → r'Avg? Accuracy:...' matched
            #                   'Avg Accuracy' and 'Av Accuracy' but NEVER
            #                   'AVN Accuracy' (completely different string).
            if ('AVN Accuracy:' in line or 'Avg Accuracy:' in line) and not metrics['avn_accuracy']:
                match = re.search(r'(?:AVN|Avg) Accuracy:\s*([\d.]+)%?', line)
                if match:
                    metrics['avn_accuracy'] = float(match.group(1))

            # Buffer size
            if 'buffer_size' in line and not metrics['buffer_size']:
                match = re.search(r"buffer_size['\"]?\s*:\s*(\d+)", line)
                if match:
                    metrics['buffer_size'] = int(match.group(1))

            # Rewards
            if 'avn_rewards:' in line:
                matched   = re.search(r'matched=(\d+)', line)
                unmatched = re.search(r'unmatched=(\d+)', line)
                duplicate = re.search(r'duplicate=(\d+)', line)

                if matched and not metrics['matched_rewards']:
                    metrics['matched_rewards'] = int(matched.group(1))
                if unmatched and not metrics['unmatched_rewards']:
                    metrics['unmatched_rewards'] = int(unmatched.group(1))
                if duplicate and not metrics['duplicates']:
                    metrics['duplicates'] = int(duplicate.group(1))

        return metrics

    except Exception as e:
        print(f"Metric extraction error: {e}")
        return None


# ── FIX BUG-2: service status now parsed per line, not on the whole blob ───
def get_service_status():
    """Get supervisor service status — per-line parsing to avoid false positives"""
    services = {}

    # Maps supervisord program name → friendly API key
    service_map = {
        'quasar_engine':  'quasar_engine',
        'features':       'features',
        'rewards':        'rewards',
    }

    try:
        result = subprocess.run(
            ['supervisorctl', 'status'],
            capture_output=True, text=True, timeout=5
        )

        if result.returncode != 0 or not result.stdout.strip():
            raise RuntimeError(f"supervisorctl error: {result.stderr.strip()}")

        # ── FIXED: iterate line by line so each service is evaluated
        #           against only its OWN status token, not the entire output.
        for line in result.stdout.splitlines():
            parts = line.split()
            if not parts:
                continue

            program_name = parts[0]
            status_token = parts[1] if len(parts) > 1 else 'UNKNOWN'

            friendly = service_map.get(program_name)
            if friendly is None:
                continue   # not a service we track — skip

            if status_token == 'RUNNING':
                services[friendly] = 'active'
            elif status_token in ('STOPPED', 'FATAL', 'EXITED', 'BACKOFF'):
                services[friendly] = 'inactive'
            else:
                services[friendly] = 'unknown'

    except Exception as ex:
        print(f"⚠️  supervisorctl error: {ex}")
        # Fallback: mark everything unknown rather than falsely green
        for friendly in ('quasar_engine', 'features', 'rewards'):
            if friendly not in services:
                services[friendly] = 'unknown'

    return services


def get_system_resources():
    """Get system resource usage"""
    try:
        cpu_percent    = psutil.cpu_percent(interval=1)
        memory         = psutil.virtual_memory()
        disk           = psutil.disk_usage('/')

        quasar_memory  = 0
        process_count  = 0
        for proc in psutil.process_iter(['name', 'memory_info', 'cmdline']):
            try:
                cmdline = ' '.join(proc.info.get('cmdline', []) or [])
                if 'quasar' in cmdline.lower() or 'quasar' in (proc.info.get('name', '') or '').lower():
                    quasar_memory += proc.info['memory_info'].rss / (1024 ** 3)
                    process_count += 1
            except Exception:
                pass

        return {
            'cpu_percent':       round(cpu_percent, 1),
            'memory_percent':    round(memory.percent, 1),
            'memory_used_gb':    round(memory.used    / (1024 ** 3), 2),
            'memory_total_gb':   round(memory.total   / (1024 ** 3), 2),
            'disk_percent':      round(disk.percent, 1),
            'disk_used_gb':      round(disk.used      / (1024 ** 3), 2),
            'disk_total_gb':     round(disk.total      / (1024 ** 3), 2),
            'quasar_memory_gb':  round(quasar_memory, 2),
            'process_count':     process_count
        }
    except Exception as e:
        print(f"Resource error: {e}")
        return None


def get_checkpoint_info():
    """Get latest checkpoint information"""
    try:
        if not os.path.exists(CHECKPOINT_DIR):
            return None

        checkpoints = [f for f in os.listdir(CHECKPOINT_DIR) if f.endswith('.pt')]
        if not checkpoints:
            return None

        checkpoints.sort(key=lambda x: os.path.getmtime(os.path.join(CHECKPOINT_DIR, x)))
        latest   = checkpoints[-1]
        filepath = os.path.join(CHECKPOINT_DIR, latest)
        size_mb  = os.path.getsize(filepath) / (1024 ** 2)
        modified = datetime.fromtimestamp(os.path.getmtime(filepath))

        match = re.search(r'step_(\d+)', latest)
        step  = int(match.group(1)) if match else None

        return {
            'filename': latest,
            'size_mb':  round(size_mb, 2),
            'modified': modified.strftime('%Y-%m-%d %H:%M:%S'),
            'step':     step,
            'count':    len(checkpoints)
        }
    except Exception as e:
        print(f"Checkpoint error: {e}")
        return None


def get_process_count():
    """Count running quasar processes"""
    try:
        result = subprocess.run(
            ['ps', 'aux'],
            capture_output=True, text=True, timeout=2
        )
        count = sum(
            1 for line in result.stdout.split('\n')
            if 'quasar' in line.lower() and 'python' in line.lower() and 'grep' not in line
        )
        return count
    except Exception:
        return None


# ============================================================================
# API ENDPOINTS
# ============================================================================

APP_DIR = '/home/user/app'

@app.route('/')
def index():
    """Main dashboard page.
    FIX: render_template() requires dashboard.html inside a templates/ subfolder.
         dashboard.html lives at /home/user/app/dashboard.html (the app root),
         so send_from_directory is used instead — no folder restructuring needed.
    """
    if os.path.exists(os.path.join(APP_DIR, 'dashboard.html')):
        return send_from_directory(APP_DIR, 'dashboard.html')

    # Soft fallback while dashboard.html is absent
    return """<!DOCTYPE html>
<html>
<head>
    <title>K1RL QUASAR — VOLATILITY 75 (1s)</title>
    <meta charset="utf-8">
    <style>body{background:#0a0a0a;color:#00ffff;font-family:monospace;text-align:center;padding:50px}.pulse{animation:pulse 2s infinite}@keyframes pulse{0%,100%{opacity:1}50%{opacity:.4}}.sub{color:#888;font-size:0.8em;margin-top:-10px;letter-spacing:3px}</style>
</head>
<body>
    <h1>🌌 K1RL QUASAR</h1>
    <p class="sub">VOLATILITY 75 INDEX | DERIV</p>
    <p class="pulse">dashboard.html not found at /home/user/app/ — retrying…</p>
    <script>setTimeout(()=>location.reload(),4000)</script>
</body>
</html>""", 200


@app.route('/api/metrics')
def api_metrics():
    """Get current metrics"""
    metrics       = extract_latest_metrics()
    services      = get_service_status()
    resources     = get_system_resources()
    checkpoint    = get_checkpoint_info()
    process_count = get_process_count()

    response = {
        'metrics':       metrics or {},
        'services':      services,
        'resources':     resources or {},
        'health':        resources or {},   # alias for compatibility
        'checkpoint':    checkpoint,
        'process_count': process_count,
        'timestamp':     datetime.now().isoformat(),
        'asset':         'VOLATILITY 75 INDEX (1S) | DERIV',   # ← V75_1S: display name for dashboard subtitle
        'visitors': {
            'active_now':      len(active_visitors),
            'all_time_unique': visitor_stats['total_unique']
        }
    }

    return jsonify(response)


@app.route('/api/visitors')
def api_visitors():
    """Get current visitor statistics"""
    cleanup_inactive_visitors()
    return jsonify({
        'active_now':       len(active_visitors),
        'all_time_unique':  visitor_stats['total_unique'],
        'total_pageviews':  visitor_stats['total_pageviews'],
        'timestamp':        datetime.now().isoformat()
    })


# ── FIX BUG-3: validate against LOG_FILES which now includes 'quasar_engine'
#               so /logs/quasar_engine → 200 instead of 400 ──────────────────
@app.route('/api/logs/<service>')
@app.route('/logs/<service>')
def api_logs(service):
    """Get recent logs for a service"""
    if service not in LOG_FILES:
        return jsonify({'error': f'Invalid service: {service}. '
                                 f'Valid options: {list(LOG_FILES.keys())}'}), 400

    log_file = LOG_FILES[service]
    if not os.path.exists(log_file):
        return f"Log file not found: {log_file}", 404

    try:
        result = subprocess.run(
            ['tail', '-100', log_file],
            capture_output=True, text=True, timeout=5
        )
        lines = result.stdout.split('\n')

        if request.headers.get('Accept', '').startswith('application/json'):
            return jsonify({
                'service': service,
                'lines':   lines[-100:],
                'count':   len(lines)
            })

        return '\n'.join(lines[-100:]), 200, {'Content-Type': 'text/plain'}

    except Exception as e:
        return jsonify({'error': str(e)}), 500


@app.route('/api/stream/logs')
def stream_logs():
    """Stream logs in real-time using Server-Sent Events"""
    def generate():
        try:
            proc = subprocess.Popen(
                ['tail', '-f', LOG_FILES['quasar']],
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
                text=True
            )
            try:
                for line in iter(proc.stdout.readline, ''):
                    if line:
                        yield (f"data: {json.dumps({'line': line.strip(), 'timestamp': datetime.now().isoformat()})}\n\n")
            except Exception:
                pass
            finally:
                proc.kill()
        except Exception as e:
            yield f"data: {json.dumps({'error': str(e)})}\n\n"

    return Response(generate(), mimetype='text/event-stream')


@app.route('/api/health')
@app.route('/health')
def health():
    """Health check endpoint"""
    return jsonify({
        'status':    'ok',
        'platform':  'huggingface-spaces',
        'timestamp': datetime.now().isoformat()
    })


# ============================================================================
# ACCURACY API ENDPOINTS
# ============================================================================

@app.route('/api/accuracy/current')
def api_accuracy_current():
    if not ACCURACY_AVAILABLE:
        metrics = extract_latest_metrics()
        if metrics and metrics.get('avn_accuracy') is not None:
            return jsonify({
                'accuracy':  metrics['avn_accuracy'],
                'timestamp': metrics['timestamp'],
                'iteration': 0
            })
        return jsonify({'error': 'accuracy not available'}), 503
    return jsonify(accuracy_api.get_current_accuracy())


@app.route('/api/accuracy/history')
def api_accuracy_history():
    if not ACCURACY_AVAILABLE:
        return jsonify({'error': 'accuracy history not available'}), 503
    return jsonify(accuracy_api.get_accuracy_history())


@app.route('/api/accuracy/stats')
def api_accuracy_stats():
    if not ACCURACY_AVAILABLE:
        metrics = extract_latest_metrics()
        if metrics and metrics.get('avn_accuracy') is not None:
            acc = metrics['avn_accuracy']
            return jsonify({
                'current':    acc,
                'average':    acc,
                'min':        acc,
                'max':        acc,
                'count':      1,
                'last_10_avg': acc,
                'trend':      'stable'
            })
        return jsonify({'error': 'accuracy stats not available'}), 503
    return jsonify(accuracy_api.get_accuracy_stats())


# ============================================================================
# MAIN
# ============================================================================

if __name__ == '__main__':
    print("=" * 70)
    print("K1RL QUANT - Quantitative Intelligence Observatory")
    print("HuggingFace Spaces Edition")
    print("=" * 70)

    load_visitor_data()
    print(f"📊 Starting with {visitor_stats['total_unique']} total unique visitors")
    print(f"Starting server on http://0.0.0.0:7860")
    print("=" * 70)

    app.run(
        host='0.0.0.0',
        port=7860,
        debug=False,
        threaded=True
    )