2017-07-17 20:19:13 +00:00
|
|
|
package prompt
|
|
|
|
|
|
|
|
import "strings"
|
|
|
|
|
2017-08-04 11:30:50 +00:00
|
|
|
type CompletionFilter func([]Suggest, string, bool) []Suggest
|
2017-07-18 10:04:11 +00:00
|
|
|
|
2017-08-04 11:30:50 +00:00
|
|
|
func FilterHasPrefix(completions []Suggest, sub string, ignoreCase bool) []Suggest {
|
2017-07-17 20:19:13 +00:00
|
|
|
if sub == "" {
|
|
|
|
return completions
|
|
|
|
}
|
|
|
|
if ignoreCase {
|
|
|
|
sub = strings.ToUpper(sub)
|
|
|
|
}
|
|
|
|
|
2017-08-04 11:30:50 +00:00
|
|
|
ret := make([]Suggest, 0, len(completions))
|
2017-07-18 17:12:22 +00:00
|
|
|
for i := range completions {
|
|
|
|
c := completions[i].Text
|
2017-07-17 20:19:13 +00:00
|
|
|
if ignoreCase {
|
2017-07-18 17:12:22 +00:00
|
|
|
c = strings.ToUpper(c)
|
2017-07-17 20:19:13 +00:00
|
|
|
}
|
2017-07-18 17:12:22 +00:00
|
|
|
if strings.HasPrefix(c, sub) {
|
2017-07-17 20:19:13 +00:00
|
|
|
ret = append(ret, completions[i])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return ret
|
|
|
|
}
|
|
|
|
|
2017-08-04 11:30:50 +00:00
|
|
|
func FilterHasSuffix(completions []Suggest, sub string, ignoreCase bool) []Suggest {
|
2017-07-17 20:19:13 +00:00
|
|
|
if sub == "" {
|
|
|
|
return completions
|
|
|
|
}
|
|
|
|
if ignoreCase {
|
|
|
|
sub = strings.ToUpper(sub)
|
|
|
|
}
|
|
|
|
|
2017-08-04 11:30:50 +00:00
|
|
|
ret := make([]Suggest, 0, len(completions))
|
2017-07-18 17:12:22 +00:00
|
|
|
for i := range completions {
|
|
|
|
c := completions[i].Text
|
2017-07-17 20:19:13 +00:00
|
|
|
if ignoreCase {
|
2017-07-18 17:12:22 +00:00
|
|
|
c = strings.ToUpper(c)
|
2017-07-17 20:19:13 +00:00
|
|
|
}
|
2017-07-18 17:12:22 +00:00
|
|
|
if strings.HasSuffix(c, sub) {
|
2017-07-17 20:19:13 +00:00
|
|
|
ret = append(ret, completions[i])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return ret
|
|
|
|
}
|
|
|
|
|
2017-08-04 11:30:50 +00:00
|
|
|
func FilterContains(completions []Suggest, sub string, ignoreCase bool) []Suggest {
|
2017-07-17 20:19:13 +00:00
|
|
|
if sub == "" {
|
|
|
|
return completions
|
|
|
|
}
|
|
|
|
if ignoreCase {
|
|
|
|
sub = strings.ToUpper(sub)
|
|
|
|
}
|
|
|
|
|
2017-08-04 11:30:50 +00:00
|
|
|
ret := make([]Suggest, 0, len(completions))
|
2017-07-18 17:12:22 +00:00
|
|
|
for i := range completions {
|
|
|
|
c := completions[i].Text
|
2017-07-17 20:19:13 +00:00
|
|
|
if ignoreCase {
|
2017-07-18 17:12:22 +00:00
|
|
|
c = strings.ToUpper(c)
|
2017-07-17 20:19:13 +00:00
|
|
|
}
|
2017-07-18 17:12:22 +00:00
|
|
|
if strings.Contains(c, sub) {
|
2017-07-18 02:36:25 +00:00
|
|
|
ret = append(ret, completions[i])
|
2017-07-17 20:19:13 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return ret
|
|
|
|
}
|