local vfn = vim.fn local utils = require('go.utils') local create_cmd = function(cmd, func, opt) opt = vim.tbl_extend('force', { desc = 'go.nvim ' .. cmd }, opt or {}) vim.api.nvim_create_user_command(cmd, func, opt) end local dap_config = function() create_cmd('BreakCondition', function(_) local dap = utils.load_plugin('nvim-dap', 'dap') if not dap then return end dap.set_breakpoint(vfn.input('Breakpoint condition: ')) end) create_cmd('LogPoint', function(_) local dap = utils.load_plugin('nvim-dap', 'dap') if not dap then return end dap.set_breakpoint(nil, nil, vfn.input('Log message: ')) end) create_cmd('ReplRun', function(_) local dap = utils.load_plugin('nvim-dap', 'dap') if not dap then return end dap.repl.run_last() end) create_cmd('ReplToggle', function(_) local dap = utils.load_plugin('nvim-dap', 'dap') if not dap then return end dap.repl.toggle() end) create_cmd('ReplOpen', function(_) local dap = utils.load_plugin('nvim-dap', 'dap') if not dap then return end dap.repl.open() vim.cmd('split') end) create_cmd('DapRerun', function(_) local dap = utils.load_plugin('nvim-dap', 'dap') if not dap then return end dap.disconnect() dap.close() dap.run_last() end) local gdap = require('go.dap') create_cmd('GoDebug', function(opts) local dap = utils.load_plugin('nvim-dap', 'dap') if not dap then return end gdap.run(unpack(opts.fargs)) end, { complete = function(a, l) return package.loaded.go.dbg_complete(a, l) end, nargs = '*' }) create_cmd('GoCreateLaunch', function(_) require('go.launch').config() end) create_cmd('GoBreakSave', function(_) gdap.save_brks() end) create_cmd('GoBreakLoad', function(_) local dap = utils.load_plugin('nvim-dap', 'dap') if not dap then return end gdap.load_brks() end) create_cmd('DapStop', function(_) require('go.dap').stop() end) create_cmd('GoDbgConfig', function(_) require('go.launch').config() end) create_cmd('GoBreakToggle', function(_) local dap = utils.load_plugin('nvim-dap', 'dap') if not dap then return end gdap.breakpt() end) create_cmd('GoDbgKeys', function(_) gdap.debug_keys() end) create_cmd('DapUiFloat', function(_) require('dapui').float_element() end) create_cmd('DapUiToggle', function(_) require('dapui').toggle() end) create_cmd('GoDbgStop', function(_) gdap.stop(true) end) create_cmd('GoDbgContinue', function(_) local dap = utils.load_plugin('nvim-dap', 'dap') if not dap then return end dap.continue() end) end return { add_cmds = function() vim.cmd([[ augroup go.filetype autocmd! autocmd FileType go setlocal omnifunc=v:lua.vim.lsp.omnifunc autocmd FileType go au QuickFixCmdPost [^l]* nested cwindow autocmd FileType go au QuickFixCmdPost l* nested lwindow augroup end ]]) create_cmd('GoMake', function(_) require('go.asyncmake').make() end) create_cmd('GoFmt', function(opts) require('go.format').gofmt(unpack(opts.fargs)) end, { nargs = '*' }) create_cmd('GoImport', function(opts) require('go.format').goimport(unpack(opts.fargs)) end, { complete = function(a, l) return package.loaded.go.doc_complete(a, l) end, nargs = '*' }) create_cmd('GoGet', function(opts) require('go.goget').run(opts.fargs) end, { nargs = '*' }) local gobin = _GO_NVIM_CFG.go local cmd = string.format([[command! -nargs=* GoGenerate :setl makeprg=%s\ generate | lua require'go.asyncmake'.make()]], gobin) vim.cmd(cmd) cmd = string.format( [[command! -nargs=* -complete=customlist,v:lua.package.loaded.go.package_complete GoBuild :setl makeprg=%s\ build | lua require'go.asyncmake'.make()]], gobin ) vim.cmd(cmd) cmd = string.format( [[command! -nargs=* -complete=customlist,v:lua.package.loaded.go.package_complete GoVet :setl makeprg=%s\ vet | lua require'go.asyncmake'.make()]], gobin ) vim.cmd(cmd) cmd = string.format( [[command! -nargs=* -complete=customlist,v:lua.package.loaded.go.package_complete GoRun :setl makeprg=%s\ run | lua require'go.asyncmake'.make()]], gobin ) vim.cmd(cmd) create_cmd('GoStop', function(opts) require('go.asyncmake').stopjob(unpack(opts.fargs)) end, { nargs = '*' }) -- if you want to output to quickfix -- example to running test in split buffer -- vim.cmd( -- [[command! -nargs=* -complete=custom,v:lua.package.loaded.go.package_complete GoTest :setl makeprg=go\ test\ -v\ | lua require'go.runner'.make()]]) create_cmd('GoTest', function(opts) require('go.gotest').test(unpack(opts.fargs)) end, { complete = function(a, l) return package.loaded.go.package_complete(a, l) end, nargs = '*' }) create_cmd('GoTestSum', function(opts) if opts.fargs[1] == '-w' then return require('go.gotestsum').watch() end require('go.gotestsum').run(unpack(opts.fargs)) end, { complete = function(a, l) return package.loaded.go.package_complete(a, l) end, nargs = '*' }) create_cmd('GoCoverage', function(opts) require('go.coverage').run(unpack(opts.fargs)) end, { complete = function(a, l) return package.loaded.go.package_complete(a, l) end, nargs = '*' }) create_cmd('GoPkgOutline', function(opts) require('go.package').outline(unpack(opts.fargs)) end, { complete = function(a, l) return package.loaded.go.package_complete(a, l) end, nargs = '*' }) -- vim.cmd([[command! GoTestCompile :setl makeprg=go\ build | :GoMake]]) --print-issued-lines=false vim.cmd( [[command! GoLint :setl makeprg=golangci-lint\ run\ --print-issued-lines=false\ --exclude-use-default=false | :GoMake]] ) create_cmd('GoProject', function(opts) require('go.project').setup() end) create_cmd('GoCheat', function(opts) require('go.chtsh').run(unpack(opts.fargs)) end, { nargs = '*' }) -- e.g. GoTestFunc unit create_cmd('GoTestFunc', function(opts) require('go.gotest').test_func(unpack(opts.fargs)) end, { nargs = '*' }) create_cmd('GoTestSubCase', function(opts) require('go.gotest').test_tblcase(unpack(opts.fargs)) end, { nargs = '*' }) -- e.g. GoTestFile unit -- command! -nargs=* -complete=custom,v:lua.package.loaded.go.package_complete GoTestFile lua require('go.gotest').test_file() create_cmd('GoTestFile', function(opts) require('go.gotest').test_file(unpack(opts.fargs)) end, { complete = function(a, l) return package.loaded.go.package_complete(a, l) end, nargs = '*' }) create_cmd('GoTestPkg', function(opts) require('go.gotest').test_package(unpack(opts.fargs)) end, { complete = function(a, l) return package.loaded.go.package_complete(a, l) end, nargs = '*' }) create_cmd('GoAddTest', function(opts) require('go.gotests').fun_test(unpack(opts.fargs)) end) create_cmd('GoAddExpTest', function(opts) require('go.gotests').exported_test(unpack(opts.fargs)) end, { nargs = '*' }) create_cmd('GoAddAllTest', function(opts) require('go.gotests').all_test(unpack(opts.fargs)) end, { nargs = '*' }) create_cmd('GoModVendor', function(opts) require('go.mod').run('vendor', unpack(opts.fargs)) end, { nargs = '*' }) create_cmd('GoModInit', function(opts) require('go.mod').run('init', unpack(opts.fargs)) end, { nargs = '*' }) create_cmd('GoEnv', function(opts) require('go.env').load_env(unpack(opts.fargs)) end, { nargs = '*' }) create_cmd('GoCodeLenAct', function(_) require('go.codelens').run_action() end) create_cmd('GoCodeAction', function(t) if t.range ~= 0 then require('go.codeaction').run_range_code_action({ t.line1, t.line2 }) else require('go.codeaction').run_code_action() end end, { range = true }) create_cmd('GoModifyTag', function(opts) require('go.tags').modify(unpack(opts.fargs)) end, { complete = function(a, l) return package.loaded.go.modify_tags_complete(a, l) end, nargs = '*' }) create_cmd('GoAddTag', function(opts) require('go.tags').add(unpack(opts.fargs)) end, { complete = function(a, l) return package.loaded.go.add_tags_complete(a, l) end, nargs = '*' }) create_cmd('GoRmTag', function(opts) require('go.tags').rm(unpack(opts.fargs)) end, { nargs = '*' }) create_cmd('GoImpl', function(opts) require('go.impl').run(unpack(opts.fargs)) end, { complete = function(a, l) return package.loaded.go.impl_complete(a, l) end, nargs = '*' }) create_cmd('GoDoc', function(opts) require('go.godoc').run(unpack(opts.fargs)) end, { complete = function(a, l) return package.loaded.go.doc_complete(a, l) end, nargs = '*' }) create_cmd('GoInstallBinary', function(opts) require('go.install').install(unpack(opts.fargs)) end, { complete = function(a, l) return package.loaded.go.tools_complete(a, l) end, nargs = '*' }) create_cmd('GoUpdateBinary', function(opts) require('go.install').update(unpack(opts.fargs)) end, { complete = function(a, l) return package.loaded.go.tools_complete(a, l) end, nargs = '*' }) create_cmd('GoInstallBinaries', function(_) require('go.install').install_all() end) create_cmd('GoUpdateBinaries', function(_) require('go.install').update_all() end) create_cmd('GoClearTag', function(_) require('go.tags').clear() end) create_cmd('GoCmt', function(_) require('go.comment').gen() end) create_cmd('GoRename', function(_) require('go.rename').lsprename() end) create_cmd('GoIfErr', function(_) require('go.iferr').run() end) create_cmd('GoFillStruct', function(_) require('go.reftool').fillstruct() end) create_cmd('GoFillSwitch', function(_) require('go.reftool').fillswitch() end) create_cmd('GoFixPlurals', function(_) require('go.fixplurals').fixplurals() end) create_cmd('GoAlt', function(opts) require('go.alternate').switch(opts.bang, '') end, { bang = true }) create_cmd('GoAltV', function(opts) require('go.alternate').switch(opts.bang, 'vsplit') end, { bang = true }) create_cmd('GoAltS', function(opts) require('go.alternate').switch(opts.bang, 'split') end, { bang = true }) create_cmd('GoModTidy', function(_) require('go.gopls').tidy() end) create_cmd('GoListImports', function(_) print(vim.inspect(require('go.gopls').list_imports())) end) create_cmd('GoCallstack', function(_) require('go.guru').callstack(-1) end) create_cmd('GoChannel', function(_) require('go.guru').channel_peers(-1) end) if _GO_NVIM_CFG.dap_debug then dap_config() end create_cmd('GoMockGen', require('go.mockgen').run, { nargs = '*', -- bang = true, complete = function(_, _, _) -- return completion candidates as a list-like table return { '-p', '-d', '-i', '-s' } end, }) create_cmd('GoEnv', function(opts) require('go.env').load_env(unpack(opts.fargs)) end, { nargs = '*' }) create_cmd('GoGenReturn', function() require('go.lsp').hover_returns() end) create_cmd('GoJson2Struct', function(opts) require('go.json2struct').run(opts) end, { nargs = '*', bang = true, register = true, -- complete = function(ArgLead, CmdLine, CursorPos) complete = function(_, _, _) return { 'myStruct' } end, range = true, }) create_cmd('Gomvp', function(opts) require('go.gomvp').run(opts.fargs) end, { nargs = '*' }) create_cmd('Govulnckeck', function(opts) require('go.govulncheck').run(opts.fargs) end, { nargs = '*' }) create_cmd('GoEnum', function(opts) require('go.enum').run(unpack(opts.fargs)) end, { nargs = '*' }) end, }