2015-06-05 19:49:22 -06:00
|
|
|
#include <set>
|
|
|
|
#include <map>
|
|
|
|
#include "Console.h"
|
|
|
|
#include "Core.h"
|
|
|
|
#include "DataDefs.h"
|
|
|
|
#include "Export.h"
|
|
|
|
#include "PluginManager.h"
|
|
|
|
#include "VTableInterpose.h"
|
|
|
|
#include "uicommon.h"
|
2015-06-06 08:31:12 -06:00
|
|
|
#include "modules/Gui.h"
|
|
|
|
|
|
|
|
#include "df/building_tradedepotst.h"
|
|
|
|
#include "df/viewscreen_dwarfmodest.h"
|
2015-06-09 11:52:37 -06:00
|
|
|
#include "df/viewscreen_layer_militaryst.h"
|
2015-06-05 19:49:22 -06:00
|
|
|
#include "df/viewscreen_tradegoodsst.h"
|
|
|
|
|
|
|
|
using namespace DFHack;
|
2015-06-06 08:31:12 -06:00
|
|
|
using namespace df::enums;
|
|
|
|
using std::string;
|
|
|
|
using std::vector;
|
2015-06-05 19:49:22 -06:00
|
|
|
|
|
|
|
DFHACK_PLUGIN("confirm");
|
|
|
|
DFHACK_PLUGIN_IS_ENABLED(is_enabled);
|
|
|
|
REQUIRE_GLOBAL(gps);
|
2015-06-06 08:31:12 -06:00
|
|
|
REQUIRE_GLOBAL(ui);
|
2015-06-05 19:49:22 -06:00
|
|
|
|
|
|
|
typedef std::set<df::interface_key> ikey_set;
|
2015-06-06 08:31:12 -06:00
|
|
|
command_result df_confirm (color_ostream &out, vector <string> & parameters);
|
2015-06-05 19:49:22 -06:00
|
|
|
|
2015-06-06 08:31:12 -06:00
|
|
|
static std::multimap<string, VMethodInterposeLinkBase> hooks;
|
2015-06-05 19:49:22 -06:00
|
|
|
|
2015-06-09 11:52:37 -06:00
|
|
|
#define IMPLEMENT_CONFIRMATION_HOOKS(cls) IMPLEMENT_CONFIRMATION_HOOKS_PRIO(cls, 0)
|
|
|
|
#define IMPLEMENT_CONFIRMATION_HOOKS_PRIO(cls, prio) \
|
2015-06-05 19:49:22 -06:00
|
|
|
static cls cls##_instance; \
|
|
|
|
struct cls##_hooks : cls::screen_type { \
|
|
|
|
typedef cls::screen_type interpose_base; \
|
|
|
|
DEFINE_VMETHOD_INTERPOSE(void, feed, (ikey_set *input)) \
|
|
|
|
{ \
|
|
|
|
cls##_instance.screen = this; \
|
|
|
|
if (!cls##_instance.feed(input)) \
|
|
|
|
INTERPOSE_NEXT(feed)(input); \
|
|
|
|
} \
|
|
|
|
DEFINE_VMETHOD_INTERPOSE(void, render, ()) \
|
|
|
|
{ \
|
|
|
|
cls##_instance.screen = this; \
|
|
|
|
INTERPOSE_NEXT(render)(); \
|
|
|
|
cls##_instance.render(); \
|
|
|
|
} \
|
|
|
|
DEFINE_VMETHOD_INTERPOSE(bool, key_conflict, (df::interface_key key)) \
|
|
|
|
{ \
|
|
|
|
return cls##_instance.key_conflict(key) || INTERPOSE_NEXT(key_conflict)(key); \
|
|
|
|
} \
|
|
|
|
}; \
|
2015-06-09 11:52:37 -06:00
|
|
|
IMPLEMENT_VMETHOD_INTERPOSE_PRIO(cls##_hooks, feed, prio); \
|
|
|
|
IMPLEMENT_VMETHOD_INTERPOSE_PRIO(cls##_hooks, render, prio); \
|
|
|
|
IMPLEMENT_VMETHOD_INTERPOSE_PRIO(cls##_hooks, key_conflict, prio);
|
2015-06-05 19:49:22 -06:00
|
|
|
|
|
|
|
template <class T>
|
|
|
|
class confirmation {
|
|
|
|
public:
|
|
|
|
enum cstate { INACTIVE, ACTIVE, SELECTED };
|
|
|
|
typedef T screen_type;
|
|
|
|
screen_type *screen;
|
|
|
|
bool feed (ikey_set *input) {
|
|
|
|
if (state == INACTIVE)
|
|
|
|
{
|
|
|
|
for (auto it = input->begin(); it != input->end(); ++it)
|
|
|
|
{
|
|
|
|
if (intercept_key(*it))
|
|
|
|
{
|
|
|
|
last_key = *it;
|
|
|
|
state = ACTIVE;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
else if (state == ACTIVE)
|
|
|
|
{
|
|
|
|
if (input->count(df::interface_key::LEAVESCREEN))
|
|
|
|
state = INACTIVE;
|
|
|
|
else if (input->count(df::interface_key::SELECT))
|
|
|
|
state = SELECTED;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
bool key_conflict (df::interface_key key)
|
|
|
|
{
|
|
|
|
if (key == df::interface_key::SELECT || key == df::interface_key::LEAVESCREEN)
|
|
|
|
return false;
|
|
|
|
return state == ACTIVE;
|
|
|
|
}
|
|
|
|
void render() {
|
2015-06-06 08:31:12 -06:00
|
|
|
static vector<string> lines;
|
2015-06-05 19:49:22 -06:00
|
|
|
Screen::Pen corner_ul = Screen::Pen((char)201, COLOR_GREY, COLOR_BLACK);
|
|
|
|
Screen::Pen corner_ur = Screen::Pen((char)187, COLOR_GREY, COLOR_BLACK);
|
|
|
|
Screen::Pen corner_dl = Screen::Pen((char)200, COLOR_GREY, COLOR_BLACK);
|
|
|
|
Screen::Pen corner_dr = Screen::Pen((char)188, COLOR_GREY, COLOR_BLACK);
|
|
|
|
Screen::Pen border_ud = Screen::Pen((char)205, COLOR_GREY, COLOR_BLACK);
|
|
|
|
Screen::Pen border_lr = Screen::Pen((char)186, COLOR_GREY, COLOR_BLACK);
|
|
|
|
if (state == ACTIVE)
|
|
|
|
{
|
|
|
|
split_string(&lines, get_message(), "\n");
|
|
|
|
size_t max_length = 30;
|
|
|
|
for (auto it = lines.begin(); it != lines.end(); ++it)
|
|
|
|
max_length = std::max(max_length, it->size());
|
|
|
|
int width = max_length + 4;
|
|
|
|
int height = lines.size() + 4;
|
|
|
|
int x1 = (gps->dimx / 2) - (width / 2);
|
|
|
|
int x2 = x1 + width - 1;
|
|
|
|
int y1 = (gps->dimy / 2) - (height / 2);
|
|
|
|
int y2 = y1 + height - 1;
|
|
|
|
for (int x = x1; x <= x2; x++)
|
|
|
|
{
|
|
|
|
Screen::paintTile(border_ud, x, y1);
|
|
|
|
Screen::paintTile(border_ud, x, y2);
|
|
|
|
}
|
|
|
|
for (int y = y1; y <= y2; y++)
|
|
|
|
{
|
|
|
|
Screen::paintTile(border_lr, x1, y);
|
|
|
|
Screen::paintTile(border_lr, x2, y);
|
|
|
|
}
|
|
|
|
Screen::paintTile(corner_ul, x1, y1);
|
|
|
|
Screen::paintTile(corner_ur, x2, y1);
|
|
|
|
Screen::paintTile(corner_dl, x1, y2);
|
|
|
|
Screen::paintTile(corner_dr, x2, y2);
|
2015-06-06 08:31:12 -06:00
|
|
|
string title = " " + get_title() + " ";
|
|
|
|
Screen::paintString(Screen::Pen(' ', COLOR_DARKGREY, COLOR_BLACK),
|
|
|
|
x2 - 6, y1, "DFHack");
|
2015-06-05 19:49:22 -06:00
|
|
|
Screen::paintString(Screen::Pen(' ', COLOR_BLACK, COLOR_GREY),
|
|
|
|
(gps->dimx / 2) - (title.size() / 2), y1, title);
|
|
|
|
int x = x1 + 2;
|
|
|
|
OutputString(COLOR_LIGHTGREEN, x, y2, Screen::getKeyDisplay(df::interface_key::LEAVESCREEN));
|
|
|
|
OutputString(COLOR_WHITE, x, y2, ": Cancel");
|
2015-06-09 11:52:37 -06:00
|
|
|
x = x2 - 2 - 3 - Screen::getKeyDisplay(df::interface_key::SELECT).size();
|
2015-06-05 19:49:22 -06:00
|
|
|
OutputString(COLOR_LIGHTGREEN, x, y2, Screen::getKeyDisplay(df::interface_key::SELECT));
|
|
|
|
OutputString(COLOR_WHITE, x, y2, ": Ok");
|
|
|
|
Screen::fillRect(Screen::Pen(' ', COLOR_BLACK, COLOR_BLACK), x1 + 1, y1 + 1, x2 - 1, y2 - 1);
|
|
|
|
for (size_t i = 0; i < lines.size(); i++)
|
|
|
|
{
|
|
|
|
Screen::paintString(Screen::Pen(' ', get_color(), COLOR_BLACK), x1 + 2, y1 + 2 + i, lines[i]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else if (state == SELECTED)
|
|
|
|
{
|
|
|
|
ikey_set tmp;
|
|
|
|
tmp.insert(last_key);
|
|
|
|
screen->feed(&tmp);
|
|
|
|
state = INACTIVE;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
virtual bool intercept_key (df::interface_key key) = 0;
|
2015-06-06 08:31:12 -06:00
|
|
|
virtual string get_title() { return "Confirm"; }
|
|
|
|
virtual string get_message() = 0;
|
2015-06-05 19:49:22 -06:00
|
|
|
virtual UIColor get_color() { return COLOR_YELLOW; }
|
|
|
|
protected:
|
|
|
|
cstate state;
|
|
|
|
df::interface_key last_key;
|
|
|
|
};
|
|
|
|
|
2015-06-09 11:52:37 -06:00
|
|
|
class trade_confirmation : public confirmation<df::viewscreen_tradegoodsst> {
|
|
|
|
public:
|
|
|
|
virtual bool intercept_key (df::interface_key key) { return key == df::interface_key::TRADE_TRADE; }
|
|
|
|
virtual string get_id() { return "trade"; }
|
|
|
|
virtual string get_title() { return "Confirm trade"; }
|
|
|
|
virtual string get_message() { return "Are you sure you want to trade the selected goods?"; }
|
|
|
|
};
|
|
|
|
IMPLEMENT_CONFIRMATION_HOOKS(trade_confirmation);
|
|
|
|
|
|
|
|
class trade_cancel_confirmation : public confirmation<df::viewscreen_tradegoodsst> {
|
|
|
|
public:
|
|
|
|
virtual bool intercept_key (df::interface_key key)
|
|
|
|
{
|
|
|
|
if (key == df::interface_key::LEAVESCREEN)
|
|
|
|
{
|
|
|
|
#define check_list(list) for (auto it = screen->list.begin(); it != screen->list.end(); ++it) if (*it) return true
|
|
|
|
check_list(trader_selected);
|
|
|
|
check_list(broker_selected);
|
|
|
|
#undef check_list
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
virtual string get_id() { return "trade-cancel"; }
|
|
|
|
virtual string get_title() { return "Cancel trade"; }
|
|
|
|
virtual string get_message() { return "Are you sure you want leave this screen?\nSelected items will not be saved."; }
|
|
|
|
};
|
|
|
|
IMPLEMENT_CONFIRMATION_HOOKS_PRIO(trade_cancel_confirmation, -1);
|
|
|
|
|
2015-06-05 19:49:22 -06:00
|
|
|
class trade_seize_confirmation : public confirmation<df::viewscreen_tradegoodsst> {
|
|
|
|
public:
|
|
|
|
virtual bool intercept_key (df::interface_key key) { return key == df::interface_key::TRADE_SEIZE; }
|
2015-06-06 08:31:12 -06:00
|
|
|
virtual string get_id() { return "trade-seize"; }
|
|
|
|
virtual string get_title() { return "Confirm seize"; }
|
|
|
|
virtual string get_message() { return "Are you sure you want to sieze these goods?"; }
|
2015-06-05 19:49:22 -06:00
|
|
|
};
|
|
|
|
IMPLEMENT_CONFIRMATION_HOOKS(trade_seize_confirmation);
|
|
|
|
|
|
|
|
class trade_offer_confirmation : public confirmation<df::viewscreen_tradegoodsst> {
|
|
|
|
public:
|
|
|
|
virtual bool intercept_key (df::interface_key key) { return key == df::interface_key::TRADE_OFFER; }
|
2015-06-06 08:31:12 -06:00
|
|
|
virtual string get_id() { return "trade-offer"; }
|
|
|
|
virtual string get_title() { return "Confirm offer"; }
|
|
|
|
virtual string get_message() { return "Are you sure you want to offer these goods?\nYou will receive no payment."; }
|
2015-06-05 19:49:22 -06:00
|
|
|
};
|
|
|
|
IMPLEMENT_CONFIRMATION_HOOKS(trade_offer_confirmation);
|
|
|
|
|
2015-06-06 08:31:12 -06:00
|
|
|
class hauling_route_delete_confirmation : public confirmation<df::viewscreen_dwarfmodest> {
|
|
|
|
public:
|
|
|
|
virtual bool intercept_key (df::interface_key key)
|
|
|
|
{
|
|
|
|
if (ui->main.mode == ui_sidebar_mode::Hauling && ui->hauling.view_routes.size())
|
|
|
|
return key == df::interface_key::D_HAULING_REMOVE;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
virtual string get_id() { return "haul-delete"; }
|
|
|
|
virtual string get_title() { return "Confirm deletion"; }
|
|
|
|
virtual string get_message()
|
|
|
|
{
|
|
|
|
std::string type = (ui->hauling.view_stops[ui->hauling.cursor_top]) ? "stop" : "route";
|
|
|
|
return std::string("Are you sure you want to delete this ") + type + "?";
|
|
|
|
}
|
|
|
|
};
|
|
|
|
IMPLEMENT_CONFIRMATION_HOOKS(hauling_route_delete_confirmation);
|
|
|
|
|
|
|
|
class depot_remove_confirmation : public confirmation<df::viewscreen_dwarfmodest> {
|
|
|
|
public:
|
|
|
|
virtual bool intercept_key (df::interface_key key)
|
|
|
|
{
|
|
|
|
df::building_tradedepotst *depot = virtual_cast<df::building_tradedepotst>(Gui::getAnyBuilding(screen));
|
|
|
|
if (depot && key == df::interface_key::DESTROYBUILDING)
|
|
|
|
{
|
|
|
|
for (auto it = ui->caravans.begin(); it != ui->caravans.end(); ++it)
|
|
|
|
{
|
|
|
|
if ((**it).time_remaining)
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
virtual string get_id() { return "depot-remove"; }
|
|
|
|
virtual string get_title() { return "Confirm depot removal"; }
|
|
|
|
virtual string get_message()
|
|
|
|
{
|
|
|
|
return "Are you sure you want to remove this depot?\n"
|
|
|
|
"Merchants are present and will lose profits.";
|
|
|
|
}
|
|
|
|
};
|
|
|
|
IMPLEMENT_CONFIRMATION_HOOKS(depot_remove_confirmation);
|
|
|
|
|
2015-06-09 11:52:37 -06:00
|
|
|
class squad_disband_confirmation : public confirmation<df::viewscreen_layer_militaryst> {
|
|
|
|
public:
|
|
|
|
virtual bool intercept_key (df::interface_key key)
|
|
|
|
{
|
|
|
|
return screen->num_squads && key == df::interface_key::D_MILITARY_DISBAND_SQUAD;
|
|
|
|
}
|
|
|
|
virtual string get_id() { return "squad-disband"; }
|
|
|
|
virtual string get_title() { return "Disband squad"; }
|
|
|
|
virtual string get_message() { return "Are you sure you want to disband this squad?"; }
|
|
|
|
};
|
|
|
|
IMPLEMENT_CONFIRMATION_HOOKS(squad_disband_confirmation);
|
|
|
|
|
|
|
|
class note_delete_confirmation : public confirmation<df::viewscreen_dwarfmodest> {
|
|
|
|
public:
|
|
|
|
virtual bool intercept_key (df::interface_key key)
|
|
|
|
{
|
|
|
|
return ui->main.mode == ui_sidebar_mode::NotesPoints && key == df::interface_key::D_NOTE_DELETE;
|
|
|
|
}
|
|
|
|
virtual string get_id() { return "note-delete"; }
|
|
|
|
virtual string get_title() { return "Delete note"; }
|
|
|
|
virtual string get_message() { return "Are you sure you want to delete this note?"; }
|
|
|
|
};
|
|
|
|
IMPLEMENT_CONFIRMATION_HOOKS(note_delete_confirmation);
|
|
|
|
|
|
|
|
class route_delete_confirmation : public confirmation<df::viewscreen_dwarfmodest> {
|
|
|
|
public:
|
|
|
|
virtual bool intercept_key (df::interface_key key)
|
|
|
|
{
|
|
|
|
return ui->main.mode == ui_sidebar_mode::NotesRoutes && key == df::interface_key::D_NOTE_ROUTE_DELETE;
|
|
|
|
}
|
|
|
|
virtual string get_id() { return "route-delete"; }
|
|
|
|
virtual string get_title() { return "Delete route"; }
|
|
|
|
virtual string get_message() { return "Are you sure you want to delete this route?"; }
|
|
|
|
};
|
|
|
|
IMPLEMENT_CONFIRMATION_HOOKS(route_delete_confirmation);
|
|
|
|
|
2015-06-05 19:49:22 -06:00
|
|
|
#define CHOOK(cls) \
|
2015-06-06 08:31:12 -06:00
|
|
|
HOOK_ACTION(cls, cls##_hooks, render) \
|
|
|
|
HOOK_ACTION(cls, cls##_hooks, feed) \
|
|
|
|
HOOK_ACTION(cls, cls##_hooks, key_conflict)
|
2015-06-05 19:49:22 -06:00
|
|
|
|
|
|
|
#define CHOOKS \
|
2015-06-09 11:52:37 -06:00
|
|
|
CHOOK(trade_confirmation) \
|
|
|
|
CHOOK(trade_cancel_confirmation) \
|
2015-06-05 19:49:22 -06:00
|
|
|
CHOOK(trade_seize_confirmation) \
|
2015-06-06 08:31:12 -06:00
|
|
|
CHOOK(trade_offer_confirmation) \
|
|
|
|
CHOOK(hauling_route_delete_confirmation) \
|
2015-06-09 11:52:37 -06:00
|
|
|
CHOOK(depot_remove_confirmation) \
|
|
|
|
CHOOK(squad_disband_confirmation) \
|
|
|
|
CHOOK(note_delete_confirmation) \
|
|
|
|
CHOOK(route_delete_confirmation)
|
2015-06-05 19:49:22 -06:00
|
|
|
|
2015-06-06 08:31:12 -06:00
|
|
|
DFhackCExport command_result plugin_init (color_ostream &out, vector <PluginCommand> &commands)
|
2015-06-05 19:49:22 -06:00
|
|
|
{
|
2015-06-06 08:31:12 -06:00
|
|
|
vector<string> hook_names;
|
|
|
|
#define HOOK_ACTION(cls, hookcls, method) hooks.insert(std::pair<string, VMethodInterposeLinkBase>(cls##_instance.get_id(), INTERPOSE_HOOK(hookcls, method))); \
|
2015-06-05 19:49:22 -06:00
|
|
|
if (std::find(hook_names.begin(), hook_names.end(), cls##_instance.get_id()) == hook_names.end()) hook_names.push_back(cls##_instance.get_id());
|
|
|
|
CHOOKS
|
|
|
|
#undef HOOK_ACTION
|
2015-06-06 08:31:12 -06:00
|
|
|
string help =
|
2015-06-05 19:49:22 -06:00
|
|
|
" confirmation enable|disable option|all ...\n"
|
|
|
|
"Available options:\n " + join_strings(", ", hook_names);
|
|
|
|
commands.push_back(PluginCommand(
|
|
|
|
"confirm",
|
|
|
|
"Confirmation dialogs",
|
|
|
|
df_confirm,
|
|
|
|
false, //allow non-interactive use
|
|
|
|
help.c_str()
|
|
|
|
));
|
|
|
|
return CR_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
DFhackCExport command_result plugin_enable ( color_ostream &out, bool enable)
|
|
|
|
{
|
|
|
|
if (is_enabled != enable)
|
|
|
|
{
|
2015-06-06 08:31:12 -06:00
|
|
|
#define HOOK_ACTION(cls, hookcls, method) !INTERPOSE_HOOK(hookcls, method).apply(enable) ||
|
2015-06-05 19:49:22 -06:00
|
|
|
if (CHOOKS 0)
|
|
|
|
return CR_FAILURE;
|
|
|
|
#undef HOOK_ACTION
|
|
|
|
is_enabled = enable;
|
|
|
|
}
|
|
|
|
|
|
|
|
return CR_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
DFhackCExport command_result plugin_shutdown (color_ostream &out)
|
|
|
|
{
|
2015-06-06 08:31:12 -06:00
|
|
|
#define HOOK_ACTION(cls, hookcls, method) INTERPOSE_HOOK(hookcls, method).remove();
|
2015-06-05 19:49:22 -06:00
|
|
|
CHOOKS;
|
|
|
|
#undef HOOK_ACTION
|
|
|
|
return CR_OK;
|
|
|
|
}
|
|
|
|
|
2015-06-06 08:31:12 -06:00
|
|
|
void enable_conf (color_ostream &out, string name, bool state)
|
2015-06-05 19:49:22 -06:00
|
|
|
{
|
|
|
|
bool found = false;
|
|
|
|
for (auto it = hooks.begin(); it != hooks.end(); ++it)
|
|
|
|
{
|
|
|
|
if (it->first == name)
|
|
|
|
{
|
|
|
|
found = true;
|
|
|
|
it->second.apply(state);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (!found)
|
|
|
|
out.printerr("Unrecognized option: %s\n", name.c_str());
|
|
|
|
}
|
|
|
|
|
2015-06-06 08:31:12 -06:00
|
|
|
command_result df_confirm (color_ostream &out, vector <string> & parameters)
|
2015-06-05 19:49:22 -06:00
|
|
|
{
|
|
|
|
CoreSuspender suspend;
|
|
|
|
bool state = true;
|
|
|
|
for (auto it = parameters.begin(); it != parameters.end(); ++it)
|
|
|
|
{
|
|
|
|
if (*it == "enable")
|
|
|
|
state = true;
|
|
|
|
else if (*it == "disable")
|
|
|
|
state = false;
|
|
|
|
else if (*it == "all")
|
|
|
|
{
|
|
|
|
for (auto it = hooks.begin(); it != hooks.end(); ++it)
|
|
|
|
{
|
|
|
|
it->second.apply(state);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
enable_conf(out, *it, state);
|
|
|
|
}
|
|
|
|
return CR_OK;
|
|
|
|
}
|