From 586b1f63cc09ab3d1c70dbfec099351a9c54528c Mon Sep 17 00:00:00 2001
From: Jimmy Monin <jimmy@monin.net>
Date: Mon, 11 Sep 2017 22:07:03 +0200
Subject: [PATCH] Configure web_logs based on YunoHost domains: - add existing
 domains at installation - add hooks to process added/removed domains

---
 hooks/post_domain_add    | 12 ++++++++++++
 hooks/post_domain_remove | 10 ++++++++++
 scripts/_common.sh       | 14 ++++++++++++++
 scripts/install          |  3 +++
 scripts/upgrade          |  5 +++++
 5 files changed, 44 insertions(+)
 create mode 100644 hooks/post_domain_add
 create mode 100644 hooks/post_domain_remove

diff --git a/hooks/post_domain_add b/hooks/post_domain_add
new file mode 100644
index 0000000..8716a9f
--- /dev/null
+++ b/hooks/post_domain_add
@@ -0,0 +1,12 @@
+#!/bin/bash
+
+# Retrieve arguments
+domain=$1
+domain_label=${domain//\./_} # Replace "." by "_" for the domain label
+# Add a web_log entry for this new domain
+cat >> /opt/netdata/etc/netdata/python.d/web_log.conf <<EOF
+${domain_label}_log:
+  name: '${domain_label}'
+  path: '/var/log/nginx/$domain-access.log'
+
+EOF
\ No newline at end of file
diff --git a/hooks/post_domain_remove b/hooks/post_domain_remove
new file mode 100644
index 0000000..c0db7ee
--- /dev/null
+++ b/hooks/post_domain_remove
@@ -0,0 +1,10 @@
+#!/bin/bash
+
+# Retrieve arguments
+domain=$1
+
+# Look for the domain entry, and delete the related lines
+# by deleting every line between:
+# - the domain label (with "." replaced by "_")
+# - and the domain file specification (need to replace "." by "\." in order not to get interpreted by sed)
+sed --in-place "/${domain//\./_}_log/,/${domain//\./\\\.}/d" /opt/netdata/etc/netdata/python.d/web_log.conf
diff --git a/scripts/_common.sh b/scripts/_common.sh
index 7887fb5..5b96a65 100644
--- a/scripts/_common.sh
+++ b/scripts/_common.sh
@@ -32,6 +32,20 @@ configure_netdata() {
   mv ./netdata-uninstaller.sh /opt/netdata/etc/netdata
 }
 
+# Add a web_log entry for every YunoHost domain
+netdata_add_yunohost_web_logs () {
+  echo "# ------------YUNOHOST DOMAINS---------------" >> /opt/netdata/etc/netdata/python.d/web_log.conf
+  for domain in $(yunohost domain list --output-as plain); do
+  domain_label=${domain//\./_} # Replace "." by "_" for the domain label
+  cat >> /opt/netdata/etc/netdata/python.d/web_log.conf <<EOF
+${domain_label}_log:
+  name: '${domain_label}'
+  path: '/var/log/nginx/$domain-access.log'
+
+EOF
+done
+}
+
 # ============= FUTURE YUNOHOST HELPER =============
 # Delete a file checksum from the app settings
 #
diff --git a/scripts/install b/scripts/install
index 14f213c..19e3f4d 100644
--- a/scripts/install
+++ b/scripts/install
@@ -93,6 +93,9 @@ ynh_add_nginx_config
 pushd $NETDATA_TMPDIR
 ./netdata-installer.sh --install /opt --dont-wait
 
+# Add a web_log entry for every YunoHost domain
+netdata_add_yunohost_web_logs
+
 configure_netdata
 
 # Store the app.src file
diff --git a/scripts/upgrade b/scripts/upgrade
index 0d281b7..26948d6 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -91,6 +91,11 @@ exec 3>${tmp}
 pushd $NETDATA_TMPDIR
 ./netdata-installer.sh --install /opt --dont-wait >&3 2>&3 || ynh_die "FAILED TO COMPILE/INSTALL NETDATA"
 
+# If YunoHost web logs have never been set in netdata configuration, set them
+if [ -z "$(grep "YUNOHOST" /opt/netdata/etc/netdata/python.d/web_log.conf)" ] ; then
+  netdata_add_yunohost_web_logs
+fi
+
 configure_netdata
 
 # Store the app.src file