chore: add some utils (#228)

* chore: add some utils

* chore: add tests

Author: medcl

core/util/http.go

@@ -0,0 +1,31 @@
+/* Copyright © INFINI LTD. All rights reserved.
+ * Web: https://infinilabs.com
+ * Email: hello#infini.ltd */
+
+package util
+
+import (
+	"bytes"
+	"infini.sh/framework/core/errors"
+	"io/ioutil"
+	"net/http"
+)
+
+func ReadBody(r *http.Request) ([]byte, error) {
+	if r.ContentLength > 0 && r.Body != nil {
+		content, err := ioutil.ReadAll(r.Body)
+		_ = r.Body.Close()
+		if err != nil {
+			return nil, err
+		}
+		if len(content) == 0 {
+			return nil, errors.NewWithCode(err, errors.BodyEmpty, r.URL.String())
+		}
+
+		// Replace r.Body so it can be read again later
+		r.Body = ioutil.NopCloser(bytes.NewBuffer(content))
+
+		return content, nil
+	}
+	return nil, errors.Error("request body is nil")
+}

core/util/set.go

@@ -0,0 +1,16 @@
+/* Copyright © INFINI LTD. All rights reserved.
+ * Web: https://infinilabs.com
+ * Email: hello#infini.ltd */
+
+package util
+
+import "github.com/emirpasic/gods/sets/hashset"
+
+func IsSuperset(a, b *hashset.Set) bool {
+	for _, item := range b.Values() {
+		if !a.Contains(item) {
+			return false
+		}
+	}
+	return true
+}

core/util/set_test.go

@@ -0,0 +1,89 @@
+/* Copyright © INFINI LTD. All rights reserved.
+ * Web: https://infinilabs.com
+ * Email: hello#infini.ltd */
+
+package util
+
+import (
+	"github.com/emirpasic/gods/sets/hashset"
+	"testing"
+)
+
+func newSet(values ...interface{}) *hashset.Set {
+	s := hashset.New()
+	s.Add(values...)
+	return s
+}
+
+func TestIsSuperset(t *testing.T) {
+	tests := []struct {
+		name     string
+		a        *hashset.Set
+		b        *hashset.Set
+		expected bool
+	}{
+		{
+			name:     "both empty",
+			a:        newSet(),
+			b:        newSet(),
+			expected: true, // empty set is superset of empty
+		},
+		{
+			name:     "a empty, b non-empty",
+			a:        newSet(),
+			b:        newSet("x"),
+			expected: false,
+		},
+		{
+			name:     "a non-empty, b empty",
+			a:        newSet("x", "y"),
+			b:        newSet(),
+			expected: true,
+		},
+		{
+			name:     "a == b",
+			a:        newSet(1, 2, 3),
+			b:        newSet(1, 2, 3),
+			expected: true,
+		},
+		{
+			name:     "a larger than b (true superset)",
+			a:        newSet(1, 2, 3, 4),
+			b:        newSet(2, 3),
+			expected: true,
+		},
+		{
+			name:     "a missing one element from b",
+			a:        newSet(1, 2),
+			b:        newSet(1, 2, 3),
+			expected: false,
+		},
+		{
+			name:     "different types - string elements",
+			a:        newSet("a", "b", "c"),
+			b:        newSet("b", "c"),
+			expected: true,
+		},
+		{
+			name:     "b contains element not in a",
+			a:        newSet("a", "b"),
+			b:        newSet("a", "b", "c"),
+			expected: false,
+		},
+		{
+			name:     "subset partially overlapping",
+			a:        newSet("x", "y"),
+			b:        newSet("y", "z"),
+			expected: false,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			result := IsSuperset(tt.a, tt.b)
+			if result != tt.expected {
+				t.Errorf("IsSuperset(%v, %v) = %v; want %v", tt.a.Values(), tt.b.Values(), result, tt.expected)
+			}
+		})
+	}
+}

core/util/stringutil.go

@@ -619,7 +619,7 @@ const (
 	lowercaseChars = "abcdefghijklmnopqrstuvwxyz"
 	uppercaseChars = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
 	digitChars     = "0123456789"
-	specialChars   = "!@#%^&*_+-=?"
+	specialChars   = "!@#%^&$*_+-=?,.~`;:|()<>"
 
 	allChars    = lowercaseChars + uppercaseChars + digitChars + specialChars
 	simpleChars = lowercaseChars + digitChars