import frappe
from frappe import _
from lms.lms.utils import has_course_instructor_role, has_course_moderator_role, check_profile_restriction, get_restriction_details


def get_context(context):
    context.no_cache = 1
    context.live_courses, context.upcoming_courses = get_courses()
    context.restriction = check_profile_restriction()
    portal_course_creation = frappe.db.get_single_value("LMS Settings", "portal_course_creation")
    context.show_creators_section = frappe.session.user != "Guest" and \
        (portal_course_creation == "Anyone" or has_course_instructor_role())
    context.show_review_section = has_course_moderator_role() and frappe.session.user != "Guest"

    if context.restriction:
        context.restriction_details = get_restriction_details()

    context.metatags = {
        "title": _("Course List"),
        "image": frappe.db.get_single_value("Website Settings", "banner_image"),
        "description": "This page lists all the courses published on our website",
        "keywords": "All Courses, Courses, Learn"
    }


def get_courses():
    courses = frappe.get_all("LMS Course",
        filters={"published": True},
        fields=["name", "upcoming", "title", "image", "enable_certification",
        "paid_certificate", "price_certificate", "currency"])

    live_courses, upcoming_courses = [], []
    for course in courses:
        if course.upcoming:
            upcoming_courses.append(course)
        else:
            live_courses.append(course)
    return live_courses, upcoming_courses