chore: import upstream snapshot with attribution

pkg/dir/dir.go

@@ -0,0 +1,69 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you 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 dir
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+)
+
+func CreateDirIfNotExist(path string) error {
+	return os.MkdirAll(path, os.ModePerm)
+}
+
+func CheckDirExist(path string) bool {
+	f, err := os.Stat(path)
+	return err == nil && f.IsDir()
+}
+
+func CheckFileExist(path string) bool {
+	f, err := os.Stat(path)
+	return err == nil && !f.IsDir()
+}
+
+func DirSize(path string) (int64, error) {
+	var size int64
+	err := filepath.Walk(path, func(_ string, info os.FileInfo, err error) error {
+		if !info.IsDir() {
+			size += info.Size()
+		}
+		return err
+	})
+	return size, err
+}
+
+func FormatFileSize(fileSize int64) (size string) {
+	switch {
+	case fileSize < 1024:
+		// return strconv.FormatInt(fileSize, 10) + "B"
+		return fmt.Sprintf("%.2f B", float64(fileSize)/float64(1))
+	case fileSize < (1024 * 1024):
+		return fmt.Sprintf("%.2f KB", float64(fileSize)/float64(1024))
+	case fileSize < (1024 * 1024 * 1024):
+		return fmt.Sprintf("%.2f MB", float64(fileSize)/float64(1024*1024))
+	case fileSize < (1024 * 1024 * 1024 * 1024):
+		return fmt.Sprintf("%.2f GB", float64(fileSize)/float64(1024*1024*1024))
+	case fileSize < (1024 * 1024 * 1024 * 1024 * 1024):
+		return fmt.Sprintf("%.2f TB", float64(fileSize)/float64(1024*1024*1024*1024))
+	default: // if fileSize < (1024 * 1024 * 1024 * 1024 * 1024 * 1024)
+		return fmt.Sprintf("%.2f EB", float64(fileSize)/float64(1024*1024*1024*1024*1024))
+	}
+}