import os
import traceback
from pathlib import Path

from module.const import MAX_FOLDER_NAME_SIZE
from module.download import download_file
from module.get_canvas import get_extra_assignment_files
from module.helpers import make_valid_filename, shorten_file_name
from module.items import CanvasModuleItem, CanvasModule
from module.singlefile import download_page


def download_module_item(module: CanvasModule, item: CanvasModuleItem, modules_dir, cookies_path):
    try:
        module_name = make_valid_filename(str(module.name))
        module_name = shorten_file_name(module_name, len(module_name) - MAX_FOLDER_NAME_SIZE)
        output_dir = Path(modules_dir, module_name)
        output_dir.mkdir(parents=True, exist_ok=True)

        if not item.url:
            return

        # Download attached files
        for file in item.attached_files:
            file.download(output_dir / file.filename)

        # Download the module page.
        html_filename = make_valid_filename(str(item.title)) + ".html"
        if not (output_dir / html_filename).exists():
            download_page(item.url, cookies_path, output_dir, html_filename)
    except:
        # TODO: wrap all threaded funcs in this try/catch
        traceback.print_exc()


def download_assignment(cookies_path, cookie_jar, base_assign_dir, assignment):
    assignment_title = make_valid_filename(str(assignment.title))
    assignment_title = shorten_file_name(assignment_title, len(assignment_title) - MAX_FOLDER_NAME_SIZE)
    assign_dir = os.path.join(base_assign_dir, assignment_title)

    if assignment.html_url != "":
        if not os.path.exists(assign_dir):
            os.makedirs(assign_dir)

        assignment_page_path = os.path.join(assign_dir, "assignment.html")

        if not os.path.exists(assignment_page_path):
            download_page(assignment.html_url, cookies_path, assign_dir, "assignment.html")

        extra_files = get_extra_assignment_files(assignment.description, cookie_jar)
        for name, url in extra_files:
            download_file(url, Path(assign_dir, name), cookie_jar)

    for submission in assignment.submissions:
        download_submission(assignment, submission, assign_dir, cookies_path)


def download_submission(assignment, submission, assign_dir, cookies_path):
    submission_dir = assign_dir

    if len(assignment.submissions) != 1:
        submission_dir = os.path.join(assign_dir, str(submission.user_id))

    if submission.preview_url != "":
        if not os.path.exists(submission_dir):
            os.makedirs(submission_dir)

        submission_page_dir = os.path.join(submission_dir, "submission.html")

        if not os.path.exists(submission_page_dir):
            download_page(submission.preview_url, cookies_path, submission_dir, "submission.html")

    if (submission.attempt != 1 and assignment.updated_url != "" and assignment.html_url != ""
            and assignment.html_url.rstrip("/") != assignment.updated_url.rstrip("/")):
        submission_dir = os.path.join(assign_dir, "attempts")

        if not os.path.exists(submission_dir):
            os.makedirs(submission_dir)

        for i in range(submission.attempt):
            filename = "attempt_" + str(i + 1) + ".html"
            submission_page_attempt_dir = os.path.join(submission_dir, filename)

            if not os.path.exists(submission_page_attempt_dir):
                download_page(assignment.updated_url + "/history?version=" + str(i + 1), cookies_path, submission_dir, filename)