ryzom/server/src/ai_data_service/commands.cpp

   1 // Ryzom - MMORPG Framework <http://dev.ryzom.com/projects/ryzom/>
   2 // Copyright (C) 2010  Winch Gate Property Limited
   3 //
   4 // This program is free software: you can redistribute it and/or modify
   5 // it under the terms of the GNU Affero General Public License as
   6 // published by the Free Software Foundation, either version 3 of the
   7 // License, or (at your option) any later version.
   8 //
   9 // This program is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU Affero General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU Affero General Public License
  15 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17
  18
  19
  20 // Nel Misc
  21 #include "nel/net/service.h"
  22 #include "nel/misc/command.h"
  23
  24 // Game share
  25 #include "game_share/macro_manager.h"
  26
  27 // locals
  28 #include "ai_files.h"
  29 #include "ai_manager.h"
  30 #include "ai_service.h"
  31
  32 using namespace NLMISC;
  33 using namespace NLNET;
  34 using namespace std;
  35
  36
  37 //-------------------------------------------------------------------------
  38 // utility routines
  39
  40 static void argsToManagerVector(const vector<string> &args, vector<CAIManager *> &theManagers)
  41 {
  42         if(args.size()==0)
  43         {
  44                 for (uint i=0;i<CAIManager::numManagers();++i)
  45                         theManagers.push_back(CAIManager::getManagerByIdx(i));
  46         }
  47         else
  48         {
  49                 for (uint i=0;i<args.size();++i)
  50                 {
  51                         int id=CAIManager::nameToId(args[i]);
  52                         if (i==-1)
  53                                 nlwarning("Can't find a manger record for: %s",args[i].c_str());
  54                         else
  55                                 theManagers.push_back(CAIManager::getManagerById(id));
  56                 }
  57         }
  58 }
  59
  60 //-------------------------------------------------------------------------
  61 // scanning the hard disk for interesting files, compiling
  62 // where nessessary     and allerting ai_service(s) of data file changes
  63
  64 NLMISC_COMMAND(aiRescanDirectories,"Rescan disk file directories","")
  65 {
  66         if(args.size()!=0) return false;
  67         COMMAND_MACRO_RECORD_TEST
  68
  69         CAIFiles::scan();
  70
  71         return true;
  72 }
  73
  74 NLMISC_COMMAND(aiMake,"scan the hard disk for interesting files, recompile as necessary and allert services","")
  75 {
  76         COMMAND_MACRO_RECORD_TEST
  77
  78         // rescan the source and object directories for files
  79         CAIFiles::scan();
  80
  81         // build the list of ai managers to act on
  82         vector<CAIManager *> theManagers;
  83         argsToManagerVector(args,theManagers);
  84
  85         // do the work
  86         for (uint i=0;i<theManagers.size();++i)
  87         {
  88                 if (theManagers[i]->needCompile())
  89                 {
  90                         //nlinfo("Compiling manager: %04d: %s",theManagers[i]->id(),theManagers[i]->name());
  91                         theManagers[i]->compile();
  92                 }
  93                 else
  94                         nlinfo("Manager is up to date: %04d: %s",theManagers[i]->id(),theManagers[i]->name().c_str());
  95         }
  96
  97         return true;
  98 }
  99
 100 NLMISC_COMMAND(aiClean,"delete all generated files on the hard disk (next 'make' will rebuild everything)","")
 101 {
 102         COMMAND_MACRO_RECORD_TEST
 103
 104         // rescan the source and object directories for files
 105         CAIFiles::scan();
 106
 107         // build the list of ai managers to act on
 108         vector<CAIManager *> theManagers;
 109         argsToManagerVector(args,theManagers);
 110
 111         // do the work
 112         for (uint i=0;i<theManagers.size();++i)
 113         {
 114                 nlinfo("Cleaning: %04d: %s",theManagers[i]->id(),theManagers[i]->name().c_str());
 115                 theManagers[i]->clean();
 116         }
 117
 118         return true;
 119 }
 120
 121 //-------------------------------------------------------------------------
 122
 123 NLMISC_COMMAND(aiDisplayServices,"list the ai services, their parameters and their lists of running managers","[<service id>[...]]")
 124 {
 125         if(args.size()!=0) return false;
 126         COMMAND_MACRO_RECORD_TEST
 127
 128         return true;
 129 }
 130
 131 NLMISC_COMMAND(aiDisplayManagers,"Display the known ai manager list","[<manager name|id>[...]]")
 132 {
 133         COMMAND_MACRO_RECORD_TEST
 134
 135         // build the list of ai managers to act on
 136         vector<CAIManager *> theManagers;
 137         argsToManagerVector(args,theManagers);
 138
 139         // do the work
 140         for (uint i=0;i<theManagers.size();++i)
 141                 theManagers[i]->display();
 142
 143         return true;
 144 }
 145
 146 //-------------------------------------------------------------------------
 147 NLMISC_COMMAND(aiOpenAllManagers,"Open all managers and automatically assign to ai services","")
 148 {
 149         if(args.size()!=0) return false;
 150         COMMAND_MACRO_RECORD_TEST
 151
 152         for (uint i=0;i<CAIManager::numManagers();++i)
 153                 if (!CAIManager::getManagerByIdx(i)->isOpen())
 154                         CAIManager::getManagerByIdx(i)->open();
 155
 156         return true;
 157 }
 158
 159 NLMISC_COMMAND(aiOpenManagers,"Open managers and automatically assign to ai services","<manager name|id>[...]")
 160 {
 161         if(args.size()<1) return false;
 162         COMMAND_MACRO_RECORD_TEST
 163
 164         for (uint i=0;i<args.size();++i)
 165         {
 166                 int id=CAIManager::nameToId(args[i]);
 167                 if (i==-1)
 168                         nlwarning("Can't find a manger record for: %s",args[i].c_str());
 169                 else
 170                         CAIManager::getManagerById(id)->open();
 171         }
 172
 173         return true;
 174 }
 175
 176 NLMISC_COMMAND(aiAssignManagers,"Open managers and assign to specified ai service","<service><manager name|id>[...]")
 177 {
 178         if(args.size()<2) return false;
 179         COMMAND_MACRO_RECORD_TEST
 180
 181         // convert args[0] to service id and verify existance in services list
 182         NLNET::TServiceId service(atoi(args[0].c_str()));
 183         if (service.get()==0 && args[0]!="0")
 184         {
 185                 nlwarning("Invalid service id: %s",args[0].c_str());
 186                 return false;
 187         }
 188
 189         // iterate through args assigning managers to service
 190         for (uint i=1;i<args.size();++i)
 191         {
 192                 int id=CAIManager::nameToId(args[i]);
 193                 if (i==-1)
 194                         nlwarning("Can't find a manger record for: %s",args[i].c_str());
 195                 else
 196                         CAIManager::getManagerById(id)->assign(service);
 197         }
 198
 199         return true;
 200 }
 201
 202 NLMISC_COMMAND(aiReassignManagers,"Save & close managers on existing services and open on specified ai service","<service><manager name|id>[...]")
 203 {
 204         if(args.size()<2) return false;
 205         COMMAND_MACRO_RECORD_TEST
 206
 207         // convert args[0] to service id and verify existance in services list
 208         NLNET::TServiceId service(atoi(args[0].c_str()));
 209         if (service.get()==0 && args[0]!="0")
 210         {
 211                 nlwarning("Invalid service id: %s",args[0].c_str());
 212                 return false;
 213         }
 214
 215         // iterate through args reassigning managers to service
 216         for (uint i=1;i<args.size();++i)
 217         {
 218                 int id=CAIManager::nameToId(args[i]);
 219                 if (i==-1)
 220                         nlwarning("Can't find a manger record for: %s",args[i].c_str());
 221                 else
 222                         CAIManager::getManagerById(id)->reassign(service);
 223         }
 224
 225         return true;
 226 }
 227
 228 //-------------------------------------------------------------------------
 229 NLMISC_COMMAND(aiCloseAllManagers,"Close all managers across all ai services on the shard","")
 230 {
 231         if(args.size()!=0) return false;
 232         COMMAND_MACRO_RECORD_TEST
 233
 234         for (uint i=0;i<CAIManager::numManagers();++i)
 235                 if (CAIManager::getManagerByIdx(i)->isOpen())
 236                         CAIManager::getManagerByIdx(i)->close();
 237
 238         return true;
 239 }
 240
 241 NLMISC_COMMAND(aiCloseManagers,"Close listed managers","<manager name|id>[...]")
 242 {
 243         if(args.size()<1) return false;
 244         COMMAND_MACRO_RECORD_TEST
 245
 246         for (uint i=0;i<args.size();++i)
 247         {
 248                 int id=CAIManager::nameToId(args[i]);
 249                 if (i==-1)
 250                         nlwarning("Can't find a manger record for: %s",args[i].c_str());
 251                 else
 252                         CAIManager::getManagerById(id)->close();
 253         }
 254
 255         return true;
 256 }
 257
 258 //-------------------------------------------------------------------------