From cccc42cb92730529060b25c1c7f273b4c5f7ddfb Mon Sep 17 00:00:00 2001
From: antonbormotov2 <anton.bormotov@bonance.com>
Date: Thu, 17 Nov 2022 17:23:17 +0700
Subject: [PATCH] feat(hostname): support bool attr "clean" available in hosts
 salt state

---
 hostsfile/init.sls | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/hostsfile/init.sls b/hostsfile/init.sls
index 89dd0c7..4fb4ae8 100644
--- a/hostsfile/init.sls
+++ b/hostsfile/init.sls
@@ -22,16 +22,21 @@
 {%- do hosts.update(pillar_hosts) %}
 
 {%- set domain = salt['pillar.get']('hostsfile:domain', '') %}
+{%- set shorthost = salt['pillar.get']('hostsfile:shorthost', False) %}
 {%- for name, addrlist in hosts.items() %}
     {% if addrlist %}
 {{ name }}-host-entry:
   host.present:
     - ip: {{ addrlist if addrlist is string else addrlist|first }}
     - names:
+        {%- if shorthost and '.' in name %}
+      - {{ name.split('.')[0] }}
+        {%- endif %}
       - {{ name }}
         {%- if domain %}
       - {{ name }}.{{ domain }}
         {%- endif %}
+    - clean: salt['pillar.get']('hostsfile:clean', False)
     {%- endif %}
 {% endfor %}