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

new: custom prompt

Browse source
This commit is contained in:
evilsocket 2018-01-11 16:38:02 +01:00
commit 7dccb87aa8
2 changed files with 91 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

13
session/session.go
View file

@ -32,6 +32,7 @@ type Session struct {
Queue *packets.Queue `json:"-"`
Input *readline.Instance `json:"-"`
Active bool `json:"active"`
Prompt Prompt `json:"-"`
CoreHandlers []CommandHandler `json:"-"`
Modules []Module `json:"-"`
@ -44,6 +45,7 @@ func New() (*Session, error) {
var err error
s := &Session{
Prompt: NewPrompt(),
Env: nil,
Active: false,
Queue: nil,
@ -161,6 +163,8 @@ func (s *Session) Start() error {
return err
}
s.Env.Set(PromptVariable, DefaultPrompt)
s.Env.Set("iface.name", s.Interface.Name())
s.Env.Set("iface.address", s.Interface.IpAddress)
s.Env.Set("iface.mac", s.Interface.HwAddress)
@ -224,14 +228,7 @@ func (s *Session) Start() error {
}
func (s *Session) ReadLine() (string, error) {
prompt := core.FG_WHITE + core.BG_YELLOW + " " + s.Interface.CIDR() +
core.FG_BLACK +
" > " +
s.Interface.IpAddress +
" " + core.RESET +
core.BOLD + " » " + core.RESET
s.Input.SetPrompt(prompt)
s.Input.SetPrompt(s.Prompt.Render(s))
s.Input.Refresh()
return s.Input.Readline()
}