import os
from subprocess import PIPE, run, CalledProcessError
import tempfile

from django.template.loader import get_template

from django_tex.exceptions import TexError
from django.conf import settings

DEFAULT_INTERPRETER = "lualatex"


def run_tex(source, template_name=None):
    with tempfile.TemporaryDirectory() as tempdir:
        return run_tex_in_directory(source, tempdir, template_name=template_name)


def run_tex_in_directory(source, directory, template_name=None):
    filename = "texput.tex"
    command = getattr(settings, "LATEX_INTERPRETER", DEFAULT_INTERPRETER)
    latex_interpreter_options = getattr(settings, "LATEX_INTERPRETER_OPTIONS", "")
    with open(os.path.join(directory, filename), "x", encoding="utf-8") as f:
        f.write(source)
    args = f'cd "{directory}" && {command} -interaction=batchmode {latex_interpreter_options} {filename}'
    try:
        run(args, shell=True, stdout=PIPE, stderr=PIPE, check=True)
    except CalledProcessError as called_process_error:
        try:
            with open(
                os.path.join(directory, "texput.log"), "r", encoding="utf-8"
            ) as f:
                log = f.read()
        except FileNotFoundError:
            raise called_process_error
        else:
            raise TexError(log=log, source=source, template_name=template_name)
    with open(os.path.join(directory, "texput.pdf"), "rb") as f:
        pdf = f.read()
    return pdf


def compile_template_to_pdf(template_name, context):
    source = render_template_with_context(template_name, context)
    return run_tex(source, template_name=template_name)


def render_template_with_context(template_name, context):
    template = get_template(template_name, using="tex")
    return template.render(context)

import os
from subprocess import PIPE, run, CalledProcessError
import tempfile

from django.template.loader import get_template

from django_tex.exceptions import TexError
from django.conf import settings

DEFAULT_INTERPRETER = "lualatex"


def run_tex(source, template_name=None):
    with tempfile.TemporaryDirectory() as tempdir:
        return run_tex_in_directory(source, tempdir, template_name=template_name)


def run_tex_in_directory(source, directory, template_name=None):
    filename = "texput.tex"
    command = getattr(settings, "LATEX_INTERPRETER", DEFAULT_INTERPRETER)
    latex_interpreter_options = getattr(settings, "LATEX_INTERPRETER_OPTIONS", "")
    with open(os.path.join(directory, filename), "x", encoding="utf-8") as f:
        f.write(source)
    args = f'cd "{directory}" && {command} -interaction=batchmode {latex_interpreter_options} {filename}'
    try:
        run(args, shell=True, stdout=PIPE, stderr=PIPE, check=True)
    except CalledProcessError as called_process_error:
        try:
            with open(
                os.path.join(directory, "texput.log"), "r", encoding="utf-8"
            ) as f:
                log = f.read()
        except FileNotFoundError:
            raise called_process_error
        else:
            raise TexError(log=log, source=source, template_name=template_name)
    with open(os.path.join(directory, "texput.pdf"), "rb") as f:
        pdf = f.read()
    return pdf


def compile_template_to_pdf(template_name, context):
    source = render_template_with_context(template_name, context)
    return run_tex(source, template_name=template_name)


def render_template_with_context(template_name, context):
    template = get_template(template_name, using="tex")
    return template.render(context)
