mirror of
https://github.com/bettercap/bettercap
synced 2025-08-14 18:57:17 -07:00
refact: new ListParam utility method for modules CSV parameters
This commit is contained in:
parent
fb593af38d
commit
90c1550a19
3 changed files with 46 additions and 40 deletions
|
@ -2,6 +2,7 @@ package session
|
|||
|
||||
import (
|
||||
"fmt"
|
||||
"strings"
|
||||
"sync"
|
||||
)
|
||||
|
||||
|
@ -53,6 +54,23 @@ func (m *SessionModule) Param(name string) *ModuleParam {
|
|||
return m.params[name]
|
||||
}
|
||||
|
||||
func (m SessionModule) ListParam(name string) (err error, values []string) {
|
||||
values = make([]string, 0)
|
||||
list := ""
|
||||
if err, list = m.StringParam(name); err != nil {
|
||||
return
|
||||
} else {
|
||||
parts := strings.Split(list, ",")
|
||||
for _, part := range parts {
|
||||
part = strings.Trim(part, "\t\n\r ")
|
||||
if part != "" {
|
||||
values = append(values, part)
|
||||
}
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (m SessionModule) StringParam(name string) (error, string) {
|
||||
if p, found := m.params[name]; found == true {
|
||||
if err, v := p.Get(m.Session); err != nil {
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue