From 42381cde353aaa3fd6ae60e5adec9aa6a906c069 Mon Sep 17 00:00:00 2001
From: Silvio Rhatto <rhatto@riseup.net>
Date: Wed, 30 May 2018 21:01:49 -0300
Subject: Support for pre and post commands at certbot class

---
 manifests/init.pp | 24 +++++++++++++++++-------
 1 file changed, 17 insertions(+), 7 deletions(-)

diff --git a/manifests/init.pp b/manifests/init.pp
index 96e16b1..483511f 100644
--- a/manifests/init.pp
+++ b/manifests/init.pp
@@ -1,10 +1,12 @@
 class certbot(
-  $script_base = '/usr/bin',
-  $basedir     = '/var/spool/certbot',
-  $owner       = 'www-data',
-  $pre_hook    = '',
-  $post_hook   = '',
-  $plugin      = 'webroot',
+  $script_base  = '/usr/bin',
+  $basedir      = '/var/spool/certbot',
+  $owner        = 'www-data',
+  $pre_hook     = '',
+  $post_hook    = '',
+  $pre_command  = '',
+  $post_command = '',
+  $plugin       = 'webroot',
 ) {
 
   $tool = $::lsbdistcodename ? {
@@ -20,6 +22,14 @@ class certbot(
     $real_post_hook = "--post-hook \"${post_hook}\""
   }
 
+  if $pre_command != '' {
+    $real_pre_command = "${pre_command} &&"
+  }
+
+  if $post_command != '' {
+    $real_post_command = "&& ${post_command}"
+  }
+
   # Certbot support
   file { $basedir:
     ensure  => directory,
@@ -38,7 +48,7 @@ class certbot(
 
   # Chosing an arbitrary minute within the hour in the hope that won't overload Let's Encrypt servers
   cron { 'certbot-renew':
-    command => "${script_base}/${tool} renew --${plugin} --quiet -n ${real_pre_hook} ${real_post_hook}",
+    command => "${real_pre_command}${script_base}/${tool} renew --${plugin} --quiet -n ${real_pre_hook} ${real_post_hook}${real_post_command}",
     user    => 'root',
     hour    => [ 5, 23 ],
     minute  => "28",
-- 
cgit v1.2.3