intercept/routes/gps.py

"""GPS routes for gpsd daemon support."""

from __future__ import annotations

import queue
import time
from collections.abc import Generator

from flask import Blueprint, Response, jsonify

from utils.gps import (
    GPSPosition,
    GPSSkyData,
    get_current_position,
    get_gps_reader,
    start_gpsd,
    stop_gps,
)
from utils.logging import get_logger
from utils.sse import format_sse

logger = get_logger('intercept.gps')

gps_bp = Blueprint('gps', __name__, url_prefix='/gps')

# Queue for SSE position updates
_gps_queue: queue.Queue = queue.Queue(maxsize=100)


def _position_callback(position: GPSPosition) -> None:
    """Callback to queue position updates for SSE stream."""
    try:
        _gps_queue.put_nowait({'type': 'position', **position.to_dict()})
    except queue.Full:
        # Discard oldest if queue is full
        try:
            _gps_queue.get_nowait()
            _gps_queue.put_nowait({'type': 'position', **position.to_dict()})
        except queue.Empty:
            pass


def _sky_callback(sky: GPSSkyData) -> None:
    """Callback to queue sky data updates for SSE stream."""
    try:
        _gps_queue.put_nowait({'type': 'sky', **sky.to_dict()})
    except queue.Full:
        try:
            _gps_queue.get_nowait()
            _gps_queue.put_nowait({'type': 'sky', **sky.to_dict()})
        except queue.Empty:
            pass


@gps_bp.route('/auto-connect', methods=['POST'])
def auto_connect_gps():
    """
    Automatically connect to gpsd if available.

    Called on page load to seamlessly enable GPS if gpsd is running.
    Returns current status if already connected.
    """
    import socket

    # Check if already running
    reader = get_gps_reader()
    if reader and reader.is_running:
        position = reader.position
        sky = reader.sky
        return jsonify({
            'status': 'connected',
            'source': 'gpsd',
            'has_fix': position is not None,
            'position': position.to_dict() if position else None,
            'sky': sky.to_dict() if sky else None,
        })

    # Try to connect to gpsd on localhost:2947
    host = 'localhost'
    port = 2947

    # First check if gpsd is reachable
    try:
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.settimeout(1.0)
        sock.connect((host, port))
        sock.close()
    except Exception:
        return jsonify({
            'status': 'unavailable',
            'message': 'gpsd not running'
        })

    # Clear the queue
    while not _gps_queue.empty():
        try:
            _gps_queue.get_nowait()
        except queue.Empty:
            break

    # Start the gpsd client
    success = start_gpsd(host, port,
                         callback=_position_callback,
                         sky_callback=_sky_callback)

    if success:
        return jsonify({
            'status': 'connected',
            'source': 'gpsd',
            'has_fix': False,
            'position': None,
            'sky': None,
        })
    else:
        return jsonify({
            'status': 'unavailable',
            'message': 'Failed to connect to gpsd'
        })


@gps_bp.route('/stop', methods=['POST'])
def stop_gps_reader():
    """Stop GPS client."""
    reader = get_gps_reader()
    if reader:
        reader.remove_callback(_position_callback)
        reader.remove_sky_callback(_sky_callback)

    stop_gps()

    return jsonify({'status': 'stopped'})


@gps_bp.route('/status')
def get_gps_status():
    """Get current GPS client status."""
    reader = get_gps_reader()

    if not reader:
        return jsonify({
            'running': False,
            'device': None,
            'position': None,
            'sky': None,
            'error': None,
            'message': 'GPS client not started'
        })

    position = reader.position
    sky = reader.sky
    return jsonify({
        'running': reader.is_running,
        'device': reader.device_path,
        'position': position.to_dict() if position else None,
        'sky': sky.to_dict() if sky else None,
        'last_update': reader.last_update.isoformat() if reader.last_update else None,
        'error': reader.error,
        'message': 'Waiting for GPS fix - ensure GPS has clear view of sky' if reader.is_running and not position else None
    })


@gps_bp.route('/position')
def get_position():
    """Get current GPS position."""
    position = get_current_position()

    if position:
        return jsonify({
            'status': 'ok',
            'position': position.to_dict()
        })
    else:
        reader = get_gps_reader()
        if not reader or not reader.is_running:
            return jsonify({
                'status': 'error',
                'message': 'GPS client not running'
            }), 400
        else:
            return jsonify({
                'status': 'waiting',
                'message': 'Waiting for GPS fix - ensure GPS has clear view of sky'
            })


@gps_bp.route('/satellites')
def get_satellites():
    """Get current satellite sky view data."""
    reader = get_gps_reader()

    if not reader or not reader.is_running:
        return jsonify({
            'status': 'error',
            'message': 'GPS client not running'
        }), 400

    sky = reader.sky
    if sky:
        return jsonify({
            'status': 'ok',
            'sky': sky.to_dict()
        })
    else:
        return jsonify({
            'status': 'waiting',
            'message': 'Waiting for satellite data'
        })


@gps_bp.route('/stream')
def stream_gps():
    """SSE stream of GPS position and sky updates."""
    def generate() -> Generator[str, None, None]:
        last_keepalive = time.time()
        keepalive_interval = 30.0

        while True:
            try:
                data = _gps_queue.get(timeout=1)
                last_keepalive = time.time()
                yield format_sse(data)
            except queue.Empty:
                now = time.time()
                if now - last_keepalive >= keepalive_interval:
                    yield format_sse({'type': 'keepalive'})
                    last_keepalive = now

    response = Response(generate(), mimetype='text/event-stream')
    response.headers['Cache-Control'] = 'no-cache'
    response.headers['X-Accel-Buffering'] = 'no'
    response.headers['Connection'] = 'keep-alive'
    return response