from collections import deque

from django.template import Context
from django.test.utils import override_settings

from cms.api import add_plugin, create_page
from cms.models import CMSPlugin
from cms.plugin_rendering import (
    ContentRenderer,
    LegacyRenderer,
    StructureRenderer,
)
from cms.test_utils.testcases import CMSTestCase
from cms.test_utils.util.context_managers import override_placeholder_conf
from cms.toolbar.toolbar import CMSToolbar
from cms.toolbar.utils import get_object_edit_url


class TestStructureRenderer(CMSTestCase):
    renderer_class = StructureRenderer

    def get_renderer(self, path=None, language='en', page=None):
        if page and path is None:
            path = page.get_absolute_url(language)
        request = self.get_request(path, language, page=page)
        return self.renderer_class(request)

    def test_get_plugin_class_cache(self):
        plugin = CMSPlugin(plugin_type='MultiColumnPlugin')
        renderer = self.get_renderer()
        plugin_class = renderer.get_plugin_class(plugin)
        self.assertIn('MultiColumnPlugin', renderer._cached_plugin_classes)
        self.assertEqual(plugin_class.__name__, 'MultiColumnPlugin')
        self.assertEqual(renderer._cached_plugin_classes['MultiColumnPlugin'], plugin_class)

    def test_get_placeholder_plugin_menu(self):
        cms_page = create_page("page", 'nav_playground.html', "en")
        superuser = self.get_superuser()
        placeholder_1 = cms_page.get_placeholders("en").get(slot='body')

        with self.login_user_context(superuser):
            renderer = self.get_renderer()
            plugin_menu = renderer.get_placeholder_plugin_menu(placeholder_1)
            expected = '<div class="cms-submenu-item cms-submenu-item-title"><span>Multi Columns</span></div>'
            self.assertTrue(expected in plugin_menu)

    def test_render_placeholder_toolbar_js(self):
        cms_page = create_page("page", 'nav_playground.html', "en")
        renderer = self.get_renderer()
        placeholder = cms_page.get_placeholders("en").get(slot='body')
        content = renderer.get_placeholder_toolbar_js(placeholder, cms_page)

        expected_bits = [
            '"MultiColumnPlugin"',
            '"addPluginHelpTitle": "Add plugin to placeholder \\"Body\\""',
            '"name": "Body"',
            f'"placeholder_id": "{placeholder.pk}"',
        ]

        for bit in expected_bits:
            self.assertIn(bit, content)

    def test_render_placeholder_toolbar_js_escaping(self):
        cms_page = create_page("page", 'nav_playground.html', "en")
        renderer = self.get_renderer()
        placeholder = cms_page.get_placeholders("en").get(slot='body')

        conf = {placeholder.slot: {'name': 'Content-with-dash'}}

        with override_placeholder_conf(CMS_PLACEHOLDER_CONF=conf):
            content = renderer.get_placeholder_toolbar_js(placeholder, cms_page)

        expected_bits = [
            '"MultiColumnPlugin"',
            '"addPluginHelpTitle": "Add plugin to placeholder \\"Content-with-dash\\""',
            '"name": "Content-with-dash"',
            f'"placeholder_id": "{placeholder.pk}"',
        ]

        for bit in expected_bits:
            self.assertIn(bit, content)


class TestContentRenderer(TestStructureRenderer):
    renderer_class = ContentRenderer

    @override_settings(CMS_PLACEHOLDER_CACHE=True)
    def test_placeholder_cache_enabled(self):
        cms_page = create_page("page", 'nav_playground.html', "en")
        request_path = cms_page.get_absolute_url('en')
        renderer = self.get_renderer(request_path, page=cms_page)
        self.assertTrue(renderer.placeholder_cache_is_enabled())

    @override_settings(CMS_PLACEHOLDER_CACHE=True)
    def test_placeholder_cache_disabled(self):
        with override_settings(CMS_PLACEHOLDER_CACHE=False):
            # Placeholder cache has been explicitly disabled
            renderer = self.get_renderer()
            self.assertFalse(renderer.placeholder_cache_is_enabled())

        with self.login_user_context(self.get_staff_user_with_no_permissions()):
            # Placeholder cache is disabled for all staff users
            renderer = self.get_renderer()
            self.assertFalse(renderer.placeholder_cache_is_enabled())

    def test_preload_placeholders_for_page_with_inherit_off(self):
        cms_page = create_page("page", 'nav_playground.html', "en")
        placeholder_1 = cms_page.get_placeholders("en").get(slot='body')
        placeholder_1_plugin_1 = add_plugin(
            placeholder_1,
            plugin_type='LinkPlugin',
            language='en',
            name='Link #1',
            external_link='https://www.django-cms.org',
        )
        placeholder_1_plugin_2 = add_plugin(
            placeholder_1,
            plugin_type='LinkPlugin',
            language='en',
            name='Link #2',
            external_link='https://www.django-cms.org',
        )
        placeholder_2 = cms_page.get_placeholders("en").get(slot='right-column')
        placeholder_2_plugin_1 = add_plugin(
            placeholder_2,
            plugin_type='LinkPlugin',
            language='en',
            name='Link #3',
            external_link='https://www.django-cms.org',
        )
        renderer = self.get_renderer(page=cms_page)
        renderer._preload_placeholders_for_page(cms_page)

        self.assertIn(cms_page.pk, renderer._placeholders_by_page_cache)
        self.assertIn(placeholder_1.slot, renderer._placeholders_by_page_cache[cms_page.pk])
        self.assertIn(placeholder_2.slot, renderer._placeholders_by_page_cache[cms_page.pk])

        cache = renderer._placeholders_by_page_cache[cms_page.pk]

        self.assertEqual(cache[placeholder_1.slot], placeholder_1)
        self.assertEqual(
            cache[placeholder_1.slot]._plugins_cache, deque([placeholder_1_plugin_1, placeholder_1_plugin_2])
        )
        self.assertEqual(cache[placeholder_2.slot], placeholder_2)
        self.assertEqual(cache[placeholder_2.slot]._plugins_cache, deque([placeholder_2_plugin_1]))


class TestExceptionCatchers(CMSTestCase):
    renderer_class = ContentRenderer

    def setUp(self):
        self.user = self.get_superuser()

        self.cms_page = create_page("page", 'nav_playground.html', "en")
        self.placeholder_1 = self.cms_page.get_placeholders("en").get(slot='body')
        add_plugin(
            self.placeholder_1,
            plugin_type='LinkPlugin',
            language='en',
            name='Link #1',
            external_link='https://www.django-cms.org',
        )
        add_plugin(
            self.placeholder_1,
            plugin_type='BuggyPlugin',
            language='en',
        )
        self.placeholder_2 = self.cms_page.get_placeholders("en").get(slot='right-column')
        self.placeholder_2_plugin_1 = add_plugin(
            self.placeholder_2,
            plugin_type='LinkPlugin',
            language='en',
            name='Link #3',
            external_link='https://www.django-cms.org',
        )

        self.request = self.get_request(
            get_object_edit_url(self.cms_page.get_admin_content('en')),
            'en',
            page=self.cms_page,
        )
        self.request.toolbar = CMSToolbar(self.request)
        self.renderer = self.renderer_class(self.request)

    def test_non_existent_plugins_creates_error_log_and_does_not_fail(self):
        self.placeholder_2_plugin_1.plugin_type = 'NonExistingPlugin'
        self.placeholder_2_plugin_1.save()
        with self.assertLogs("cms.utils.plugins", level="ERROR") as logs:
            plugin_context = Context()
            self.renderer.render_placeholder(self.placeholder_2, plugin_context, "en")
            self.assertEqual(len(logs.output), 1)
            self.assertIn("Plugin not installed: NonExistingPlugin", logs.output[0])
        self.placeholder_2_plugin_1.plugin_type = 'LinkPlugin'
        self.placeholder_2_plugin_1.save()

    def test_exception_in_plugin_render_creates_error_log_and_does_not_fail(self):
        with self.assertLogs("cms.plugin_rendering", level="ERROR") as logs:
            plugin_context = Context()
            self.renderer.render_placeholder(self.placeholder_1, plugin_context, "en")
            self.assertEqual(len(logs.output), 1)
            self.assertIn("ZeroDivisionError: division by zero", logs.output[0])

    def test_exception_in_template_rendering_creates_error_log_and_does_not_fail(self):
        from cms.test_utils.project.pluginapp.plugins.link.cms_plugins import (
            LinkPlugin,
        )
        link_template = LinkPlugin.render_template
        LinkPlugin.render_template = "pluginapp/link/bugs.html"
        with self.assertLogs("cms.plugin_rendering", level="ERROR") as logs:
            plugin_context = Context()
            self.renderer.render_placeholder(self.placeholder_1, plugin_context, "en")
            self.assertEqual(len(logs.output), 1)
            self.assertIn("TemplateSyntaxError", logs.output[0])
            self.assertIn("Invalid block tag on line 1", logs.output[0])
        LinkPlugin.render_template = link_template

    def test_exception_in_plugin_render_shows_error_in_edit_mode(self):
        plugin_context = Context()
        with self.assertLogs("cms.plugin_rendering", level="ERROR"):
            markup = self.renderer.render_placeholder(self.placeholder_1, plugin_context, "en", editable=True)
        self.assertTrue(self.renderer.toolbar.edit_mode_active)
        self.assertIn('<div class="cms-rendering-exception">', markup)
        self.assertIn("ZeroDivisionError", markup)
        self.assertIn("division by zero", markup)

    def test_exception_in_plugin_render_is_silent_in_preview_mode(self):
        plugin_context = Context()
        with self.assertLogs("cms.plugin_rendering", level="ERROR"):
            markup = self.renderer.render_placeholder(self.placeholder_1, plugin_context, "en", editable=False)
        self.assertEqual('', markup)

    def test_exception_in_plugin_render_is_silent_in_live_mode(self):
        plugin_context = Context()
        self.request = self.get_request(
            self.cms_page.get_absolute_url('en'),
            'en',
            page=self.cms_page,
        )
        self.request.toolbar = CMSToolbar(self.request)
        self.renderer = self.renderer_class(self.request)

        with self.assertLogs("cms.plugin_rendering", level="ERROR"):
            markup = self.renderer.render_placeholder(self.placeholder_1, plugin_context, "en", editable=False)
        self.assertFalse(self.renderer.toolbar.edit_mode_active)
        self.assertEqual('', markup)

    @override_settings(CMS_CATCH_PLUGIN_500_EXCEPTION=False)
    def test_exception_in_plugin_render_is_raised__in_live_mode(self):
        plugin_context = Context()
        self.request = self.get_request(
            self.cms_page.get_absolute_url('en'),
            'en',
            page=self.cms_page,
        )
        self.request.toolbar = CMSToolbar(self.request)
        self.renderer = self.renderer_class(self.request)

        with self.assertRaises(ZeroDivisionError):
            with self.assertLogs("cms.plugin_rendering", level="ERROR"):
                self.renderer.render_placeholder(self.placeholder_1, plugin_context, "en", editable=False)


class TestLegacyRenderer(TestContentRenderer):
    renderer_class = LegacyRenderer


class TestLegacyRendererExceptionCatcher(TestExceptionCatchers):
    renderer_class = LegacyRenderer