Merge pull request #134 from projectdiscovery/feat-index-any

Adding IndexAny strings helper

strings/stringsutil.go

@@ -282,3 +282,13 @@ func Truncate(data string, maxSize int) string {
 	}
 	return data
 }
+
+// IndexAny returns the index of the first instance of any of the specified substrings in s, or -1 if s does not contain any of the substrings.
+func IndexAny(s string, seps ...string) (int, string) {
+	for _, sep := range seps {
+		if idx := strings.Index(s, sep); idx >= 0 {
+			return idx, sep
+		}
+	}
+	return -1, ""
+}

strings/stringsutil_test.go

@@ -359,3 +359,21 @@ func TestTruncate(t *testing.T) {
 		require.Equalf(t, test.result, res, "test:%s maxsize: %d result: %s", test.test, test.maxSize, res)
 	}
 }
+
+func TestIndexAny(t *testing.T) {
+	tests := []struct {
+		s           string
+		seps        []string
+		expectedIdx int
+		expectedSep string
+	}{
+		{"abcdefg", []string{"a", "b"}, 0, "a"},
+		{"abcdefg", []string{"z", "b"}, 1, "b"},
+		{"abcdefg", []string{"z", "zz"}, -1, ""},
+	}
+	for _, test := range tests {
+		idx, sep := IndexAny(test.s, test.seps...)
+		require.Equal(t, test.expectedIdx, idx)
+		require.Equal(t, test.expectedSep, sep)
+	}
+}