From: Amolith <amolith@secluded.site>
To: ~whereswaldon/arbor-dev@lists.sr.ht
Cc: Amolith <amolith@secluded.site>
Subject: [PATCH 06/12] Add Prometheus and Caddy targets
Date: Wed, 20 Jul 2022 23:09:55 -0400
Message-Id: <20220721031001.97027-7-amolith@secluded.site>
In-Reply-To: <20220721031001.97027-1-amolith@secluded.site>
References: <20220721031001.97027-1-amolith@secluded.site>
MIME-Version: 1.0
Content-Transfer-Encoding: 8bit

Signed-off-by: Amolith <amolith@secluded.site>
---
 setup.go | 23 +++++++++++++++++++++--
 1 file changed, 21 insertions(+), 2 deletions(-)

diff --git a/setup.go b/setup.go
index 1c1c602..e9f4cbc 100644
--- a/setup.go
+++ b/setup.go
@@ -23,8 +23,8 @@ var files embed.FS
 
 // Handles initial configuration of the instance
 func Deploy() error {
-	mg.Deps(Users, Sudoers, Apt, Firewall)
-
+	// Run the various mage targets
+	mg.Deps(Users, Sudoers, Apt, Firewall, Prometheus, Caddy)
 
 	// Finally override the ssh config to disallow root login.
 	// This is the point of no return and will create a zombie
@@ -156,6 +156,25 @@ func Firewall() error {
 	return sh.Run("ufw", "enable")
 }
 
+func Prometheus() error {
+	log.Println("Running target Prometheus")
+	err := config("files/prometheus.yml", "/etc/prometheus/prometheus.yml")
+	if err != nil {
+		return err
+	}
+	return sh.Run("systemctl", "restart", "prometheus")
+}
+
+// Copy Caddyfile to system and restart daemon
+func Caddy() error {
+	log.Println("Running target Caddy")
+	err := config("files/Caddyfile", "/etc/caddy/Caddyfile")
+	if err != nil {
+		return err
+	}
+	return sh.Run("systemctl", "restart", "caddy")
+}
+
 // config() takes source template and destination file arguments and executes
 // the template replacing `hostname` with the system's hostname.
 func config(rfile string, wfile string) error {
-- 
2.37.1