mirror of https://github.com/Hilbis/Hilbish
109 lines
2.2 KiB
Go
109 lines
2.2 KiB
Go
// 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"
|
|
"strings"
|
|
|
|
"hilbish/util"
|
|
"github.com/yuin/gopher-lua"
|
|
)
|
|
|
|
func Loader(L *lua.LState) int {
|
|
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.`)
|
|
|
|
L.Push(mod)
|
|
return 1
|
|
}
|
|
|
|
func luaErr(L *lua.LState, msg string) {
|
|
L.Error(lua.LString(msg), 2)
|
|
}
|
|
|
|
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)
|
|
|
|
err := os.Chdir(strings.TrimSpace(path))
|
|
if err != nil {
|
|
e := err.(*os.PathError).Err.Error()
|
|
luaErr(L, e)
|
|
}
|
|
|
|
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
|
|
}
|