init

Signed-off-by: storyicon <yuanchao@bilibili.com>

pkg/util/util.go

@@ -0,0 +1,106 @@
+// Copyright 2021 storyicon@foxmail.com
+//
+// 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 util
+
+import (
+	"os"
+	"os/exec"
+	"path/filepath"
+	"regexp"
+	"runtime"
+	"strings"
+	"syscall"
+)
+
+// ContainsEmpty is used to check whether items contains empty string
+func ContainsEmpty(items ...string) bool {
+	for _, item := range items {
+		if item == "" {
+			return true
+		}
+	}
+	return false
+}
+
+// SetToSlice is used to convert set<string> to slice<string>
+func SetToSlice(set map[string]struct{}) []string {
+	data := make([]string, 0, len(set))
+	for key := range set {
+		data = append(data, key)
+	}
+	return data
+}
+
+// GetMapKeys is used to get the keys of map
+func GetMapKeys(dict map[string]string) []string {
+	data := make([]string, 0, len(dict))
+	for key := range dict {
+		data = append(data, key)
+	}
+	return data
+}
+
+// GetExitCode is used to parse exit code from cmd error
+func GetExitCode(err error) int {
+	if exitErr, ok := err.(*exec.ExitError); ok {
+		// The program has exited with an exit code != 0
+		// This works on both Unix and Windows. Although package
+		// syscall is generally platform dependent, WaitStatus is
+		// defined for both Unix and Windows and in both cases has
+		// an ExitStatus() method with the same signature.
+		if status, ok := exitErr.Sys().(syscall.WaitStatus); ok {
+			return status.ExitStatus()
+		}
+	}
+	return 1
+}
+
+var regexpEnvironmentVar = regexp.MustCompile(`\$[A-Za-z_]+`)
+
+// RenderPathWithEnv is used to render path with environment
+func RenderPathWithEnv(path string) string {
+	matches := regexpEnvironmentVar.FindAllString(path, -1)
+	for _, match := range matches {
+		path = strings.ReplaceAll(path, match, os.Getenv(match[1:]))
+	}
+	return filepath.Clean(path)
+}
+
+// SplitGoPackageVersion is used to split go package version
+func SplitGoPackageVersion(pkg string) (path string, version string, ok bool) {
+	i := strings.Index(pkg, "@")
+	if i == -1 {
+		return "", "", false
+	}
+	return pkg[:i], pkg[i+1:], true
+}
+
+// JoinGoPackageVersion is used to join go path and versions
+func JoinGoPackageVersion(path, version string) string {
+	return strings.Join([]string{
+		path, version,
+	}, "@")
+}
+
+// GetBinaryFileName is used to get os based binary file name
+func GetBinaryFileName(name string) string {
+	if runtime.GOOS == "windows" {
+		if !strings.HasSuffix(name, ".exe") {
+			return name + ".exe"
+		}
+		return name
+	}
+	return name
+}