-module(erlydtl_unparser).
-export([unparse/1, unparse/2]).

unparse(DjangoParseTree, undefined) ->
    unparse(DjangoParseTree);
unparse(DjangoParseTree, true) ->
    Text = unparse(DjangoParseTree),
    Trimmed = re:replace(Text, <<"(^\\s+)|(\\s+$)|\n">>, <<"">>, [global, multiline]),
    Joined = join_iolist(Trimmed, " "),
    binary_to_list(iolist_to_binary(Joined)).

join_iolist(IOList, Sep) ->
    join_iolist(IOList, Sep, []).

join_iolist([[]|IOList], Sep, Acc) ->
    join_iolist(IOList, Sep, Acc);
join_iolist([Data|IOList], Sep, Acc) ->
    join_iolist(IOList, Sep, [Sep, Data|Acc]);
join_iolist([], _, [_|Acc]) ->
    lists:reverse(Acc);
join_iolist(IOList, _, Acc) ->
    lists:reverse([IOList|Acc]).


unparse(DjangoParseTree) ->
    do_unparse(DjangoParseTree).

do_unparse(DjangoParseTree) ->
    do_unparse(DjangoParseTree, []).

do_unparse([], Acc) ->
    lists:flatten(lists:reverse(Acc));
do_unparse([{'extends', Value}|Rest], Acc) ->
    do_unparse(Rest, [["{% extends ", unparse_value(Value), " %}"]|Acc]);
do_unparse([{'autoescape', OnOrOff, Contents}|Rest], Acc) ->
    do_unparse(Rest, [["{% autoescape ", unparse_identifier(OnOrOff), " %}", do_unparse(Contents), "{% endautoescape %}"]|Acc]);
do_unparse([{'block', Identifier, Contents}|Rest], Acc) ->
    do_unparse(Rest, [["{% block ", unparse_identifier(Identifier), " %}", do_unparse(Contents), "{% endblock %}"]|Acc]);
do_unparse([{'blocktrans', Args, Contents, undefined}|Rest], Acc) ->
    do_unparse(Rest, [["{% blocktrans ", unparse_blocktrans_args(Args), " %}", do_unparse(Contents), "{% endblocktrans %}"]|Acc]);
do_unparse([{'blocktrans', Args, Contents, PluralContents}|Rest], Acc) ->
    do_unparse(Rest, [["{% blocktrans ", unparse_blocktrans_args(Args), " %}",
                    do_unparse(Contents),
                    "{% plural %}",
                    do_unparse(PluralContents),
                    "{% endblocktrans %}"]|Acc]);
do_unparse([{'call', Identifier}|Rest], Acc) ->
    do_unparse(Rest, [["{% call ", unparse_identifier(Identifier), " %}"]|Acc]);
do_unparse([{'call', Identifier, With}|Rest], Acc) ->
    do_unparse(Rest, [["{% call ", unparse_identifier(Identifier), " with ", unparse_args(With), " %}"]|Acc]);
do_unparse([{'comment', Contents}|Rest], Acc) ->
    do_unparse(Rest, [["{% comment %}", do_unparse(Contents), "{% endcomment %}"]|Acc]);
do_unparse([{'comment_tag', _Pos, Text}|Rest], Acc) ->
    do_unparse(Rest, [["{#", Text, "#}"]|Acc]);
do_unparse([{'cycle', Names}|Rest], Acc) ->
    do_unparse(Rest, [["{% cycle ", do_unparse(Names), " %}"]|Acc]);
do_unparse([{'cycle_compat', Names}|Rest], Acc) ->
    do_unparse(Rest, [["{% cycle ", unparse_cycle_compat_names(Names), " %}"]|Acc]);
do_unparse([{'date', 'now', Value}|Rest], Acc) ->
    do_unparse(Rest, [["{% now ", unparse_value(Value), " %}"]|Acc]);
do_unparse([{'filter', FilterList, Contents}|Rest], Acc) ->
    do_unparse(Rest, [["{% filter ", unparse_filters(FilterList), " %}", do_unparse(Contents), "{% endfilter %}"]|Acc]);
do_unparse([{'firstof', Vars}|Rest], Acc) ->
    do_unparse(Rest, [["{% firstof ", do_unparse(Vars), " %}"]|Acc]);
do_unparse([{'for', {'in', IteratorList, Identifier}, Contents}|Rest], Acc) ->
    do_unparse(Rest, [["{% for ", unparse_identifier(Identifier), " in ", do_unparse(IteratorList), " %}",
                    do_unparse(Contents),
                    "{% endfor %}"]|Acc]);
do_unparse([{'for', {'in', IteratorList, Identifier}, Contents, EmptyPartsContents}|Rest], Acc) ->
    do_unparse(Rest, [["{% for ", unparse_identifier(Identifier), " in ", do_unparse(IteratorList), " %}",
                    do_unparse(Contents),
                    "{% empty %}",
                    do_unparse(EmptyPartsContents),
                    "{% endfor %}"]|Acc]);
do_unparse([{'if', Expression, Contents}|Rest], Acc) ->
    do_unparse(Rest, [["{% if ", unparse_expression(Expression), " %}",
                    do_unparse(Contents),
                    "{% endif %}"]|Acc]);
do_unparse([{'ifchanged', Expression, IfContents}|Rest], Acc) ->
    do_unparse(Rest, [["{% ifchanged ", unparse_expression(Expression), " %}",
                    do_unparse(IfContents),
                    "{% endifchanged %}"]|Acc]);
do_unparse([{'ifchangedelse', Expression, IfContents, ElseContents}|Rest], Acc) ->
    do_unparse(Rest, [["{% ifchanged ", unparse_expression(Expression), " %}",
                    do_unparse(IfContents),
                    "{% else %}",
                    do_unparse(ElseContents),
                    "{% endifchanged %}"]|Acc]);
do_unparse([{'ifelse', Expression, IfContents, ElseContents}|Rest], Acc) ->
    do_unparse(Rest, [["{% if ", unparse_expression(Expression), " %}",
                    do_unparse(IfContents),
                    "{% else %}",
                    do_unparse(ElseContents),
                    "{% endif %}"]|Acc]);
do_unparse([{'ifequal', [Arg1, Arg2], Contents}|Rest], Acc) ->
    do_unparse(Rest, [["{% ifequal ", unparse_value(Arg1), " ", unparse_value(Arg2), " %}",
                    do_unparse(Contents),
                    "{% endifequal %}"]|Acc]);
do_unparse([{'ifequalelse', [Arg1, Arg2], IfContents, ElseContents}|Rest], Acc) ->
    do_unparse(Rest, [["{% ifequal ", unparse_value(Arg1), " ", unparse_value(Arg2), " %}",
                    do_unparse(IfContents),
                    "{% else %}",
                    do_unparse(ElseContents),
                    "{% endifequal %}"]|Acc]);
do_unparse([{'ifnotequal', [Arg1, Arg2], Contents}|Rest], Acc) ->
    do_unparse(Rest, [["{% ifnotequal ", unparse_value(Arg1), " ", unparse_value(Arg2), " %}",
                    do_unparse(Contents),
                    "{% endifnotequal %}"]|Acc]);
do_unparse([{'ifnotequalelse', [Arg1, Arg2], IfContents, ElseContents}|Rest], Acc) ->
    do_unparse(Rest, [["{% ifnotequal ", unparse_value(Arg1), " ", unparse_value(Arg2), " %}",
                    do_unparse(IfContents),
                    "{% else %}",
                    do_unparse(ElseContents),
                    "{% endifnotequal %}"]|Acc]);
do_unparse([{'include', Value, []}|Rest], Acc) ->
    do_unparse(Rest, [["{% include ", unparse_value(Value), " %}"]|Acc]);
do_unparse([{'include', Value, Args}|Rest], Acc) ->
    do_unparse(Rest, [["{% include ", unparse_value(Value), " with ", unparse_args(Args)]|Acc]);
do_unparse([{'include_only', Value, []}|Rest], Acc) ->
    do_unparse(Rest, [["{% include ", unparse_value(Value), " only %}"]|Acc]);
do_unparse([{'include_only', Value, Args}|Rest], Acc) ->
    do_unparse(Rest, [["{% include ", unparse_value(Value), " with ", unparse_args(Args), " only %}"]|Acc]);
do_unparse([{'regroup', {Variable, Identifier1, Identifier2}, Contents}|Rest], Acc) ->
    do_unparse(Rest, [["{% regroup ", unparse_value(Variable), " by ", unparse_identifier(Identifier1), " as ", unparse_identifier(Identifier2), " %}",
                    do_unparse(Contents),
                    "{% endregroup %}"]|Acc]);
do_unparse([{'spaceless', Contents}|Rest], Acc) ->
    do_unparse(Rest, [["{% spaceless %}", do_unparse(Contents), "{% endspaceless %}"]|Acc]);
do_unparse([{'ssi', Arg}|Rest], Acc) ->
    do_unparse(Rest, [["{% ssi ", unparse_value(Arg), " %}"]|Acc]);
do_unparse([{'ssi_parsed', Arg}|Rest], Acc) ->
    do_unparse(Rest, [["{% ssi ", unparse_value(Arg), " parsed %}"]|Acc]);
do_unparse([{'string', _, String}|Rest], Acc) ->
    do_unparse(Rest, [[String]|Acc]);
do_unparse([{'tag', Identifier, []}|Rest], Acc) ->
    do_unparse(Rest, [["{% ", unparse_identifier(Identifier), " %}"]|Acc]);
do_unparse([{'tag', Identifier, Args}|Rest], Acc) ->
    do_unparse(Rest, [["{% ", unparse_identifier(Identifier), " ", unparse_args(Args), " %}"]|Acc]);
do_unparse([{'templatetag', Identifier}|Rest], Acc) ->
    do_unparse(Rest, [["{% templatetag ", unparse_identifier(Identifier), " %}"]|Acc]);
do_unparse([{'trans', Value}|Rest], Acc) ->
    do_unparse(Rest, [["{% trans ", unparse_value(Value), " %}"]|Acc]);
do_unparse([{'widthratio', Numerator, Denominator, Scale}|Rest], Acc) ->
    do_unparse(Rest, [["{% widthratio ", unparse_value(Numerator), " ", unparse_value(Denominator), " ", unparse_value(Scale), " %}"]|Acc]);
do_unparse([{'with', Args, Contents}|Rest], Acc) ->
    do_unparse(Rest, [["{% with ", unparse_args(Args), " %}",
                    do_unparse(Contents),
                    "{% endwidth %}"]|Acc]);
do_unparse([ValueToken|Rest], Acc) ->
    do_unparse(Rest, [["{{ ", unparse_value(ValueToken), " }}"]|Acc]).


unparse_identifier({identifier, _, Name}) ->
    atom_to_list(Name).

unparse_filters(FilterList) ->
    unparse_filters(FilterList, []).

unparse_filters([], Acc) ->
    lists:reverse(Acc);
unparse_filters([Filter], Acc) ->
    unparse_filters([], [unparse_filter(Filter)|Acc]);
unparse_filters([Filter|Rest], Acc) ->
    unparse_filters(Rest, lists:reverse([unparse_filter(Filter), "|"], Acc)).

unparse_filter([Identifier]) ->
    unparse_identifier(Identifier);
unparse_filter([Identifier, Arg]) ->
    [unparse_identifier(Identifier), ":", unparse_value(Arg)].

unparse_expression({'expr', "in", Arg1, Arg2}) ->
    [unparse_value(Arg1), " in ", unparse_value(Arg2)];
unparse_expression({'expr', "not", {'expr', "in", Arg1, Arg2}}) ->
    [unparse_value(Arg1), " not in ", unparse_value(Arg2)];
unparse_expression({'expr', "not", Expr}) ->
    ["not ", unparse_expression(Expr)];
unparse_expression({'expr', "eq", Arg1, Arg2}) ->
    [unparse_value(Arg1), " == ", unparse_value(Arg2)];
unparse_expression({'expr', "ne", Arg1, Arg2}) ->
    [unparse_value(Arg1), " != ", unparse_value(Arg2)];
unparse_expression({'expr', "ge", Arg1, Arg2}) ->
    [unparse_value(Arg1), " >= ", unparse_value(Arg2)];
unparse_expression({'expr', "le", Arg1, Arg2}) ->
    [unparse_value(Arg1), " <= ", unparse_value(Arg2)];
unparse_expression({'expr', "gt", Arg1, Arg2}) ->
    [unparse_value(Arg1), " > ", unparse_value(Arg2)];
unparse_expression({'expr', "lt", Arg1, Arg2}) ->
    [unparse_value(Arg1), " < ", unparse_value(Arg2)];
unparse_expression({'expr', "or", Arg1, Arg2}) ->
    [unparse_expression(Arg1), " or ", unparse_expression(Arg2)];
unparse_expression({'expr', "and", Arg1, Arg2}) ->
    [unparse_expression(Arg1), " and ", unparse_expression(Arg2)];
unparse_expression(Other) ->
    unparse_value(Other).

unparse_value({'string_literal', _, Value}) ->
    Value;
unparse_value({'number_literal', _, Value}) ->
    Value;
unparse_value({'apply_filter', Variable, Filter}) ->
    [unparse_value(Variable), "|", unparse_filter(Filter)];
unparse_value({'attribute', {Variable, Identifier}}) ->
    [unparse_value(Variable), ".", unparse_identifier(Identifier)];
unparse_value({'variable', Identifier}) ->
    unparse_identifier(Identifier).

unparse_args(Args) ->
    unparse_args(Args, []).

unparse_args([], Acc) ->
    collect_args_acc(Acc);
unparse_args([{{identifier, _, Name}, Value}|Args], Acc) ->
    unparse_args(Args, [[atom_to_list(Name), "=", unparse_value(Value)]|Acc]).

unparse_cycle_compat_names(Names) ->
    unparse_cycle_compat_names(Names, []).

unparse_cycle_compat_names([], Acc) ->
    lists:reverse(Acc);
unparse_cycle_compat_names([{identifier, _, Name}], Acc) ->
    unparse_cycle_compat_names([], [atom_to_list(Name)|Acc]);
unparse_cycle_compat_names([{identifier, _, Name}|Rest], Acc) ->
    unparse_cycle_compat_names(Rest, lists:reverse([atom_to_list(Name), ", "], Acc)).

unparse_blocktrans_args(Args) ->
    unparse_blocktrans_args(Args, []).

unparse_blocktrans_args([], Acc) ->
    collect_args_acc(Acc);
unparse_blocktrans_args([{args, WithArgs}|Args], Acc) ->
    unparse_blocktrans_args(
      Args, [["with ", unparse_args(WithArgs)]|Acc]);
unparse_blocktrans_args([{count, Count}|Args], Acc) ->
    unparse_blocktrans_args(
      Args, [["count ", unparse_args([Count])]|Acc]);
unparse_blocktrans_args([{context, Context}|Args], Acc) ->
    unparse_blocktrans_args(
      Args, [["context ", unparse_value(Context)]|Acc]);
unparse_blocktrans_args([trimmed|Args], Acc) ->
    unparse_blocktrans_args(
      Args, ["trimmed"|Acc]).

collect_args_acc(Acc) ->
    lists:flatten(string:join(lists:reverse(Acc), " ")).