mirror of
https://github.com/bettercap/bettercap
synced 2025-07-05 20:42:09 -07:00
37 lines
854 B
Go
37 lines
854 B
Go
package session
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/evilsocket/bettercap/core"
|
|
"regexp"
|
|
"strconv"
|
|
)
|
|
|
|
type ModuleHandler struct {
|
|
Name string
|
|
Description string
|
|
Parser *regexp.Regexp
|
|
Exec func(args []string) error
|
|
}
|
|
|
|
func NewModuleHandler(name string, expr string, desc string, exec func(args []string) error) ModuleHandler {
|
|
return ModuleHandler{
|
|
Name: name,
|
|
Description: desc,
|
|
Parser: regexp.MustCompile(expr),
|
|
Exec: exec,
|
|
}
|
|
}
|
|
|
|
func (h *ModuleHandler) Help(padding int) string {
|
|
return fmt.Sprintf(" "+core.Bold("%"+strconv.Itoa(padding)+"s")+" : %s\n", h.Name, h.Description)
|
|
}
|
|
|
|
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)]
|
|
} else {
|
|
return false, nil
|
|
}
|
|
}
|