backend/api/sources.py

from flask import Blueprint, request, jsonify, current_app
from flask_jwt_extended import jwt_required, get_jwt_identity
from services.content_service import ContentService
import pandas as pd

sources_bp = Blueprint('sources', __name__)

@sources_bp.route('/', methods=['OPTIONS'])
@sources_bp.route('', methods=['OPTIONS'])
def handle_options():
    """Handle OPTIONS requests for preflight CORS checks."""
    return '', 200

@sources_bp.route('/', methods=['GET'])
@sources_bp.route('', methods=['GET'])
@jwt_required()
def get_sources():
    """
    Get all sources for the current user.
    
    Returns:
        JSON: List of sources
    """
    try:
        user_id = get_jwt_identity()
        
        # Check if Supabase client is initialized
        if not hasattr(current_app, 'supabase') or current_app.supabase is None:
            # Add CORS headers to error response
            response_data = jsonify({
                'success': False,
                'message': 'Database connection not initialized'
            })
            response_data.headers.add('Access-Control-Allow-Origin', 'http://localhost:3000')
            response_data.headers.add('Access-Control-Allow-Credentials', 'true')
            return response_data, 500
        
        # Fetch sources from Supabase
        response = (
            current_app.supabase
            .table("Source")
            .select("*")
            .eq("user_id", user_id)
            .execute()
        )
        
        sources = response.data if response.data else []
        
        # Add CORS headers explicitly
        response_data = jsonify({
            'success': True,
            'sources': sources
        })
        response_data.headers.add('Access-Control-Allow-Origin', 'http://localhost:3000')
        response_data.headers.add('Access-Control-Allow-Credentials', 'true')
        return response_data, 200
        
    except Exception as e:
        current_app.logger.error(f"Get sources error: {str(e)}")
        # Add CORS headers to error response
        response_data = jsonify({
            'success': False,
            'message': 'An error occurred while fetching sources'
        })
        response_data.headers.add('Access-Control-Allow-Origin', 'http://localhost:3000')
        response_data.headers.add('Access-Control-Allow-Credentials', 'true')
        return response_data, 500

@sources_bp.route('/', methods=['POST'])
@sources_bp.route('', methods=['POST'])
@jwt_required()
def add_source():
    """
    Add a new source for the current user.
    
    Request Body:
        source (str): Source URL
        
    Returns:
        JSON: Add source result
    """
    try:
        user_id = get_jwt_identity()
        data = request.get_json()
        
        # Validate required fields
        if not data or 'source' not in data:
            return jsonify({
                'success': False,
                'message': 'Source URL is required'
            }), 400
        
        source_url = data['source']
        
        # Use content service to add source
        try:
            content_service = ContentService()
            result = content_service.add_rss_source(source_url, user_id)
            
            return jsonify({
                'success': True,
                'message': result
            }), 201
        except Exception as e:
            # If content service fails, just store in database directly
            current_app.logger.warning(f"Content service failed, storing in database directly: {str(e)}")
            
            # Store source directly in Supabase
            response = (
                current_app.supabase
                .table("Source")
                .insert({
                    "url": source_url,
                    "user_id": user_id,
                    "created_at": "now()"
                })
                .execute()
            )
            
            if response.data:
                return jsonify({
                    'success': True,
                    'message': 'Source added successfully'
                }), 201
            else:
                raise Exception("Failed to store source in database")
        
    except Exception as e:
        current_app.logger.error(f"Add source error: {str(e)}")
        return jsonify({
            'success': False,
            'message': f'An error occurred while adding source: {str(e)}'
        }), 500

@sources_bp.route('/<source_id>', methods=['OPTIONS'])
def handle_source_options(source_id):
    """Handle OPTIONS requests for preflight CORS checks for specific source."""
    return '', 200

@sources_bp.route('/<source_id>', methods=['DELETE'])
@jwt_required()
def delete_source(source_id):
    """
    Delete a source.
    
    Path Parameters:
        source_id (str): Source ID
        
    Returns:
        JSON: Delete source result
    """
    try:
        user_id = get_jwt_identity()
        
        # Delete source from Supabase
        response = (
            current_app.supabase
            .table("Source")
            .delete()
            .eq("id", source_id)
            .eq("user_id", user_id)
            .execute()
        )
        
        if response.data:
            return jsonify({
                'success': True,
                'message': 'Source deleted successfully'
            }), 200
        else:
            return jsonify({
                'success': False,
                'message': 'Source not found or unauthorized'
            }), 404
        
    except Exception as e:
        current_app.logger.error(f"Delete source error: {str(e)}")
        return jsonify({
            'success': False,
            'message': 'An error occurred while deleting source'
        }), 500