From afa48b1028a228ae6cfde130a17d28f0eb0ce180 Mon Sep 17 00:00:00 2001
From: Evilham <cvs@evilham.com>
Date: Wed, 4 Aug 2021 21:00:52 +0200
Subject: [PATCH] [__single_binary_service] Support customisation of systemd
 units

Requested by pedro
---
 manifest           | 14 ++++++++++----
 parameter/optional |  1 +
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/manifest b/manifest
index be967eb..fe9ef74 100755
--- a/manifest
+++ b/manifest
@@ -60,6 +60,8 @@ if [ -z "${SERVICE_DESCRIPTION}" ]; then
 	SERVICE_DESCRIPTION="cdist-managed '${SERVICE_NAME}' service"
 fi
 
+SERVICE_DEFINITION="$(cat "${__object}/parameter/service-definition" 2>/dev/null || true)"
+
 DOWNLOAD_URL="$(cat "${__object}/parameter/url")"
 CHECKSUM="$(cat "${__object}/parameter/checksum")"
 SHOULD_VERSION="$(cat "${__object}/parameter/version")"
@@ -102,10 +104,8 @@ INIT="$(cat "${__global}/explorer/init")"
 # TODO: Support non-systemd
 case "${INIT}" in
 	systemd)
-		__systemd_unit "${SERVICE_NAME}.service" \
-			--source "-" \
-			--state "${STATE}" \
-			--enablement-state "enabled" <<EOF
+		if [ -z "${SERVICE_DEFINITION}" ]; then
+			SERVICE_DEFINITION="$(cat <<EOF
 [Unit]
 Description=${SERVICE_DESCRIPTION}
 After=network.target
@@ -121,6 +121,12 @@ Restart=always
 [Install]
 WantedBy=multi-user.target
 EOF
+)"
+		fi
+		echo ${SERVICE_DEFINITION} | __systemd_unit "${SERVICE_NAME}.service" \
+			--source "-" \
+			--state "${STATE}" \
+			--enablement-state "enabled"
 	service_require="${service_require} __systemd_unit/${SERVICE_NAME}.service"
 
   ;;
diff --git a/parameter/optional b/parameter/optional
index 7669591..7c2ca06 100644
--- a/parameter/optional
+++ b/parameter/optional
@@ -6,4 +6,5 @@ binary
 service-args
 service-exec
 service-description
+service-definition
 unpack-args