github.com/aergoio/aergo@v1.3.1/cmd/brick/exec/search.go

package exec

import (
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"strings"

	"github.com/aergoio/aergo/cmd/brick/context"
)

var index = make(map[string]map[string]string)

func Candidates(cmd string, chunks []context.Chunk, current int, symbol string) map[string]string {
	if ret := search(cmd, chunks, current, symbol); ret != nil {
		return ret
	}

	ret := make(map[string]string)

	// if there is no suggestion, then return general
	if descr, ok := context.Symbols[symbol]; ok {
		ret[symbol] = descr
	}

	return ret
}

// search contract args using get abi
func extractContractAndFuncName(cmd string, chunks []context.Chunk) (string, string) {
	var contractName string
	var funcName string

	executor := GetExecutor(cmd)
	if executor != nil {
		symbols := strings.Fields(executor.Syntax())

		for i, symbol := range symbols {
			if len(chunks) <= i {
				break
			}
			if symbol == context.ContractSymbol {
				// compare with symbol in syntax and extract contract name
				contractName = chunks[i].Text
			} else if symbol == context.FunctionSymbol {
				// extract function name
				funcName = chunks[i].Text
			}
		}
	}

	return contractName, funcName
}

func search(cmd string, chunks []context.Chunk, current int, symbol string) map[string]string {
	if keywords, ok := index[symbol]; ok {
		return keywords
	}

	if symbol == context.FunctionSymbol {
		contractName, _ := extractContractAndFuncName(cmd, chunks)
		if contractName != "" {
			return searchFuncHint(contractName)
		}
	} else if symbol == context.ContractArgsSymbol {
		contractName, funcName := extractContractAndFuncName(cmd, chunks)
		if contractName != "" && funcName != "" {
			// search abi using contract and function name
			return searchAbiHint(contractName, funcName)
		}
	} else if symbol == context.PathSymbol {
		if len(chunks) <= current { //there is no word yet
			return searchInPath(context.Chunk{Text: ".", Accent: false})
		}
		return searchInPath(chunks[current])
	}

	return nil
}

func searchFuncHint(contractName string) map[string]string {
	// read receipt and extract abi functions
	ret := make(map[string]string)

	abi, err := context.Get().GetABI(contractName)
	if err != nil {
		ret["<error>"] = err.Error()
		return ret
	}
	for _, contractFunc := range abi.Functions {
		// gather functions
		ret[contractFunc.Name] = ""
	}

	return ret
}

func searchAbiHint(contractName, funcName string) map[string]string {
	abi, err := context.Get().GetABI(contractName)
	if err != nil {
		return nil
	}

	for _, contractFunc := range abi.Functions {
		if contractFunc.Name == funcName {
			argsHint := "`["
			for i, funcArg := range contractFunc.GetArguments() {
				argsHint += funcArg.Name
				if i+1 != len(contractFunc.GetArguments()) {
					argsHint += ", "
				}
			}
			argsHint += "]`"

			ret := make(map[string]string)
			ret[argsHint] = context.Symbols[context.ContractArgsSymbol]
			return ret
		}
	}

	return nil
}

func searchInPath(chunk context.Chunk) map[string]string {

	if strings.HasSuffix(chunk.Text, ".") {
		// attach file sperator, to get files in this relative path
		chunk.Text = fmt.Sprintf("%s%c", chunk.Text, filepath.Separator)
	}
	ret := make(map[string]string)

	// extract parent directory path
	dir := filepath.Dir(chunk.Text)

	// navigate file list in the parent directory
	fileInfo, err := ioutil.ReadDir(dir)
	if err != nil {
		ret[err.Error()] = ""
		return ret
	}

	// detatch last base path
	// other function internally use filepath.Clean() that remove text . or ..
	// it makes prompt filter hard to match suggestions and the input

	currentDir, _ := filepath.Split(chunk.Text)

	for _, file := range fileInfo {
		// generate suggestion text
		fullPath := currentDir + file.Name()

		if file.IsDir() {
			fullPath += string(os.PathSeparator)
		}
		if chunk.Accent {
			fullPath = "`" + fullPath // attach accent again
		}
		// if contains white space...
		if wsIdx := strings.LastIndex(fullPath, " "); wsIdx != -1 {
			// cut it because auto completer will switch text only after whitespace
			fullPath = fullPath[wsIdx+1:]
		}

		ret[fullPath] = ""
	}

	return ret
}

func Index(symbol, text string) {
	if _, ok := index[symbol]; !ok {
		index[symbol] = make(map[string]string)
	}
	index[symbol][text] = symbol
}