initial commit - from neko open source repository..

internal/utils/array.go

@@ -0,0 +1,24 @@
+package utils
+
+import (
+	"reflect"
+)
+
+func ArrayIn(val interface{}, array interface{}) (exists bool, index int) {
+	exists = false
+	index = -1
+
+	switch reflect.TypeOf(array).Kind() {
+	case reflect.Slice:
+		s := reflect.ValueOf(array)
+		for i := 0; i < s.Len(); i++ {
+			if reflect.DeepEqual(val, s.Index(i).Interface()) == true {
+				index = i
+				exists = true
+				return
+			}
+		}
+	}
+
+	return
+}

internal/utils/color.go

@@ -0,0 +1,34 @@
+package utils
+
+import (
+	"fmt"
+	"regexp"
+)
+
+const (
+	char = "&"
+)
+
+// Colors: http://www.lihaoyi.com/post/BuildyourownCommandLinewithANSIescapecodes.html
+var re = regexp.MustCompile(char + `(?m)([0-9]{1,2};[0-9]{1,2}|[0-9]{1,2})`)
+
+func Color(str string) string {
+	result := ""
+	lastIndex := 0
+
+	for _, v := range re.FindAllSubmatchIndex([]byte(str), -1) {
+		groups := []string{}
+		for i := 0; i < len(v); i += 2 {
+			groups = append(groups, str[v[i]:v[i+1]])
+		}
+
+		result += str[lastIndex:v[0]] + "\033[" + groups[1] + "m"
+		lastIndex = v[1]
+	}
+
+	return result + str[lastIndex:]
+}
+
+func Colorf(format string, a ...interface{}) string {
+	return fmt.Sprintf(Color(format), a...)
+}

internal/utils/ip.go

@@ -0,0 +1,35 @@
+package utils
+
+import (
+	"bytes"
+	"io/ioutil"
+	"net/http"
+)
+
+// dig @resolver1.opendns.com ANY myip.opendns.com +short -4
+
+func GetIP() (string, error) {
+	rsp, err := http.Get("http://checkip.amazonaws.com")
+	if err != nil {
+		return "", err
+	}
+	defer rsp.Body.Close()
+
+	buf, err := ioutil.ReadAll(rsp.Body)
+	if err != nil {
+		return "", err
+	}
+
+	return string(bytes.TrimSpace(buf)), nil
+}
+
+func ReadUserIP(r *http.Request) string {
+	IPAddress := r.Header.Get("X-Real-Ip")
+	if IPAddress == "" {
+		IPAddress = r.Header.Get("X-Forwarded-For")
+	}
+	if IPAddress == "" {
+		IPAddress = r.RemoteAddr
+	}
+	return IPAddress
+}

internal/utils/json.go

@@ -0,0 +1,10 @@
+package utils
+
+import "encoding/json"
+
+func Unmarshal(in interface{}, raw []byte, callback func() error) error {
+	if err := json.Unmarshal(raw, &in); err != nil {
+		return err
+	}
+	return callback()
+}

internal/utils/uid.go

@@ -0,0 +1,98 @@
+package utils
+
+import (
+	"crypto/rand"
+	"fmt"
+	"math"
+)
+
+const (
+	defaultAlphabet = "_-0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ" // len=64
+	defaultSize     = 21
+	defaultMaskSize = 5
+)
+
+// Generator function
+type Generator func([]byte) (int, error)
+
+// BytesGenerator is the default bytes generator
+var BytesGenerator Generator = rand.Read
+
+func initMasks(params ...int) []uint {
+	var size int
+	if len(params) == 0 {
+		size = defaultMaskSize
+	} else {
+		size = params[0]
+	}
+	masks := make([]uint, size)
+	for i := 0; i < size; i++ {
+		shift := 3 + i
+		masks[i] = (2 << uint(shift)) - 1
+	}
+	return masks
+}
+
+func getMask(alphabet string, masks []uint) int {
+	for i := 0; i < len(masks); i++ {
+		curr := int(masks[i])
+		if curr >= len(alphabet)-1 {
+			return curr
+		}
+	}
+	return 0
+}
+
+// GenerateUID is a low-level function to change alphabet and ID size.
+func GenerateUID(alphabet string, size int) (string, error) {
+	if len(alphabet) == 0 || len(alphabet) > 255 {
+		return "", fmt.Errorf("alphabet must not empty and contain no more than 255 chars. Current len is %d", len(alphabet))
+	}
+	if size <= 0 {
+		return "", fmt.Errorf("size must be positive integer")
+	}
+
+	masks := initMasks(size)
+	mask := getMask(alphabet, masks)
+	ceilArg := 1.6 * float64(mask*size) / float64(len(alphabet))
+	step := int(math.Ceil(ceilArg))
+
+	id := make([]byte, size)
+	bytes := make([]byte, step)
+	for j := 0; ; {
+		_, err := BytesGenerator(bytes)
+		if err != nil {
+			return "", err
+		}
+		for i := 0; i < step; i++ {
+			currByte := bytes[i] & byte(mask)
+			if currByte < byte(len(alphabet)) {
+				id[j] = alphabet[currByte]
+				j++
+				if j == size {
+					return string(id[:size]), nil
+				}
+			}
+		}
+	}
+}
+
+// NewUID generates secure URL-friendly unique ID.
+func NewUID(param ...int) (string, error) {
+	var size int
+	if len(param) == 0 {
+		size = defaultSize
+	} else {
+		size = param[0]
+	}
+	bytes := make([]byte, size)
+	_, err := BytesGenerator(bytes)
+	if err != nil {
+		return "", err
+	}
+	id := make([]byte, size)
+	for i := 0; i < size; i++ {
+		id[i] = defaultAlphabet[bytes[i]&63]
+	}
+	return string(id[:size]), nil
+}