from django.conf import settings

from django.template.loader import render_to_string

from django.utils.safestring import mark_safe

import mptt_comments

register = template.Library()

class BaseMpttCommentNode(BaseCommentNode):

    root_node = None

    def __init__(self, ctype=None, object_pk_expr=None, object_expr=None, as_varname=None, comment=None):

        super(BaseMpttCommentNode, self). __init__(ctype=ctype, object_pk_expr=object_pk_expr, object_expr=object_expr, as_varname=as_varname, comment=comment)

        self.comment_model = mptt_comments.get_model()

    def get_root_node(self, context):

        if not self.root_node:

            ctype, object_pk = self.get_target_ctype_pk(context)

            self.root_node = self.comment_model.objects.get_root_comment(ctype, object_pk)

        return self.root_node

objects = {}

class MpttCommentFormNode(BaseMpttCommentNode):

    global objects

    """Insert a form for the comment model into the context."""

    def get_form(self, context):

        ctype, object_pk = self.get_target_ctype_pk(context)

        key = str(ctype)+'_'+str(object_pk)

        if objects.has_key(key):

            return mptt_comments.get_form()(objects[key], parent_comment=self.get_root_node(context))

        elif object_pk:

            objects[key] = ctype.get_object_for_this_type(pk=object_pk)

            return mptt_comments.get_form()(objects[key], parent_comment=self.get_root_node(context))

        else:

    def render(self, context):

        context[self.as_varname] = self.get_form(context)

class MpttCommentListNode(BaseMpttCommentNode):

    offset = getattr(settings, 'MPTT_COMMENTS_OFFSET', 20)

    cutoff_level = getattr(settings, 'MPTT_COMMENTS_CUTOFF', 3)

    def get_query_set(self, context):

        related = getattr(settings, 'MPTT_COMMENTS_SELECT_RELATED', None)

            qs = qs.select_related(*related)

        return qs

    def get_context_value_from_queryset(self, context, qs):

        return list(qs[:self.offset])

    def render(self, context):

        qs = self.get_query_set(context)

        context[self.as_varname] = self.get_context_value_from_queryset(context, qs)

        context['collapse_levels_above'] = 2

        context['cutoff_level'] = self.cutoff_level

        context['bottom_level'] = self.bottom_level

def get_mptt_comment_list(parser, token):

    """

    Gets the list of comments for the given params and populates the template

def mptt_comments_media():

    return mark_safe( render_to_string( ('comments/comments_media.html',) , { }) )

def display_comment_toplevel_for(target):

    model = target.__class__

    template_list = [

        "comments/%s_%s_display_comments_toplevel.html" % tuple(str(model._meta).split(".")),

        "comments/%s_display_comments_toplevel.html" % model._meta.app_label,

    return render_to_string(

        template_list, {

            "object" : target

        # RequestContext(context['request'], {})

    )

register.filter(children_count)

register.tag(get_mptt_comment_form)

register.simple_tag(mptt_comment_form_target)