from re import I
import frappe
from . import utils
from frappe.utils import cstr

from community.www import batch

def get_context(context):
    utils.get_common_context(context)

    chapter_index = frappe.form_dict.get("chapter")
    lesson_index = frappe.form_dict.get("lesson")
    lesson_number = f"{chapter_index}.{lesson_index}"

    course_name = context.course.name
    if not chapter_index or not lesson_index:
        if context.batch:
            index_ = get_lesson_index(context.course, context.batch, frappe.session.user) or "1.1"
        else:
            index_ = "1.1"
        frappe.local.flags.redirect_location = context.course.get_learn_url(index_) + context.course.query_parameter
        raise frappe.Redirect

    context.lesson = context.course.get_lesson(chapter_index, lesson_index)
    context.lesson_index = lesson_index
    context.chapter_index = chapter_index

    outline = context.course.get_outline()
    prev_ = outline.get_prev(lesson_number)
    next_ = outline.get_next(lesson_number)
    context.prev_chap = get_chapter_title(course_name, prev_)
    context.next_chap = get_chapter_title(course_name, next_)
    context.next_url = context.course.get_learn_url(next_) and  context.course.get_learn_url(next_) + context.course.query_parameter
    context.prev_url = context.course.get_learn_url(prev_) and context.course.get_learn_url(prev_) + context.course.query_parameter

    context.page_extensions = get_page_extensions()

    context.page_context = {
        "course": context.course.name,
        "batch": context.get("batch") and context.batch.name,
        "lesson": context.lesson.name,
        "is_member": context.membership is not None
    }

def get_chapter_title(course_name, lesson_number):
    if not lesson_number:
        return
    lesson_split = cstr(lesson_number).split(".")
    chapter_index = lesson_split[0]
    lesson_index = lesson_split[1]
    chapter_name = frappe.db.get_value("Chapter", {"course": course_name, "index_": chapter_index}, "name")
    return frappe.db.get_value("Lesson", {"chapter": chapter_name, "index_": lesson_index}, "title")

def get_lesson_index(course, batch, user):
    lesson = batch.get_current_lesson(user)
    return lesson and course.get_lesson_index(lesson)

def get_page_extensions():
    default_value = ["community.plugins.PageExtension"]
    classnames = frappe.get_hooks("community_lesson_page_extensions") or default_value
    extensions = [frappe.get_attr(name)() for name in classnames]
    return extensions