#!/usr/bin/env python3

from flask import url_for
from .util import live_server_setup


def test_language_switching(client, live_server, measure_memory_usage, datastore_path):
    """
    Test that the language switching functionality works correctly.

    1. Switch to Italian using the /set-language endpoint
    2. Verify that Italian text appears on the page
    3. Switch back to English and verify English text appears
    """

    # Step 1: Set the language to Italian using the /set-language endpoint
    res = client.get(
        url_for("set_language", locale="it"),
        follow_redirects=True
    )

    assert res.status_code == 200

    # Step 2: Request the index page - should be in Italian
    # The session cookie should be maintained by the test client
    res = client.get(
        url_for("watchlist.index"),
        follow_redirects=True
    )

    assert res.status_code == 200

    # Check for Italian text - "Annulla" (translation of "Cancel")
    assert b"Annulla" in res.data, "Expected Italian text 'Annulla' not found after setting language to Italian"

    assert b'Modifiche testo/HTML, JSON e PDF' in res.data, "Expected italian from processors.available_processors()"

    # Step 3: Switch back to English
    res = client.get(
        url_for("set_language", locale="en"),
        follow_redirects=True
    )

    assert res.status_code == 200

    # Request the index page - should now be in English
    res = client.get(
        url_for("watchlist.index"),
        follow_redirects=True
    )

    assert res.status_code == 200

    # Check for English text
    assert b"Cancel" in res.data, "Expected English text 'Cancel' not found after switching back to English"


def test_invalid_locale(client, live_server, measure_memory_usage, datastore_path):
    """
    Test that setting an invalid locale doesn't break the application.
    The app should ignore invalid locales and continue working.
    """

    # First set to English
    res = client.get(
        url_for("set_language", locale="en"),
        follow_redirects=True
    )

    assert res.status_code == 200

    # Try to set an invalid locale
    res = client.get(
        url_for("set_language", locale="invalid_locale_xyz"),
        follow_redirects=True
    )

    assert res.status_code == 200

    # Should still be able to access the page (should stay in English since invalid locale was ignored)
    res = client.get(
        url_for("watchlist.index"),
        follow_redirects=True
    )

    assert res.status_code == 200
    assert b"Cancel" in res.data, "Should remain in English when invalid locale is provided"


def test_language_persistence_in_session(client, live_server, measure_memory_usage, datastore_path):
    """
    Test that the language preference persists across multiple requests
    within the same session.
    """

    # Set language to Italian
    res = client.get(
        url_for("set_language", locale="it"),
        follow_redirects=True
    )

    assert res.status_code == 200

    # Make multiple requests - language should persist
    for _ in range(3):
        res = client.get(
            url_for("watchlist.index"),
            follow_redirects=True
        )

        assert res.status_code == 200
        assert b"Annulla" in res.data, "Italian text should persist across requests"


def test_set_language_with_redirect(client, live_server, measure_memory_usage, datastore_path):
    """
    Test that changing language keeps the user on the same page.
    Example: User is on /settings, changes language, stays on /settings.
    """
    from flask import url_for

    # Set language with a redirect parameter (simulating language change from /settings)
    res = client.get(
        url_for("set_language", locale="de", redirect="/settings"),
        follow_redirects=False
    )

    # Should redirect back to settings
    assert res.status_code in [302, 303]
    assert '/settings' in res.location

    # Verify language was set in session
    with client.session_transaction() as sess:
        assert sess.get('locale') == 'de'

    # Test with invalid locale (should still redirect safely)
    res = client.get(
        url_for("set_language", locale="invalid_locale", redirect="/settings"),
        follow_redirects=False
    )
    assert res.status_code in [302, 303]
    assert '/settings' in res.location

    # Test with malicious redirect (should default to watchlist)
    res = client.get(
        url_for("set_language", locale="en", redirect="https://evil.com"),
        follow_redirects=False
    )
    assert res.status_code in [302, 303]
    # Should not redirect to evil.com
    assert 'evil.com' not in res.location