import wagtail.admin.rich_text.editors.draftail.features as draftail_features
from wagtail.admin.rich_text.converters.html_to_contentstate import (
    InlineStyleElementHandler,
)
from wagtail import hooks

from basicknowledge.models import BasicKnowledge
from books.models import ContentBlockSnapshot
from books.models.chapter import Chapter
from books.models.contentblock import ContentBlock
from core.logger import get_logger

logger = get_logger(__name__)


# 1. Use the register_rich_text_features hook.
@hooks.register("register_rich_text_features")
def register_brand_feature(features):
    """
    Registering the feature, which uses the `BRAND` Draft.js inline style type,
    and is stored as HTML with a `<span class="brand">` tag.
    """
    feature_name = "brand"
    type_ = "BRAND"

    # 2. Configure how Draftail handles the feature in its toolbar.
    control = {
        "type": type_,
        "label": "Grün",
        "description": "Grün",
        "style": {"color": "#17A887", "font-weight": "600"},
    }

    # 3. Call register_editor_plugin to register the configuration for Draftail.
    features.register_editor_plugin(
        "draftail", feature_name, draftail_features.InlineStyleFeature(control)
    )

    # 4.configure the content transform from the DB to the editor and back.
    db_conversion = {
        "from_database_format": {
            'span[class="brand"]': InlineStyleElementHandler(type_)
        },
        "to_database_format": {"style_map": {type_: 'span class="brand""'}},
    }

    # 5. Call register_converter_rule to register the content transformation conversion.
    features.register_converter_rule("contentstate", feature_name, db_conversion)

    # 6. (optional) Add the feature to the default features list to make it available
    # on rich text fields that do not specify an explicit 'features' list
    features.default_features.append(feature_name)


@hooks.register("register_rich_text_features")
def register_secondary_feature(features):
    """
    Registering the feature, which uses the `SECONDARY` Draft.js inline style type,
    and is stored as HTML with a `<span class="secondary">` tag.
    """
    feature_name = "secondary"
    type_ = "SECONDARY"

    # 2. Configure how Draftail handles the feature in its toolbar.
    control = {
        "type": type_,
        "label": "Blau",
        "description": "Blau",
        "style": {"color": "#078CC6", "font-weight": "600"},
    }

    # 3. Call register_editor_plugin to register the configuration for Draftail.
    features.register_editor_plugin(
        "draftail", feature_name, draftail_features.InlineStyleFeature(control)
    )

    # 4.configure the content transform from the DB to the editor and back.
    db_conversion = {
        "from_database_format": {
            'span[class="secondary"]': InlineStyleElementHandler(type_)
        },
        "to_database_format": {"style_map": {type_: 'span class="secondary"'}},
    }

    # 5. Call register_converter_rule to register the content transformation conversion.
    features.register_converter_rule("contentstate", feature_name, db_conversion)

    # 6. (optional) Add the feature to the default features list to make it available
    # on rich text fields that do not specify an explicit 'features' list
    features.default_features.append(feature_name)


@hooks.register("construct_explorer_page_queryset")
def remove_page_types_from_menu(parent_page, pages, request):
    return (
        pages.not_type(ContentBlockSnapshot)
        .not_type(BasicKnowledge)
        .exclude(contentblock__user_created=True)
    )


@hooks.register("after_copy_page")
def after_copy_hook(request, page, new_page):
    # todo: find every ContentBlock further down in the tree, see if there are any Surveys or Assignments and copy them and reassign them
    if type(page.specific) == ContentBlock:
        # logger.debug("It's a content block")
        content_block: ContentBlock = new_page.specific
        # logger.debug(f"duplicatin {content_block.title, content_block.pk}")
        content_block.duplicate_attached_entities()

    else:
        # logger.debug(f"It's something else {type(page.specific)}, {ContentBlock}")
        content_blocks = new_page.specific.get_content_blocks()
        for content_block in content_blocks:
            content_block.duplicate_attached_entities()


@hooks.register("after_move_page")
def after_move_hook(request, page):
    logger.debug(f"after moving the page {page.title}")
    if type(page.specific) == ContentBlock:
        logger.debug("it's a content block")
        page.specific.reassign_entities()
    if type(page.specific) == Chapter:
        logger.debug("it's a chapter")
        content_blocks = page.specific.get_content_blocks()
        logger.debug(page.id)
        logger.debug(page.specific.get_content_blocks())
        logger.debug(page.get_children())
        for content_block in content_blocks:
            content_block.reassign_entities()


@hooks.register("after_edit_page")
def after_edit_hook(request, page):
    logger.debug(f"After edit page {page.title}, {type(page).__name__}")


@hooks.register("after_create_page")
def after_create_hook(request, page):
    # reassign assignment and survey module
    if type(page.specific) == ContentBlock:
        content_block = page.specific
        content_block.reassign_entities()