support TrustedValue

Author: iQQBot

components/scrubber/scrubber.go

@@ -202,12 +202,19 @@ func (s *scrubberImpl) Struct(val any) error {
 	return nil
 }
 
-func (s *scrubberImpl) deepCopyStruct(fieldName string, src reflect.Value, scrubTag string) reflect.Value {
+func (s *scrubberImpl) deepCopyStruct(fieldName string, src reflect.Value, scrubTag string, skipScrub bool) reflect.Value {
 	if src.Kind() == reflect.Ptr && src.IsNil() {
 		return reflect.New(src.Type()).Elem()
 	}
 
-	if src.Kind() == reflect.String {
+	if src.CanInterface() {
+		value := src.Interface()
+		if _, ok := value.(TrustedValue); ok {
+			skipScrub = true
+		}
+	}
+
+	if src.Kind() == reflect.String && !skipScrub {
 		dst := reflect.New(src.Type())
 		var (
 			setExplicitValue bool
@@ -241,7 +248,6 @@ func (s *scrubberImpl) deepCopyStruct(fieldName string, src reflect.Value, scrub
 	}
 
 	switch src.Kind() {
-
 	case reflect.Struct:
 		dst := reflect.New(src.Type())
 		t := src.Type()
@@ -275,7 +281,7 @@ func (s *scrubberImpl) deepCopyStruct(fieldName string, src reflect.Value, scrub
 			}
 
 			tagValue := f.Tag.Get("scrub")
-			copied := s.deepCopyStruct(f.Name, srcValue, tagValue)
+			copied := s.deepCopyStruct(f.Name, srcValue, tagValue, skipScrub)
 			dstValue.Set(copied)
 		}
 		return dst.Elem()
@@ -285,31 +291,31 @@ func (s *scrubberImpl) deepCopyStruct(fieldName string, src reflect.Value, scrub
 		keys := src.MapKeys()
 		for i := 0; i < src.Len(); i++ {
 			mValue := src.MapIndex(keys[i])
-			dst.SetMapIndex(keys[i], s.deepCopyStruct(keys[i].String(), mValue, ""))
+			dst.SetMapIndex(keys[i], s.deepCopyStruct(keys[i].String(), mValue, "", skipScrub))
 		}
 		return dst
 
 	case reflect.Slice:
 		dst := reflect.MakeSlice(src.Type(), src.Len(), src.Cap())
 		for i := 0; i < src.Len(); i++ {
-			dst.Index(i).Set(s.deepCopyStruct(fieldName, src.Index(i), ""))
+			dst.Index(i).Set(s.deepCopyStruct(fieldName, src.Index(i), "", skipScrub))
 		}
 		return dst
 
 	case reflect.Array:
 		if src.Len() == 0 {
-			return src // can not access to src.Index(0)
+			return src
 		}
 
 		dst := reflect.New(src.Type()).Elem()
 		for i := 0; i < src.Len(); i++ {
-			dst.Index(i).Set(s.deepCopyStruct(fieldName, src.Index(i), ""))
+			dst.Index(i).Set(s.deepCopyStruct(fieldName, src.Index(i), "", skipScrub))
 		}
 		return dst
 
 	case reflect.Ptr:
 		dst := reflect.New(src.Elem().Type())
-		copied := s.deepCopyStruct(fieldName, src.Elem(), scrubTag)
+		copied := s.deepCopyStruct(fieldName, src.Elem(), scrubTag, skipScrub)
 		dst.Elem().Set(copied)
 		return dst
 
@@ -322,7 +328,7 @@ func (s *scrubberImpl) deepCopyStruct(fieldName string, src reflect.Value, scrub
 
 // Struct implements Scrubber
 func (s *scrubberImpl) DeepCopyStruct(val any) any {
-	return s.deepCopyStruct("", reflect.ValueOf(val), "").Interface()
+	return s.deepCopyStruct("", reflect.ValueOf(val), "", false).Interface()
 }
 
 func (s *scrubberImpl) scrubJsonObject(val map[string]interface{}) error {