github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-14 10:46:57 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

new: preloading and exporting caplets from api.rest

Browse source
This commit is contained in:
evilsocket 2019-03-19 19:10:40 +01:00
commit 3edf80fc99
No known key found for this signature in database
GPG key ID: 1564D7F30393A456
3 changed files with 67 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

29
caplets/manager.go
View file

@ -16,22 +16,22 @@ var (
cacheLock = sync.Mutex{}
)
func List() []Caplet {
caplets := make([]Caplet, 0)
func List() []*Caplet {
caplets := make([]*Caplet, 0)
for _, searchPath := range LoadPaths {
files, _ := filepath.Glob(searchPath + "/*" + Suffix)
files2, _ := filepath.Glob(searchPath + "/*/*" + Suffix)
for _, fileName := range append(files, files2...) {
if stats, err := os.Stat(fileName); err == nil {
if _, err := os.Stat(fileName); err == nil {
base := strings.Replace(fileName, searchPath+"/", "", -1)
base = strings.Replace(base, Suffix, "", -1)
caplets = append(caplets, Caplet{
Name: base,
Path: fileName,
Size: stats.Size(),
})
if err, caplet := Load(base); err != nil {
fmt.Fprintf(os.Stderr, "wtf: %v\n", err)
} else {
caplets = append(caplets, caplet)
}
}
}
}
@ -51,6 +51,7 @@ func Load(name string) (error, *Caplet) {
return nil, caplet
}
baseName := name
names := []string{}
if !strings.HasSuffix(name, Suffix) {
name += Suffix
@ -64,16 +65,18 @@ func Load(name string) (error, *Caplet) {
names = append(names, name)
}
for _, filename := range names {
if fs.Exists(filename) {
for _, fileName := range names {
if stats, err := os.Stat(fileName); err == nil {
cap := &Caplet{
Path: filename,
Name: baseName,
Path: fileName,
Code: make([]string, 0),
Size: stats.Size(),
}
cache[name] = cap
if reader, err := fs.LineReader(filename); err != nil {
return fmt.Errorf("error reading caplet %s: %v", filename, err), nil
if reader, err := fs.LineReader(fileName); err != nil {
return fmt.Errorf("error reading caplet %s: %v", fileName, err), nil
} else {
for line := range reader {
if line == "" || line[0] == '#' {