infrastructure for hiding armok tools

develop
Myk Taylor 2023-04-17 09:39:15 -07:00
parent 4e01ac1f83
commit e9f6695ace
No known key found for this signature in database
GPG Key ID: 8A39CA0FA0C16E78
6 changed files with 67 additions and 12 deletions

@ -21,7 +21,7 @@ for the tag assignment spreadsheet.
"why" tags "why" tags
---------- ----------
- `armok <armok-tag-index>`: Tools that give you complete control over an aspect of the game or provide access to information that the game intentionally keeps hidden. - `armok <armok-tag-index>`: Tools that give you complete control over an aspect of the game or provide access to information that the game intentionally keeps hidden. Players that do not wish to see these tools listed in DFHack command lists can hide them in the ``Preferences`` tab of `gui/control-panel`.
- `auto <auto-tag-index>`: Tools that run in the background and automatically manage routine, toilsome aspects of your fortress. - `auto <auto-tag-index>`: Tools that run in the background and automatically manage routine, toilsome aspects of your fortress.
- `bugfix <bugfix-tag-index>`: Tools that fix specific bugs, either permanently or on-demand. - `bugfix <bugfix-tag-index>`: Tools that fix specific bugs, either permanently or on-demand.
- `design <design-tag-index>`: Tools that help you design your fort. - `design <design-tag-index>`: Tools that help you design your fort.

@ -40,7 +40,8 @@ changelog.txt uses a syntax similar to RST, with a few special sequences:
## Misc Improvements ## Misc Improvements
- `buildingplan`: minimized planner panel stays minimized until you change it again - `buildingplan`: minimized planner panel stays minimized until you change it again
- ``toggle-kbd-cursor``: add hotkey for toggling the keyboard cursor (Alt-K) - ``toggle-kbd-cursor``: add hotkey for toggling the keyboard cursor (Alt-K)
- `gui/control-panel`: add option for hiding the terminal console by default - `gui/control-panel`: add preference option for hiding the terminal console on startup
- `gui/control-panel`: add preference option for hiding "armok" tools in command lists
## Documentation ## Documentation

@ -58,6 +58,11 @@ function dfhack.getHideConsoleOnStartup()
return dfhack.HIDE_CONSOLE_ON_STARTUP return dfhack.HIDE_CONSOLE_ON_STARTUP
end end
dfhack.HIDE_ARMOK_TOOLS = false
function dfhack.getHideArmokTools()
return dfhack.HIDE_ARMOK_TOOLS
end
-- Error handling -- Error handling
safecall = dfhack.safecall safecall = dfhack.safecall

@ -788,7 +788,11 @@ function ls(filter_str, skip_tags, show_dev_commands, exclude_strs)
table.insert(excludes, {str=argparse.stringList(exclude_strs)}) table.insert(excludes, {str=argparse.stringList(exclude_strs)})
end end
if not show_dev_commands then if not show_dev_commands then
table.insert(excludes, {tag='dev'}) local dev_tags = {'dev', 'unavailable'}
if dfhack.getHideArmokTools() then
table.insert(dev_tags, 'armok')
end
table.insert(excludes, {tag=dev_tags})
end end
list_entries(skip_tags, include, excludes) list_entries(skip_tags, include, excludes)
end end
@ -813,7 +817,16 @@ function tags(tag)
local skip_tags = true local skip_tags = true
local include = {entry_type={ENTRY_TYPES.COMMAND}, tag=tag} local include = {entry_type={ENTRY_TYPES.COMMAND}, tag=tag}
list_entries(skip_tags, include)
local excludes = {tag={}}
if tag ~= 'unavailable' then
table.insert(excludes.tag, 'unavailable')
end
if tag ~= 'armok' and dfhack.getHideArmokTools() then
table.insert(excludes.tag, 'armok')
end
list_entries(skip_tags, include, excludes)
end end
return _ENV return _ENV

@ -49,7 +49,22 @@ static int cleanupHotkeys(lua_State *) {
return 0; return 0;
} }
static void add_binding_if_valid(const string &sym, const string &cmdline, df::viewscreen *screen, bool filtermenu) { static bool should_hide_armok(color_ostream &out, const string &cmdline) {
bool should_hide = false;
auto L = Lua::Core::State;
Lua::StackUnwinder top(L);
Lua::CallLuaModuleFunction(out, L, "plugins.hotkeys", "should_hide_armok", 1, 1,
[&](lua_State *L){
Lua::Push(L, cmdline);
}, [&](lua_State *L){
should_hide = lua_toboolean(L, -1);
});
return should_hide;
}
static void add_binding_if_valid(color_ostream &out, const string &sym, const string &cmdline, df::viewscreen *screen, bool filtermenu) {
if (!can_invoke(cmdline, screen)) if (!can_invoke(cmdline, screen))
return; return;
@ -59,6 +74,11 @@ static void add_binding_if_valid(const string &sym, const string &cmdline, df::v
return; return;
} }
if (should_hide_armok(out, cmdline)) {
DEBUG(log).print("filtering out armok keybinding\n");
return;
}
current_bindings[sym] = cmdline; current_bindings[sym] = cmdline;
sorted_keys.push_back(sym); sorted_keys.push_back(sym);
string keyspec = sym + "@" + MENU_SCREEN_FOCUS_STRING; string keyspec = sym + "@" + MENU_SCREEN_FOCUS_STRING;
@ -67,7 +87,7 @@ static void add_binding_if_valid(const string &sym, const string &cmdline, df::v
Core::getInstance().AddKeyBinding(keyspec, binding); Core::getInstance().AddKeyBinding(keyspec, binding);
} }
static void find_active_keybindings(df::viewscreen *screen, bool filtermenu) { static void find_active_keybindings(color_ostream &out, df::viewscreen *screen, bool filtermenu) {
DEBUG(log).print("scanning for active keybindings\n"); DEBUG(log).print("scanning for active keybindings\n");
if (valid) if (valid)
cleanupHotkeys(NULL); cleanupHotkeys(NULL);
@ -103,7 +123,7 @@ static void find_active_keybindings(df::viewscreen *screen, bool filtermenu) {
string::size_type colon_pos = invoke_cmd->find(":"); string::size_type colon_pos = invoke_cmd->find(":");
// colons at location 0 are for commands like ":lua" // colons at location 0 are for commands like ":lua"
if (colon_pos == string::npos || colon_pos == 0) { if (colon_pos == string::npos || colon_pos == 0) {
add_binding_if_valid(sym, *invoke_cmd, screen, filtermenu); add_binding_if_valid(out, sym, *invoke_cmd, screen, filtermenu);
} }
else { else {
vector<string> tokens; vector<string> tokens;
@ -111,7 +131,7 @@ static void find_active_keybindings(df::viewscreen *screen, bool filtermenu) {
string focus = tokens[0].substr(1); string focus = tokens[0].substr(1);
if(Gui::matchFocusString(focus)) { if(Gui::matchFocusString(focus)) {
auto cmdline = trim(tokens[1]); auto cmdline = trim(tokens[1]);
add_binding_if_valid(sym, cmdline, screen, filtermenu); add_binding_if_valid(out, sym, cmdline, screen, filtermenu);
} }
} }
} }
@ -124,7 +144,10 @@ static void find_active_keybindings(df::viewscreen *screen, bool filtermenu) {
} }
static int getHotkeys(lua_State *L) { static int getHotkeys(lua_State *L) {
find_active_keybindings(Gui::getCurViewscreen(true), true); color_ostream *out = Lua::GetOutput(L);
if (!out)
out = &Core::getInstance().getConsole();
find_active_keybindings(*out, Gui::getCurViewscreen(true), true);
Lua::PushVector(L, sorted_keys); Lua::PushVector(L, sorted_keys);
Lua::Push(L, current_bindings); Lua::Push(L, current_bindings);
return 2; return 2;
@ -140,7 +163,7 @@ static void list(color_ostream &out) {
DEBUG(log).print("listing active hotkeys\n"); DEBUG(log).print("listing active hotkeys\n");
bool was_valid = valid; bool was_valid = valid;
if (!valid) if (!valid)
find_active_keybindings(Gui::getCurViewscreen(true), false); find_active_keybindings(out, Gui::getCurViewscreen(true), false);
out.print("Valid keybindings for the current focus:\n %s\n", out.print("Valid keybindings for the current focus:\n %s\n",
join_strings("\n", Gui::getCurFocus(true)).c_str()); join_strings("\n", Gui::getCurFocus(true)).c_str());
@ -176,6 +199,8 @@ static command_result hotkeys_cmd(color_ostream &out, vector <string> & paramete
return Core::getInstance().runCommand(out, INVOKE_MENU_COMMAND ); return Core::getInstance().runCommand(out, INVOKE_MENU_COMMAND );
} }
CoreSuspender guard;
if (parameters[0] == "list") { if (parameters[0] == "list") {
list(out); list(out);
return CR_OK; return CR_OK;
@ -185,8 +210,6 @@ static command_result hotkeys_cmd(color_ostream &out, vector <string> & paramete
if (parameters.size() != 2 || parameters[0] != "invoke") if (parameters.size() != 2 || parameters[0] != "invoke")
return CR_WRONG_USAGE; return CR_WRONG_USAGE;
CoreSuspender guard;
int index = string_to_int(parameters[1], -1); int index = string_to_int(parameters[1], -1);
if (index < 0) if (index < 0)
return CR_WRONG_USAGE; return CR_WRONG_USAGE;

@ -5,6 +5,19 @@ local helpdb = require('helpdb')
local overlay = require('plugins.overlay') local overlay = require('plugins.overlay')
local widgets = require('gui.widgets') local widgets = require('gui.widgets')
local function get_command(cmdline)
local first_word = cmdline:trim():split(' +')[1]
if first_word:startswith(':') then first_word = first_word:sub(2) end
return first_word
end
function should_hide_armok(cmdline)
local first_word = get_command(cmdline)
return dfhack.getHideArmokTools() and
helpdb.is_entry(first_word) and
helpdb.get_entry_tags(first_word).armok
end
-- ----------------- -- -- ----------------- --
-- HotspotMenuWidget -- -- HotspotMenuWidget --
-- ----------------- -- -- ----------------- --