2
0

CModHandler.cpp 35 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083
  1. /*
  2. * CModHandler.cpp, part of VCMI engine
  3. *
  4. * Authors: listed in file AUTHORS in main folder
  5. *
  6. * License: GNU General Public License v2.0 or later
  7. * Full text of license available in license.txt file, in main folder
  8. *
  9. */
  10. #include "StdInc.h"
  11. #include "CModHandler.h"
  12. #include "mapObjects/CObjectClassesHandler.h"
  13. #include "filesystem/FileStream.h"
  14. #include "filesystem/AdapterLoaders.h"
  15. #include "filesystem/CFilesystemLoader.h"
  16. #include "CCreatureHandler.h"
  17. #include "CArtHandler.h"
  18. #include "CTownHandler.h"
  19. #include "CHeroHandler.h"
  20. #include "mapObjects/CObjectHandler.h"
  21. #include "StringConstants.h"
  22. #include "CStopWatch.h"
  23. #include "IHandlerBase.h"
  24. #include "spells/CSpellHandler.h"
  25. #include "CSkillHandler.h"
  26. #include "ScriptHandler.h"
  27. #include "BattleFieldHandler.h"
  28. #include "ObstacleHandler.h"
  29. #include <vstd/StringUtils.h>
  30. CIdentifierStorage::CIdentifierStorage():
  31. state(LOADING)
  32. {
  33. }
  34. CIdentifierStorage::~CIdentifierStorage()
  35. {
  36. }
  37. void CIdentifierStorage::checkIdentifier(std::string & ID)
  38. {
  39. if (boost::algorithm::ends_with(ID, "."))
  40. logMod->warn("BIG WARNING: identifier %s seems to be broken!", ID);
  41. else
  42. {
  43. size_t pos = 0;
  44. do
  45. {
  46. if (std::tolower(ID[pos]) != ID[pos] ) //Not in camelCase
  47. {
  48. logMod->warn("Warning: identifier %s is not in camelCase!", ID);
  49. ID[pos] = std::tolower(ID[pos]);// Try to fix the ID
  50. }
  51. pos = ID.find('.', pos);
  52. }
  53. while(pos++ != std::string::npos);
  54. }
  55. }
  56. CIdentifierStorage::ObjectCallback::ObjectCallback(
  57. std::string localScope, std::string remoteScope, std::string type,
  58. std::string name, const std::function<void(si32)> & callback,
  59. bool optional):
  60. localScope(localScope),
  61. remoteScope(remoteScope),
  62. type(type),
  63. name(name),
  64. callback(callback),
  65. optional(optional)
  66. {}
  67. void CIdentifierStorage::requestIdentifier(ObjectCallback callback)
  68. {
  69. checkIdentifier(callback.type);
  70. checkIdentifier(callback.name);
  71. assert(!callback.localScope.empty());
  72. if (state != FINISHED) // enqueue request if loading is still in progress
  73. scheduledRequests.push_back(callback);
  74. else // execute immediately for "late" requests
  75. resolveIdentifier(callback);
  76. }
  77. void CIdentifierStorage::requestIdentifier(std::string scope, std::string type, std::string name, const std::function<void(si32)> & callback)
  78. {
  79. auto pair = vstd::splitStringToPair(name, ':'); // remoteScope:name
  80. requestIdentifier(ObjectCallback(scope, pair.first, type, pair.second, callback, false));
  81. }
  82. void CIdentifierStorage::requestIdentifier(std::string scope, std::string fullName, const std::function<void(si32)>& callback)
  83. {
  84. auto scopeAndFullName = vstd::splitStringToPair(fullName, ':');
  85. auto typeAndName = vstd::splitStringToPair(scopeAndFullName.second, '.');
  86. requestIdentifier(ObjectCallback(scope, scopeAndFullName.first, typeAndName.first, typeAndName.second, callback, false));
  87. }
  88. void CIdentifierStorage::requestIdentifier(std::string type, const JsonNode & name, const std::function<void(si32)> & callback)
  89. {
  90. auto pair = vstd::splitStringToPair(name.String(), ':'); // remoteScope:name
  91. requestIdentifier(ObjectCallback(name.meta, pair.first, type, pair.second, callback, false));
  92. }
  93. void CIdentifierStorage::requestIdentifier(const JsonNode & name, const std::function<void(si32)> & callback)
  94. {
  95. auto pair = vstd::splitStringToPair(name.String(), ':'); // remoteScope:<type.name>
  96. auto pair2 = vstd::splitStringToPair(pair.second, '.'); // type.name
  97. requestIdentifier(ObjectCallback(name.meta, pair.first, pair2.first, pair2.second, callback, false));
  98. }
  99. void CIdentifierStorage::tryRequestIdentifier(std::string scope, std::string type, std::string name, const std::function<void(si32)> & callback)
  100. {
  101. auto pair = vstd::splitStringToPair(name, ':'); // remoteScope:name
  102. requestIdentifier(ObjectCallback(scope, pair.first, type, pair.second, callback, true));
  103. }
  104. void CIdentifierStorage::tryRequestIdentifier(std::string type, const JsonNode & name, const std::function<void(si32)> & callback)
  105. {
  106. auto pair = vstd::splitStringToPair(name.String(), ':'); // remoteScope:name
  107. requestIdentifier(ObjectCallback(name.meta, pair.first, type, pair.second, callback, true));
  108. }
  109. boost::optional<si32> CIdentifierStorage::getIdentifier(std::string scope, std::string type, std::string name, bool silent)
  110. {
  111. auto pair = vstd::splitStringToPair(name, ':'); // remoteScope:name
  112. auto idList = getPossibleIdentifiers(ObjectCallback(scope, pair.first, type, pair.second, std::function<void(si32)>(), silent));
  113. if (idList.size() == 1)
  114. return idList.front().id;
  115. if (!silent)
  116. logMod->error("Failed to resolve identifier %s of type %s from mod %s", name , type ,scope);
  117. return boost::optional<si32>();
  118. }
  119. boost::optional<si32> CIdentifierStorage::getIdentifier(std::string type, const JsonNode & name, bool silent)
  120. {
  121. auto pair = vstd::splitStringToPair(name.String(), ':'); // remoteScope:name
  122. auto idList = getPossibleIdentifiers(ObjectCallback(name.meta, pair.first, type, pair.second, std::function<void(si32)>(), silent));
  123. if (idList.size() == 1)
  124. return idList.front().id;
  125. if (!silent)
  126. logMod->error("Failed to resolve identifier %s of type %s from mod %s", name.String(), type, name.meta);
  127. return boost::optional<si32>();
  128. }
  129. boost::optional<si32> CIdentifierStorage::getIdentifier(const JsonNode & name, bool silent)
  130. {
  131. auto pair = vstd::splitStringToPair(name.String(), ':'); // remoteScope:<type.name>
  132. auto pair2 = vstd::splitStringToPair(pair.second, '.'); // type.name
  133. auto idList = getPossibleIdentifiers(ObjectCallback(name.meta, pair.first, pair2.first, pair2.second, std::function<void(si32)>(), silent));
  134. if (idList.size() == 1)
  135. return idList.front().id;
  136. if (!silent)
  137. logMod->error("Failed to resolve identifier %s of type %s from mod %s", name.String(), pair2.first, name.meta);
  138. return boost::optional<si32>();
  139. }
  140. boost::optional<si32> CIdentifierStorage::getIdentifier(std::string scope, std::string fullName, bool silent)
  141. {
  142. auto pair = vstd::splitStringToPair(fullName, ':'); // remoteScope:<type.name>
  143. auto pair2 = vstd::splitStringToPair(pair.second, '.'); // type.name
  144. auto idList = getPossibleIdentifiers(ObjectCallback(scope, pair.first, pair2.first, pair2.second, std::function<void(si32)>(), silent));
  145. if (idList.size() == 1)
  146. return idList.front().id;
  147. if (!silent)
  148. logMod->error("Failed to resolve identifier %s of type %s from mod %s", fullName, pair2.first, scope);
  149. return boost::optional<si32>();
  150. }
  151. void CIdentifierStorage::registerObject(std::string scope, std::string type, std::string name, si32 identifier)
  152. {
  153. ObjectData data;
  154. data.scope = scope;
  155. data.id = identifier;
  156. std::string fullID = type + '.' + name;
  157. checkIdentifier(fullID);
  158. std::pair<const std::string, ObjectData> mapping = std::make_pair(fullID, data);
  159. if(!vstd::containsMapping(registeredObjects, mapping))
  160. {
  161. logMod->trace("registered %s as %s:%s", fullID, scope, identifier);
  162. registeredObjects.insert(mapping);
  163. }
  164. }
  165. std::vector<CIdentifierStorage::ObjectData> CIdentifierStorage::getPossibleIdentifiers(const ObjectCallback & request)
  166. {
  167. std::set<std::string> allowedScopes;
  168. bool isValidScope = true;
  169. if (request.remoteScope.empty())
  170. {
  171. // normally ID's from all required mods, own mod and virtual "core" mod are allowed
  172. if(request.localScope != "core" && !request.localScope.empty())
  173. {
  174. allowedScopes = VLC->modh->getModDependencies(request.localScope, isValidScope);
  175. if(!isValidScope)
  176. return std::vector<ObjectData>();
  177. }
  178. allowedScopes.insert(request.localScope);
  179. allowedScopes.insert("core");
  180. }
  181. else
  182. {
  183. //...unless destination mod was specified explicitly
  184. //note: getModDependencies does not work for "core" by design
  185. //for map format support core mod has access to any mod
  186. //TODO: better solution for access from map?
  187. if(request.localScope == "core" || request.localScope == "")
  188. {
  189. allowedScopes.insert(request.remoteScope);
  190. }
  191. else
  192. {
  193. // allow only available to all core mod or dependencies
  194. auto myDeps = VLC->modh->getModDependencies(request.localScope, isValidScope);
  195. if(!isValidScope)
  196. return std::vector<ObjectData>();
  197. if(request.remoteScope == "core" || request.remoteScope == request.localScope || myDeps.count(request.remoteScope))
  198. allowedScopes.insert(request.remoteScope);
  199. }
  200. }
  201. std::string fullID = request.type + '.' + request.name;
  202. auto entries = registeredObjects.equal_range(fullID);
  203. if (entries.first != entries.second)
  204. {
  205. std::vector<ObjectData> locatedIDs;
  206. for (auto it = entries.first; it != entries.second; it++)
  207. {
  208. if (vstd::contains(allowedScopes, it->second.scope))
  209. {
  210. locatedIDs.push_back(it->second);
  211. }
  212. }
  213. return locatedIDs;
  214. }
  215. return std::vector<ObjectData>();
  216. }
  217. bool CIdentifierStorage::resolveIdentifier(const ObjectCallback & request)
  218. {
  219. auto identifiers = getPossibleIdentifiers(request);
  220. if (identifiers.size() == 1) // normally resolved ID
  221. {
  222. request.callback(identifiers.front().id);
  223. return true;
  224. }
  225. if (request.optional && identifiers.empty()) // failed to resolve optinal ID
  226. {
  227. return true;
  228. }
  229. // error found. Try to generate some debug info
  230. if (identifiers.size() == 0)
  231. logMod->error("Unknown identifier!");
  232. else
  233. logMod->error("Ambiguous identifier request!");
  234. logMod->error("Request for %s.%s from mod %s", request.type, request.name, request.localScope);
  235. for (auto id : identifiers)
  236. {
  237. logMod->error("\tID is available in mod %s", id.scope);
  238. }
  239. return false;
  240. }
  241. void CIdentifierStorage::finalize()
  242. {
  243. state = FINALIZING;
  244. bool errorsFound = false;
  245. //Note: we may receive new requests during resolution phase -> end may change -> range for can't be used
  246. for(auto it = scheduledRequests.begin(); it != scheduledRequests.end(); it++)
  247. {
  248. errorsFound |= !resolveIdentifier(*it);
  249. }
  250. if (errorsFound)
  251. {
  252. for(auto object : registeredObjects)
  253. {
  254. logMod->trace("%s : %s -> %d", object.second.scope, object.first, object.second.id);
  255. }
  256. logMod->error("All known identifiers were dumped into log file");
  257. }
  258. assert(errorsFound == false);
  259. state = FINISHED;
  260. }
  261. ContentTypeHandler::ContentTypeHandler(IHandlerBase * handler, std::string objectName):
  262. handler(handler),
  263. objectName(objectName),
  264. originalData(handler->loadLegacyData((size_t)VLC->modh->settings.data["textData"][objectName].Float()))
  265. {
  266. for(auto & node : originalData)
  267. {
  268. node.setMeta("core");
  269. }
  270. }
  271. bool ContentTypeHandler::preloadModData(std::string modName, std::vector<std::string> fileList, bool validate)
  272. {
  273. bool result;
  274. JsonNode data = JsonUtils::assembleFromFiles(fileList, result);
  275. data.setMeta(modName);
  276. ModInfo & modInfo = modData[modName];
  277. for(auto entry : data.Struct())
  278. {
  279. size_t colon = entry.first.find(':');
  280. if (colon == std::string::npos)
  281. {
  282. // normal object, local to this mod
  283. modInfo.modData[entry.first].swap(entry.second);
  284. }
  285. else
  286. {
  287. std::string remoteName = entry.first.substr(0, colon);
  288. std::string objectName = entry.first.substr(colon + 1);
  289. // patching this mod? Send warning and continue - this situation can be handled normally
  290. if (remoteName == modName)
  291. logMod->warn("Redundant namespace definition for %s", objectName);
  292. logMod->trace("Patching object %s (%s) from %s", objectName, remoteName, modName);
  293. JsonNode & remoteConf = modData[remoteName].patches[objectName];
  294. JsonUtils::merge(remoteConf, entry.second);
  295. }
  296. }
  297. return result;
  298. }
  299. bool ContentTypeHandler::loadMod(std::string modName, bool validate)
  300. {
  301. ModInfo & modInfo = modData[modName];
  302. bool result = true;
  303. auto performValidate = [&,this](JsonNode & data, const std::string & name){
  304. handler->beforeValidate(data);
  305. if (validate)
  306. result &= JsonUtils::validate(data, "vcmi:" + objectName, name);
  307. };
  308. // apply patches
  309. if (!modInfo.patches.isNull())
  310. JsonUtils::merge(modInfo.modData, modInfo.patches);
  311. for(auto & entry : modInfo.modData.Struct())
  312. {
  313. const std::string & name = entry.first;
  314. JsonNode & data = entry.second;
  315. if (vstd::contains(data.Struct(), "index") && !data["index"].isNull())
  316. {
  317. // try to add H3 object data
  318. size_t index = static_cast<size_t>(data["index"].Float());
  319. if(originalData.size() > index)
  320. {
  321. logMod->trace("found original data in loadMod(%s) at index %d", name, index);
  322. JsonUtils::merge(originalData[index], data);
  323. std::swap(originalData[index], data);
  324. originalData[index].clear(); // do not use same data twice (same ID)
  325. }
  326. else
  327. {
  328. logMod->warn("no original data in loadMod(%s) at index %d", name, index);
  329. }
  330. performValidate(data, name);
  331. handler->loadObject(modName, name, data, index);
  332. }
  333. else
  334. {
  335. // normal new object
  336. logMod->trace("no index in loadMod(%s)", name);
  337. performValidate(data,name);
  338. handler->loadObject(modName, name, data);
  339. }
  340. }
  341. return result;
  342. }
  343. void ContentTypeHandler::loadCustom()
  344. {
  345. handler->loadCustom();
  346. }
  347. void ContentTypeHandler::afterLoadFinalization()
  348. {
  349. handler->afterLoadFinalization();
  350. }
  351. CContentHandler::CContentHandler()
  352. {
  353. }
  354. void CContentHandler::init()
  355. {
  356. handlers.insert(std::make_pair("heroClasses", ContentTypeHandler(&VLC->heroh->classes, "heroClass")));
  357. handlers.insert(std::make_pair("artifacts", ContentTypeHandler(VLC->arth, "artifact")));
  358. handlers.insert(std::make_pair("creatures", ContentTypeHandler(VLC->creh, "creature")));
  359. handlers.insert(std::make_pair("factions", ContentTypeHandler(VLC->townh, "faction")));
  360. handlers.insert(std::make_pair("objects", ContentTypeHandler(VLC->objtypeh, "object")));
  361. handlers.insert(std::make_pair("heroes", ContentTypeHandler(VLC->heroh, "hero")));
  362. handlers.insert(std::make_pair("spells", ContentTypeHandler(VLC->spellh, "spell")));
  363. handlers.insert(std::make_pair("skills", ContentTypeHandler(VLC->skillh, "skill")));
  364. handlers.insert(std::make_pair("templates", ContentTypeHandler((IHandlerBase *)VLC->tplh, "template")));
  365. handlers.insert(std::make_pair("scripts", ContentTypeHandler(VLC->scriptHandler, "script")));
  366. handlers.insert(std::make_pair("battlefields", ContentTypeHandler(VLC->battlefieldsHandler, "battlefield")));
  367. handlers.insert(std::make_pair("obstacles", ContentTypeHandler(VLC->obstacleHandler, "obstacle")));
  368. //TODO: any other types of moddables?
  369. }
  370. bool CContentHandler::preloadModData(std::string modName, JsonNode modConfig, bool validate)
  371. {
  372. bool result = true;
  373. for(auto & handler : handlers)
  374. {
  375. result &= handler.second.preloadModData(modName, modConfig[handler.first].convertTo<std::vector<std::string> >(), validate);
  376. }
  377. return result;
  378. }
  379. bool CContentHandler::loadMod(std::string modName, bool validate)
  380. {
  381. bool result = true;
  382. for(auto & handler : handlers)
  383. {
  384. result &= handler.second.loadMod(modName, validate);
  385. }
  386. return result;
  387. }
  388. void CContentHandler::loadCustom()
  389. {
  390. for(auto & handler : handlers)
  391. {
  392. handler.second.loadCustom();
  393. }
  394. }
  395. void CContentHandler::afterLoadFinalization()
  396. {
  397. for(auto & handler : handlers)
  398. {
  399. handler.second.afterLoadFinalization();
  400. }
  401. }
  402. void CContentHandler::preloadData(CModInfo & mod)
  403. {
  404. bool validate = (mod.validation != CModInfo::PASSED);
  405. // print message in format [<8-symbols checksum>] <modname>
  406. logMod->info("\t\t[%08x]%s", mod.checksum, mod.name);
  407. if (validate && mod.identifier != "core")
  408. {
  409. if (!JsonUtils::validate(mod.config, "vcmi:mod", mod.identifier))
  410. mod.validation = CModInfo::FAILED;
  411. }
  412. if (!preloadModData(mod.identifier, mod.config, validate))
  413. mod.validation = CModInfo::FAILED;
  414. }
  415. void CContentHandler::load(CModInfo & mod)
  416. {
  417. bool validate = (mod.validation != CModInfo::PASSED);
  418. if (!loadMod(mod.identifier, validate))
  419. mod.validation = CModInfo::FAILED;
  420. if (validate)
  421. {
  422. if (mod.validation != CModInfo::FAILED)
  423. logMod->info("\t\t[DONE] %s", mod.name);
  424. else
  425. logMod->error("\t\t[FAIL] %s", mod.name);
  426. }
  427. else
  428. logMod->info("\t\t[SKIP] %s", mod.name);
  429. }
  430. const ContentTypeHandler & CContentHandler::operator[](const std::string & name) const
  431. {
  432. return handlers.at(name);
  433. }
  434. static JsonNode loadModSettings(std::string path)
  435. {
  436. if (CResourceHandler::get("local")->existsResource(ResourceID(path)))
  437. {
  438. return JsonNode(ResourceID(path, EResType::TEXT));
  439. }
  440. // Probably new install. Create initial configuration
  441. CResourceHandler::get("local")->createResource(path);
  442. return JsonNode();
  443. }
  444. JsonNode addMeta(JsonNode config, std::string meta)
  445. {
  446. config.setMeta(meta);
  447. return config;
  448. }
  449. CModInfo::CModInfo():
  450. checksum(0),
  451. enabled(false),
  452. validation(PENDING)
  453. {
  454. }
  455. CModInfo::CModInfo(std::string identifier,const JsonNode & local, const JsonNode & config):
  456. identifier(identifier),
  457. name(config["name"].String()),
  458. description(config["description"].String()),
  459. dependencies(config["depends"].convertTo<std::set<std::string> >()),
  460. conflicts(config["conflicts"].convertTo<std::set<std::string> >()),
  461. checksum(0),
  462. enabled(false),
  463. validation(PENDING),
  464. config(addMeta(config, identifier))
  465. {
  466. loadLocalData(local);
  467. }
  468. JsonNode CModInfo::saveLocalData() const
  469. {
  470. std::ostringstream stream;
  471. stream << std::noshowbase << std::hex << std::setw(8) << std::setfill('0') << checksum;
  472. JsonNode conf;
  473. conf["active"].Bool() = enabled;
  474. conf["validated"].Bool() = validation != FAILED;
  475. conf["checksum"].String() = stream.str();
  476. return conf;
  477. }
  478. std::string CModInfo::getModDir(std::string name)
  479. {
  480. return "MODS/" + boost::algorithm::replace_all_copy(name, ".", "/MODS/");
  481. }
  482. std::string CModInfo::getModFile(std::string name)
  483. {
  484. return getModDir(name) + "/mod.json";
  485. }
  486. void CModInfo::updateChecksum(ui32 newChecksum)
  487. {
  488. // comment-out next line to force validation of all mods ignoring checksum
  489. if (newChecksum != checksum)
  490. {
  491. checksum = newChecksum;
  492. validation = PENDING;
  493. }
  494. }
  495. void CModInfo::loadLocalData(const JsonNode & data)
  496. {
  497. bool validated = false;
  498. enabled = true;
  499. checksum = 0;
  500. if (data.getType() == JsonNode::JsonType::DATA_BOOL)
  501. {
  502. enabled = data.Bool();
  503. }
  504. if (data.getType() == JsonNode::JsonType::DATA_STRUCT)
  505. {
  506. enabled = data["active"].Bool();
  507. validated = data["validated"].Bool();
  508. checksum = strtol(data["checksum"].String().c_str(), nullptr, 16);
  509. }
  510. if (enabled)
  511. validation = validated ? PASSED : PENDING;
  512. else
  513. validation = validated ? PASSED : FAILED;
  514. }
  515. CModHandler::CModHandler() : content(std::make_shared<CContentHandler>())
  516. {
  517. modules.COMMANDERS = false;
  518. modules.STACK_ARTIFACT = false;
  519. modules.STACK_EXP = false;
  520. modules.MITHRIL = false;
  521. for (int i = 0; i < GameConstants::RESOURCE_QUANTITY; ++i)
  522. {
  523. identifiers.registerObject("core", "resource", GameConstants::RESOURCE_NAMES[i], i);
  524. }
  525. for(int i=0; i<GameConstants::PRIMARY_SKILLS; ++i)
  526. {
  527. identifiers.registerObject("core", "primSkill", PrimarySkill::names[i], i);
  528. identifiers.registerObject("core", "primarySkill", PrimarySkill::names[i], i);
  529. }
  530. }
  531. CModHandler::~CModHandler()
  532. {
  533. }
  534. void CModHandler::loadConfigFromFile (std::string name)
  535. {
  536. std::string paths;
  537. for(auto& p : CResourceHandler::get()->getResourceNames(ResourceID("config/" + name)))
  538. {
  539. paths += p.string() + ", ";
  540. }
  541. paths = paths.substr(0, paths.size() - 2);
  542. logMod->debug("Loading hardcoded features settings from [%s], result:", paths);
  543. settings.data = JsonUtils::assembleFromFiles("config/" + name);
  544. const JsonNode & hardcodedFeatures = settings.data["hardcodedFeatures"];
  545. settings.MAX_HEROES_AVAILABLE_PER_PLAYER = static_cast<int>(hardcodedFeatures["MAX_HEROES_AVAILABLE_PER_PLAYER"].Integer());
  546. logMod->debug("\tMAX_HEROES_AVAILABLE_PER_PLAYER\t%d", settings.MAX_HEROES_AVAILABLE_PER_PLAYER);
  547. settings.MAX_HEROES_ON_MAP_PER_PLAYER = static_cast<int>(hardcodedFeatures["MAX_HEROES_ON_MAP_PER_PLAYER"].Integer());
  548. logMod->debug("\tMAX_HEROES_ON_MAP_PER_PLAYER\t%d", settings.MAX_HEROES_ON_MAP_PER_PLAYER);
  549. settings.CREEP_SIZE = static_cast<int>(hardcodedFeatures["CREEP_SIZE"].Integer());
  550. logMod->debug("\tCREEP_SIZE\t%d", settings.CREEP_SIZE);
  551. settings.WEEKLY_GROWTH = static_cast<int>(hardcodedFeatures["WEEKLY_GROWTH_PERCENT"].Integer());
  552. logMod->debug("\tWEEKLY_GROWTH\t%d", settings.WEEKLY_GROWTH);
  553. settings.NEUTRAL_STACK_EXP = static_cast<int>(hardcodedFeatures["NEUTRAL_STACK_EXP_DAILY"].Integer());
  554. logMod->debug("\tNEUTRAL_STACK_EXP\t%d", settings.NEUTRAL_STACK_EXP);
  555. settings.MAX_BUILDING_PER_TURN = static_cast<int>(hardcodedFeatures["MAX_BUILDING_PER_TURN"].Integer());
  556. logMod->debug("\tMAX_BUILDING_PER_TURN\t%d", settings.MAX_BUILDING_PER_TURN);
  557. settings.DWELLINGS_ACCUMULATE_CREATURES = hardcodedFeatures["DWELLINGS_ACCUMULATE_CREATURES"].Bool();
  558. logMod->debug("\tDWELLINGS_ACCUMULATE_CREATURES\t%d", static_cast<int>(settings.DWELLINGS_ACCUMULATE_CREATURES));
  559. settings.ALL_CREATURES_GET_DOUBLE_MONTHS = hardcodedFeatures["ALL_CREATURES_GET_DOUBLE_MONTHS"].Bool();
  560. logMod->debug("\tALL_CREATURES_GET_DOUBLE_MONTHS\t%d", static_cast<int>(settings.ALL_CREATURES_GET_DOUBLE_MONTHS));
  561. settings.WINNING_HERO_WITH_NO_TROOPS_RETREATS = hardcodedFeatures["WINNING_HERO_WITH_NO_TROOPS_RETREATS"].Bool();
  562. logMod->debug("\tWINNING_HERO_WITH_NO_TROOPS_RETREATS\t%d", static_cast<int>(settings.WINNING_HERO_WITH_NO_TROOPS_RETREATS));
  563. settings.BLACK_MARKET_MONTHLY_ARTIFACTS_CHANGE = hardcodedFeatures["BLACK_MARKET_MONTHLY_ARTIFACTS_CHANGE"].Bool();
  564. logMod->debug("\tBLACK_MARKET_MONTHLY_ARTIFACTS_CHANGE\t%d", static_cast<int>(settings.BLACK_MARKET_MONTHLY_ARTIFACTS_CHANGE));
  565. settings.NO_RANDOM_SPECIAL_WEEKS_AND_MONTHS = hardcodedFeatures["NO_RANDOM_SPECIAL_WEEKS_AND_MONTHS"].Bool();
  566. logMod->debug("\tNO_RANDOM_SPECIAL_WEEKS_AND_MONTHS\t%d", static_cast<int>(settings.NO_RANDOM_SPECIAL_WEEKS_AND_MONTHS));
  567. const JsonNode & gameModules = settings.data["modules"];
  568. modules.STACK_EXP = gameModules["STACK_EXPERIENCE"].Bool();
  569. logMod->debug("\tSTACK_EXP\t%d", static_cast<int>(modules.STACK_EXP));
  570. modules.STACK_ARTIFACT = gameModules["STACK_ARTIFACTS"].Bool();
  571. logMod->debug("\tSTACK_ARTIFACT\t%d", static_cast<int>(modules.STACK_ARTIFACT));
  572. modules.COMMANDERS = gameModules["COMMANDERS"].Bool();
  573. logMod->debug("\tCOMMANDERS\t%d", static_cast<int>(modules.COMMANDERS));
  574. modules.MITHRIL = gameModules["MITHRIL"].Bool();
  575. logMod->debug("\tMITHRIL\t%d", static_cast<int>(modules.MITHRIL));
  576. }
  577. // currentList is passed by value to get current list of depending mods
  578. bool CModHandler::hasCircularDependency(TModID modID, std::set <TModID> currentList) const
  579. {
  580. const CModInfo & mod = allMods.at(modID);
  581. // Mod already present? We found a loop
  582. if (vstd::contains(currentList, modID))
  583. {
  584. logMod->error("Error: Circular dependency detected! Printing dependency list:");
  585. logMod->error("\t%s -> ", mod.name);
  586. return true;
  587. }
  588. currentList.insert(modID);
  589. // recursively check every dependency of this mod
  590. for(const TModID & dependency : mod.dependencies)
  591. {
  592. if (hasCircularDependency(dependency, currentList))
  593. {
  594. logMod->error("\t%s ->\n", mod.name); // conflict detected, print dependency list
  595. return true;
  596. }
  597. }
  598. return false;
  599. }
  600. bool CModHandler::checkDependencies(const std::vector <TModID> & input) const
  601. {
  602. for(const TModID & id : input)
  603. {
  604. const CModInfo & mod = allMods.at(id);
  605. for(const TModID & dep : mod.dependencies)
  606. {
  607. if(!vstd::contains(input, dep))
  608. {
  609. logMod->error("Error: Mod %s requires missing %s!", mod.name, dep);
  610. return false;
  611. }
  612. }
  613. for(const TModID & conflicting : mod.conflicts)
  614. {
  615. if(vstd::contains(input, conflicting))
  616. {
  617. logMod->error("Error: Mod %s conflicts with %s!", mod.name, allMods.at(conflicting).name);
  618. return false;
  619. }
  620. }
  621. if(hasCircularDependency(id))
  622. return false;
  623. }
  624. return true;
  625. }
  626. // Returned vector affects the resource loaders call order (see CFilesystemList::load).
  627. // The loaders call order matters when dependent mod overrides resources in its dependencies.
  628. std::vector <TModID> CModHandler::validateAndSortDependencies(std::vector <TModID> modsToResolve) const
  629. {
  630. // Topological sort algorithm.
  631. // TODO: Investigate possible ways to improve performance.
  632. boost::range::sort(modsToResolve); // Sort mods per name
  633. std::vector <TModID> sortedValidMods; // Vector keeps order of elements (LIFO)
  634. sortedValidMods.reserve(modsToResolve.size()); // push_back calls won't cause memory reallocation
  635. std::set <TModID> resolvedModIDs; // Use a set for validation for performance reason, but set does not keep order of elements
  636. // Mod is resolved if it has not dependencies or all its dependencies are already resolved
  637. auto isResolved = [&](const CModInfo & mod) -> CModInfo::EValidationStatus
  638. {
  639. if(mod.dependencies.size() > resolvedModIDs.size())
  640. return CModInfo::PENDING;
  641. for(const TModID & dependency : mod.dependencies)
  642. {
  643. if(!vstd::contains(resolvedModIDs, dependency))
  644. return CModInfo::PENDING;
  645. }
  646. return CModInfo::PASSED;
  647. };
  648. while(true)
  649. {
  650. std::set <TModID> resolvedOnCurrentTreeLevel;
  651. for(auto it = modsToResolve.begin(); it != modsToResolve.end();) // One iteration - one level of mods tree
  652. {
  653. if(isResolved(allMods.at(*it)) == CModInfo::PASSED)
  654. {
  655. resolvedOnCurrentTreeLevel.insert(*it); // Not to the resolvedModIDs, so current node childs will be resolved on the next iteration
  656. sortedValidMods.push_back(*it);
  657. it = modsToResolve.erase(it);
  658. continue;
  659. }
  660. it++;
  661. }
  662. if(resolvedOnCurrentTreeLevel.size())
  663. {
  664. resolvedModIDs.insert(resolvedOnCurrentTreeLevel.begin(), resolvedOnCurrentTreeLevel.end());
  665. continue;
  666. }
  667. // If there're no valid mods on the current mods tree level, no more mod can be resolved, should be end.
  668. break;
  669. }
  670. // Left mods have unresolved dependencies, output all to log.
  671. for(const auto & brokenModID : modsToResolve)
  672. {
  673. const CModInfo & brokenMod = allMods.at(brokenModID);
  674. for(const TModID & dependency : brokenMod.dependencies)
  675. {
  676. if(!vstd::contains(resolvedModIDs, dependency))
  677. logMod->error("Mod '%s' will not work: it depends on mod '%s', which is not installed.", brokenMod.name, dependency);
  678. }
  679. }
  680. return sortedValidMods;
  681. }
  682. std::vector<std::string> CModHandler::getModList(std::string path)
  683. {
  684. std::string modDir = boost::to_upper_copy(path + "MODS/");
  685. size_t depth = boost::range::count(modDir, '/');
  686. auto list = CResourceHandler::get("initial")->getFilteredFiles([&](const ResourceID & id) -> bool
  687. {
  688. if (id.getType() != EResType::DIRECTORY)
  689. return false;
  690. if (!boost::algorithm::starts_with(id.getName(), modDir))
  691. return false;
  692. if (boost::range::count(id.getName(), '/') != depth )
  693. return false;
  694. return true;
  695. });
  696. //storage for found mods
  697. std::vector<std::string> foundMods;
  698. for (auto & entry : list)
  699. {
  700. std::string name = entry.getName();
  701. name.erase(0, modDir.size()); //Remove path prefix
  702. if (!name.empty())
  703. foundMods.push_back(name);
  704. }
  705. return foundMods;
  706. }
  707. void CModHandler::loadMods(std::string path, std::string parent, const JsonNode & modSettings, bool enableMods)
  708. {
  709. for(std::string modName : getModList(path))
  710. loadOneMod(modName, parent, modSettings, enableMods);
  711. }
  712. void CModHandler::loadOneMod(std::string modName, std::string parent, const JsonNode & modSettings, bool enableMods)
  713. {
  714. boost::to_lower(modName);
  715. std::string modFullName = parent.empty() ? modName : parent + '.' + modName;
  716. if(CResourceHandler::get("initial")->existsResource(ResourceID(CModInfo::getModFile(modFullName))))
  717. {
  718. CModInfo mod(modFullName, modSettings[modName], JsonNode(ResourceID(CModInfo::getModFile(modFullName))));
  719. if (!parent.empty()) // this is submod, add parent to dependencies
  720. mod.dependencies.insert(parent);
  721. allMods[modFullName] = mod;
  722. if (mod.enabled && enableMods)
  723. activeMods.push_back(modFullName);
  724. loadMods(CModInfo::getModDir(modFullName) + '/', modFullName, modSettings[modName]["mods"], enableMods && mod.enabled);
  725. }
  726. }
  727. void CModHandler::loadMods(bool onlyEssential)
  728. {
  729. JsonNode modConfig;
  730. if(onlyEssential)
  731. {
  732. loadOneMod("vcmi", "", modConfig, true);//only vcmi and submods
  733. }
  734. else
  735. {
  736. modConfig = loadModSettings("config/modSettings.json");
  737. loadMods("", "", modConfig["activeMods"], true);
  738. }
  739. coreMod = CModInfo("core", modConfig["core"], JsonNode(ResourceID("config/gameConfig.json")));
  740. coreMod.name = "Original game files";
  741. }
  742. std::vector<std::string> CModHandler::getAllMods()
  743. {
  744. std::vector<std::string> modlist;
  745. for (auto & entry : allMods)
  746. modlist.push_back(entry.first);
  747. return modlist;
  748. }
  749. std::vector<std::string> CModHandler::getActiveMods()
  750. {
  751. return activeMods;
  752. }
  753. static JsonNode genDefaultFS()
  754. {
  755. // default FS config for mods: directory "Content" that acts as H3 root directory
  756. JsonNode defaultFS;
  757. defaultFS[""].Vector().resize(2);
  758. defaultFS[""].Vector()[0]["type"].String() = "zip";
  759. defaultFS[""].Vector()[0]["path"].String() = "/Content.zip";
  760. defaultFS[""].Vector()[1]["type"].String() = "dir";
  761. defaultFS[""].Vector()[1]["path"].String() = "/Content";
  762. return defaultFS;
  763. }
  764. static ISimpleResourceLoader * genModFilesystem(const std::string & modName, const JsonNode & conf)
  765. {
  766. static const JsonNode defaultFS = genDefaultFS();
  767. if (!conf["filesystem"].isNull())
  768. return CResourceHandler::createFileSystem(CModInfo::getModDir(modName), conf["filesystem"]);
  769. else
  770. return CResourceHandler::createFileSystem(CModInfo::getModDir(modName), defaultFS);
  771. }
  772. static ui32 calculateModChecksum(const std::string modName, ISimpleResourceLoader * filesystem)
  773. {
  774. boost::crc_32_type modChecksum;
  775. // first - add current VCMI version into checksum to force re-validation on VCMI updates
  776. modChecksum.process_bytes(reinterpret_cast<const void*>(GameConstants::VCMI_VERSION.data()), GameConstants::VCMI_VERSION.size());
  777. // second - add mod.json into checksum because filesystem does not contains this file
  778. // FIXME: remove workaround for core mod
  779. if (modName != "core")
  780. {
  781. ResourceID modConfFile(CModInfo::getModFile(modName), EResType::TEXT);
  782. ui32 configChecksum = CResourceHandler::get("initial")->load(modConfFile)->calculateCRC32();
  783. modChecksum.process_bytes(reinterpret_cast<const void *>(&configChecksum), sizeof(configChecksum));
  784. }
  785. // third - add all detected text files from this mod into checksum
  786. auto files = filesystem->getFilteredFiles([](const ResourceID & resID)
  787. {
  788. return resID.getType() == EResType::TEXT &&
  789. ( boost::starts_with(resID.getName(), "DATA") ||
  790. boost::starts_with(resID.getName(), "CONFIG"));
  791. });
  792. for (const ResourceID & file : files)
  793. {
  794. ui32 fileChecksum = filesystem->load(file)->calculateCRC32();
  795. modChecksum.process_bytes(reinterpret_cast<const void *>(&fileChecksum), sizeof(fileChecksum));
  796. }
  797. return modChecksum.checksum();
  798. }
  799. void CModHandler::loadModFilesystems()
  800. {
  801. activeMods = validateAndSortDependencies(activeMods);
  802. coreMod.updateChecksum(calculateModChecksum("core", CResourceHandler::get("core")));
  803. for(std::string & modName : activeMods)
  804. {
  805. CModInfo & mod = allMods[modName];
  806. CResourceHandler::addFilesystem("data", modName, genModFilesystem(modName, mod.config));
  807. }
  808. }
  809. std::set<TModID> CModHandler::getModDependencies(TModID modId, bool & isModFound)
  810. {
  811. auto it = allMods.find(modId);
  812. isModFound = (it != allMods.end());
  813. if(isModFound)
  814. return it->second.dependencies;
  815. logMod->error("Mod not found: '%s'", modId);
  816. return std::set<TModID>();
  817. }
  818. void CModHandler::initializeConfig()
  819. {
  820. loadConfigFromFile("defaultMods.json");
  821. }
  822. void CModHandler::load()
  823. {
  824. CStopWatch totalTime, timer;
  825. logMod->info("\tInitializing content handler: %d ms", timer.getDiff());
  826. content->init();
  827. for(const TModID & modName : activeMods)
  828. {
  829. logMod->trace("Generating checksum for %s", modName);
  830. allMods[modName].updateChecksum(calculateModChecksum(modName, CResourceHandler::get(modName)));
  831. }
  832. // first - load virtual "core" mod that contains all data
  833. // TODO? move all data into real mods? RoE, AB, SoD, WoG
  834. content->preloadData(coreMod);
  835. for(const TModID & modName : activeMods)
  836. content->preloadData(allMods[modName]);
  837. logMod->info("\tParsing mod data: %d ms", timer.getDiff());
  838. content->load(coreMod);
  839. for(const TModID & modName : activeMods)
  840. content->load(allMods[modName]);
  841. VLC->scriptHandler->performRegistration(VLC);//todo: this should be done before any other handlers load
  842. content->loadCustom();
  843. logMod->info("\tLoading mod data: %d ms", timer.getDiff());
  844. VLC->creh->loadCrExpBon();
  845. VLC->creh->buildBonusTreeForTiers(); //do that after all new creatures are loaded
  846. identifiers.finalize();
  847. logMod->info("\tResolving identifiers: %d ms", timer.getDiff());
  848. content->afterLoadFinalization();
  849. logMod->info("\tHandlers post-load finalization: %d ms ", timer.getDiff());
  850. logMod->info("\tAll game content loaded in %d ms", totalTime.getDiff());
  851. }
  852. void CModHandler::afterLoad(bool onlyEssential)
  853. {
  854. JsonNode modSettings;
  855. for (auto & modEntry : allMods)
  856. {
  857. std::string pointer = "/" + boost::algorithm::replace_all_copy(modEntry.first, ".", "/mods/");
  858. modSettings["activeMods"].resolvePointer(pointer) = modEntry.second.saveLocalData();
  859. }
  860. modSettings["core"] = coreMod.saveLocalData();
  861. if(!onlyEssential)
  862. {
  863. FileStream file(*CResourceHandler::get()->getResourceName(ResourceID("config/modSettings.json")), std::ofstream::out | std::ofstream::trunc);
  864. file << modSettings.toJson();
  865. }
  866. }
  867. std::string CModHandler::normalizeIdentifier(const std::string & scope, const std::string & remoteScope, const std::string & identifier)
  868. {
  869. auto p = vstd::splitStringToPair(identifier, ':');
  870. if(p.first.empty())
  871. p.first = scope;
  872. if(p.first == remoteScope)
  873. p.first.clear();
  874. return p.first.empty() ? p.second : p.first + ":" + p.second;
  875. }
  876. void CModHandler::parseIdentifier(const std::string & fullIdentifier, std::string & scope, std::string & type, std::string & identifier)
  877. {
  878. auto p = vstd::splitStringToPair(fullIdentifier, ':');
  879. scope = p.first;
  880. auto p2 = vstd::splitStringToPair(p.second, '.');
  881. if(p2.first != "")
  882. {
  883. type = p2.first;
  884. identifier = p2.second;
  885. }
  886. else
  887. {
  888. type = p.second;
  889. identifier = "";
  890. }
  891. }
  892. std::string CModHandler::makeFullIdentifier(const std::string & scope, const std::string & type, const std::string & identifier)
  893. {
  894. if(type == "")
  895. logGlobal->error("Full identifier (%s %s) requires type name", scope, identifier);
  896. std::string actualScope = scope;
  897. std::string actualName = identifier;
  898. //ignore scope if identifier is scoped
  899. auto scopeAndName = vstd::splitStringToPair(identifier, ':');
  900. if(scopeAndName.first != "")
  901. {
  902. actualScope = scopeAndName.first;
  903. actualName = scopeAndName.second;
  904. }
  905. if(actualScope == "")
  906. {
  907. return actualName == "" ? type : type + "." + actualName;
  908. }
  909. else
  910. {
  911. return actualName == "" ? actualScope+ ":" + type : actualScope + ":" + type + "." + actualName;
  912. }
  913. }