elvish/pkg/eval/builtin_fn_fs.go
2019-12-23 20:00:59 +00:00

86 lines
1.6 KiB
Go

package eval
import (
"errors"
"os"
"path/filepath"
"github.com/elves/elvish/pkg/eval/vals"
"github.com/elves/elvish/pkg/store/storedefs"
"github.com/elves/elvish/pkg/util"
)
// Filesystem.
// ErrStoreNotConnected is thrown by dir-history when the store is not connected.
var ErrStoreNotConnected = errors.New("store not connected")
func init() {
addBuiltinFns(map[string]interface{}{
// Directory
"cd": cd,
"dir-history": dirs,
// Path
"path-abs": filepath.Abs,
"path-base": filepath.Base,
"path-clean": filepath.Clean,
"path-dir": filepath.Dir,
"path-ext": filepath.Ext,
"eval-symlinks": filepath.EvalSymlinks,
"tilde-abbr": tildeAbbr,
// File types
"-is-dir": isDir,
})
}
func cd(fm *Frame, args ...string) error {
var dir string
switch len(args) {
case 0:
var err error
dir, err = util.GetHome("")
if err != nil {
return err
}
case 1:
dir = args[0]
default:
return ErrArgs
}
return fm.Chdir(dir)
}
type dirHistoryEntry struct {
Path string `json:"path"`
Score float64 `json:"score"`
}
func (dirHistoryEntry) IsStructMap(vals.StructMapMarker) {}
func dirs(fm *Frame) error {
if fm.DaemonClient == nil {
return ErrStoreNotConnected
}
dirs, err := fm.DaemonClient.Dirs(storedefs.NoBlacklist)
if err != nil {
return err
}
out := fm.ports[1].Chan
for _, dir := range dirs {
out <- dirHistoryEntry{dir.Path, dir.Score}
}
return nil
}
func tildeAbbr(path string) string {
return util.TildeAbbr(path)
}
func isDir(path string) bool {
fi, err := os.Stat(path)
return err == nil && fi.Mode().IsDir()
}