summaryrefslogtreecommitdiff
path: root/predict.go
diff options
context:
space:
mode:
authorEyal Posener <[email protected]>2017-05-11 22:00:21 +0300
committerEyal Posener <[email protected]>2017-05-11 22:02:05 +0300
commit200b1b5b690cf8b9202c52b6d85e52d99a3f0b4d (patch)
tree66868830fc73a8f3a5b57ff704d6e789ed9d5fdc /predict.go
parentd6cbc0e9690035368477b3ed8aa22aece0381b77 (diff)
predict: split file
Diffstat (limited to 'predict.go')
-rw-r--r--predict.go133
1 files changed, 0 insertions, 133 deletions
diff --git a/predict.go b/predict.go
index 64e793b..8207063 100644
--- a/predict.go
+++ b/predict.go
@@ -1,14 +1,5 @@
package complete
-import (
- "io/ioutil"
- "os"
- "path/filepath"
- "strings"
-
- "github.com/posener/complete/match"
-)
-
// Predictor implements a predict method, in which given
// command line arguments returns a list of options it predicts.
type Predictor interface {
@@ -48,127 +39,3 @@ var PredictNothing Predictor
// PredictAnything expects something, but nothing particular, such as a number
// or arbitrary name.
var PredictAnything = PredictFunc(func(Args) []string { return nil })
-
-// PredictSet expects specific set of terms, given in the options argument.
-func PredictSet(options ...string) Predictor {
- return predictSet(options)
-}
-
-type predictSet []string
-
-func (p predictSet) Predict(a Args) (prediction []string) {
- for _, m := range p {
- if match.Prefix(m, a.Last) {
- prediction = append(prediction, m)
- }
- }
- return
-}
-
-// PredictDirs will search for directories in the given started to be typed
-// path, if no path was started to be typed, it will complete to directories
-// in the current working directory.
-func PredictDirs(pattern string) Predictor {
- return files(pattern, false)
-}
-
-// PredictFiles will search for files matching the given pattern in the started to
-// be typed path, if no path was started to be typed, it will complete to files that
-// match the pattern in the current working directory.
-// To match any file, use "*" as pattern. To match go files use "*.go", and so on.
-func PredictFiles(pattern string) Predictor {
- return files(pattern, true)
-}
-
-func files(pattern string, allowFiles bool) PredictFunc {
- return func(a Args) (prediction []string) {
- if strings.HasSuffix(a.Last, "/..") {
- return
- }
- dir := dirFromLast(a.Last)
- rel := !filepath.IsAbs(pattern)
- Log("looking for files in %s (last=%s)", dir, a.Last)
- files := listFiles(dir, pattern)
-
- // get wording directory for relative name
- workDir, err := os.Getwd()
- if err != nil {
- workDir = ""
- }
-
- // add dir if match
- files = append(files, dir)
-
- // add all matching files to prediction
- for _, f := range files {
- if stat, err := os.Stat(f); err != nil || (!stat.IsDir() && !allowFiles) {
- continue
- }
-
- // change file name to relative if necessary
- if rel && workDir != "" {
- f = toRel(workDir, f)
- }
-
- // test matching of file to the argument
- if match.File(f, a.Last) {
- prediction = append(prediction, f)
- }
- }
- return
- }
-}
-func listFiles(dir, pattern string) []string {
- m := map[string]bool{}
- if files, err := filepath.Glob(filepath.Join(dir, pattern)); err == nil {
- for _, f := range files {
- m[f] = true
- }
- }
- if dirs, err := ioutil.ReadDir(dir); err == nil {
- for _, d := range dirs {
- if d.IsDir() {
- m[d.Name()] = true
- }
- }
- }
- list := make([]string, 0, len(m))
- for k := range m {
- list = append(list, k)
- }
- return list
-}
-
-// toRel changes a file name to a relative name
-func toRel(wd, file string) string {
- abs, err := filepath.Abs(file)
- if err != nil {
- return file
- }
- rel, err := filepath.Rel(wd, abs)
- if err != nil {
- return file
- }
- if rel != "." {
- rel = "./" + rel
- }
- if info, err := os.Stat(rel); err == nil && info.IsDir() {
- rel += "/"
- }
- return rel
-}
-
-// dirFromLast gives the directory of the current written
-// last argument if it represents a file name being written.
-// in case that it is not, we fall back to the current directory.
-func dirFromLast(last string) string {
- if info, err := os.Stat(last); err == nil && info.IsDir() {
- return last
- }
- dir := filepath.Dir(last)
- _, err := os.Stat(dir)
- if err != nil {
- return "./"
- }
- return dir
-}