publish lancet

fileutil/file.go

@@ -0,0 +1,104 @@
+// Copyright 2021 dudaodong@gmail.com. All rights reserved.
+// Use of this source code is governed by MIT license.
+
+// Package fileutil implements some basic functions for file operations
+
+package fileutil
+
+import (
+	"errors"
+	"io"
+	"io/ioutil"
+	"os"
+)
+
+// IsFileExists checks if a file or directory exists
+func IsExist(path string) bool {
+	_, err := os.Stat(path)
+	if err == nil {
+		return true
+	}
+	if errors.Is(err, os.ErrExist) {
+		return false
+	}
+	return false
+}
+
+// CreateFile create a file in path
+func CreateFile(path string) bool {
+	file, err := os.Create(path)
+	if err != nil {
+		return false
+	}
+
+	defer file.Close()
+	return true
+}
+
+// IsFileExists checks if the path is directy or not
+func IsDir(path string) bool {
+	file, err := os.Stat(path)
+	if err != nil {
+		return false
+	}
+	return file.IsDir()
+}
+
+// RemoveFile remove the path file
+func RemoveFile(path string) error {
+	return os.Remove(path)
+}
+
+// CopyFile copy src file to dest file
+func CopyFile(srcFilePath string, dstFilePath string) error {
+	srcFile, err := os.Open(srcFilePath)
+	if err != nil {
+		return err
+	}
+	defer srcFile.Close()
+
+	distFile, err := os.Create(dstFilePath)
+	if err != nil {
+		return err
+	}
+	defer distFile.Close()
+
+	var tmp = make([]byte, 1024*4)
+	for {
+		n, err := srcFile.Read(tmp)
+		distFile.Write(tmp[:n])
+		if err != nil {
+			if err == io.EOF {
+				return nil
+			} else {
+				return err
+			}
+		}
+	}
+}
+
+// ListFileNames return all file names in the path
+func ListFileNames(path string) ([]string, error) {
+	if !IsExist(path) {
+		return []string{}, nil
+	}
+
+	fs, err := ioutil.ReadDir(path)
+	if err != nil {
+		return []string{}, err
+	}
+
+	sz := len(fs)
+	if sz == 0 {
+		return []string{}, nil
+	}
+
+	res := []string{}
+	for i := 0; i < sz; i++ {
+		if !fs[i].IsDir() {
+			res = append(res, fs[i].Name())
+		}
+	}
+
+	return res, nil
+}