649 lines
19 KiB
Go
649 lines
19 KiB
Go
// Package main is complete tool for the go command line
|
|
package main
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/posener/complete/v2"
|
|
"github.com/posener/complete/v2/predict"
|
|
)
|
|
|
|
// envVerbose is the sys env var that controls error output verbosity.
|
|
const envVerbose = "GOCOMP_VERBOSE"
|
|
|
|
var (
|
|
ellipsis = predict.Set{"./..."}
|
|
anyPackage = complete.PredictFunc(predictPackages)
|
|
goFiles = predict.Files("*.go")
|
|
anyFile = predict.Files("*")
|
|
anyGo = predict.Or(goFiles, anyPackage, ellipsis)
|
|
)
|
|
|
|
func main() {
|
|
if os.Getenv(envVerbose) != "1" {
|
|
log.SetOutput(ioutil.Discard)
|
|
}
|
|
|
|
build := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"o": anyFile,
|
|
"i": predict.Nothing,
|
|
|
|
"a": predict.Nothing,
|
|
"n": predict.Nothing,
|
|
"p": predict.Something,
|
|
"race": predict.Nothing,
|
|
"msan": predict.Nothing,
|
|
"v": predict.Nothing,
|
|
"work": predict.Nothing,
|
|
"x": predict.Nothing,
|
|
"asmflags": predict.Something,
|
|
"buildmode": predict.Something,
|
|
"compiler": predict.Something,
|
|
"gccgoflags": predict.Set{"gccgo", "gc"},
|
|
"gcflags": predict.Something,
|
|
"installsuffix": predict.Something,
|
|
"ldflags": predict.Something,
|
|
"linkshared": predict.Nothing,
|
|
"pkgdir": anyPackage,
|
|
"tags": predict.Something,
|
|
"toolexec": predict.Something,
|
|
},
|
|
Args: anyGo,
|
|
}
|
|
|
|
run := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"exec": predict.Something,
|
|
},
|
|
Args: goFiles,
|
|
}
|
|
|
|
test := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"args": predict.Something,
|
|
"c": predict.Nothing,
|
|
"exec": predict.Something,
|
|
|
|
"bench": predictBenchmark,
|
|
"benchtime": predict.Something,
|
|
"count": predict.Something,
|
|
"cover": predict.Nothing,
|
|
"covermode": predict.Set{"set", "count", "atomic"},
|
|
"coverpkg": predict.Dirs("*"),
|
|
"cpu": predict.Something,
|
|
"run": predictTest,
|
|
"short": predict.Nothing,
|
|
"timeout": predict.Something,
|
|
|
|
"benchmem": predict.Nothing,
|
|
"blockprofile": predict.Files("*.out"),
|
|
"blockprofilerate": predict.Something,
|
|
"coverprofile": predict.Files("*.out"),
|
|
"cpuprofile": predict.Files("*.out"),
|
|
"memprofile": predict.Files("*.out"),
|
|
"memprofilerate": predict.Something,
|
|
"mutexprofile": predict.Files("*.out"),
|
|
"mutexprofilefraction": predict.Something,
|
|
"outputdir": predict.Dirs("*"),
|
|
"trace": predict.Files("*.out"),
|
|
},
|
|
Args: anyGo,
|
|
}
|
|
|
|
fmt := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"n": predict.Nothing,
|
|
"x": predict.Nothing,
|
|
},
|
|
Args: anyGo,
|
|
}
|
|
|
|
get := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"d": predict.Nothing,
|
|
"f": predict.Nothing,
|
|
"fix": predict.Nothing,
|
|
"insecure": predict.Nothing,
|
|
"t": predict.Nothing,
|
|
"u": predict.Nothing,
|
|
},
|
|
Args: anyGo,
|
|
}
|
|
|
|
generate := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"n": predict.Nothing,
|
|
"x": predict.Nothing,
|
|
"v": predict.Nothing,
|
|
"run": predict.Something,
|
|
},
|
|
Args: anyGo,
|
|
}
|
|
|
|
vet := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"n": predict.Nothing,
|
|
"x": predict.Nothing,
|
|
},
|
|
Args: anyGo,
|
|
}
|
|
|
|
list := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"e": predict.Nothing,
|
|
"f": predict.Something,
|
|
"json": predict.Nothing,
|
|
},
|
|
Args: predict.Or(anyPackage, ellipsis),
|
|
}
|
|
|
|
doc := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"c": predict.Nothing,
|
|
"cmd": predict.Nothing,
|
|
"u": predict.Nothing,
|
|
},
|
|
Args: anyPackage,
|
|
}
|
|
|
|
tool := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"n": predict.Nothing,
|
|
},
|
|
Sub: map[string]*complete.Command{
|
|
"addr2line": {
|
|
Args: anyFile,
|
|
},
|
|
"asm": {
|
|
Flags: map[string]complete.Predictor{
|
|
"D": predict.Something,
|
|
"I": predict.Dirs("*"),
|
|
"S": predict.Nothing,
|
|
"V": predict.Nothing,
|
|
"debug": predict.Nothing,
|
|
"dynlink": predict.Nothing,
|
|
"e": predict.Nothing,
|
|
"o": anyFile,
|
|
"shared": predict.Nothing,
|
|
"trimpath": predict.Nothing,
|
|
},
|
|
Args: predict.Files("*.s"),
|
|
},
|
|
"cgo": {
|
|
Flags: map[string]complete.Predictor{
|
|
"debug-define": predict.Nothing,
|
|
"debug-gcc": predict.Nothing,
|
|
"dynimport": anyFile,
|
|
"dynlinker": predict.Nothing,
|
|
"dynout": anyFile,
|
|
"dynpackage": anyPackage,
|
|
"exportheader": predict.Dirs("*"),
|
|
"gccgo": predict.Nothing,
|
|
"gccgopkgpath": predict.Dirs("*"),
|
|
"gccgoprefix": predict.Something,
|
|
"godefs": predict.Nothing,
|
|
"import_runtime_cgo": predict.Nothing,
|
|
"import_syscall": predict.Nothing,
|
|
"importpath": predict.Dirs("*"),
|
|
"objdir": predict.Dirs("*"),
|
|
"srcdir": predict.Dirs("*"),
|
|
},
|
|
Args: goFiles,
|
|
},
|
|
"compile": {
|
|
Flags: map[string]complete.Predictor{
|
|
"%": predict.Nothing,
|
|
"+": predict.Nothing,
|
|
"B": predict.Nothing,
|
|
"D": predict.Dirs("*"),
|
|
"E": predict.Nothing,
|
|
"I": predict.Dirs("*"),
|
|
"K": predict.Nothing,
|
|
"N": predict.Nothing,
|
|
"S": predict.Nothing,
|
|
"V": predict.Nothing,
|
|
"W": predict.Nothing,
|
|
"asmhdr": anyFile,
|
|
"bench": anyFile,
|
|
"buildid": predict.Nothing,
|
|
"complete": predict.Nothing,
|
|
"cpuprofile": anyFile,
|
|
"d": predict.Nothing,
|
|
"dynlink": predict.Nothing,
|
|
"e": predict.Nothing,
|
|
"f": predict.Nothing,
|
|
"h": predict.Nothing,
|
|
"i": predict.Nothing,
|
|
"importmap": predict.Something,
|
|
"installsuffix": predict.Something,
|
|
"j": predict.Nothing,
|
|
"l": predict.Nothing,
|
|
"largemodel": predict.Nothing,
|
|
"linkobj": anyFile,
|
|
"live": predict.Nothing,
|
|
"m": predict.Nothing,
|
|
"memprofile": predict.Nothing,
|
|
"memprofilerate": predict.Something,
|
|
"msan": predict.Nothing,
|
|
"nolocalimports": predict.Nothing,
|
|
"o": anyFile,
|
|
"p": predict.Dirs("*"),
|
|
"pack": predict.Nothing,
|
|
"r": predict.Nothing,
|
|
"race": predict.Nothing,
|
|
"s": predict.Nothing,
|
|
"shared": predict.Nothing,
|
|
"traceprofile": anyFile,
|
|
"trimpath": predict.Something,
|
|
"u": predict.Nothing,
|
|
"v": predict.Nothing,
|
|
"w": predict.Nothing,
|
|
"wb": predict.Nothing,
|
|
},
|
|
Args: goFiles,
|
|
},
|
|
"cover": {
|
|
Flags: map[string]complete.Predictor{
|
|
"func": predict.Something,
|
|
"html": predict.Something,
|
|
"mode": predict.Set{"set", "count", "atomic"},
|
|
"o": anyFile,
|
|
"var": predict.Something,
|
|
},
|
|
Args: anyFile,
|
|
},
|
|
"dist": {
|
|
Sub: map[string]*complete.Command{
|
|
"banner": {Flags: map[string]complete.Predictor{"v": predict.Nothing}},
|
|
"bootstrap": {Flags: map[string]complete.Predictor{"v": predict.Nothing}},
|
|
"clean": {Flags: map[string]complete.Predictor{"v": predict.Nothing}},
|
|
"env": {Flags: map[string]complete.Predictor{"v": predict.Nothing, "p": predict.Nothing}},
|
|
"install": {Flags: map[string]complete.Predictor{"v": predict.Nothing}, Args: predict.Dirs("*")},
|
|
"list": {Flags: map[string]complete.Predictor{"v": predict.Nothing, "json": predict.Nothing}},
|
|
"test": {Flags: map[string]complete.Predictor{"v": predict.Nothing, "h": predict.Nothing}},
|
|
"version": {Flags: map[string]complete.Predictor{"v": predict.Nothing}},
|
|
},
|
|
},
|
|
"doc": doc,
|
|
"fix": {
|
|
Flags: map[string]complete.Predictor{
|
|
"diff": predict.Nothing,
|
|
"force": predict.Something,
|
|
"r": predict.Set{"context", "gotypes", "netipv6zone", "printerconfig"},
|
|
},
|
|
Args: anyGo,
|
|
},
|
|
"link": {
|
|
Flags: map[string]complete.Predictor{
|
|
"B": predict.Something, // note
|
|
"D": predict.Something, // address (default -1)
|
|
"E": predict.Something, // entry symbol name
|
|
"H": predict.Something, // header type
|
|
"I": predict.Something, // linker binary
|
|
"L": predict.Dirs("*"), // directory
|
|
"R": predict.Something, // quantum (default -1)
|
|
"T": predict.Something, // address (default -1)
|
|
"V": predict.Nothing,
|
|
"X": predict.Something,
|
|
"a": predict.Something,
|
|
"buildid": predict.Something, // build id
|
|
"buildmode": predict.Something,
|
|
"c": predict.Nothing,
|
|
"cpuprofile": anyFile,
|
|
"d": predict.Nothing,
|
|
"debugtramp": predict.Something, // int
|
|
"dumpdep": predict.Nothing,
|
|
"extar": predict.Something,
|
|
"extld": predict.Something,
|
|
"extldflags": predict.Something, // flags
|
|
"f": predict.Nothing,
|
|
"g": predict.Nothing,
|
|
"importcfg": anyFile,
|
|
"installsuffix": predict.Something, // dir suffix
|
|
"k": predict.Something, // symbol
|
|
"libgcc": predict.Something, // maybe "none"
|
|
"linkmode": predict.Something, // mode
|
|
"linkshared": predict.Nothing,
|
|
"memprofile": anyFile,
|
|
"memprofilerate": predict.Something, // rate
|
|
"msan": predict.Nothing,
|
|
"n": predict.Nothing,
|
|
"o": predict.Something,
|
|
"pluginpath": predict.Something,
|
|
"r": predict.Something, // "dir1:dir2:..."
|
|
"race": predict.Nothing,
|
|
"s": predict.Nothing,
|
|
"tmpdir": predict.Dirs("*"),
|
|
"u": predict.Nothing,
|
|
"v": predict.Nothing,
|
|
"w": predict.Nothing,
|
|
// "h": predict.Something, // halt on error
|
|
},
|
|
Args: predict.Or(
|
|
predict.Files("*.a"),
|
|
predict.Files("*.o"),
|
|
),
|
|
},
|
|
"nm": {
|
|
Flags: map[string]complete.Predictor{
|
|
"n": predict.Nothing,
|
|
"size": predict.Nothing,
|
|
"sort": predict.Something,
|
|
"type": predict.Nothing,
|
|
},
|
|
Args: anyGo,
|
|
},
|
|
"objdump": {
|
|
Flags: map[string]complete.Predictor{
|
|
"s": predict.Something,
|
|
"S": predict.Nothing,
|
|
},
|
|
Args: anyFile,
|
|
},
|
|
"pack": {
|
|
/* this lacks the positional aspect of all these params */
|
|
Flags: map[string]complete.Predictor{
|
|
"c": predict.Nothing,
|
|
"p": predict.Nothing,
|
|
"r": predict.Nothing,
|
|
"t": predict.Nothing,
|
|
"x": predict.Nothing,
|
|
"cv": predict.Nothing,
|
|
"pv": predict.Nothing,
|
|
"rv": predict.Nothing,
|
|
"tv": predict.Nothing,
|
|
"xv": predict.Nothing,
|
|
},
|
|
Args: predict.Or(
|
|
predict.Files("*.a"),
|
|
predict.Files("*.o"),
|
|
),
|
|
},
|
|
"pprof": {
|
|
Flags: map[string]complete.Predictor{
|
|
"callgrind": predict.Nothing,
|
|
"disasm": predict.Something,
|
|
"dot": predict.Nothing,
|
|
"eog": predict.Nothing,
|
|
"evince": predict.Nothing,
|
|
"gif": predict.Nothing,
|
|
"gv": predict.Nothing,
|
|
"list": predict.Something,
|
|
"pdf": predict.Nothing,
|
|
"peek": predict.Something,
|
|
"png": predict.Nothing,
|
|
"proto": predict.Nothing,
|
|
"ps": predict.Nothing,
|
|
"raw": predict.Nothing,
|
|
"svg": predict.Nothing,
|
|
"tags": predict.Nothing,
|
|
"text": predict.Nothing,
|
|
"top": predict.Nothing,
|
|
"tree": predict.Nothing,
|
|
"web": predict.Nothing,
|
|
"weblist": predict.Something,
|
|
"output": anyFile,
|
|
"functions": predict.Nothing,
|
|
"files": predict.Nothing,
|
|
"lines": predict.Nothing,
|
|
"addresses": predict.Nothing,
|
|
"base": predict.Something,
|
|
"drop_negative": predict.Nothing,
|
|
"cum": predict.Nothing,
|
|
"seconds": predict.Something,
|
|
"nodecount": predict.Something,
|
|
"nodefraction": predict.Something,
|
|
"edgefraction": predict.Something,
|
|
"sample_index": predict.Nothing,
|
|
"mean": predict.Nothing,
|
|
"inuse_space": predict.Nothing,
|
|
"inuse_objects": predict.Nothing,
|
|
"alloc_space": predict.Nothing,
|
|
"alloc_objects": predict.Nothing,
|
|
"total_delay": predict.Nothing,
|
|
"contentions": predict.Nothing,
|
|
"mean_delay": predict.Nothing,
|
|
"runtime": predict.Nothing,
|
|
"focus": predict.Something,
|
|
"ignore": predict.Something,
|
|
"tagfocus": predict.Something,
|
|
"tagignore": predict.Something,
|
|
"call_tree": predict.Nothing,
|
|
"unit": predict.Something,
|
|
"divide_by": predict.Something,
|
|
"buildid": predict.Something,
|
|
"tools": predict.Dirs("*"),
|
|
"help": predict.Nothing,
|
|
},
|
|
Args: anyFile,
|
|
},
|
|
"tour": {
|
|
Flags: map[string]complete.Predictor{
|
|
"http": predict.Something,
|
|
"openbrowser": predict.Nothing,
|
|
},
|
|
},
|
|
"trace": {
|
|
Flags: map[string]complete.Predictor{
|
|
"http": predict.Something,
|
|
"pprof": predict.Set{"net", "sync", "syscall", "sched"},
|
|
},
|
|
Args: anyFile,
|
|
},
|
|
"vet": {
|
|
Flags: map[string]complete.Predictor{
|
|
"all": predict.Nothing,
|
|
"asmdecl": predict.Nothing,
|
|
"assign": predict.Nothing,
|
|
"atomic": predict.Nothing,
|
|
"bool": predict.Nothing,
|
|
"buildtags": predict.Nothing,
|
|
"cgocall": predict.Nothing,
|
|
"composites": predict.Nothing,
|
|
"compositewhitelist": predict.Nothing,
|
|
"copylocks": predict.Nothing,
|
|
"httpresponse": predict.Nothing,
|
|
"lostcancel": predict.Nothing,
|
|
"methods": predict.Nothing,
|
|
"nilfunc": predict.Nothing,
|
|
"printf": predict.Nothing,
|
|
"printfuncs": predict.Something,
|
|
"rangeloops": predict.Nothing,
|
|
"shadow": predict.Nothing,
|
|
"shadowstrict": predict.Nothing,
|
|
"shift": predict.Nothing,
|
|
"structtags": predict.Nothing,
|
|
"tags": predict.Something,
|
|
"tests": predict.Nothing,
|
|
"unreachable": predict.Nothing,
|
|
"unsafeptr": predict.Nothing,
|
|
"unusedfuncs": predict.Something,
|
|
"unusedresult": predict.Nothing,
|
|
"unusedstringmethods": predict.Something,
|
|
"v": predict.Nothing,
|
|
},
|
|
Args: anyGo,
|
|
},
|
|
},
|
|
}
|
|
|
|
clean := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"i": predict.Nothing,
|
|
"r": predict.Nothing,
|
|
"n": predict.Nothing,
|
|
"x": predict.Nothing,
|
|
"cache": predict.Nothing,
|
|
"testcache": predict.Nothing,
|
|
"modcache": predict.Nothing,
|
|
},
|
|
Args: predict.Or(anyPackage, ellipsis),
|
|
}
|
|
|
|
env := &complete.Command{
|
|
Args: predict.Something,
|
|
}
|
|
|
|
bug := &complete.Command{}
|
|
version := &complete.Command{}
|
|
|
|
fix := &complete.Command{
|
|
Args: anyGo,
|
|
}
|
|
|
|
modDownload := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"json": predict.Nothing,
|
|
},
|
|
Args: anyPackage,
|
|
}
|
|
|
|
modEdit := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"fmt": predict.Nothing,
|
|
"module": predict.Nothing,
|
|
"print": predict.Nothing,
|
|
|
|
"exclude": anyPackage,
|
|
"dropexclude": anyPackage,
|
|
"replace": anyPackage,
|
|
"dropreplace": anyPackage,
|
|
"require": anyPackage,
|
|
"droprequire": anyPackage,
|
|
},
|
|
Args: predict.Files("go.mod"),
|
|
}
|
|
|
|
modGraph := &complete.Command{}
|
|
|
|
modInit := &complete.Command{
|
|
Args: predict.Something,
|
|
}
|
|
|
|
modTidy := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"v": predict.Nothing,
|
|
},
|
|
}
|
|
|
|
modVendor := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"v": predict.Nothing,
|
|
},
|
|
}
|
|
|
|
modVerify := &complete.Command{}
|
|
|
|
modWhy := &complete.Command{
|
|
Flags: map[string]complete.Predictor{
|
|
"m": predict.Nothing,
|
|
"vendor": predict.Nothing,
|
|
},
|
|
Args: anyPackage,
|
|
}
|
|
|
|
modHelp := &complete.Command{
|
|
Sub: map[string]*complete.Command{
|
|
"download": {},
|
|
"edit": {},
|
|
"graph": {},
|
|
"init": {},
|
|
"tidy": {},
|
|
"vendor": {},
|
|
"verify": {},
|
|
"why": {},
|
|
},
|
|
}
|
|
|
|
mod := &complete.Command{
|
|
Sub: map[string]*complete.Command{
|
|
"download": modDownload,
|
|
"edit": modEdit,
|
|
"graph": modGraph,
|
|
"init": modInit,
|
|
"tidy": modTidy,
|
|
"vendor": modVendor,
|
|
"verify": modVerify,
|
|
"why": modWhy,
|
|
"help": modHelp,
|
|
},
|
|
}
|
|
|
|
help := &complete.Command{
|
|
Sub: map[string]*complete.Command{
|
|
"bug": {},
|
|
"build": {},
|
|
"clean": {},
|
|
"doc": {},
|
|
"env": {},
|
|
"fix": {},
|
|
"fmt": {},
|
|
"generate": {},
|
|
"get": {},
|
|
"install": {},
|
|
"list": {},
|
|
"mod": modHelp,
|
|
"run": {},
|
|
"test": {},
|
|
"tool": {},
|
|
"version": {},
|
|
"vet": {},
|
|
"buildmode": {},
|
|
"c": {},
|
|
"cache": {},
|
|
"environment": {},
|
|
"filetype": {},
|
|
"go.mod": {},
|
|
"gopath": {},
|
|
"gopath-get": {},
|
|
"goproxy": {},
|
|
"importpath": {},
|
|
"modules": {},
|
|
"module-get": {},
|
|
"packages": {},
|
|
"testflag": {},
|
|
"testfunc": {},
|
|
},
|
|
}
|
|
|
|
// commands that also accepts the build flags
|
|
for name, options := range build.Flags {
|
|
test.Flags[name] = options
|
|
run.Flags[name] = options
|
|
list.Flags[name] = options
|
|
vet.Flags[name] = options
|
|
get.Flags[name] = options
|
|
}
|
|
|
|
gogo := &complete.Command{
|
|
Sub: map[string]*complete.Command{
|
|
"build": build,
|
|
"install": build, // install and build have the same flags
|
|
"run": run,
|
|
"test": test,
|
|
"fmt": fmt,
|
|
"get": get,
|
|
"generate": generate,
|
|
"vet": vet,
|
|
"list": list,
|
|
"doc": doc,
|
|
"tool": tool,
|
|
"clean": clean,
|
|
"env": env,
|
|
"bug": bug,
|
|
"fix": fix,
|
|
"version": version,
|
|
"mod": mod,
|
|
"help": help,
|
|
},
|
|
Flags: map[string]complete.Predictor{
|
|
"h": predict.Nothing,
|
|
},
|
|
}
|
|
|
|
gogo.Complete("go")
|
|
}
|