{#
# vi:syntax=jinja
#}

{#- Generates one known_hosts entry per given key #}
{%- macro known_host_entry(host, host_names, keys) %}

{#-   Get IPv4 and IPv6 addresses from the DNS #}
{%-   set ip4 = salt['dig.A'](host) -%}
{%-   set ip6 = salt['dig.AAAA'](host) -%}

{#-   The host names to use are to be found within the dict 'host_names'. #}
{#-   If there are none, the host is used directly. #}
{%-   set names = [host_names.get(host, host)] -%}

{#-   Extract the hostname from the FQDN and add it to the names. #}
{%-   if use_hostnames is iterable -%}
{%-     for name in names | sort -%}
{%-       if salt["match.{}".format(hostnames_tgt_type)](hostnames_target, minion_id=name) -%}
{%-         set hostname = name.split('.')|first -%}
{%-         if hostname not in names -%}
{%-           do names.append(hostname) -%}
{%-         endif -%}
{%-       endif -%}
{%-     endfor -%}
{%-   endif -%}

{#-   Append IP addresses and aliases (if they are not already present) #}
{%-   for ip in (ip4 + ip6)|sort -%}
{%-     do names.append(ip) -%}
{%-     for alias in aliases_ips.get(ip, []) -%}
{%-       if alias not in names -%}
{%-         do names.append(alias) -%}
{%-       endif -%}
{%-     endfor -%}
{%-   endfor -%}

{#-   Write one line per key; join the names together #}
{%-   for line in keys.split('\n') -%}
{%-     if line -%}
{{        ','.join(names) }} {{ line }}
{%      endif -%}
{%-   endfor -%}
{%- endmacro -%}

{#- Pre-fetch pillar data #}
{%- set target = salt['pillar.get']('openssh:known_hosts:target', '*') -%}
{%- set tgt_type = salt['pillar.get']('openssh:known_hosts:tgt_type', 'glob') -%}
{%- set keys_function = salt['pillar.get']('openssh:known_hosts:mine_keys_function', 'public_ssh_host_keys') -%}
{%- set hostname_function = salt['pillar.get']('openssh:known_hosts:mine_hostname_function', 'public_ssh_hostname') -%}
{%- set use_hostnames = salt['pillar.get']('openssh:known_hosts:hostnames', False) -%}
{%- set hostnames_target_default = '*' if grains['domain'] == '' else "*.{}".format(grains['domain']) -%}
{%- set hostnames_target = salt['pillar.get']('openssh:known_hosts:hostnames:target', hostnames_target_default) -%}
{%- set hostnames_tgt_type = salt['pillar.get']('openssh:known_hosts:hostnames:tgt_type', 'glob') -%}

{#- Lookup IP of all aliases so that when we have a matching IP, we inject the alias name
    in the SSH known_hosts entry -#}
{%- set aliases = salt['pillar.get']('openssh:known_hosts:aliases', []) -%}
{%- set aliases_ips = {} -%}
{%- for alias in aliases -%}
  {%- for ip in salt['dig.A'](alias) + salt['dig.AAAA'](alias) -%}
    {%- do aliases_ips.setdefault(ip, []).append(alias) -%}
  {%- endfor -%}
{%- endfor -%}

{#- Loop over targetted minions -#}
{%- set host_keys = salt['mine.get'](target, keys_function, tgt_type=tgt_type) -%}
{%- set host_names = salt['mine.get'](target, hostname_function, tgt_type=tgt_type) -%}
{%- for host, keys in host_keys|dictsort -%}
{{ known_host_entry(host, host_names, keys) }}
{%- endfor -%}
