From 69d1f35d515be4af2317bcf0f9fd40bb4516a6e7 Mon Sep 17 00:00:00 2001
From: Stefano Da Ros <stefano.daros@signavio.com>
Date: Thu, 7 Feb 2019 15:24:01 +0100
Subject: [PATCH] Add support for defining -config-dir as argument when
 installing as a service

---
 main.go | 57 +++++++++++++++++++++++++++++++++++++++++++--------------
 1 file changed, 43 insertions(+), 14 deletions(-)

diff --git a/main.go b/main.go
index 742089d..d7f9ae1 100644
--- a/main.go
+++ b/main.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"fmt"
 	"net/http"
 	"os"
 
@@ -75,12 +76,53 @@ func (a *App) run() {
 
 func main() {
 	a := &App{}
+	serviceControl, ok := viper.Get("service").(string)
+	if ok && serviceControl != "" {
+		switch serviceControl {
+		case "install":
+			configDir, ok := viper.Get("configDir").(string)
+			if !ok {
+				msg := "the -config-dir argument must be provided " +
+					"when installing the workflow connector as a service "
+				logger.Error(fmt.Errorf(msg))
+				os.Exit(1)
+			}
+			svc := newService(a, "-config-dir", configDir)
+			if err := service.Control(svc, serviceControl); err != nil {
+				logger.Error(err)
+				os.Exit(1)
+			}
+			err := svc.Run()
+			if err != nil {
+				logger.Errorf("unable to run the service: %s\n", err)
+				os.Exit(1)
+			}
+		default:
+			// Execute user specified control on service
+			svc := newService(a)
+			if err := service.Control(svc, serviceControl); err != nil {
+				logger.Error(err)
+				os.Exit(1)
+			}
+			err := svc.Run()
+			if err != nil {
+				logger.Errorf("unable to run the service: %s\n", err)
+				os.Exit(1)
+			}
+		}
+		return
+	}
+}
+
+func newService(a *App, args ...string) (svc service.Service) {
+	var arguments []string
 	svc, err := service.New(
 		a,
 		&service.Config{
 			Name:        config.Options.Name,
 			DisplayName: config.Options.DisplayName,
 			Description: config.Options.Description,
+			Arguments:   append(arguments, args...),
 		},
 	)
 	if err != nil {
@@ -92,18 +134,5 @@ func main() {
 		logger.Errorf("unable to initialize logger: %s\n", err)
 		os.Exit(1)
 	}
-	serviceControl, ok := viper.Get("service").(string)
-	if ok && serviceControl != "" {
-		// Execute user specified control on service
-		if err := service.Control(svc, serviceControl); err != nil {
-			logger.Error(err)
-			os.Exit(1)
-		}
-		return
-	}
-	err = svc.Run()
-	if err != nil {
-		logger.Errorf("unable to run the service: %s\n", err)
-		os.Exit(1)
-	}
+	return
 }