#!/usr/sbin/nft -f

{% if zone is vyos_defined %}
    chain VYOS_ZONE_FORWARD {
        type filter hook forward priority -1; policy accept;
{%     if firewall.state_policy is vyos_defined %}
        jump VYOS_STATE_POLICY
{%     endif %}
{%     for zone_name, zone_conf in zone.items() %}
{%         if zone_conf.ipv4 %}
{%             if 'local_zone' not in zone_conf %}
        oifname { {{ zone_conf.interface | join(',') }} } counter jump VZONE_{{ zone_name }}
{%             endif %}
{%         endif %}
{%     endfor %}
    }
    chain VYOS_ZONE_LOCAL {
        type filter hook input priority -1; policy accept;
{%     if firewall.state_policy is vyos_defined %}
        jump VYOS_STATE_POLICY
{%     endif %}
{%     for zone_name, zone_conf in zone.items() %}
{%         if zone_conf.ipv4 %}
{%             if 'local_zone' in zone_conf %}
        counter jump VZONE_{{ zone_name }}_IN
{%             endif %}
{%         endif %}
{%     endfor %}
    }
    chain VYOS_ZONE_OUTPUT {
        type filter hook output priority -1; policy accept;
{%     if firewall.state_policy is vyos_defined %}
        jump VYOS_STATE_POLICY
{%     endif %}
{%     for zone_name, zone_conf in zone.items() %}
{%         if zone_conf.ipv4 %}
{%             if 'local_zone' in zone_conf %}
        counter jump VZONE_{{ zone_name }}_OUT
{%             endif %}
{%         endif %}
{%     endfor %}
    }
{%     for zone_name, zone_conf in zone.items() if zone_conf.ipv4 %}
{%         if zone_conf.local_zone is vyos_defined %}
    chain VZONE_{{ zone_name }}_IN {
        iifname lo counter return
{%             for from_zone, from_conf in zone_conf.from.items() if from_conf.firewall.name is vyos_defined %}
        iifname { {{ zone[from_zone].interface | join(",") }} } counter jump NAME_{{ from_conf.firewall.name }}
        iifname { {{ zone[from_zone].interface | join(",") }} } counter return
{%             endfor %}
        {{ zone_conf | nft_default_rule('zone_' + zone_name) }}
    }
    chain VZONE_{{ zone_name }}_OUT {
        oifname lo counter return
{%             for from_zone, from_conf in zone_conf.from_local.items() if from_conf.firewall.name is vyos_defined %}
        oifname { {{ zone[from_zone].interface | join(",") }} } counter jump NAME_{{ from_conf.firewall.name }}
        oifname { {{ zone[from_zone].interface | join(",") }} } counter return
{%             endfor %}
        {{ zone_conf | nft_default_rule('zone_' + zone_name) }}
    }
{%         else %}
    chain VZONE_{{ zone_name }} {
        iifname { {{ zone_conf.interface | join(",") }} } counter {{ zone_conf | nft_intra_zone_action(ipv6=False) }}
{%             if zone_conf.intra_zone_filtering is vyos_defined %}
        iifname { {{ zone_conf.interface | join(",") }} } counter return
{%             endif %}
{%             for from_zone, from_conf in zone_conf.from.items() if from_conf.firewall.name is vyos_defined %}
{%                 if zone[from_zone].local_zone is not defined %}
        iifname { {{ zone[from_zone].interface | join(",") }} } counter jump NAME_{{ from_conf.firewall.name }}
        iifname { {{ zone[from_zone].interface | join(",") }} } counter return
{%                 endif %}
{%             endfor %}
        {{ zone_conf | nft_default_rule('zone_' + zone_name) }}
    }
{%         endif %}
{%     endfor %}
{% endif %}