github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-21 05:53:20 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

new: ble, can, hid and wifi modules will now set a custom prompt (closes #1117)

Browse source
This commit is contained in:
Simone Margaritelli 2024-08-17 12:10:38 +02:00
commit 6282fe3451
7 changed files with 80 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

18
session/session.go
View file

@ -458,10 +458,18 @@ func (s *Session) Run(line string) error {
}
// is it a module command?
for _, m := range s.Modules {
for _, h := range m.Handlers() {
if parsed, args := h.Parse(line); parsed {
return h.Exec(args)
for _, mod := range s.Modules {
for _, modHandler := range mod.Handlers() {
if parsed, args := modHandler.Parse(line); parsed {
if err := modHandler.Exec(args); err != nil {
return err
} else if prompt := mod.Prompt(); prompt != "" {
// if the module handler has been executed successfully and
// the module overrides the prompt, set it
s.Env.Set(PromptVariable, prompt)
s.Refresh()
}
return nil
}
}
}
@ -478,5 +486,5 @@ func (s *Session) Run(line string) error {
return nil
}
return fmt.Errorf("unknown or invalid syntax \"%s%s%s\", type %shelp%s for the help menu.", tui.BOLD, line, tui.RESET, tui.BOLD, tui.RESET)
return fmt.Errorf("unknown or invalid syntax \"%s%s%s\", type %shelp%s for the help menu", tui.BOLD, line, tui.RESET, tui.BOLD, tui.RESET)
}