Merge pull request #98 from projectdiscovery/issue-97-two-ways-map

Adding key lookup with value helper

maps/generic_map.go

@@ -1,7 +1,7 @@
 package mapsutil
 
 // Map wraps a generic map type
-type Map[K comparable, V any] map[K]V
+type Map[K, V comparable] map[K]V
 
 // Has checks if the current map has the provided key
 func (m Map[K, V]) Has(key K) bool {
@@ -32,3 +32,15 @@ func (m Map[K, V]) Merge(n map[K]V) {
 		m[k] = v
 	}
 }
+
+// GetKeyWithValue returns the first key having value
+func (m Map[K, V]) GetKeyWithValue(value V) (K, bool) {
+	var zero K
+	for k, v := range m {
+		if v == value {
+			return k, true
+		}
+	}
+
+	return zero, false
+}

maps/generic_map_test.go

@@ -3,6 +3,8 @@ package mapsutil
 import (
 	"reflect"
 	"testing"
+
+	"github.com/stretchr/testify/require"
 )
 
 func TestMapHas(t *testing.T) {
@@ -66,3 +68,53 @@ func TestMapMerge(t *testing.T) {
 		t.Errorf("Merge(%v) = %v, expected %v", n, m, expected)
 	}
 }
+
+func TestMap_GetKeyWithValue(t *testing.T) {
+	type testCase[K, V comparable] struct {
+		InputMap    Map[K, V]
+		Value       V
+		ExpectedKey K
+		ExpectedOk  bool
+	}
+
+	genericMap := Map[string, string]{"a": "a", "b": "b", "c": "c"}
+
+	testCases := []testCase[string, string]{
+		{
+			InputMap:    genericMap,
+			Value:       "b",
+			ExpectedKey: "b",
+			ExpectedOk:  true,
+		},
+		{
+			InputMap:    genericMap,
+			Value:       "d",
+			ExpectedKey: "",
+			ExpectedOk:  false,
+		},
+		{
+			InputMap:    genericMap,
+			Value:       "b",
+			ExpectedKey: "b",
+			ExpectedOk:  true,
+		},
+		{
+			InputMap:    genericMap,
+			Value:       "d",
+			ExpectedKey: "",
+			ExpectedOk:  false,
+		},
+		{
+			InputMap:    genericMap,
+			Value:       "value",
+			ExpectedKey: "",
+			ExpectedOk:  false,
+		},
+	}
+
+	for _, tc := range testCases {
+		key, ok := tc.InputMap.GetKeyWithValue(tc.Value)
+		require.Equal(t, tc.ExpectedKey, key)
+		require.Equal(t, tc.ExpectedOk, ok)
+	}
+}

maps/synclock_map.go

@@ -11,7 +11,7 @@ var (
 )
 
 // SyncLock adds sync and lock capabilities to generic map
-type SyncLockMap[K comparable, V any] struct {
+type SyncLockMap[K, V comparable] struct {
 	ReadOnly atomic.Bool
 	mu       sync.RWMutex
 	Map      Map[K, V]