mirror of
https://github.com/drakkan/sftpgo.git
synced 2024-11-26 09:30:28 +00:00
92 lines
2.2 KiB
Go
92 lines
2.2 KiB
Go
// Package utils provides some common utility methods
|
|
package utils
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/drakkan/sftpgo/logger"
|
|
)
|
|
|
|
const logSender = "utils"
|
|
|
|
// IsStringInSlice searches a string in a slice and returns true if the string is found
|
|
func IsStringInSlice(obj string, list []string) bool {
|
|
for _, v := range list {
|
|
if v == obj {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// GetTimeAsMsSinceEpoch returns unix timestamp as milliseconds from a time struct
|
|
func GetTimeAsMsSinceEpoch(t time.Time) int64 {
|
|
return t.UnixNano() / 1000000
|
|
}
|
|
|
|
// ScanDirContents returns the number of files contained in a directory, their size and a slice with the file paths
|
|
func ScanDirContents(path string) (int, int64, []string, error) {
|
|
var numFiles int
|
|
var size int64
|
|
var fileList []string
|
|
var err error
|
|
numFiles = 0
|
|
size = 0
|
|
isDir, err := isDirectory(path)
|
|
if err == nil && isDir {
|
|
err = filepath.Walk(path, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if info != nil && info.Mode().IsRegular() {
|
|
size += info.Size()
|
|
numFiles++
|
|
fileList = append(fileList, path)
|
|
}
|
|
return err
|
|
})
|
|
}
|
|
|
|
return numFiles, size, fileList, err
|
|
}
|
|
|
|
func isDirectory(path string) (bool, error) {
|
|
fileInfo, err := os.Stat(path)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return fileInfo.IsDir(), err
|
|
}
|
|
|
|
// SetPathPermissions call os.Chown on unix, it does nothing on windows
|
|
func SetPathPermissions(path string, uid int, gid int) {
|
|
if runtime.GOOS != "windows" {
|
|
if err := os.Chown(path, uid, gid); err != nil {
|
|
logger.Warn(logSender, "error chowning path %v: %v", path, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
// GetEnvVar retrieves the value of the environment variable named
|
|
// by the key. If the variable is present in the environment the it
|
|
// returns the fallback value
|
|
func GetEnvVar(key, fallback string) string {
|
|
if value, ok := os.LookupEnv(key); ok {
|
|
return value
|
|
}
|
|
return fallback
|
|
}
|
|
|
|
// GetEnvVarAsInt retrieves the value of the environment variable named
|
|
// by the key and returns its value or fallback
|
|
func GetEnvVarAsInt(key string, fallback int) int {
|
|
stringValue := GetEnvVar(key, strconv.Itoa(fallback))
|
|
if value, err := strconv.Atoi(stringValue); err == nil {
|
|
return value
|
|
}
|
|
return fallback
|
|
}
|