views.py

import json
import os
import zipfile
from io import BytesIO
from django.http import HttpResponse
from django.shortcuts import redirect, render
from django.urls import reverse
from .forms import *
from .merge import merge

AUTOCONFIG_JS = """pref("general.config.filename", "firefox.cfg");
pref("general.config.obscure_value", 0);"""


def get_forms(request, FormClasses):
    forms = []
    invalid_data = False
    for i in range(len(FormClasses)):
        FormClass = FormClasses[i]
        name = FormClass.id
        session_data = request.session.get(name + "_data", None)
        if request.POST.get("form_name", "") == name:
            post_data = request.POST
            form = FormClass(post_data)
            if not form.is_valid():
                invalid_data = True
            # we save the invalid data anyway,
            # so the state which forms were valid is correct
            request.session[name + "_data"] = post_data
        else:
            form = FormClass(session_data)
        if not i == len(FormClasses) - 1:  # last item
            form.next = FormClasses[i + 1].id
        else:
            form.next = "finish"
        forms.append(form)
    return forms, invalid_data


def main(request):
    form_classes = PROFILES.get(
        request.session.get("profile", sorted(PROFILES)[0]), ["empty", []]
    )[1]
    forms, invalid_data = get_forms(request, form_classes)

    # are all forms finished?
    finished = True
    for form in forms:
        if not form.is_valid():
            finished = False
            break

    if request.POST:
        if request.POST.get("profile", None):
            profile_name = request.POST.get("profile", "default")
            request.session.clear()
            if profile_name in PROFILES:
                request.session["profile"] = profile_name
                form_classes = PROFILES.get(profile_name)[1]
                forms, invalid_data = get_forms(request, form_classes)
            return redirect(reverse("main") + "#" + forms[0].id)
        # start over again
        elif request.POST.get("reset", None) == "reset":
            request.session.clear()

        # redirect to the current form or to the next one?
        next = request.POST.get("next", "")
        form_name = request.POST.get("form_name", "")
        save = "save" in request.POST
        if "profile" not in request.session and (next or save):
            request.session["profile"] = sorted(PROFILES)[0]

        if next and not invalid_data:
            return redirect(reverse("main") + "#" + next)
        else:
            return redirect(reverse("main") + "#" + form_name)
    else:
        # nothing posted, just render the current page
        (
            prefs_js,
            addons,
            files_inline,
            enterprise_policy,
        ) = generate_prefsjs_and_addonlist(forms, False)
        return render(
            request,
            "main.html",
            {
                "profiles": [(name, PROFILES[name][0]) for name in sorted(PROFILES)],
                "active_profile": request.session.get("profile"),
                "forms": forms,
                "prefs_js": prefs_js,
                "enterprise_policy": enterprise_policy,
                "filenames": addons + list(files_inline.keys()),
                "finished": finished,
            },
        )


def generate_prefsjs_and_addonlist(forms, prefsjs_only, pref_type="user_pref"):
    config = {}
    addons = []
    files_inline = {}
    enterprise_policy = {}
    for form in forms:
        (
            form_config,
            form_addons,
            files_inline,
            form_enterprise_policy,
        ) = form.get_config_and_addons()
        for key in form_config:
            config[key] = form_config[key]
        addons += form_addons
        enterprise_policy = merge(enterprise_policy, form_enterprise_policy)
    addons = sorted(addons)

    enterprise_policy = {"policies": enterprise_policy}
    enterprise_policy = json.dumps(enterprise_policy, indent=2)

    prefs = ""
    if addons and not prefsjs_only:
        # allow preinstalled addons in the profile
        config["extensions.autoDisableScopes"] = 14
    for key in sorted(config):
        value = config[key]
        if isinstance(value, str):
            value = '"{0}"'.format(value)
        elif isinstance(value, bool):
            value = "true" if value else "false"
        else:
            value = str(value)
        prefs += pref_type + '("{key}", {value});\r\n'.format(key=key, value=value)

    return prefs, addons, files_inline, enterprise_policy


def download(request, what):
    form_classes = PROFILES.get(
        request.session.get("profile", sorted(PROFILES)[0]), ["empty", []]
    )[1]
    forms, invalid_data = get_forms(request, form_classes)

    prefsjs_only = False
    prefsjs_text = False
    addons_only = False
    as_enterprise_policy = False
    pref_type = "user_pref"

    zipfilename = "profile.zip"

    if what == "prefs.js":
        prefsjs_only = True
    elif what == "prefs.js.txt":
        prefsjs_only = True
        prefsjs_text = True
    elif what == "addons.zip":
        addons_only = True
        zipfilename = "addons.zip"
    elif what == "enterprise_policy.zip":
        as_enterprise_policy = True
        zipfilename = "enterprise_policy.zip"
        pref_type = "pref"

    if invalid_data:
        return redirect(reverse("main") + "#finish")

    prefs, addons, files_inline, enterprise_policy = generate_prefsjs_and_addonlist(
        forms, prefsjs_only, pref_type
    )

    if prefsjs_only:
        response = HttpResponse(prefs, content_type="text/plain")
        if prefsjs_text:
            response["Content-Disposition"] = 'filename="prefs.js"'
        else:
            response["Content-Disposition"] = 'attachment; filename="prefs.js"'
    elif as_enterprise_policy:
        memoryFile = BytesIO()
        zip_file = zipfile.ZipFile(memoryFile, "w", zipfile.ZIP_DEFLATED)
        autoconfig_header = "// IMPORTANT: Start your code on the 2nd line"
        zip_info = zipfile.ZipInfo("firefox.cfg")
        zip_info.external_attr |= 0o644 << 16
        zip_file.writestr(
            zip_info,
            autoconfig_header + "\n" + prefs,
            compress_type=zipfile.ZIP_DEFLATED,
        )
        zip_info = zipfile.ZipInfo("defaults/pref/autoconfig.js")
        zip_info.external_attr |= 0o644 << 16
        zip_file.writestr(zip_info, AUTOCONFIG_JS, compress_type=zipfile.ZIP_DEFLATED)
        zip_info = zipfile.ZipInfo("distribution/policies.json")
        zip_info.external_attr |= 0o644 << 16
        zip_file.writestr(
            zip_info, enterprise_policy, compress_type=zipfile.ZIP_DEFLATED
        )
        zip_file.close()

        memoryFile.seek(0)
        response = HttpResponse(memoryFile.read(), content_type="application/zip")
        response["Content-Disposition"] = 'attachment; filename="' + zipfilename + '"'
    else:
        memoryFile = BytesIO()
        zip_file = zipfile.ZipFile(memoryFile, "w", zipfile.ZIP_DEFLATED)
        if not addons_only:
            zip_info = zipfile.ZipInfo("prefs.js")
            zip_info.external_attr |= 0o644 << 16
            zip_file.writestr(zip_info, prefs, compress_type=zipfile.ZIP_DEFLATED)

        for addon in addons:
            zip_file.write(
                os.path.join("extensions", addon), compress_type=zipfile.ZIP_DEFLATED
            )

        for file in files_inline:
            zip_info = zipfile.ZipInfo(file)
            zip_info.external_attr |= 0o644 << 16
            zip_file.writestr(
                zip_info, files_inline[file], compress_type=zipfile.ZIP_DEFLATED
            )
        zip_file.close()

        memoryFile.seek(0)
        response = HttpResponse(memoryFile.read(), content_type="application/zip")
        response["Content-Disposition"] = 'attachment; filename="' + zipfilename + '"'

    return response