mirror of
https://github.com/bettercap/bettercap.git
synced 2025-03-12 04:36:03 -07:00
34 lines
752 B
Go
34 lines
752 B
Go
package session
|
|
|
|
import (
|
|
"github.com/bettercap/readline"
|
|
"regexp"
|
|
)
|
|
|
|
type CommandHandler struct {
|
|
Name string
|
|
Description string
|
|
Completer *readline.PrefixCompleter
|
|
Parser *regexp.Regexp
|
|
Exec func(args []string, s *Session) error
|
|
}
|
|
|
|
func NewCommandHandler(name string, expr string, desc string, exec func(args []string, s *Session) error) CommandHandler {
|
|
return CommandHandler{
|
|
Name: name,
|
|
Description: desc,
|
|
Completer: nil,
|
|
Parser: regexp.MustCompile(expr),
|
|
Exec: exec,
|
|
}
|
|
}
|
|
|
|
func (h *CommandHandler) Parse(line string) (bool, []string) {
|
|
result := h.Parser.FindStringSubmatch(line)
|
|
if len(result) == h.Parser.NumSubexp()+1 {
|
|
return true, result[1:]
|
|
} else {
|
|
return false, nil
|
|
}
|
|
}
|