mirror of
https://github.com/1Panel-dev/1Panel.git
synced 2024-11-15 20:20:11 +08:00
64 lines
1.1 KiB
Go
64 lines
1.1 KiB
Go
package files
|
|
|
|
import (
|
|
"github.com/spf13/afero"
|
|
"io"
|
|
"io/fs"
|
|
"os"
|
|
"path"
|
|
)
|
|
|
|
type FileOp struct {
|
|
Fs afero.Fs
|
|
}
|
|
|
|
func NewFileOp() FileOp {
|
|
return FileOp{
|
|
Fs: afero.NewOsFs(),
|
|
}
|
|
}
|
|
|
|
func (f FileOp) CreateDir(dst string, mode fs.FileMode) error {
|
|
return f.Fs.MkdirAll(dst, mode)
|
|
}
|
|
|
|
func (f FileOp) DeleteDir(dst string) error {
|
|
return f.Fs.RemoveAll(dst)
|
|
}
|
|
|
|
func (f FileOp) Stat(dst string) bool {
|
|
info, _ := f.Fs.Stat(dst)
|
|
if info != nil {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (f FileOp) DeleteFile(dst string) error {
|
|
return f.Fs.Remove(dst)
|
|
}
|
|
|
|
func (f FileOp) WriteFile(dst string, in io.Reader, mode fs.FileMode) error {
|
|
dir, _ := path.Split(dst)
|
|
if err := f.Fs.MkdirAll(dir, mode); err != nil {
|
|
return err
|
|
}
|
|
file, err := f.Fs.OpenFile(dst, os.O_RDWR|os.O_CREATE|os.O_TRUNC, mode)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer file.Close()
|
|
|
|
if _, err = io.Copy(file, in); err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err = file.Stat(); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (f FileOp) Chmod(dst string, mode fs.FileMode) error {
|
|
return f.Fs.Chmod(dst, mode)
|
|
}
|