predict/predict.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

// Package predict provides helper functions for completion predictors.
package predict

import "github.com/posener/complete/v2"

// Set predicts a set of predefined values.
type Set []string

func (p Set) Predict(_ string) (options []string) {
	return p
}

var (
	// Something is used to indicate that does not completes somthing. Such that other prediction
	// wont be applied.
	Something = Set{""}

	// Nothing is used to indicate that does not completes anything.
	Nothing = Set{}
)

// Or unions prediction functions, so that the result predication is the union of their
// predications.
func Or(ps ...complete.Predictor) complete.Predictor {
	return complete.PredictFunc(func(prefix string) (options []string) {
		for _, p := range ps {
			if p == nil {
				continue
			}
			options = append(options, p.Predict(prefix)...)
		}
		return
	})
}