From 61fb790224edd1e7d167f972fbf9ae9d2ad89dd2 Mon Sep 17 00:00:00 2001
From: Mario Alegre <mario@git.alemor.org>
Date: Sun, 20 Dec 2020 00:39:48 -0500
Subject: [PATCH] bugfixing

---
 bin/wg-addpeer | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/bin/wg-addpeer b/bin/wg-addpeer
index cf689c2..275d9a9 100755
--- a/bin/wg-addpeer
+++ b/bin/wg-addpeer
@@ -35,10 +35,14 @@ dest_key=$(sshp sudo wg show $wg_dev public-key)
 echo "Configuring peer on local host ..."
 sudo wg set $wg_dev peer "${dest_key}" endpoint $dest_fqdn:$dest_port allowed-ips $dest_wgip/32
 echo "Configuring peer hostname on local host ..."
-echo -e "$dest_wgip\t$dest_name.$wg_domain" | sudo tee -a /etc/hosts >/dev/null
+search="^[0-9.]+\s+$dest_name.$wg_domain"
+replace="$dest_wgip\t$dest_name.$wg_domain"
+(grep -qE "$search" /etc/hosts && sudo sed -E "s/$search/$replace/" /etc/hosts || echo -e "$replace" | cat /etc/hosts -) | sudo tee /etc/hosts >/dev/null
 
 # add peer on dest
 echo "Configuring peer on remote host ..."
 sshp sudo wg set $wg_dev peer "'${host_key}'" endpoint $host_fqdn:$host_port allowed-ips $host_wgip/32
 echo "Configuring peer hostname on remote host ..."
-sshp "echo -e '$host_wgip\t$host_name.$wg_domain' | sudo tee -a /etc/hosts >/dev/null"
+search="^[0-9.]+\s+$host_name.$wg_domain"
+replace="$host_wgip\t$host_name.$wg_domain"
+sshp "(grep -qE '$search' /etc/hosts && sudo sed -E 's/$search/$replace/' /etc/hosts || echo -e '$replace' | cat /etc/hosts -) | sudo tee /etc/hosts >/dev/null"