go2rtc/pkg/creds/secrets.go

package creds

import (
	"io"
	"net/http"
	"regexp"
	"slices"
	"strings"
	"sync"
)

func AddSecret(value string) {
	if value == "" {
		return
	}

	secretsMu.Lock()
	defer secretsMu.Unlock()

	if slices.Contains(secrets, value) {
		return
	}

	secrets = append(secrets, value)
	secretsReplacer = nil
}

var secrets []string
var secretsMu sync.Mutex
var secretsReplacer *strings.Replacer
var userinfoRegexp *regexp.Regexp

func getReplacer() *strings.Replacer {
	secretsMu.Lock()
	defer secretsMu.Unlock()

	if secretsReplacer == nil {
		oldnew := make([]string, 0, 2*len(secrets))
		for _, s := range secrets {
			oldnew = append(oldnew, s, "***")
		}
		secretsReplacer = strings.NewReplacer(oldnew...)
	}

	if userinfoRegexp == nil {
		userinfoRegexp = regexp.MustCompile(`://[` + userinfo + `]+@`)
	}

	return secretsReplacer
}

// Uniform Resource Identifier (URI)
// https://datatracker.ietf.org/doc/html/rfc3986
const (
	unreserved = `A-Za-z0-9-._~`
	subdelims  = `!$&'()*+,;=`
	userinfo   = unreserved + subdelims + `%:`
)

func SecretString(s string) string {
	re := getReplacer()
	s = userinfoRegexp.ReplaceAllString(s, `://***@`)
	return re.Replace(s)
}

func SecretWrite(w io.Writer, s string) (n int, err error) {
	re := getReplacer()
	s = userinfoRegexp.ReplaceAllString(s, `://***@`)
	return re.WriteString(w, s)
}

func SecretWriter(w io.Writer) io.Writer {
	return &secretWriter{w}
}

type secretWriter struct {
	w io.Writer
}

func (s *secretWriter) Write(b []byte) (int, error) {
	return SecretWrite(s.w, string(b))
}

func SecretResponse(w http.ResponseWriter) http.ResponseWriter {
	return &secretResponse{w}
}

type secretResponse struct {
	http.ResponseWriter
}

func (s *secretResponse) Write(b []byte) (int, error) {
	return SecretWrite(s.ResponseWriter, string(b))
}