Added internal package secret

Author: asmyasnikov

internal/secret/password.go

@@ -0,0 +1,19 @@
+package secret
+
+func Password(password string) string {
+	var (
+		bytes         = []byte(password)
+		startPosition = 3
+		endPosition   = len(bytes) - 2
+	)
+	if startPosition > endPosition {
+		for i := range bytes {
+			bytes[i] = '*'
+		}
+	} else {
+		for i := startPosition; i < endPosition; i++ {
+			bytes[i] = '*'
+		}
+	}
+	return string(bytes)
+}

internal/secret/password_test.go

@@ -0,0 +1,27 @@
+package secret
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestPassword(t *testing.T) {
+	for _, tt := range []struct {
+		password string
+		exp      string
+	}{
+		{
+			password: "test",
+			exp:      "****",
+		},
+		{
+			password: "test-long-password",
+			exp:      "tes*************rd",
+		},
+	} {
+		t.Run("", func(t *testing.T) {
+			require.Equal(t, tt.exp, Password(tt.password))
+		})
+	}
+}

log/secret.go renamed to internal/secret/token.go

@@ -1,20 +1,20 @@
-package log
+package secret
 
 import (
 	"bytes"
 	"fmt"
 	"hash/crc32"
 )
 
-func Secret(secret string) string {
+func Token(token string) string {
 	var mask bytes.Buffer
-	if len(secret) > 16 {
-		mask.WriteString(secret[:4])
+	if len(token) > 16 {
+		mask.WriteString(token[:4])
 		mask.WriteString("****")
-		mask.WriteString(secret[len(secret)-4:])
+		mask.WriteString(token[len(token)-4:])
 	} else {
 		mask.WriteString("****")
 	}
-	mask.WriteString(fmt.Sprintf("(CRC-32c: %08X)", crc32.Checksum([]byte(secret), crc32.IEEETable)))
+	mask.WriteString(fmt.Sprintf("(CRC-32c: %08X)", crc32.Checksum([]byte(token), crc32.IEEETable)))
 	return mask.String()
 }

internal/secret/token_test.go

@@ -0,0 +1,23 @@
+package secret
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestToken(t *testing.T) {
+	for _, tt := range []struct {
+		token string
+		exp   string
+	}{
+		{
+			token: "t1.9euelZqOnJiKlcuPzciQyJjNyZ6OzO3rnpWaj53JlZKcy8aTyoyYm8-Uz8ABguCyAUPnxXz391GzzByWXoOxBT7krgydRAkaqyL7G4VVZCQDN1lZ8Lxk_sKbBMpbRn51r8dxNJNzr80ai5AQ", //nolint:lll
+			exp:   "t1.9****i5AQ(CRC-32c: B83EA6E4)",
+		},
+	} {
+		t.Run("", func(t *testing.T) {
+			require.Equal(t, tt.exp, Token(tt.token))
+		})
+	}
+}

log/driver.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"time"
 
+	"github.com/ydb-platform/ydb-go-sdk/v3/internal/secret"
 	"github.com/ydb-platform/ydb-go-sdk/v3/trace"
 )
 
@@ -444,13 +445,13 @@ func internalDriver(l *wrapper, d trace.Detailer) (t trace.Driver) { //nolint:go
 			if info.Error == nil {
 				l.Log(ctx, "done",
 					latency(start),
-					String("token", Secret(info.Token)),
+					String("token", secret.Token(info.Token)),
 				)
 			} else {
 				l.Log(WithLevel(ctx, ERROR), "done",
 					Error(info.Error),
 					latency(start),
-					String("token", Secret(info.Token)),
+					String("token", secret.Token(info.Token)),
 					version(),
 				)
 			}