summaryrefslogtreecommitdiff
path: root/predict.go
diff options
context:
space:
mode:
authorEyal Posener <[email protected]>2017-05-11 22:23:41 +0300
committerGitHub <[email protected]>2017-05-11 22:23:41 +0300
commitbd51afb7f204fb907fb9a16fec4491028e50c8c7 (patch)
tree9c731042ef4f7e6d9fba2ae1225c36fe398e768f /predict.go
parentd3bbb859d52b45987e3cd2098e28423f32edd999 (diff)
parent026906bf106ac1c18117443a26e6efa196bd9dae (diff)
Merge pull request #13 from posener/improves
Improves
Diffstat (limited to 'predict.go')
-rw-r--r--predict.go125
1 files changed, 0 insertions, 125 deletions
diff --git a/predict.go b/predict.go
index 9e1cce9..8207063 100644
--- a/predict.go
+++ b/predict.go
@@ -1,12 +1,5 @@
package complete
-import (
- "os"
- "path/filepath"
-
- "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 {
@@ -46,121 +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, true, 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, false, true)
-}
-
-// PredictFilesOrDirs any file or directory that matches the pattern
-func PredictFilesOrDirs(pattern string) Predictor {
- return files(pattern, true, true)
-}
-
-func files(pattern string, allowDirs, allowFiles bool) PredictFunc {
- return func(a Args) (prediction []string) {
- dir := dirFromLast(a.Last)
- Log("looking for files in %s (last=%s)", dir, a.Last)
- files, err := filepath.Glob(filepath.Join(dir, pattern))
- if err != nil {
- Log("failed glob operation with pattern '%s': %s", pattern, err)
- }
- if allowDirs {
- files = append(files, dir)
- }
- files = selectByType(files, allowDirs, allowFiles)
- if !filepath.IsAbs(pattern) {
- filesToRel(files)
- }
- // add all matching files to prediction
- for _, f := range files {
- if match.File(f, a.Last) {
- prediction = append(prediction, f)
- }
- }
- return
- }
-}
-
-func selectByType(names []string, allowDirs bool, allowFiles bool) []string {
- filtered := make([]string, 0, len(names))
- for _, name := range names {
- stat, err := os.Stat(name)
- if err != nil {
- continue
- }
- if (stat.IsDir() && !allowDirs) || (!stat.IsDir() && !allowFiles) {
- continue
- }
- filtered = append(filtered, name)
- }
- return filtered
-}
-
-// filesToRel, change list of files to their names in the relative
-// to current directory form.
-func filesToRel(files []string) {
- wd, err := os.Getwd()
- if err != nil {
- return
- }
- for i := range files {
- abs, err := filepath.Abs(files[i])
- if err != nil {
- continue
- }
- rel, err := filepath.Rel(wd, abs)
- if err != nil {
- continue
- }
- if rel != "." {
- rel = "./" + rel
- }
- if info, err := os.Stat(rel); err == nil && info.IsDir() {
- rel += "/"
- }
- files[i] = rel
- }
- return
-}
-
-// 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
-}