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

Cleaner handling of modules start/stop

Browse source
This commit is contained in:
evilsocket 2017-11-17 17:06:41 +01:00
parent 1ee605d01d
commit b1678b5902
6 changed files with 66 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

26
session/module_handler.go
View file

@ -15,12 +15,18 @@ type ModuleHandler struct {
}
func NewModuleHandler(name string, expr string, desc string, exec func(args []string) error) ModuleHandler {
return ModuleHandler{
h := ModuleHandler{
Name: name,
Description: desc,
Parser: regexp.MustCompile(expr),
Parser: nil,
Exec: exec,
}
if expr != "" {
h.Parser = regexp.MustCompile(expr)
}
return h
}
func (h *ModuleHandler) Help(padding int) string {
@ -28,10 +34,18 @@ func (h *ModuleHandler) Help(padding int) string {
}
func (h *ModuleHandler) Parse(line string) (bool, []string) {
result := h.Parser.FindStringSubmatch(line)
if len(result) == h.Parser.NumSubexp()+1 {
return true, result[1:len(result)]
if h.Parser == nil {
if line == h.Name {
return true, nil
} else {
return false, nil
}
} else {
return false, nil
result := h.Parser.FindStringSubmatch(line)
if len(result) == h.Parser.NumSubexp()+1 {
return true, result[1:len(result)]
} else {
return false, nil
}
}
}