lf/comp.go

package main

import (
	"io/ioutil"
	"log"
	"os"
	"path"
	"strings"
)

var (
	gCmdWords = []string{"set", "map", "cmd"}
	gOptWords = []string{
		"preview",
		"nopreview",
		"preview!",
		"hidden",
		"nohidden",
		"hidden!",
		"tabstop",
		"scrolloff",
		"sortby",
		"showinfo",
		"opener",
		"ratios",
	}
)

func matchLongest(s1, s2 string) string {
	i := 0
	for ; i < len(s1) && i < len(s2); i++ {
		if s1[i] != s2[i] {
			break
		}
	}
	return s1[:i]
}

func matchWord(s string, words []string) string {
	var match string

	for _, w := range words {
		if strings.HasPrefix(w, s) {
			if match != "" {
				match = matchLongest(match, w)
			} else {
				match = w + " "
			}
		}
	}

	if match != "" {
		return match
	}

	return s
}

func matchExec(s string) string {
	var match string

	paths := strings.Split(envPath, ":")

	for _, p := range paths {
		if _, err := os.Stat(p); os.IsNotExist(err) {
			continue
		}

		fi, err := ioutil.ReadDir(p)
		if err != nil {
			log.Print(err)
		}

		for _, f := range fi {
			if strings.HasPrefix(f.Name(), s) {
				f, err = os.Stat(path.Join(p, f.Name()))
				if err != nil {
					log.Print(err)
				}

				if !f.Mode().IsRegular() || f.Mode()&0111 == 0 {
					continue
				}
				if match != "" {
					match = matchLongest(match, f.Name())
				} else {
					match = f.Name() + " "
				}
			}
		}
	}

	if match != "" {
		return match
	}

	return s
}

func matchFile(s string) string {
	var match string

	wd, err := os.Getwd()
	if err != nil {
		log.Print(err)
	}

	fi, err := ioutil.ReadDir(wd)
	if err != nil {
		log.Print(err)
	}

	for _, f := range fi {
		if strings.HasPrefix(f.Name(), s) {
			if match != "" {
				match = matchLongest(match, f.Name())
			} else {
				match = f.Name() + " "
			}
		}
	}

	if match != "" {
		return match
	}

	return s
}

func compCmd(acc []rune) []rune {
	if len(acc) == 0 || acc[len(acc)-1] == ' ' {
		return acc
	}

	s := string(acc)
	f := strings.Fields(s)

	switch len(f) {
	case 0: // do nothing
	case 1:
		words := gCmdWords
		for c, _ := range gOpts.cmds {
			words = append(words, c)
		}
		return []rune(matchWord(s, words))
	default:
		switch f[0] {
		case "set":
			opt := matchWord(f[1], gOptWords)
			ret := []rune(f[0])
			ret = append(ret, ' ')
			ret = append(ret, []rune(opt)...)
			return ret
		case "map", "cmd": // do nothing
		default:
			ret := []rune(f[0])
			ret = append(ret, ' ')
			for i := 1; i < len(f); i++ {
				name := matchFile(f[i])
				ret = append(ret, []rune(name)...)
			}
			return ret
		}
	}

	return acc
}

func compShell(acc []rune) []rune {
	if len(acc) == 0 || acc[len(acc)-1] == ' ' {
		return acc
	}

	s := string(acc)
	f := strings.Fields(s)

	switch len(f) {
	case 0: // do nothing
	case 1:
		return []rune(matchExec(s))
	default:
		ret := []rune(f[0])
		ret = append(ret, ' ')
		for i := 1; i < len(f); i++ {
			name := matchFile(f[i])
			ret = append(ret, []rune(name)...)
		}
		return ret
	}

	return acc
}