mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 22:30:22 -04:00
75 lines
1.4 KiB
Go
75 lines
1.4 KiB
Go
package driver
|
|
|
|
import (
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"sort"
|
|
)
|
|
|
|
// ReadFile works the same as ioutil.ReadFile with the Driver abstraction
|
|
func ReadFile(r Driver, filename string) ([]byte, error) {
|
|
f, err := r.Open(filename)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer f.Close()
|
|
|
|
data, err := ioutil.ReadAll(f)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return data, nil
|
|
}
|
|
|
|
// WriteFile works the same as ioutil.WriteFile with the Driver abstraction
|
|
func WriteFile(r Driver, filename string, data []byte, perm os.FileMode) error {
|
|
f, err := r.OpenFile(filename, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, perm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
|
|
n, err := f.Write(data)
|
|
if err != nil {
|
|
return err
|
|
} else if n != len(data) {
|
|
return io.ErrShortWrite
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// ReadDir works the same as ioutil.ReadDir with the Driver abstraction
|
|
func ReadDir(r Driver, dirname string) ([]os.FileInfo, error) {
|
|
f, err := r.Open(dirname)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer f.Close()
|
|
|
|
dirs, err := f.Readdir(-1)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
sort.Sort(fileInfos(dirs))
|
|
return dirs, nil
|
|
}
|
|
|
|
// Simple implementation of the sort.Interface for os.FileInfo
|
|
type fileInfos []os.FileInfo
|
|
|
|
func (fis fileInfos) Len() int {
|
|
return len(fis)
|
|
}
|
|
|
|
func (fis fileInfos) Less(i, j int) bool {
|
|
return fis[i].Name() < fis[j].Name()
|
|
}
|
|
|
|
func (fis fileInfos) Swap(i, j int) {
|
|
fis[i], fis[j] = fis[j], fis[i]
|
|
}
|