/
rust.lua
197 lines (184 loc) · 9.25 KB
/
rust.lua
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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
--- Rust language actions
local M = {}
-- Frontend - options displayed on telescope
M.options = {
{ text = "Build and run program", value = "option1" },
{ text = "Build program", value = "option2" },
{ text = "Run program", value = "option3" },
{ text = "Build solution", value = "option4" },
{ text = "", value = "separator" },
{ text = "Cargo build and run", value = "option5" },
{ text = "Cargo build", value = "option6" },
{ text = "Cargo run", value = "option7" },
{ text = "", value = "separator" },
{ text = "Cargo build --workspace and run", value = "option8" },
{ text = "Cargo build --workspace", value = "option9" }
}
-- Backend - overseer tasks performed on option selected
function M.action(selected_option)
local utils = require("compiler.utils")
local overseer = require("overseer")
local entry_point = utils.os_path(vim.fn.getcwd() .. "/main.rs") -- working_directory/main.rs
local output_dir = utils.os_path(vim.fn.getcwd() .. "/bin/") -- working_directory/bin/
local output = utils.os_path(vim.fn.getcwd() .. "/bin/program") -- working_directory/bin/program
local arguments = "-D warnings -g" -- arguments can be overriden in .solution
local final_message = "--task finished--"
if selected_option == "option1" then
local task = overseer.new_task({
name = "- Rust compiler",
strategy = { "orchestrator",
tasks = {{ "shell", name = "- Build & run program → " .. entry_point,
cmd = "rm -f " .. output .. " || true" .. -- clean
" && mkdir -p " .. output_dir .. -- mkdir
" && rustc " .. entry_point .. " -o " .. output .. " " .. arguments .. -- compile
" && " .. output .. -- run
" && echo " .. entry_point .. -- echo
" && echo '" .. final_message .. "'"
},},},})
task:start()
vim.cmd("OverseerOpen")
elseif selected_option == "option2" then
local task = overseer.new_task({
name = "- Rust compiler",
strategy = { "orchestrator",
tasks = {{ "shell", name = "- Build program → " .. entry_point,
cmd = "rm -f " .. output .. " || true" .. -- clean
" && mkdir -p " .. output_dir .. -- mkdir
" && rustc " .. entry_point .. " -o " .. output .. " " .. arguments .. -- compile
" && echo " .. entry_point .. -- echo
" && echo '" .. final_message .. "'"
},},},})
task:start()
vim.cmd("OverseerOpen")
elseif selected_option == "option3" then
local task = overseer.new_task({
name = "- Rust compiler",
strategy = { "orchestrator",
tasks = {{ "shell", name = "- Run program → " .. entry_point,
cmd = output .. -- run
" && echo " .. output .. -- echo
" && echo '" .. final_message .. "'"
},},},})
task:start()
vim.cmd("OverseerOpen")
elseif selected_option == "option4" then
local entry_points
local task = {}
local tasks = {}
local executables = {}
-- if .solution file exists in working dir
local solution_file = utils.get_solution_file()
if solution_file then
local config = utils.parse_solution_file(solution_file)
for entry, variables in pairs(config) do
if entry == "executables" then goto continue end
entry_point = utils.os_path(variables.entry_point)
output = utils.os_path(variables.output)
output_dir = utils.os_path(output:match("^(.-[/\\])[^/\\]*$"))
arguments = variables.arguments or arguments -- optional
task = { "shell", name = "- Build program → " .. entry_point,
cmd = "rm -f " .. output .. " || true" .. -- clean
" && mkdir -p " .. output_dir .. -- mkdir
" && rustc " .. entry_point .. " -o " .. output .. " " .. arguments .. -- compile
" && echo " .. entry_point .. -- echo
" && echo '" .. final_message .. "'"
}
table.insert(tasks, task) -- store all the tasks we've created
::continue::
end
local solution_executables = config["executables"]
if solution_executables then
for entry, executable in pairs(solution_executables) do
task = { "shell", name = "- Run program → " .. executable,
cmd = executable .. -- run
" && echo " .. executable .. -- echo
" && echo '" .. final_message .. "'"
}
table.insert(executables, task) -- store all the executables we've created
end
end
task = overseer.new_task({
name = "- Rust compiler", strategy = { "orchestrator",
tasks = {
tasks, -- Build all the programs in the solution in parallel
executables -- Then run the solution executable(s)
}}})
task:start()
vim.cmd("OverseerOpen")
else -- If no .solution file
-- Create a list of all entry point files in the working directory
entry_points = utils.find_files(vim.fn.getcwd(), "main.rs")
for _, entry_point in ipairs(entry_points) do
entry_point = utils.os_path(entry_point)
output_dir = utils.os_path(entry_point:match("^(.-[/\\])[^/\\]*$") .. "bin") -- entry_point/bin
output = utils.os_path(output_dir .. "/program") -- entry_point/bin/program
task = { "shell", name = "- Build program → " .. entry_point,
cmd = "rm -f " .. output .. " || true" .. -- clean
" && mkdir -p " .. output_dir .. -- mkdir
" && rustc " .. entry_point .. " -o " .. output .. " " .. arguments .. -- compile
" && echo " .. entry_point .. -- echo
" && echo '" .. final_message .. "'"
}
table.insert(tasks, task) -- store all the tasks we've created
end
task = overseer.new_task({ -- run all tasks we've created in parallel
name = "- Rust compiler", strategy = { "orchestrator", tasks = tasks }
})
task:start()
vim.cmd("OverseerOpen")
end
elseif selected_option == "option5" then
local task = overseer.new_task({
name = "- Rust compiler",
strategy = { "orchestrator",
tasks = {{ "shell", name = "- Cargo build & run → " .. "Cargo.toml",
cmd = "cargo build " .. -- compile
" && cargo run" .. --run
" && echo '" .. final_message .. "'" -- echo
},},},})
task:start()
vim.cmd("OverseerOpen")
elseif selected_option == "option6" then
local task = overseer.new_task({
name = "- Rust compiler",
strategy = { "orchestrator",
tasks = {{ "shell", name = "- Cargo build → " .. "Cargo.toml",
cmd = "cargo build " .. -- compile
" && echo '" .. final_message .. "'" -- echo
},},},})
task:start()
vim.cmd("OverseerOpen")
elseif selected_option == "option7" then
local task = overseer.new_task({
name = "- Rust compiler",
strategy = { "orchestrator",
tasks = {{ "shell", name = "- Cargo run → " .. "Cargo.toml",
cmd = "cargo run " .. -- run
" && echo '" .. final_message .. "'" -- echo
},},},})
task:start()
vim.cmd("OverseerOpen")
elseif selected_option == "option8" then
local task = overseer.new_task({
name = "- Rust compiler",
strategy = { "orchestrator",
tasks = {{ "shell", name = "- Cargo build --workspace & run → " .. "Cargo.toml",
cmd = "cargo build --workspace " .. -- compile
" && cargo run" ..
" && echo '" .. final_message .. "'" -- echo
},},},})
task:start()
vim.cmd("OverseerOpen")
elseif selected_option == "option9" then
local task = overseer.new_task({
name = "- Rust compiler",
strategy = { "orchestrator",
tasks = {{ "shell", name = "- Cargo build --workspace → " .. "Cargo.toml",
cmd = "cargo build --workspace" .. -- compile
" && echo '" .. final_message .. "'" -- echo
},},},})
task:start()
vim.cmd("OverseerOpen")
end
end
return M