117 lines
2.6 KiB
Go
117 lines
2.6 KiB
Go
package os
|
|
|
|
import "os"
|
|
import "io"
|
|
import "time"
|
|
import "io/fs"
|
|
import "html/template"
|
|
import "git.tebibyte.media/sashakoshka/step"
|
|
|
|
var _ step.FuncProvider = new(Provider)
|
|
|
|
// FileInfo is like the fs.FileInfo interface, but it directly contains the
|
|
// values instead of having getters for them.
|
|
type FileInfo struct {
|
|
Name string
|
|
Size int64
|
|
Mode fs.FileMode
|
|
ModTime time.Time
|
|
IsDir bool
|
|
}
|
|
|
|
// Provider provides OS functions.
|
|
type Provider struct {
|
|
|
|
}
|
|
|
|
// FuncMap fulfills the step.FuncProvider interface.
|
|
func (this *Provider) FuncMap () template.FuncMap {
|
|
return template.FuncMap {
|
|
"env": funcEnv,
|
|
"exists": funcExists,
|
|
"isFile": funcIsFile,
|
|
"isDir": funcIsDir,
|
|
"statFile": funcStatFile,
|
|
"readFile": funcReadFile,
|
|
"readDir": funcReadDir,
|
|
"writeFile": funcWriteFile,
|
|
"appendFile": funcAppendFile,
|
|
"renameFile": funcRenameFile,
|
|
"removeFile": funcRemoveFile,
|
|
}
|
|
}
|
|
|
|
func funcEnv (name string) string {
|
|
return os.Getenv(name)
|
|
}
|
|
|
|
func funcExists (name string) (bool) {
|
|
_, err := os.Stat(name)
|
|
return err == nil
|
|
}
|
|
|
|
func funcIsFile (name string) bool {
|
|
info, err := os.Stat(name)
|
|
if err != nil { return false }
|
|
return info.Mode().IsRegular()
|
|
}
|
|
|
|
func funcIsDir (name string) bool {
|
|
info, err := os.Stat(name)
|
|
if err != nil { return false }
|
|
return info.IsDir()
|
|
}
|
|
|
|
func funcStatFile (name string) (FileInfo, error) {
|
|
info, err := os.Stat(name)
|
|
if err != nil { return FileInfo { }, err }
|
|
return FileInfo {
|
|
Name: info.Name(),
|
|
Size: info.Size(),
|
|
Mode: info.Mode(),
|
|
ModTime: info.ModTime(),
|
|
IsDir: info.IsDir(),
|
|
}, nil
|
|
}
|
|
|
|
func funcReadFile (name string) (string, error) {
|
|
file, err := os.Open(name)
|
|
if err != nil { return "", err }
|
|
defer file.Close()
|
|
buffer, err := io.ReadAll(file)
|
|
if err != nil { return "", err }
|
|
return string(buffer), nil
|
|
}
|
|
|
|
func funcReadDir (name string) ([]string, error) {
|
|
entries, err := os.ReadDir(name)
|
|
if err != nil { return nil, err }
|
|
names := make([]string, len(entries))
|
|
for index, entry := range entries {
|
|
names[index] = entry.Name()
|
|
}
|
|
return names, nil
|
|
}
|
|
|
|
func funcWriteFile (name, content string) error {
|
|
file, err := os.Create(name)
|
|
if err != nil { return err }
|
|
_, err = io.WriteString(file, content)
|
|
return err
|
|
}
|
|
|
|
func funcAppendFile (name, content string) error {
|
|
file, err := os.OpenFile(name, os.O_CREATE | os.O_WRONLY | os.O_APPEND, 644)
|
|
if err != nil { return err }
|
|
_, err = io.WriteString(file, content)
|
|
return err
|
|
}
|
|
|
|
func funcRenameFile (name, newName string) error {
|
|
return os.Rename(name, newName)
|
|
}
|
|
|
|
func funcRemoveFile (name string) error {
|
|
return os.Remove(name)
|
|
}
|