Specify logger explicitly when using AddSecretsClenupLogHook

apps/gitlab-runner-helper/main.go

@@ -40,7 +40,7 @@ func main() {
 		logrus.Fatalln("Command", command, "not found")
 	}
 
-	log.AddSecretsCleanupLogHook()
+	log.AddSecretsCleanupLogHook(logrus.StandardLogger())
 	log.ConfigureLogging(app)
 
 	if err := app.Run(os.Args); err != nil {

log/secrets_cleanup.go

@@ -17,6 +17,10 @@ func (s *SecretsCleanupHook) Fire(entry *logrus.Entry) error {
 	return nil
 }
 
-func AddSecretsCleanupLogHook() {
-	logrus.AddHook(new(SecretsCleanupHook))
+func AddSecretsCleanupLogHook(logger *logrus.Logger) {
+	if logger == nil {
+		logger = logrus.StandardLogger()
+	}
+
+	logger.AddHook(new(SecretsCleanupHook))
 }

log/secrets_cleanup_test.go

@@ -32,7 +32,7 @@ func TestSecretsCleanupHook(t *testing.T) {
 
 			logger := logrus.New()
 			logger.Out = buffer
-			logger.AddHook(new(SecretsCleanupHook))
+			AddSecretsCleanupLogHook(logger)
 
 			logger.Errorln(test.message)