from django import template
from django.utils.safestring import mark_safe

from whoisdb.helpers import getWhoisObjectFields

register = template.Library()

@register.filter
def linkObject(value):
	return mark_safe('<a href="%s">%s</a>' % (value.get_absolute_url(), str(value)))

@register.filter
def linkObjects(value):
	links = []
	for obj in value:
		if hasattr(obj, "get_absolute_url"):
			links.append('<a href="%s">%s</a>' % (obj.get_absolute_url(), str(obj)))
		else:
			links.append(str(obj))

	return mark_safe(", ".join(links))

@register.filter
def getFields(value, user):
	print("User is", user)

	owner = value.canEdit(user)

	return getWhoisObjectFields(value, owner)
	#return [("Handle", value.handle), ("Auth", value.auth)]


@register.filter
def userCanEdit(value, user):
	if hasattr(value, "canEdit"):
		return value.canEdit(user)
	return False