import os
import sys

parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
sys.path.append(parent_dir)

from jinja2 import Environment, FileSystemLoader
from wtforms.widgets import CheckboxInput, FileInput
from wtforms.fields import StringField, FileField
from wtforms.validators import InputRequired
from wtforms import Form

from atst.filters import iconSvg

env = Environment(loader=FileSystemLoader("templates"))

env.filters["iconSvg"] = iconSvg

# override tojson as identity function to prevent
# wrapping strings in extra quotes
env.filters["tojson"] = lambda x: x


class InitialValueForm(Form):
    datafield = StringField(label="initialvalue value", default="initialvalue")

    errorfield = StringField(
        label="error", validators=[InputRequired(message="Test Error Message")]
    )


checkbox_template = env.get_template("components/checkbox_input.html")
ci_macro = getattr(checkbox_template.module, "CheckboxInput")
checkbox_input_form = InitialValueForm()
checkbox_input_form.datafield.widget = CheckboxInput()
rendered_checkbox_macro = ci_macro(checkbox_input_form.datafield)
with open("js/test_templates/checkbox_input_template.html", "w") as fh:
    fh.write(rendered_checkbox_macro)

upload_template = env.get_template("components/upload_input.html")
up_macro = getattr(upload_template.module, "UploadInput")
rendered_upload_macro = up_macro(InitialValueForm().datafield)
with open("js/test_templates/upload_input_template.html", "w") as fh:
    fh.write(rendered_upload_macro)

erroredform = InitialValueForm()
erroredform.validate()
rendered_upload_error_macro = up_macro(erroredform.errorfield)
with open("js/test_templates/upload_input_error_template.html", "w") as fh:
    fh.write(rendered_upload_error_macro)