github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-14 02:36:57 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: fixed bug in prompt parsing (fixes #58)

Browse source
This commit is contained in:
evilsocket 2018-02-20 02:39:59 +01:00
commit b28f8a40ec
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
session/prompt.go
View file

@ -56,7 +56,7 @@ var PromptCallbacks = map[string]func(s *Session) string{
},
}
var envRe = regexp.MustCompile("{env\\.(.+)}")
var envRe = regexp.MustCompile("{env\\.([^}]+)}")
type Prompt struct {
}
@ -79,11 +79,11 @@ func (p Prompt) Render(s *Session) string {
prompt = strings.Replace(prompt, tok, cb(s), -1)
}
m := envRe.FindStringSubmatch(prompt)
if len(m) == 2 {
name := m[1]
m := envRe.FindAllString(prompt, -1)
for _, match := range m {
name := strings.Trim(strings.Replace(match, "env.", "", -1), "{}")
_, value := s.Env.Get(name)
prompt = strings.Replace(prompt, m[0], value, -1)
prompt = strings.Replace(prompt, match, value, -1)
}
// make sure an user error does not screw all terminal