github.com/nilium/gitlab-runner@v12.5.0+incompatible/log/secrets_cleanup.go (about)

     1  package log
     2  
     3  import (
     4  	"github.com/sirupsen/logrus"
     5  
     6  	"gitlab.com/gitlab-org/gitlab-runner/helpers/url"
     7  )
     8  
     9  type SecretsCleanupHook struct{}
    10  
    11  func (s *SecretsCleanupHook) Levels() []logrus.Level {
    12  	return logrus.AllLevels
    13  }
    14  
    15  func (s *SecretsCleanupHook) Fire(entry *logrus.Entry) error {
    16  	entry.Message = url_helpers.ScrubSecrets(entry.Message)
    17  	return nil
    18  }
    19  
    20  func AddSecretsCleanupLogHook(logger *logrus.Logger) {
    21  	if logger == nil {
    22  		logger = logrus.StandardLogger()
    23  	}
    24  
    25  	logger.AddHook(new(SecretsCleanupHook))
    26  }