180 lines
5.0 KiB
Lua
180 lines
5.0 KiB
Lua
local _ENV = mkmodule('plugins.hotkeys')
|
|
|
|
local gui = require('gui')
|
|
local helpdb = require('helpdb')
|
|
local overlay = require('plugins.overlay')
|
|
local widgets = require('gui.widgets')
|
|
|
|
-- ----------------- --
|
|
-- HotspotMenuWidget --
|
|
-- ----------------- --
|
|
|
|
HotspotMenuWidget = defclass(HotspotMenuWidget, overlay.OverlayWidget)
|
|
HotspotMenuWidget.ATTRS{
|
|
default_pos={x=1,y=2},
|
|
hotspot=true,
|
|
viewscreens={'dwarfmode'},
|
|
overlay_onupdate_max_freq_seconds=0,
|
|
frame={w=2, h=1}
|
|
}
|
|
|
|
function HotspotMenuWidget:init()
|
|
self:addviews{widgets.Label{text='!!'}}
|
|
self.mouseover = false
|
|
end
|
|
|
|
function HotspotMenuWidget:overlay_onupdate()
|
|
local hasMouse = self:getMousePos()
|
|
if hasMouse and not self.mouseover then
|
|
self.mouseover = true
|
|
return true
|
|
end
|
|
self.mouseover = hasMouse
|
|
end
|
|
|
|
function HotspotMenuWidget:overlay_trigger()
|
|
local hotkeys, bindings = getHotkeys()
|
|
return MenuScreen{
|
|
hotspot_frame=self.frame,
|
|
hotkeys=hotkeys,
|
|
bindings=bindings}:show()
|
|
end
|
|
|
|
-- register the menu hotspot with the overlay
|
|
OVERLAY_WIDGETS = {menu=HotspotMenuWidget}
|
|
|
|
-- ---------- --
|
|
-- MenuScreen --
|
|
-- ---------- --
|
|
|
|
local ARROW = string.char(26)
|
|
local MENU_WIDTH = 42
|
|
local MENU_HEIGHT = 12
|
|
|
|
MenuScreen = defclass(MenuScreen, gui.Screen)
|
|
MenuScreen.ATTRS{
|
|
focus_path='hotkeys/menu',
|
|
hotspot_frame=DEFAULT_NIL,
|
|
hotkeys=DEFAULT_NIL,
|
|
bindings=DEFAULT_NIL,
|
|
}
|
|
|
|
function MenuScreen:init()
|
|
self.mouseover = false
|
|
|
|
local list_frame = copyall(self.hotspot_frame)
|
|
list_frame.w = MENU_WIDTH
|
|
list_frame.h = MENU_HEIGHT
|
|
|
|
local help_frame = {w=MENU_WIDTH, l=list_frame.l, r=list_frame.r}
|
|
if list_frame.t then
|
|
help_frame.t = list_frame.t + MENU_HEIGHT + 1
|
|
else
|
|
help_frame.b = list_frame.b + MENU_HEIGHT + 1
|
|
end
|
|
|
|
local bindings = self.bindings
|
|
local choices = {}
|
|
for _,hotkey in ipairs(self.hotkeys) do
|
|
local command = bindings[hotkey]
|
|
local choice_text = command .. (' (%s)'):format(hotkey)
|
|
local choice = {
|
|
icon=ARROW, text=choice_text, command=command}
|
|
table.insert(choices, list_frame.b and 1 or #choices + 1, choice)
|
|
end
|
|
|
|
self:addviews{
|
|
widgets.ResizingPanel{
|
|
autoarrange_subviews=true,
|
|
frame=list_frame,
|
|
frame_style=gui.GREY_LINE_FRAME,
|
|
frame_background=gui.CLEAR_PEN,
|
|
subviews={
|
|
widgets.List{
|
|
view_id='list',
|
|
choices=choices,
|
|
icon_width=2,
|
|
on_select=self:callback('onSelect'),
|
|
on_submit=self:callback('onSubmit'),
|
|
on_submit2=self:callback('onSubmit2'),
|
|
},
|
|
},
|
|
},
|
|
widgets.ResizingPanel{
|
|
view_id='help_panel',
|
|
autoarrange_subviews=true,
|
|
frame=help_frame,
|
|
frame_style=gui.GREY_LINE_FRAME,
|
|
frame_background=gui.CLEAR_PEN,
|
|
subviews={
|
|
widgets.WrappedLabel{
|
|
view_id='help',
|
|
text_to_wrap='',
|
|
scroll_keys={},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
end
|
|
|
|
function MenuScreen:onSelect(_, choice)
|
|
if not choice or #self.subviews == 0 then return end
|
|
local first_word = choice.command:trim():split(' +')[1]
|
|
self.subviews.help.text_to_wrap = helpdb.is_entry(first_word) and
|
|
helpdb.get_entry_short_help(first_word) or 'Command not found'
|
|
self.subviews.help_panel:updateLayout()
|
|
end
|
|
|
|
function MenuScreen:onSubmit(_, choice)
|
|
if not choice then return end
|
|
dfhack.screen.hideGuard(self, dfhack.run_command, choice.command)
|
|
self:dismiss()
|
|
end
|
|
|
|
function MenuScreen:onSubmit2(_, choice)
|
|
if not choice then return end
|
|
self:dismiss()
|
|
dfhack.run_script('gui/launcher', choice.command)
|
|
end
|
|
|
|
function MenuScreen:onInput(keys)
|
|
if keys.LEAVESCREEN then
|
|
self:dismiss()
|
|
return true
|
|
elseif keys.STANDARDSCROLL_RIGHT then
|
|
self:onSubmit2(self.subviews.list:getSelected())
|
|
return true
|
|
elseif keys._MOUSE_L then
|
|
local list = self.subviews.list
|
|
local x = list:getMousePos()
|
|
if x == 0 then -- clicked on icon
|
|
self:onSubmit2(list:getSelected())
|
|
return true
|
|
end
|
|
end
|
|
return self:inputToSubviews(keys)
|
|
end
|
|
|
|
function MenuScreen:onRenderFrame(dc, rect)
|
|
self:renderParent()
|
|
end
|
|
|
|
function MenuScreen:onRenderBody(dc)
|
|
local list = self.subviews.list
|
|
local idx = list:getIdxUnderMouse()
|
|
if idx and idx ~= self.last_mouse_idx then
|
|
-- focus follows mouse, but if cursor keys were used to change the
|
|
-- selection, don't override the selection until the mouse moves to
|
|
-- another item
|
|
list:setSelected(idx)
|
|
self.mouseover = true
|
|
self.last_mouse_idx = idx
|
|
elseif not idx and self.mouseover then
|
|
-- once the mouse has entered the list area, leaving it again should
|
|
-- close the menu screen
|
|
self:dismiss()
|
|
end
|
|
end
|
|
|
|
return _ENV
|