import django_tables2 as tables
from django_tables2.utils import Accessor

from .models import VRF, RIR, Aggregate, Role, Prefix, IPAddress, VLAN


RIR_EDIT_LINK = """
{% if perms.ipam.change_rir %}<a href="{% url 'ipam:rir_edit' slug=record.slug %}">Edit</a>{% endif %}
"""

UTILIZATION_GRAPH = """
{% with record.get_utilization as percentage %}
<div class="progress text-center">
    {% if percentage < 15 %}<span style="font-size: 12px;">{{ percentage }}%</span>{% endif %}
    <div class="progress-bar progress-bar-{% if percentage >= 90 %}danger{% elif percentage >= 75 %}warning{% else %}success{% endif %}" role="progressbar" aria-valuenow="{{ percentage }}" aria-valuemin="0" aria-valuemax="100" style="width: {{ percentage }}%">
        {% if percentage >= 15 %}{{ percentage }}%{% endif %}
    </div>
</div>
{% endwith %}
"""

ROLE_EDIT_LINK = """
{% if perms.ipam.change_role %}<a href="{% url 'ipam:role_edit' slug=record.slug %}">Edit</a>{% endif %}
"""

PREFIX_LINK = """
{% if record.has_children %}
    <span style="padding-left: {{ record.depth }}0px "><i class="fa fa-caret-right"></i></a>
{% else %}
    <span style="padding-left: {{ record.depth }}9px">
{% endif %}
    <a href="{% if record.pk %}{% url 'ipam:prefix' pk=record.pk %}{% else %}{% url 'ipam:prefix_add' %}?prefix={{ record }}{% if parent.vrf %}&vrf={{ parent.vrf.pk }}{% endif %}{% if parent.site %}&site={{ parent.site.pk }}{% endif %}{% endif %}">{{ record.prefix }}</a>
</span>
"""

PREFIX_LINK_BRIEF = """
<span style="padding-left: {{ record.depth }}0px">
    <a href="{% if record.pk %}{% url 'ipam:prefix' pk=record.pk %}{% else %}{% url 'ipam:prefix_add' %}?prefix={{ record }}{% if parent.vrf %}&vrf={{ parent.vrf.pk }}{% endif %}{% if parent.site %}&site={{ parent.site.pk }}{% endif %}{% endif %}">{{ record.prefix }}</a>
</span>
"""

STATUS_LABEL = """
{% if record.pk %}
    <span class="label label-{{ record.status.get_bootstrap_class_display|lower }}">{{ record.status.name }}</span>
{% else %}
    <span class="label label-success">Available</span>
{% endif %}
"""


#
# VRFs
#

class VRFTable(tables.Table):
    pk = tables.CheckBoxColumn(visible=False, default='')
    name = tables.LinkColumn('ipam:vrf', args=[Accessor('pk')], verbose_name='Name')
    rd = tables.Column(verbose_name='RD')
    description = tables.Column(sortable=False, verbose_name='Description')

    class Meta:
        model = VRF
        fields = ('pk', 'name', 'rd', 'description')
        empty_text = "No VRFs found."
        attrs = {
            'class': 'table table-hover',
        }


#
# RIRs
#

class RIRTable(tables.Table):
    pk = tables.CheckBoxColumn(visible=False, default='')
    name = tables.LinkColumn(verbose_name='Name')
    aggregate_count = tables.Column(verbose_name='Aggregates')
    slug = tables.Column(verbose_name='Slug')
    edit = tables.TemplateColumn(template_code=RIR_EDIT_LINK, verbose_name='')

    class Meta:
        model = RIR
        fields = ('pk', 'name', 'aggregate_count', 'slug', 'edit')
        empty_text = "No aggregates were found."
        attrs = {
            'class': 'table table-hover',
        }


#
# Aggregates
#

class AggregateTable(tables.Table):
    pk = tables.CheckBoxColumn(visible=False, default='')
    prefix = tables.LinkColumn('ipam:aggregate', args=[Accessor('pk')], verbose_name='Aggregate')
    rir = tables.Column(verbose_name='RIR')
    child_count = tables.Column(verbose_name='Prefixes')
    utilization = tables.TemplateColumn(UTILIZATION_GRAPH, orderable=False, verbose_name='Utilization')
    date_added = tables.DateColumn(format="Y-m-d", verbose_name='Added')
    description = tables.Column(sortable=False, verbose_name='Description')

    class Meta:
        model = Aggregate
        fields = ('pk', 'prefix', 'rir', 'child_count', 'utilization', 'date_added', 'description')
        empty_text = "No aggregates found."
        attrs = {
            'class': 'table table-hover',
        }


#
# Roles
#

class RoleTable(tables.Table):
    pk = tables.CheckBoxColumn(visible=False, default='')
    name = tables.Column(verbose_name='Name')
    prefix_count = tables.Column(accessor=Accessor('count_prefixes'), orderable=False, verbose_name='Prefixes')
    vlan_count = tables.Column(accessor=Accessor('count_vlans'), orderable=False, verbose_name='VLANs')
    slug = tables.Column(verbose_name='Slug')
    edit = tables.TemplateColumn(template_code=ROLE_EDIT_LINK, verbose_name='')

    class Meta:
        model = Role
        fields = ('pk', 'name', 'prefix_count', 'vlan_count', 'slug', 'edit')
        empty_text = "No roles were found."
        attrs = {
            'class': 'table table-hover',
        }


#
# Prefixes
#

class PrefixTable(tables.Table):
    pk = tables.CheckBoxColumn(visible=False, default='')
    status = tables.TemplateColumn(STATUS_LABEL, verbose_name='Status')
    prefix = tables.TemplateColumn(PREFIX_LINK, verbose_name='Prefix')
    vrf = tables.Column(orderable=False, default='Global', verbose_name='VRF')
    site = tables.LinkColumn('dcim:site', args=[Accessor('site.slug')], verbose_name='Site')
    role = tables.Column(verbose_name='Role')
    description = tables.Column(sortable=False, verbose_name='Description')

    class Meta:
        model = Prefix
        fields = ('pk', 'prefix', 'status', 'vrf', 'site', 'role', 'description')
        empty_text = "No prefixes found."
        attrs = {
            'class': 'table table-hover',
        }


class PrefixBriefTable(tables.Table):
    prefix = tables.TemplateColumn(PREFIX_LINK_BRIEF, verbose_name='Prefix')
    site = tables.LinkColumn('dcim:site', args=[Accessor('site.slug')], verbose_name='Site')
    status = tables.TemplateColumn(STATUS_LABEL, verbose_name='Status')
    role = tables.Column(verbose_name='Role')

    class Meta:
        model = Prefix
        fields = ('prefix', 'status', 'site', 'role')
        empty_text = "No prefixes found."
        attrs = {
            'class': 'table table-hover',
        }


#
# IPAddresses
#

class IPAddressTable(tables.Table):
    pk = tables.CheckBoxColumn(visible=False, default='')
    address = tables.LinkColumn('ipam:ipaddress', args=[Accessor('pk')], verbose_name='IP Address')
    vrf = tables.Column(orderable=False, default='Global', verbose_name='VRF')
    device = tables.LinkColumn('dcim:device', args=[Accessor('interface.device.pk')], orderable=False, verbose_name='Device')
    interface = tables.Column(orderable=False, verbose_name='Interface')
    description = tables.Column(sortable=False, verbose_name='Description')

    class Meta:
        model = IPAddress
        fields = ('pk', 'address', 'vrf', 'device', 'interface', 'description')
        empty_text = "No IP addresses found."
        attrs = {
            'class': 'table table-hover',
        }


class IPAddressBriefTable(tables.Table):
    address = tables.LinkColumn('ipam:ipaddress', args=[Accessor('pk')], verbose_name='IP Address')
    device = tables.LinkColumn('dcim:device', args=[Accessor('interface.device.pk')], orderable=False, verbose_name='Device')
    interface = tables.Column(orderable=False, verbose_name='Interface')
    nat_inside = tables.LinkColumn('ipam:ipaddress', args=[Accessor('nat_inside.pk')], orderable=False, verbose_name='NAT (Inside)')

    class Meta:
        model = IPAddress
        fields = ('address', 'device', 'interface', 'nat_inside')
        empty_text = "No IP addresses found."
        attrs = {
            'class': 'table table-hover',
        }


#
# VLANs
#

class VLANTable(tables.Table):
    pk = tables.CheckBoxColumn(visible=False, default='')
    vid = tables.LinkColumn('ipam:vlan', args=[Accessor('pk')], verbose_name='ID')
    site = tables.LinkColumn('dcim:site', args=[Accessor('site.slug')], verbose_name='Site')
    name = tables.Column(verbose_name='Name')
    status = tables.TemplateColumn(STATUS_LABEL, verbose_name='Status')
    role = tables.Column(verbose_name='Role')

    class Meta:
        model = VLAN
        fields = ('pk', 'vid', 'site', 'name', 'status', 'role')
        empty_text = "No VLANs found."
        attrs = {
            'class': 'table table-hover',
        }