You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
156 lines
3.4 KiB
Go
156 lines
3.4 KiB
Go
package nfs
|
|
|
|
import (
|
|
"git.noahlan.cn/noahlan/ntool/internal/common"
|
|
"git.noahlan.cn/noahlan/ntool/ncrypt"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
const (
|
|
// MimeSniffLen sniff Length, use for detect file mime type
|
|
MimeSniffLen = 512
|
|
)
|
|
|
|
// OSTempFile create a temp file on os.TempDir()
|
|
// The file is kept as open, the caller should close the file handle, and remove the file by name.
|
|
//
|
|
// Usage:
|
|
//
|
|
// nfs.OSTempFile("example.*.txt")
|
|
func OSTempFile(pattern string) (*os.File, error) {
|
|
return os.CreateTemp(os.TempDir(), pattern)
|
|
}
|
|
|
|
// OSTempFileWithContent create a temp file on os.TempDir() with the given content
|
|
func OSTempFileWithContent(content string) (*os.File, error) {
|
|
tmp, err := OSTempFile(ncrypt.Md5String(content))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err = os.WriteFile(tmp.Name(), []byte(content), os.ModeTemporary); err != nil {
|
|
return nil, err
|
|
}
|
|
return tmp, nil
|
|
}
|
|
|
|
// OSTempFilenameWithContent create a temp file on os.TempDir() with the given content and returns the filename (full path).
|
|
// The caller should remove the file by name after use.
|
|
func OSTempFilenameWithContent(content string) (string, error) {
|
|
tmp, err := OSTempFileWithContent(content)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
filename := tmp.Name()
|
|
if err = tmp.Close(); err != nil {
|
|
return "", err
|
|
}
|
|
return filename, nil
|
|
}
|
|
|
|
// TempFile is like os.CreateTemp, but can custom temp dir.
|
|
//
|
|
// Usage:
|
|
//
|
|
// nfs.TempFile("", "example.*.txt")
|
|
func TempFile(dir, pattern string) (*os.File, error) {
|
|
return os.CreateTemp(dir, pattern)
|
|
}
|
|
|
|
// OSTempDir creates a new temp dir on os.TempDir and return the temp dir path
|
|
//
|
|
// Usage:
|
|
//
|
|
// nfs.OSTempDir("example.*")
|
|
func OSTempDir(pattern string) (string, error) {
|
|
return os.MkdirTemp(os.TempDir(), pattern)
|
|
}
|
|
|
|
// TempDir creates a new temp dir and return the temp dir path
|
|
//
|
|
// Usage:
|
|
//
|
|
// nfs.TempDir("", "example.*")
|
|
// nfs.TempDir("testdata", "example.*")
|
|
func TempDir(dir, pattern string) (string, error) {
|
|
return os.MkdirTemp(dir, pattern)
|
|
}
|
|
|
|
// MimeType get File Mime Type name. eg "image/png"
|
|
func MimeType(path string) (mime string) {
|
|
file, err := os.Open(path)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
return ReaderMimeType(file)
|
|
}
|
|
|
|
// ReaderMimeType get the io.Reader mimeType
|
|
//
|
|
// Usage:
|
|
//
|
|
// file, err := os.Open(filepath)
|
|
// if err != nil {
|
|
// return
|
|
// }
|
|
// mime := ReaderMimeType(file)
|
|
func ReaderMimeType(r io.Reader) (mime string) {
|
|
var buf [MimeSniffLen]byte
|
|
n, _ := io.ReadFull(r, buf[:])
|
|
if n == 0 {
|
|
return ""
|
|
}
|
|
|
|
return http.DetectContentType(buf[:n])
|
|
}
|
|
|
|
// JoinPaths elements, alias of filepath.Join()
|
|
func JoinPaths(elem ...string) string {
|
|
return filepath.Join(elem...)
|
|
}
|
|
|
|
// JoinSubPaths elements, like the filepath.Join()
|
|
func JoinSubPaths(basePath string, elem ...string) string {
|
|
paths := make([]string, len(elem)+1)
|
|
paths[0] = basePath
|
|
copy(paths[1:], elem)
|
|
return filepath.Join(paths...)
|
|
}
|
|
|
|
// SlashPath alias of filepath.ToSlash
|
|
func SlashPath(path string) string {
|
|
return filepath.ToSlash(path)
|
|
}
|
|
|
|
// UnixPath like of filepath.ToSlash, but always replace
|
|
func UnixPath(path string) string {
|
|
if !strings.ContainsRune(path, '\\') {
|
|
return path
|
|
}
|
|
return strings.ReplaceAll(path, "\\", "/")
|
|
}
|
|
|
|
// ToAbsPath convert process. will expand home dir
|
|
//
|
|
// TIP: will don't check path
|
|
func ToAbsPath(p string) string {
|
|
if len(p) == 0 || IsAbsPath(p) {
|
|
return p
|
|
}
|
|
|
|
// expand home dir
|
|
if p[0] == '~' {
|
|
return common.ExpandHome(p)
|
|
}
|
|
|
|
wd, err := os.Getwd()
|
|
if err != nil {
|
|
return p
|
|
}
|
|
return filepath.Join(wd, p)
|
|
}
|