Hilbish/golibs/fs/fs.go

109 lines
2.2 KiB
Go
Raw Normal View History

// The fs module provides easy and simple access to filesystem functions and other
// things, and acts an addition to the Lua standard library's I/O and fs functions.
package fs
import (
"strconv"
"os"
2021-03-31 17:46:49 +00:00
"strings"
"hilbish/util"
"github.com/yuin/gopher-lua"
)
func Loader(L *lua.LState) int {
2021-04-28 11:26:23 +00:00
mod := L.SetFuncs(L.NewTable(), exports)
util.Document(L, mod, `The fs module provides easy and simple access to
filesystem functions and other things, and acts an
addition to the Lua standard library's I/O and fs functions.`)
2021-04-28 11:26:23 +00:00
L.Push(mod)
return 1
}
func luaErr(L *lua.LState, msg string) {
L.Error(lua.LString(msg), 2)
2021-03-31 03:56:37 +00:00
}
var exports = map[string]lua.LGFunction{
"cd": fcd,
"mkdir": fmkdir,
"stat": fstat,
"readdir": freaddir,
}
// cd(dir)
// Changes directory to `dir`
func fcd(L *lua.LState) int {
path := L.CheckString(1)
2021-03-31 17:46:49 +00:00
err := os.Chdir(strings.TrimSpace(path))
2021-03-31 03:56:37 +00:00
if err != nil {
e := err.(*os.PathError).Err.Error()
luaErr(L, e)
2021-03-31 03:56:37 +00:00
}
return 0
}
// mkdir(name, recursive)
// Makes a directory called `name`. If `recursive` is true, it will create its parent directories.
func fmkdir(L *lua.LState) int {
dirname := L.CheckString(1)
recursive := L.ToBool(2)
path := strings.TrimSpace(dirname)
var err error
if recursive {
err = os.MkdirAll(path, 0744)
} else {
err = os.Mkdir(path, 0744)
}
if err != nil {
luaErr(L, err.Error())
}
return 0
}
// stat(path)
// Returns info about `path`
func fstat(L *lua.LState) int {
path := L.CheckString(1)
pathinfo, err := os.Stat(path)
if err != nil {
luaErr(L, err.Error())
return 0
}
statTbl := L.NewTable()
L.SetField(statTbl, "name", lua.LString(pathinfo.Name()))
L.SetField(statTbl, "size", lua.LNumber(pathinfo.Size()))
L.SetField(statTbl, "mode", lua.LString("0" + strconv.FormatInt(int64(pathinfo.Mode().Perm()), 8)))
L.SetField(statTbl, "isDir", lua.LBool(pathinfo.IsDir()))
L.Push(statTbl)
return 1
}
// readdir(dir)
// Returns a table of files in `dir`
func freaddir(L *lua.LState) int {
dir := L.CheckString(1)
names := L.NewTable()
dirEntries, err := os.ReadDir(dir)
if err != nil {
luaErr(L, err.Error())
return 0
}
for _, entry := range dirEntries {
names.Append(lua.LString(entry.Name()))
}
L.Push(names)
return 1
}