github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-07-07 13:32:07 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

add readDir function to JS plugin

Browse source
This commit is contained in:
yungtravla 2019-03-08 14:20:01 +10:00 • committed by GitHub
parent 963840915b
commit 306771208a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 28 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

29
modules/utils/script_builtins.go
View file

@ -20,6 +20,33 @@ func errOtto(format string, args ...interface{}) otto.Value {
}
func init() {
// used to read a directory (returns string array)
plugin.Defines["readDir"] = func(call otto.FunctionCall) otto.Value {
argv := call.ArgumentList
argc := len(argv)
if argc != 1 {
return errOtto("readDir: expected 1 argument, %d given instead.", argc)
}
path := argv[0].String()
dir, err := ioutil.ReadDir(path)
if err != nil {
return errOtto("Could not read directory %s: %s", path, err)
}
entry_list := []string{}
for _, file := range dir {
entry_list = append( entry_list, file.Name() )
}
v, err := otto.Otto.ToValue(*call.Otto, entry_list)
if err != nil {
return errOtto("Could not convert to array: %s", err)
}
return v
}
// used to read a file ... doh
plugin.Defines["readFile"] = func(call otto.FunctionCall) otto.Value {
argv := call.ArgumentList
@ -31,7 +58,7 @@ func init() {
filename := argv[0].String()
raw, err := ioutil.ReadFile(filename)
if err != nil {
return errOtto("Could not read %s: %s", filename, err)
return errOtto("Could not read file %s: %s", filename, err)
}
v, err := otto.ToValue(string(raw))