Use github.com/coreos/go-systemd/unit for creating unit

Author: arnested

service.go

@@ -3,47 +3,39 @@ package main
 import (
 	"bytes"
 	"fmt"
-	"html/template"
+	"io"
 	"log"
 	"os"
 	"os/exec"
 	"strings"
+
+	"github.com/coreos/go-systemd/unit"
 )
 
 func installService() {
-	serviceUnit := `[Unit]
-Description=Local Docker Development DNS
-BindTo=docker.service
-After=docker.service
-
-[Service]
-Type=notify
-Environment="DOCKER_API_VERSION={{ .Version }}"
-ExecStart={{ .Executable }} service
-SuccessExitStatus=15
-Restart=on-failure
-
-[Install]
-WantedBy=docker.service
-`
-
 	version := serverVersion()
 
 	executable, err := os.Executable()
 	if err != nil {
 		panic(fmt.Errorf("could not find path of executable: %w", err))
 	}
 
-	data := struct {
-		Version    string
-		Executable string
-	}{
-		Version:    version,
-		Executable: executable,
+	serviceUnit := unit.Serialize([]*unit.UnitOption{
+		unit.NewUnitOption("Unit", "Description", "Local Docker Development DNS"),
+		unit.NewUnitOption("Unit", "BindTo", "docker.service"),
+		unit.NewUnitOption("Unit", "After", "docker.service"),
+		unit.NewUnitOption("Service", "Type", "notify"),
+		unit.NewUnitOption("Service", "Environment", "DOCKER_API_VERSION="+version),
+		unit.NewUnitOption("Service", "ExecStart", executable+" service"),
+		unit.NewUnitOption("Service", "SuccessExitStatus", "15"),
+		unit.NewUnitOption("Service", "Restart", "on-failure"),
+		unit.NewUnitOption("Install", "WantedBy", "docker.service"),
+	})
+
+	_, err = io.Copy(os.Stdout, serviceUnit)
+	if err != nil {
+		panic(fmt.Errorf("could not output service unit: %w", err))
 	}
-
-	tmpl, _ := template.New("test").Parse(serviceUnit)
-	_ = tmpl.Execute(os.Stdout, data)
 }
 
 func serverVersion() string {