First public commit

2 files changed, 144 insertions, 0 deletions

diff --git a/internal/utils/util_test.go b/internal/utils/util_test.go
new file mode 100644
index 0000000..f9daee8
--- /dev/null
+++ b/internal/utils/util_test.go
@@ -0,0 +1,39 @@
+// Copyright 2018 ThousandEyes Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package utils
+
+import (
+	"testing"
+)
+
+func TestMacColonToDash(t *testing.T) {
+	testNormMac := func(givenMac, expectedMac string) {
+		if MacColonToDash(givenMac) != expectedMac {
+			t.Errorf("Expected: %s\nGot: %s", expectedMac, givenMac)
+		}
+	}
+	testNormMac("ff:ff:ff:ff:ff:ff", "ff-ff-ff-ff-ff-ff")
+	testNormMac("ff-ff-ff-ff-ff-ff", "ff-ff-ff-ff-ff-ff")
+}
+
+func TestMacDashToColon(t *testing.T) {
+	testNormMac := func(givenMac, expectedMac string) {
+		if MacDashToColon(givenMac) != expectedMac {
+			t.Errorf("Expected: %s\nGot: %s", expectedMac, givenMac)
+		}
+	}
+	testNormMac("ff-ff-ff-ff-ff-ff", "ff:ff:ff:ff:ff:ff")
+	testNormMac("ff.ff.ff.ff.ff.ff", "ff.ff.ff.ff.ff.ff")
+}
diff --git a/internal/utils/utils.go b/internal/utils/utils.go
new file mode 100644
index 0000000..b7ee256
--- /dev/null
+++ b/internal/utils/utils.go
@@ -0,0 +1,105 @@
+// Copyright 2018 ThousandEyes Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package utils
+
+import (
+	"fmt"
+	"net"
+	"path/filepath"
+	"strings"
+)
+
+// Filter receives a slide of strings and a function that receives a string
+// and returns a bool, and returns a slide that has only the strings that
+// returned true when they were applied the received function.
+func Filter(files []string, fn func(string) bool) []string {
+	var ret []string
+	for _, f := range files {
+		if fn(f) {
+			ret = append(ret, f)
+		}
+	}
+
+	return ret
+}
+
+// StringInSlice receives a string and a slice of strings and returns true if it exists
+// there.
+func StringInSlice(a string, list []string) bool {
+	for _, b := range list {
+		if b == a {
+			return true
+		}
+	}
+	return false
+}
+
+// KeyInMap checks wheter the received key exists in the received map.
+func KeyInMap(key string, mapInput map[string]interface{}) bool {
+	_, found := mapInput[key]
+	return found
+}
+
+// MapToString provides a string representation of a map of strings.
+func MapToString(mapInput map[string]interface{}) string {
+	result := ""
+	for k, v := range mapInput {
+		if len(result) > 0 {
+			result += ", "
+		}
+		result += fmt.Sprintf("%s:%v", k, v)
+	}
+	return result
+}
+
+// BaseURLforEnvName provides an environment-sensitive method for returning
+// the BaseURL of the application.
+func BaseURLforEnvName(baseURL, environment string) string {
+	if environment != "" {
+		return filepath.Join(baseURL, "env", environment)
+	}
+	return baseURL
+}
+
+// ResolveHostname receives an IP and returns the resolved PTR. It returns an
+// empty string in case the DNS lookup fails.
+func ResolveHostname(ip string) (host string) {
+	hosts, err := net.LookupAddr(ip)
+	if err != nil {
+		return ""
+	}
+	return hosts[0]
+}
+
+// IsValidIP returns whether or not an IP is well-formed.
+func IsValidIP(ip string) bool {
+	return net.ParseIP(ip) != nil
+}
+
+// IsValidMAC returns whether or not a MAC address is well-formed.
+func IsValidMAC(mac string) bool {
+	_, err := net.ParseMAC(mac)
+	return err == nil
+}
+
+// MacColonToDash receives a mac address and replace its colons by dashes
+func MacColonToDash(mac string) string {
+	return strings.Replace(mac, ":", "-", -1)
+}
+
+// MacDashToColon receives a mac address and replace its dashes by colons
+func MacDashToColon(mac string) string {
+	return strings.Replace(mac, "-", ":", -1)
+}