import frappe

from lms.lms.utils import get_lesson_url, get_lessons, get_membership
from frappe.utils import cstr
from lms.lms.utils import redirect_to_courses_list


def get_common_context(context):
	context.no_cache = 1

	try:
		batch_name = frappe.form_dict["batch"]
	except KeyError:
		batch_name = None

	course = frappe.db.get_value(
		"LMS Course",
		frappe.form_dict["course"],
		["name", "title", "video_link", "enable_certification", "status"],
		as_dict=True,
	)
	if not course:
		redirect_to_courses_list()

	context.course = course
	context.lessons = get_lessons(course.name)
	membership = get_membership(course.name, frappe.session.user, batch_name)
	context.membership = membership
	context.progress = frappe.utils.cint(membership.progress) if membership else 0
	context.batch = membership.batch if membership and membership.batch else None
	context.course.query_parameter = (
		"?batch=" + membership.batch if membership and membership.batch else ""
	)
	context.livecode_url = get_livecode_url()


def get_livecode_url():
	return frappe.db.get_single_value("LMS Settings", "livecode_url")


def redirect_to_lesson(course, index_="1.1"):
	frappe.local.flags.redirect_location = (
		get_lesson_url(course.name, index_) + course.query_parameter
	)
	raise frappe.Redirect


def get_current_lesson_details(lesson_number, context, is_edit=False):
	details_list = list(filter(lambda x: cstr(x.number) == lesson_number, context.lessons))

	if not len(details_list):
		if is_edit:
			return None
		else:
			redirect_to_lesson(context.course)

	lesson_info = details_list[0]
	lesson_info.body = lesson_info.body.replace('"', "'")
	return lesson_info


def get_assessments(class_name, member=None):
	if not member:
		member = frappe.session.user

	assessments = frappe.get_all(
		"LMS Assessment",
		{"parent": class_name},
		["name", "assessment_type", "assessment_name"],
	)

	for assessment in assessments:
		if assessment.assessment_type == "LMS Assignment":
			assessment.title = frappe.db.get_value(
				"LMS Assignment", assessment.assessment_name, "title"
			)

			existing_submission = frappe.db.exists(
				{
					"doctype": "LMS Assignment Submission",
					"member": member,
					"assignment": assessment.assessment_name,
				}
			)

			if existing_submission:
				assessment.submission = frappe.db.get_value(
					"LMS Assignment Submission",
					existing_submission,
					["name", "status", "comments"],
					as_dict=True,
				)

			assessment.edit_url = f"/assignments/{assessment.assessment_name}"
			submission_name = existing_submission if existing_submission else "new-submission"
			assessment.url = (
				f"/assignment-submission/{assessment.assessment_name}/{submission_name}"
			)

		elif assessment.assessment_type == "LMS Quiz":
			assessment.title = frappe.db.get_value(
				"LMS Quiz", assessment.assessment_name, "title"
			)
			assessment.url = f"/quizzes/{assessment.assessment_name}"
	return assessments