github.com/elves/elvish@v0.15.0/pkg/edit/ns_helper.go

package edit

import (
	"os"
	"strings"

	"github.com/elves/elvish/pkg/eval"
	"github.com/elves/elvish/pkg/fsutil"
)

// Calls the passed function for each variable name in namespace ns that can be
// found from the top context.
func eachVariableInTop(builtin, global *eval.Ns, ns string, f func(s string)) {
	switch ns {
	case "builtin:":
		builtin.IterateNames(f)
	case "", ":":
		global.IterateNames(f)
		builtin.IterateNames(f)
	case "e:":
		fsutil.EachExternal(func(cmd string) {
			f(cmd + eval.FnSuffix)
		})
	case "E:":
		for _, s := range os.Environ() {
			if i := strings.IndexByte(s, '='); i > 0 {
				f(s[:i])
			}
		}
	default:
		segs := eval.SplitQNameSegs(ns)
		mod := global.IndexName(segs[0])
		if mod == nil {
			mod = builtin.IndexName(segs[0])
		}
		for _, seg := range segs[1:] {
			if mod == nil {
				return
			}
			mod = mod.Get().(*eval.Ns).IndexName(seg)
		}
		if mod != nil {
			mod.Get().(*eval.Ns).IterateNames(f)
		}
	}
}

// Calls the passed function for each namespace that can be used from the top
// context.
func eachNsInTop(builtin, global *eval.Ns, f func(s string)) {
	f("builtin:")
	f("e:")
	f("E:")

	global.IterateNames(func(name string) {
		if strings.HasSuffix(name, eval.NsSuffix) {
			f(name)
		}
	})

	builtin.IterateNames(func(name string) {
		if strings.HasSuffix(name, eval.NsSuffix) {
			f(name)
		}
	})
}