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
|
local Request = require("99.request")
local RequestStatus = require("99.ops.request_status")
local Mark = require("99.ops.marks")
local geo = require("99.geo")
local make_clean_up = require("99.ops.clean-up")
local Agents = require("99.extensions.agents")
local Range = geo.Range
local Point = geo.Point
--- @param context _99.RequestContext
--- @param range _99.Range
--- @param opts? _99.ops.Opts
local function over_range(context, range, opts)
opts = opts or {}
local logger = context.logger:set_area("visual")
local request = Request.new(context)
local top_mark = Mark.mark_above_range(range)
local bottom_mark = Mark.mark_point(range.buffer, range.end_)
context.marks.top_mark = top_mark
context.marks.bottom_mark = bottom_mark
logger:debug(
"visual request start",
"start",
Point.from_mark(top_mark),
"end",
Point.from_mark(bottom_mark)
)
local display_ai_status = context._99.ai_stdout_rows > 1
local top_status = RequestStatus.new(
250,
context._99.ai_stdout_rows or 1,
"Implementing",
top_mark
)
local bottom_status = RequestStatus.new(250, 1, "Implementing", bottom_mark)
local clean_up = make_clean_up(context, function()
top_status:stop()
bottom_status:stop()
context:clear_marks()
request:cancel()
end)
local full_prompt = context._99.prompts.prompts.visual_selection(range)
local additional_prompt = opts.additional_prompt
if additional_prompt then
full_prompt =
context._99.prompts.prompts.prompt(additional_prompt, full_prompt)
local rules = Agents.find_rules(context._99.rules, additional_prompt)
context:add_agent_rules(rules)
end
local additional_rules = opts.additional_rules
if additional_rules then
context:add_agent_rules(additional_rules)
end
request:add_prompt_content(full_prompt)
top_status:start()
bottom_status:start()
request:start({
on_complete = function(status, response)
vim.schedule(clean_up)
if status == "cancelled" then
logger:debug("request cancelled for visual selection, removing marks")
elseif status == "failed" then
logger:error(
"request failed for visual_selection",
"error response",
response or "no response provided"
)
elseif status == "success" then
local valid = top_mark:is_valid() and bottom_mark:is_valid()
if not valid then
logger:fatal(
-- luacheck: ignore 631
"the original visual_selection has been destroyed. You cannot delete the original visual selection during a request"
)
return
end
local new_range = Range.from_marks(top_mark, bottom_mark)
local lines = vim.split(response, "\n")
--- HACK: i am adding a new line here because above range will add a mark to the line above.
--- that way this appears to be added to "the same line" as the visual selection was
--- originally take from
table.insert(lines, 1, "")
new_range:replace_text(lines)
end
end,
on_stdout = function(line)
if display_ai_status then
top_status:push(line)
end
end,
on_stderr = function(line)
logger:debug("visual_selection#on_stderr received", "line", line)
end,
})
end
return over_range
|
