From 252176d064cf9aaa58fca9d3e4e27e8cb518ec73 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Petr=20Mr=C3=A1zek?= <peterix@gmail.com>
Date: Wed, 22 Feb 2012 22:46:12 +0100
Subject: [PATCH] Add abuse mode to mode plugin.

---
 plugins/mode.cpp | 126 +++++++++++++++++++++++++++++------------------
 1 file changed, 77 insertions(+), 49 deletions(-)

diff --git a/plugins/mode.cpp b/plugins/mode.cpp
index 1dfd42510..c275ae69d 100644
--- a/plugins/mode.cpp
+++ b/plugins/mode.cpp
@@ -69,6 +69,9 @@ void printCurrentModes(t_gamemodes gm, Console & con)
     case GAMETYPE_NONE:
         con << "NONE)" << endl;
         break;
+    default:
+        con << "!!UNKNOWN!!)" << endl;
+        break;
     }
     con << "Current game mode:\t" << gm.g_mode << " (";
     switch (gm.g_mode)
@@ -85,6 +88,9 @@ void printCurrentModes(t_gamemodes gm, Console & con)
     case GAMEMODE_NONE:
         con << "NONE)" << endl;
         break;
+    default:
+        con << "!!UNKNOWN!!)" << endl;
+        break;
     }
 }
 
@@ -92,13 +98,18 @@ command_result mode (Core * c, vector <string> & parameters)
 {
     string command = "";
     bool set = false;
+    bool abuse = false;
     t_gamemodes gm;
-    if(parameters.size())
+    for(auto iter = parameters.begin(); iter != parameters.end(); iter++)
     {
-        if(parameters[0] == "set")
+        if((*iter) == "set")
         {
             set = true;
         }
+        else if((*iter) == "abuse")
+        {
+            set = abuse = true;
+        }
         else
             return CR_WRONG_USAGE;
     }
@@ -110,57 +121,74 @@ command_result mode (Core * c, vector <string> & parameters)
     printCurrentModes(gm, c->con);
     if(set)
     {
-        if( gm.g_mode == GAMEMODE_NONE || gm.g_type == GAMETYPE_VIEW_LEGENDS )
+        if(!abuse)
         {
-            c->con.printerr("It is not safe to set modes in menus.\n");
-            return CR_FAILURE;
-        }
-        c->con << "\nPossible choices:" << endl
-               << "0 = Fortress Mode" << endl
-               << "1 = Adventurer Mode" << endl
-               << "2 = Arena Mode" << endl
-               << "3 = Arena, controlling creature" << endl
-               << "4 = Reclaim Fortress Mode" << endl
-               << "c = cancel/do nothing" << endl;
-        uint32_t select=99;
+            if( gm.g_mode == GAMEMODE_NONE || gm.g_type == GAMETYPE_VIEW_LEGENDS)
+            {
+                c->con.printerr("It is not safe to set modes in menus.\n");
+                return CR_FAILURE;
+            }
+            c->con << "\nPossible choices:" << endl
+                   << "0 = Fortress Mode" << endl
+                   << "1 = Adventurer Mode" << endl
+                   << "2 = Arena Mode" << endl
+                   << "3 = Arena, controlling creature" << endl
+                   << "4 = Reclaim Fortress Mode" << endl
+                   << "c = cancel/do nothing" << endl;
+            uint32_t select=99;
 
-        string selected;
-        input_again:
-        CommandHistory hist;
-        c->con.lineedit("Enter new mode: ",selected, hist);
-        if(selected == "c")
-            return CR_OK;
-        const char * start = selected.c_str();
-        char * end = 0;
-        select = strtol(start, &end, 10);
-        if(!end || end==start || select > 4)
-        {
-            c->con.printerr("This is not a valid selection.\n");
-            goto input_again;
+            string selected;
+            input_again:
+            CommandHistory hist;
+            c->con.lineedit("Enter new mode: ",selected, hist);
+            if(selected == "c")
+                return CR_OK;
+            const char * start = selected.c_str();
+            char * end = 0;
+            select = strtol(start, &end, 10);
+            if(!end || end==start || select > 4)
+            {
+                c->con.printerr("This is not a valid selection.\n");
+                goto input_again;
+            }
+            switch(select)
+            {
+                case 0:
+                    gm.g_mode = GAMEMODE_DWARF;
+                    gm.g_type = GAMETYPE_DWARF_MAIN;
+                    break;
+                case 1:
+                    gm.g_mode = GAMEMODE_ADVENTURE;
+                    gm.g_type = GAMETYPE_ADVENTURE_MAIN;
+                    break;
+                case 2:
+                    gm.g_mode = GAMEMODE_DWARF;
+                    gm.g_type = GAMETYPE_DWARF_ARENA;
+                    break;
+                case 3:
+                    gm.g_mode = GAMEMODE_ADVENTURE;
+                    gm.g_type = GAMETYPE_ADVENTURE_ARENA;
+                    break;
+                case 4:
+                    gm.g_mode = GAMEMODE_DWARF;
+                    gm.g_type = GAMETYPE_DWARF_RECLAIM;
+                    break;
+            }
         }
-
-        switch(select)
+        else
         {
-            case 0:
-                gm.g_mode = GAMEMODE_DWARF;
-                gm.g_type = GAMETYPE_DWARF_MAIN;
-                break;
-            case 1:
-                gm.g_mode = GAMEMODE_ADVENTURE;
-                gm.g_type = GAMETYPE_ADVENTURE_MAIN;
-                break;
-            case 2:
-                gm.g_mode = GAMEMODE_DWARF;
-                gm.g_type = GAMETYPE_DWARF_ARENA;
-                break;
-            case 3:
-                gm.g_mode = GAMEMODE_ADVENTURE;
-                gm.g_type = GAMETYPE_ADVENTURE_ARENA;
-                break;
-            case 4:
-                gm.g_mode = GAMEMODE_DWARF;
-                gm.g_type = GAMETYPE_DWARF_RECLAIM;
-                break;
+            CommandHistory hist;
+            string selected;
+            c->con.lineedit("Enter new game mode number (c for exit): ",selected, hist);
+            if(selected == "c")
+                return CR_OK;
+            const char * start = selected.c_str();
+            gm.g_mode = (GameMode) strtol(start, 0, 10);
+            c->con.lineedit("Enter new game type number (c for exit): ",selected, hist);
+            if(selected == "c")
+                return CR_OK;
+            start = selected.c_str();
+            gm.g_type = (GameType) strtol(start, 0, 10);
         }
         c->Suspend();
         world->WriteGameMode(gm);