1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
--[[
Based on https://gitlab.com/timoha/vis-go
Changes made:
- stylua format
- check if line, col in godef()
- no formatting because already handled by format.lua
- removed the goimports option
--]]
local function jump_to(path, line, col)
if path then
vis:command(string.format("e %s", path))
end
vis.win.selection:to(line, col)
end
local Gostack = { s = {}, i = 1 }
function Gostack:push(v)
self.s[self.i] = v
self.i = self.i + 1
end
function Gostack:pop()
if self.i == 1 then
return nil
end
self.i = self.i - 1
return self.s[self.i]
end
local function godef()
local win = vis.win
if win.syntax ~= "go" then
return 0
end
local file = win.file
local pos = win.selection.pos
local cmd = string.format("godef -i -o %d", pos)
local status, out, err = vis:pipe(file, { start = 0, finish = file.size }, cmd)
if status ~= 0 or not out then
if err then
vis:info(err)
end
return status
end
Gostack:push({ path = file.path, line = win.selection.line, col = win.selection.col })
local path, line, col = string.match(out, "([^:]+):([^:]+):([^:]+)")
if not path then
-- same file
line, col = string.match(out, "([^:]+):([^:]+)")
end
if line and col then
jump_to(path, line, col)
end
end
local function godef_back()
if vis.win.syntax ~= "go" then
return 0
end
local pos = Gostack:pop()
if pos then
jump_to(pos.path, pos.line, pos.col)
end
end
vis:map(vis.modes.NORMAL, "gd", godef, "Jump to Go symbol/definition")
vis:map(vis.modes.NORMAL, "gD", godef_back, "Jump back to previous Go symbol/definition")
local function gorename(argv, force, win, selection)
if win.syntax ~= "go" then
return true
end
local name = argv[1]
if not name then
vis:info("empty new name provided")
return false
end
local forceFlag = ""
if force then
forceFlag = "-force"
end
local pos = selection.pos
local f =
io.popen(string.format("gorename -offset %s:#%d -to %s %s 2>&1", win.file.path, pos, name, forceFlag), "r")
local out = f:read("*all")
local success, _, _ = f:close()
if not success then
vis:message(out)
return false
end
-- refresh current file
vis:command("e")
win.selection.pos = pos
vis:info(out)
return true
end
vis:command_register(
"gorename",
gorename,
"Perform precise type-safe renaming of identifiers in Go source code: :gorename newName"
)
|
