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
|
local Window = require("99.window")
local Consts = require("99.consts")
local Throbber = require("99.ops.throbber")
--- @alias _99.StatusWindow.State "init" | "running"
--- @class _99.StatusWindow.Opts
--- this is pure a class for testing. helps controls timings
--- @docs include
--- @field throbber_opts _99.Throbber.Opts | nil
--- options for the throbber in the top left
--- @field in_flight_interval number | nil
--- frequency in which the in-flight interval checks to see if it should be
--- displayed / removed
--- @field enable boolean | nil
--- defaults to true
--- @param opts _99.StatusWindow.Opts | nil
--- @return _99.StatusWindow.Opts
local function default_opts(opts)
opts = opts or {}
opts.throbber_opts = opts.throbber_opts
or {
throb_time = Consts.throbber_throb_time,
cooldown_time = Consts.throbber_cooldown_time,
tick_time = Consts.throbber_tick_time,
}
opts.in_flight_interval = opts.in_flight_interval
or Consts.show_in_flight_requests_loop_time
opts.enable = opts.enable == nil and true or opts.enable
return opts
end
--- @class _99.StatusWindow
--- @field opts _99.StatusWindow.Opts
--- @field state _99.StatusWindow.State
--- @field win _99.window.Window | nil
--- @field throbber _99.Throbber | nil
--- @field _99 _99.State
local StatusWindow = {}
StatusWindow.__index = StatusWindow
--- @param _99 _99.State
--- @param opts _99.StatusWindow.Opts | nil
function StatusWindow.new(_99, opts)
return setmetatable({
opts = default_opts(opts),
state = "init",
_99 = _99,
}, StatusWindow)
end
function StatusWindow:_shutdown_status_window()
if self.throbber then
self.throbber:stop()
end
local win = self.win
if win ~= nil then
Window.close(win)
end
self.win = nil
self.throbber = nil
end
function StatusWindow:_run_loop()
if self.state ~= "running" then
self:_shutdown_status_window()
return
end
vim.defer_fn(function()
self:_run_loop()
end, self.opts.in_flight_interval)
Window.refresh_active_windows()
local current_win = self.win
if current_win ~= nil and not Window.is_active_window(current_win) then
self:_shutdown_status_window()
end
local active_window = Window.has_active_status_window()
local active_other_window = Window.has_active_windows()
local active_requests = self._99.tracking:active_count()
if
active_window == false and active_other_window
or active_window and active_requests > 0
or active_window == false and active_requests == 0
then
return
end
if current_win == nil then
local ok, win = pcall(Window.status_window)
if not ok then
--- TODO: There needs to be a way to display logs for "all active requests"
--- this is its own activity and should not be added to any work set
return
end
local throb = Throbber.new(function(throb)
local count = self._99.tracking:active_count()
local win_valid = Window.valid(win)
if count == 0 or not win_valid then
return self:_shutdown_status_window()
end
--- @type string[]
local lines = {
throb .. " requests(" .. tostring(count) .. ") " .. throb,
}
for _, c in ipairs(self._99.tracking:active()) do
if c.state == "requesting" then
table.insert(lines, c.operation)
end
end
Window.resize(win, #lines[1], #lines)
vim.api.nvim_buf_set_lines(win.buf_id, 0, -1, false, lines)
end, self.opts.throbber_opts)
self.win = win
self.throbber = throb
throb:start()
end
end
function StatusWindow:start()
if not self.opts.enable then
return
end
assert(
self.state == "init",
"you cannot start an inflight request if we are not in init state: "
.. self.state
)
self.state = "running"
self:_run_loop()
end
function StatusWindow:stop()
if not self.opts.enable then
return
end
assert(
self.state == "running",
"you cannot stop a running status window if its not running"
)
self.state = "init"
end
return StatusWindow
|
