use and env file for env vars

Author: ChristianChiarulli

pkg/relay/service.go

@@ -8,7 +8,6 @@ import (
 	"github.com/pterm/pterm"
 )
 
-
 // Template for the systemd service file
 const serviceTemplate = `[Unit]
 Description=Nostr Relay Pyramid
@@ -18,19 +17,27 @@ After=network.target
 Type=simple
 User=nostr
 WorkingDirectory=/home/nostr
-Environment="DOMAIN={{.Domain}}"
-Environment="RELAY_NAME=nostr-relay-pyramid"
-Environment="RELAY_PUBKEY={{.PubKey}}"
+EnvironmentFile=/etc/systemd/system/nostr-relay-pyramid.env
 ExecStart=/usr/local/bin/nostr-relay-pyramid
 Restart=on-failure
 
 [Install]
 WantedBy=multi-user.target
 `
 
+// Template for the environment file
+const envTemplate = `
+DOMAIN={{.Domain}}
+RELAY_NAME=nostr-relay-pyramid
+RELAY_PUBKEY={{.PubKey}}
+`
+
 // Path for the systemd service file
 const serviceFilePath = "/etc/systemd/system/nostr-relay-pyramid.service"
 
+// Path for the environment file
+const envFilePath = "/etc/systemd/system/nostr-relay-pyramid.env"
+
 // Function to check if a user exists
 func userExists(username string) bool {
 	cmd := exec.Command("id", "-u", username)
@@ -40,8 +47,8 @@ func userExists(username string) bool {
 
 // Function to set up the relay service
 func SetupRelayService(domain, pubKey string) {
-
 	spinner, _ := pterm.DefaultSpinner.Start("Configuring nginx for HTTP...")
+
 	// Check if the service file exists and remove it if it does
 	if _, err := os.Stat(serviceFilePath); err == nil {
 		err = os.Remove(serviceFilePath)
@@ -50,20 +57,28 @@ func SetupRelayService(domain, pubKey string) {
 		}
 	}
 
+	// Check if the environment file exists and remove it if it does
+	if _, err := os.Stat(envFilePath); err == nil {
+		err = os.Remove(envFilePath)
+		if err != nil {
+			log.Fatalf("Error removing environment file: %v", err)
+		}
+	}
+
 	// Ensure the user for the relay service exists
 	if !userExists("nostr") {
-    spinner.UpdateText("Creating user 'nostr'...")
+		spinner.UpdateText("Creating user 'nostr'...")
 		err := exec.Command("adduser", "--disabled-login", "--gecos", "", "nostr").Run()
 		if err != nil {
 			log.Fatalf("Error creating user: %v", err)
 		}
 	} else {
-    spinner.UpdateText("User 'nostr' already exists")
+		spinner.UpdateText("User 'nostr' already exists")
 	}
 
 	// Ensure the data directory exists and set ownership
 	const dataDir = "/var/lib/nostr-relay-pyramid"
-  spinner.UpdateText("Creating data directory...")
+	spinner.UpdateText("Creating data directory...")
 	err := os.MkdirAll(dataDir, 0755)
 	if err != nil {
 		log.Fatalf("Error creating data directory: %v", err)
@@ -74,8 +89,26 @@ func SetupRelayService(domain, pubKey string) {
 		log.Fatalf("Error setting ownership of the data directory: %v", err)
 	}
 
+	// Create the environment file
+	spinner.UpdateText("Creating environment file...")
+	envFile, err := os.Create(envFilePath)
+	if err != nil {
+		log.Fatalf("Error creating environment file: %v", err)
+	}
+	defer envFile.Close()
+
+	envTmpl, err := template.New("env").Parse(envTemplate)
+	if err != nil {
+		log.Fatalf("Error parsing environment template: %v", err)
+	}
+
+	err = envTmpl.Execute(envFile, struct{ Domain, PubKey string }{Domain: domain, PubKey: pubKey})
+	if err != nil {
+		log.Fatalf("Error executing environment template: %v", err)
+	}
+
 	// Create the systemd service file
-  spinner.UpdateText("Creating service file...")
+	spinner.UpdateText("Creating service file...")
 	serviceFile, err := os.Create(serviceFilePath)
 	if err != nil {
 		log.Fatalf("Error creating service file: %v", err)
@@ -93,14 +126,14 @@ func SetupRelayService(domain, pubKey string) {
 	}
 
 	// Reload systemd to apply the new service
-  spinner.UpdateText("Reloading systemd daemon...")
+	spinner.UpdateText("Reloading systemd daemon...")
 	err = exec.Command("systemctl", "daemon-reload").Run()
 	if err != nil {
 		log.Fatalf("Error reloading systemd daemon: %v", err)
 	}
 
 	// Enable and start the nostr relay service
-  spinner.UpdateText("Enabling and starting service...")
+	spinner.UpdateText("Enabling and starting service...")
 	err = exec.Command("systemctl", "enable", "nostr-relay-pyramid").Run()
 	if err != nil {
 		log.Fatalf("Error enabling nostr relay service: %v", err)
@@ -111,6 +144,6 @@ func SetupRelayService(domain, pubKey string) {
 		log.Fatalf("Error starting nostr relay service: %v", err)
 	}
 
-  spinner.Success("Nostr relay service configured")
+	spinner.Success("Nostr relay service configured")
 }