src/Core/Editors/EditorPermission.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2008 by Sverre Rabbelier                                *
   3  *   sverre@rabbelier.nl                                                   *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 3 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  19  ***************************************************************************/
  20
  21 #include "EditorPermission.h"
  22 #include "EditorOLC.h"
  23 #include "UBSocket.h"
  24 #include "Array.h"
  25 #include "StringUtilities.h"
  26 #include "TableImpls.h"
  27 #include "Account.h"
  28 #include "AccountManager.h"
  29 #include "Permission.h"
  30 #include "PermissionManager.h"
  31 #include "GrantGroup.h"
  32 #include "GrantGroupManager.h"
  33 #include "Managers.h"
  34
  35 using mud::Permission;
  36
  37 typedef EditorPermission E;
  38 typedef CommandInfoObject<E> O;
  39 typedef CommandBinding<E> B;
  40
  41 //                                              name            function          need: object lock
  42 static O editAccount(   "Account",      &E::editAccount,        true, true);
  43 static O editGrantGroup("GrantGroup",&E::editGrantGroup,true, true);
  44 static O editGrant(             "Grant",        &E::editGrant,          true, true);
  45 static O editLogging(   "Logging",      &E::editLogging,        true, true);
  46 static O showPermission("Show",         &E::showPermission,     true, false);
  47 static O savePermission("Save",         &E::savePermission, true, true);
  48
  49 static const B commands[] = {
  50         B("account", editAccount),
  51         B("grant", editGrant),
  52         B("grantgroup", editGrantGroup),
  53         B("logging", editLogging),
  54         B("save", savePermission),
  55         B("show", showPermission),
  56 };
  57
  58 EditorPermission::EditorPermission(UBSocket* sock) :
  59 OLCEditor(sock),
  60 m_commands(commands, array_size(commands)),
  61 m_permission()
  62 {
  63         listCommands(Global::Get()->EmptyString);
  64 }
  65
  66 EditorPermission::~EditorPermission(void)
  67 {
  68
  69 }
  70
  71 std::string EditorPermission::lookup(const std::string& action)
  72 {
  73         std::string name = OLCEditor::lookup(action);
  74         if(name.size() != 0)
  75                 return name;
  76
  77         const PermissionCommand* act = (PermissionCommand*)m_commands.getObject(action);
  78         if(act)
  79                 return act->getName();
  80
  81         return Global::Get()->EmptyString;
  82 }
  83
  84 void EditorPermission::dispatch(const std::string& action, const std::string& argument)
  85 {
  86         const PermissionCommand* act = (PermissionCommand*)m_commands.getObject(action);
  87
  88         if(!act)
  89         {
  90                 OLCEditor::dispatch(action, argument);
  91                 return;
  92         }
  93
  94         if(!m_permission)
  95         {
  96                 m_sock->Send("You need to be editing a permission first.\n");
  97                 m_sock->Send("(Use the 'edit' command.)\n");
  98                 return;
  99         }
 100
 101         if(act->needLock())
 102         {
 103                 try {
 104                         m_permission->Lock();
 105                 } catch(SavableLocked& e) {
 106                         m_sock->Send("The permission you are currently editing is locked (being edited by someone else), so you cannot edit it right now.\n");
 107                         m_sock->Send("Please try again later.\n");
 108                         return;
 109                 }
 110         }
 111
 112         act->Run(this, argument);
 113         return;
 114 }
 115
 116 SavablePtr EditorPermission::getEditing()
 117 {
 118         return m_permission;
 119 }
 120
 121 TableImplPtr EditorPermission::getTable()
 122 {
 123         return db::TableImpls::Get()->PERMISSIONS;
 124 }
 125
 126 KeysPtr EditorPermission::addNew()
 127 {
 128         throw std::runtime_error("EditorPermission::addNew(), Not yet implemented.");
 129 }
 130
 131 std::vector<std::string> EditorPermission::getList()
 132 {
 133         return mud::Managers::Get()->Permission->List();
 134 }
 135
 136 void EditorPermission::setEditing(KeysPtr keys)
 137 {
 138         if(!keys->size())
 139         {
 140                 m_permission.reset();
 141                 return;
 142         }
 143
 144         mud::PermissionPtr permission = mud::Managers::Get()->Permission->GetByKeys(keys);
 145         m_permission = permission;
 146         return;
 147 }
 148
 149 std::vector<std::string> EditorPermission::getCommands()
 150 {
 151         return m_commands.getCommandsVector();
 152 }
 153
 154 void EditorPermission::editAccount(const std::string& argument)
 155 {
 156         if(argument.size() == 0)
 157         {
 158                 m_sock->Send("Please specify an account!\n");
 159                 return;
 160         }
 161
 162         try
 163         {
 164                 long id = mud::Managers::Get()->Account->lookupByName(argument);
 165                 m_sock->Sendf("Preparing to edit permission with account '%d'.\n", id);
 166                 m_account = id;
 167         }
 168         catch(RowNotFoundException& e)
 169         {
 170                 m_sock->Sendf("'%s' is not a valid account!\n", argument.c_str());
 171                 m_sock->Send(String::Get()->box(mud::Managers::Get()->Account->List(), "Accounts"));
 172         }
 173 }
 174
 175 void EditorPermission::editGrantGroup(const std::string& argument)
 176 {
 177         if(argument.size() == 0)
 178         {
 179                 m_sock->Send("Please specify a grantgroup!\n");
 180                 return;
 181         }
 182
 183         try
 184         {
 185                 long id = mud::Managers::Get()->GrantGroup->lookupByName(argument);
 186                 m_sock->Sendf("Preparing to edit permission with grangroup '%d'.\n", id);
 187                 m_grantgroup = id;
 188         }
 189         catch(RowNotFoundException& e)
 190         {
 191                 m_sock->Sendf("'%s' is not a valid grantgroup!\n", argument.c_str());
 192                 m_sock->Send(String::Get()->box(mud::Managers::Get()->GrantGroup->List(), "GrantGroups"));
 193         }
 194 }
 195
 196 void EditorPermission::editGrant(const std::string& argument)
 197 {
 198         if(!argument.compare("Enable"))
 199         {
 200                 m_permission->setGrant(true);
 201                 return;
 202         }
 203
 204         if(!argument.compare("Disable"))
 205         {
 206                 m_permission->setGrant(false);
 207                 return;
 208         }
 209
 210         m_sock->Send("Please specify a grant level!\n");
 211         m_sock->Send("Choose between 'Enable' and 'Disable'.\n");
 212         return;
 213 }
 214
 215 void EditorPermission::editLogging(const std::string& argument)
 216 {
 217         if(!argument.compare("Enable"))
 218         {
 219                 m_permission->setLog(true);
 220                 return;
 221         }
 222
 223         if(!argument.compare("Disable"))
 224         {
 225                 m_permission->setLog(false);
 226                 return;
 227         }
 228
 229         m_sock->Send("Please specify a log level!\n");
 230         m_sock->Send("Choose bewteen 'Enable' and 'Disable'.\n");
 231         return;
 232 }
 233
 234 void EditorPermission::savePermission(const std::string& argument)
 235 {
 236         m_sock->Send("Saving...\n");
 237         m_permission->Save();
 238         m_sock->Send("Saved.\n");
 239         return;
 240 }
 241
 242 void EditorPermission::showPermission(const std::string& argument)
 243 {
 244         m_sock->Send(m_permission->toString());
 245 }