JsonNode.cpp 25 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024
  1. /*
  2. * JsonNode.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 "JsonNode.h"
  12. #include "ScopeGuard.h"
  13. #include "HeroBonus.h"
  14. #include "filesystem/Filesystem.h"
  15. #include "VCMI_Lib.h" //for identifier resolution
  16. #include "CModHandler.h"
  17. #include "CGeneralTextHandler.h"
  18. #include "JsonDetail.h"
  19. using namespace JsonDetail;
  20. class LibClasses;
  21. class CModHandler;
  22. static const JsonNode nullNode;
  23. JsonNode::JsonNode(JsonType Type):
  24. type(JsonType::DATA_NULL)
  25. {
  26. setType(Type);
  27. }
  28. JsonNode::JsonNode(const char *data, size_t datasize):
  29. type(JsonType::DATA_NULL)
  30. {
  31. JsonParser parser(data, datasize);
  32. *this = parser.parse("<unknown>");
  33. }
  34. JsonNode::JsonNode(ResourceID && fileURI):
  35. type(JsonType::DATA_NULL)
  36. {
  37. auto file = CResourceHandler::get()->load(fileURI)->readAll();
  38. JsonParser parser(reinterpret_cast<char*>(file.first.get()), file.second);
  39. *this = parser.parse(fileURI.getName());
  40. }
  41. JsonNode::JsonNode(const ResourceID & fileURI):
  42. type(JsonType::DATA_NULL)
  43. {
  44. auto file = CResourceHandler::get()->load(fileURI)->readAll();
  45. JsonParser parser(reinterpret_cast<char*>(file.first.get()), file.second);
  46. *this = parser.parse(fileURI.getName());
  47. }
  48. JsonNode::JsonNode(ResourceID && fileURI, bool &isValidSyntax):
  49. type(JsonType::DATA_NULL)
  50. {
  51. auto file = CResourceHandler::get()->load(fileURI)->readAll();
  52. JsonParser parser(reinterpret_cast<char*>(file.first.get()), file.second);
  53. *this = parser.parse(fileURI.getName());
  54. isValidSyntax = parser.isValid();
  55. }
  56. JsonNode::JsonNode(const JsonNode &copy):
  57. type(JsonType::DATA_NULL),
  58. meta(copy.meta)
  59. {
  60. setType(copy.getType());
  61. switch(type)
  62. {
  63. break; case JsonType::DATA_NULL:
  64. break; case JsonType::DATA_BOOL: Bool() = copy.Bool();
  65. break; case JsonType::DATA_FLOAT: Float() = copy.Float();
  66. break; case JsonType::DATA_STRING: String() = copy.String();
  67. break; case JsonType::DATA_VECTOR: Vector() = copy.Vector();
  68. break; case JsonType::DATA_STRUCT: Struct() = copy.Struct();
  69. break; case JsonType::DATA_INTEGER:Integer() = copy.Integer();
  70. }
  71. }
  72. JsonNode::~JsonNode()
  73. {
  74. setType(JsonType::DATA_NULL);
  75. }
  76. void JsonNode::swap(JsonNode &b)
  77. {
  78. using std::swap;
  79. swap(meta, b.meta);
  80. swap(data, b.data);
  81. swap(type, b.type);
  82. }
  83. JsonNode & JsonNode::operator =(JsonNode node)
  84. {
  85. swap(node);
  86. return *this;
  87. }
  88. bool JsonNode::operator == (const JsonNode &other) const
  89. {
  90. if (getType() == other.getType())
  91. {
  92. switch(type)
  93. {
  94. case JsonType::DATA_NULL: return true;
  95. case JsonType::DATA_BOOL: return Bool() == other.Bool();
  96. case JsonType::DATA_FLOAT: return Float() == other.Float();
  97. case JsonType::DATA_STRING: return String() == other.String();
  98. case JsonType::DATA_VECTOR: return Vector() == other.Vector();
  99. case JsonType::DATA_STRUCT: return Struct() == other.Struct();
  100. case JsonType::DATA_INTEGER:return Integer()== other.Integer();
  101. }
  102. }
  103. return false;
  104. }
  105. bool JsonNode::operator != (const JsonNode &other) const
  106. {
  107. return !(*this == other);
  108. }
  109. JsonNode::JsonType JsonNode::getType() const
  110. {
  111. return type;
  112. }
  113. void JsonNode::setMeta(std::string metadata, bool recursive)
  114. {
  115. meta = metadata;
  116. if (recursive)
  117. {
  118. switch (type)
  119. {
  120. break; case JsonType::DATA_VECTOR:
  121. {
  122. for(auto & node : Vector())
  123. {
  124. node.setMeta(metadata);
  125. }
  126. }
  127. break; case JsonType::DATA_STRUCT:
  128. {
  129. for(auto & node : Struct())
  130. {
  131. node.second.setMeta(metadata);
  132. }
  133. }
  134. }
  135. }
  136. }
  137. void JsonNode::setType(JsonType Type)
  138. {
  139. if (type == Type)
  140. return;
  141. //float<->int conversion
  142. if(type == JsonType::DATA_FLOAT && Type == JsonType::DATA_INTEGER)
  143. {
  144. si64 converted = data.Float;
  145. type = Type;
  146. data.Integer = converted;
  147. return;
  148. }
  149. else if(type == JsonType::DATA_INTEGER && Type == JsonType::DATA_FLOAT)
  150. {
  151. double converted = data.Integer;
  152. type = Type;
  153. data.Float = converted;
  154. return;
  155. }
  156. //Reset node to nullptr
  157. if (Type != JsonType::DATA_NULL)
  158. setType(JsonType::DATA_NULL);
  159. switch (type)
  160. {
  161. break; case JsonType::DATA_STRING: delete data.String;
  162. break; case JsonType::DATA_VECTOR: delete data.Vector;
  163. break; case JsonType::DATA_STRUCT: delete data.Struct;
  164. break; default:
  165. break;
  166. }
  167. //Set new node type
  168. type = Type;
  169. switch(type)
  170. {
  171. break; case JsonType::DATA_NULL:
  172. break; case JsonType::DATA_BOOL: data.Bool = false;
  173. break; case JsonType::DATA_FLOAT: data.Float = 0;
  174. break; case JsonType::DATA_STRING: data.String = new std::string();
  175. break; case JsonType::DATA_VECTOR: data.Vector = new JsonVector();
  176. break; case JsonType::DATA_STRUCT: data.Struct = new JsonMap();
  177. break; case JsonType::DATA_INTEGER: data.Integer = 0;
  178. }
  179. }
  180. bool JsonNode::isNull() const
  181. {
  182. return type == JsonType::DATA_NULL;
  183. }
  184. bool JsonNode::isNumber() const
  185. {
  186. return type == JsonType::DATA_INTEGER || type == JsonType::DATA_FLOAT;
  187. }
  188. bool JsonNode::containsBaseData() const
  189. {
  190. switch(type)
  191. {
  192. case JsonType::DATA_NULL:
  193. return false;
  194. case JsonType::DATA_STRUCT:
  195. for(auto elem : *data.Struct)
  196. {
  197. if(elem.second.containsBaseData())
  198. return true;
  199. }
  200. return false;
  201. default:
  202. //other types (including vector) cannot be extended via merge
  203. return true;
  204. }
  205. }
  206. bool JsonNode::isCompact() const
  207. {
  208. switch(type)
  209. {
  210. case JsonType::DATA_VECTOR:
  211. for(JsonNode & elem : *data.Vector)
  212. {
  213. if(!elem.isCompact())
  214. return false;
  215. }
  216. return true;
  217. case JsonType::DATA_STRUCT:
  218. {
  219. int propertyCount = data.Struct->size();
  220. if(propertyCount == 0)
  221. return true;
  222. else if(propertyCount == 1)
  223. return data.Struct->begin()->second.isCompact();
  224. }
  225. return false;
  226. default:
  227. return true;
  228. }
  229. }
  230. void JsonNode::clear()
  231. {
  232. setType(JsonType::DATA_NULL);
  233. }
  234. bool & JsonNode::Bool()
  235. {
  236. setType(JsonType::DATA_BOOL);
  237. return data.Bool;
  238. }
  239. double & JsonNode::Float()
  240. {
  241. setType(JsonType::DATA_FLOAT);
  242. return data.Float;
  243. }
  244. si64 & JsonNode::Integer()
  245. {
  246. setType(JsonType::DATA_INTEGER);
  247. return data.Integer;
  248. }
  249. std::string & JsonNode::String()
  250. {
  251. setType(JsonType::DATA_STRING);
  252. return *data.String;
  253. }
  254. JsonVector & JsonNode::Vector()
  255. {
  256. setType(JsonType::DATA_VECTOR);
  257. return *data.Vector;
  258. }
  259. JsonMap & JsonNode::Struct()
  260. {
  261. setType(JsonType::DATA_STRUCT);
  262. return *data.Struct;
  263. }
  264. const bool boolDefault = false;
  265. bool JsonNode::Bool() const
  266. {
  267. if (type == JsonType::DATA_NULL)
  268. return boolDefault;
  269. assert(type == JsonType::DATA_BOOL);
  270. return data.Bool;
  271. }
  272. const double floatDefault = 0;
  273. double JsonNode::Float() const
  274. {
  275. if(type == JsonType::DATA_NULL)
  276. return floatDefault;
  277. else if(type == JsonType::DATA_INTEGER)
  278. return data.Integer;
  279. assert(type == JsonType::DATA_FLOAT);
  280. return data.Float;
  281. }
  282. const si64 integetDefault = 0;
  283. si64 JsonNode::Integer() const
  284. {
  285. if(type == JsonType::DATA_NULL)
  286. return integetDefault;
  287. else if(type == JsonType::DATA_FLOAT)
  288. return data.Float;
  289. assert(type == JsonType::DATA_INTEGER);
  290. return data.Integer;
  291. }
  292. const std::string stringDefault = std::string();
  293. const std::string & JsonNode::String() const
  294. {
  295. if (type == JsonType::DATA_NULL)
  296. return stringDefault;
  297. assert(type == JsonType::DATA_STRING);
  298. return *data.String;
  299. }
  300. const JsonVector vectorDefault = JsonVector();
  301. const JsonVector & JsonNode::Vector() const
  302. {
  303. if (type == JsonType::DATA_NULL)
  304. return vectorDefault;
  305. assert(type == JsonType::DATA_VECTOR);
  306. return *data.Vector;
  307. }
  308. const JsonMap mapDefault = JsonMap();
  309. const JsonMap & JsonNode::Struct() const
  310. {
  311. if (type == JsonType::DATA_NULL)
  312. return mapDefault;
  313. assert(type == JsonType::DATA_STRUCT);
  314. return *data.Struct;
  315. }
  316. JsonNode & JsonNode::operator[](std::string child)
  317. {
  318. return Struct()[child];
  319. }
  320. const JsonNode & JsonNode::operator[](std::string child) const
  321. {
  322. auto it = Struct().find(child);
  323. if (it != Struct().end())
  324. return it->second;
  325. return nullNode;
  326. }
  327. // to avoid duplicating const and non-const code
  328. template<typename Node>
  329. Node & resolvePointer(Node & in, const std::string & pointer)
  330. {
  331. if (pointer.empty())
  332. return in;
  333. assert(pointer[0] == '/');
  334. size_t splitPos = pointer.find('/', 1);
  335. std::string entry = pointer.substr(1, splitPos -1);
  336. std::string remainer = splitPos == std::string::npos ? "" : pointer.substr(splitPos);
  337. if (in.getType() == JsonNode::JsonType::DATA_VECTOR)
  338. {
  339. if (entry.find_first_not_of("0123456789") != std::string::npos) // non-numbers in string
  340. throw std::runtime_error("Invalid Json pointer");
  341. if (entry.size() > 1 && entry[0] == '0') // leading zeros are not allowed
  342. throw std::runtime_error("Invalid Json pointer");
  343. size_t index = boost::lexical_cast<size_t>(entry);
  344. if (in.Vector().size() > index)
  345. return in.Vector()[index].resolvePointer(remainer);
  346. }
  347. return in[entry].resolvePointer(remainer);
  348. }
  349. const JsonNode & JsonNode::resolvePointer(const std::string &jsonPointer) const
  350. {
  351. return ::resolvePointer(*this, jsonPointer);
  352. }
  353. JsonNode & JsonNode::resolvePointer(const std::string &jsonPointer)
  354. {
  355. return ::resolvePointer(*this, jsonPointer);
  356. }
  357. std::string JsonNode::toJson(bool compact) const
  358. {
  359. std::ostringstream out;
  360. JsonWriter writer(out, compact);
  361. writer.writeNode(*this);
  362. out << "\n";
  363. return out.str();
  364. }
  365. ///JsonUtils
  366. void JsonUtils::parseTypedBonusShort(const JsonVector& source, std::shared_ptr<Bonus> dest)
  367. {
  368. dest->val = source[1].Float();
  369. resolveIdentifier(source[2],dest->subtype);
  370. dest->additionalInfo = source[3].Float();
  371. dest->duration = Bonus::PERMANENT; //TODO: handle flags (as integer)
  372. dest->turnsRemain = 0;
  373. }
  374. std::shared_ptr<Bonus> JsonUtils::parseBonus (const JsonVector &ability_vec) //TODO: merge with AddAbility, create universal parser for all bonus properties
  375. {
  376. auto b = std::make_shared<Bonus>();
  377. std::string type = ability_vec[0].String();
  378. auto it = bonusNameMap.find(type);
  379. if (it == bonusNameMap.end())
  380. {
  381. logMod->error("Error: invalid ability type %s", type);
  382. return b;
  383. }
  384. b->type = it->second;
  385. parseTypedBonusShort(ability_vec, b);
  386. return b;
  387. }
  388. template <typename T>
  389. const T & parseByMap(const std::map<std::string, T> & map, const JsonNode * val, std::string err)
  390. {
  391. static T defaultValue = T();
  392. if (!val->isNull())
  393. {
  394. std::string type = val->String();
  395. auto it = map.find(type);
  396. if (it == map.end())
  397. {
  398. logMod->error("Error: invalid %s%s", err, type);
  399. return defaultValue;
  400. }
  401. else
  402. {
  403. return it->second;
  404. }
  405. }
  406. else
  407. return defaultValue;
  408. }
  409. void JsonUtils::resolveIdentifier(si32 &var, const JsonNode &node, std::string name)
  410. {
  411. const JsonNode &value = node[name];
  412. if (!value.isNull())
  413. {
  414. switch (value.getType())
  415. {
  416. case JsonNode::JsonType::DATA_INTEGER:
  417. var = value.Integer();
  418. break;
  419. case JsonNode::JsonType::DATA_FLOAT:
  420. var = value.Float();
  421. break;
  422. case JsonNode::JsonType::DATA_STRING:
  423. VLC->modh->identifiers.requestIdentifier(value, [&](si32 identifier)
  424. {
  425. var = identifier;
  426. });
  427. break;
  428. default:
  429. logMod->error("Error! Wrong identifier used for value of %s", name);
  430. }
  431. }
  432. }
  433. void JsonUtils::resolveIdentifier(const JsonNode &node, si32 &var)
  434. {
  435. switch (node.getType())
  436. {
  437. case JsonNode::JsonType::DATA_INTEGER:
  438. var = node.Integer();
  439. break;
  440. case JsonNode::JsonType::DATA_FLOAT:
  441. var = node.Float();
  442. break;
  443. case JsonNode::JsonType::DATA_STRING:
  444. VLC->modh->identifiers.requestIdentifier(node, [&](si32 identifier)
  445. {
  446. var = identifier;
  447. });
  448. break;
  449. default:
  450. logMod->error("Error! Wrong identifier used for identifier!");
  451. }
  452. }
  453. std::shared_ptr<Bonus> JsonUtils::parseBonus(const JsonNode &ability)
  454. {
  455. auto b = std::make_shared<Bonus>();
  456. if (!parseBonus(ability, b.get()))
  457. {
  458. return nullptr;
  459. }
  460. return b;
  461. }
  462. bool JsonUtils::parseBonus(const JsonNode &ability, Bonus *b)
  463. {
  464. const JsonNode *value;
  465. std::string type = ability["type"].String();
  466. auto it = bonusNameMap.find(type);
  467. if (it == bonusNameMap.end())
  468. {
  469. logMod->error("Error: invalid ability type %s", type);
  470. return false;
  471. }
  472. b->type = it->second;
  473. resolveIdentifier(b->subtype, ability, "subtype");
  474. b->val = ability["val"].Float();
  475. value = &ability["valueType"];
  476. if (!value->isNull())
  477. b->valType = static_cast<Bonus::ValueType>(parseByMap(bonusValueMap, value, "value type "));
  478. resolveIdentifier(b->additionalInfo, ability, "addInfo");
  479. b->turnsRemain = ability["turns"].Float();
  480. b->sid = ability["sourceID"].Float();
  481. b->description = ability["description"].String();
  482. value = &ability["effectRange"];
  483. if (!value->isNull())
  484. b->effectRange = static_cast<Bonus::LimitEffect>(parseByMap(bonusLimitEffect, value, "effect range "));
  485. value = &ability["duration"];
  486. if (!value->isNull())
  487. {
  488. switch (value->getType())
  489. {
  490. case JsonNode::JsonType::DATA_STRING:
  491. b->duration = parseByMap(bonusDurationMap, value, "duration type ");
  492. break;
  493. case JsonNode::JsonType::DATA_VECTOR:
  494. {
  495. ui16 dur = 0;
  496. for (const JsonNode & d : value->Vector())
  497. {
  498. dur |= parseByMap(bonusDurationMap, &d, "duration type ");
  499. }
  500. b->duration = dur;
  501. }
  502. break;
  503. default:
  504. logMod->error("Error! Wrong bonus duration format.");
  505. }
  506. }
  507. value = &ability["source"];
  508. if (!value->isNull())
  509. b->source = static_cast<Bonus::BonusSource>(parseByMap(bonusSourceMap, value, "source type "));
  510. value = &ability["limiters"];
  511. if (!value->isNull())
  512. {
  513. for (const JsonNode & limiter : value->Vector())
  514. {
  515. switch (limiter.getType())
  516. {
  517. case JsonNode::JsonType::DATA_STRING: //pre-defined limiters
  518. b->limiter = parseByMap(bonusLimiterMap, &limiter, "limiter type ");
  519. break;
  520. case JsonNode::JsonType::DATA_STRUCT: //customizable limiters
  521. {
  522. std::shared_ptr<ILimiter> l;
  523. if (limiter["type"].String() == "CREATURE_TYPE_LIMITER")
  524. {
  525. std::shared_ptr<CCreatureTypeLimiter> l2 = std::make_shared<CCreatureTypeLimiter>(); //TODO: How the hell resolve pointer to creature?
  526. const JsonVector vec = limiter["parameters"].Vector();
  527. VLC->modh->identifiers.requestIdentifier("creature", vec[0], [=](si32 creature)
  528. {
  529. l2->setCreature(CreatureID(creature));
  530. });
  531. if (vec.size() > 1)
  532. {
  533. l2->includeUpgrades = vec[1].Bool();
  534. }
  535. else
  536. l2->includeUpgrades = false;
  537. l = l2;
  538. }
  539. if (limiter["type"].String() == "HAS_ANOTHER_BONUS_LIMITER")
  540. {
  541. std::shared_ptr<HasAnotherBonusLimiter> l2 = std::make_shared<HasAnotherBonusLimiter>();
  542. const JsonVector vec = limiter["parameters"].Vector();
  543. std::string anotherBonusType = vec[0].String();
  544. auto it = bonusNameMap.find(anotherBonusType);
  545. if (it == bonusNameMap.end())
  546. {
  547. logMod->error("Error: invalid ability type %s", anotherBonusType);
  548. continue;
  549. }
  550. l2->type = it->second;
  551. if (vec.size() > 1 )
  552. {
  553. resolveIdentifier(vec[1], l2->subtype);
  554. l2->isSubtypeRelevant = true;
  555. }
  556. l = l2;
  557. }
  558. b->addLimiter(l);
  559. }
  560. break;
  561. }
  562. }
  563. }
  564. value = &ability["propagator"];
  565. if (!value->isNull())
  566. b->propagator = parseByMap(bonusPropagatorMap, value, "propagator type ");
  567. value = &ability["updater"];
  568. if (!value->isNull())
  569. {
  570. const JsonNode & updaterJson = *value;
  571. if(updaterJson["type"].String() == "GROWS_WITH_LEVEL")
  572. {
  573. std::shared_ptr<ScalingUpdater> updater = std::make_shared<ScalingUpdater>();
  574. const JsonVector param = updaterJson["parameters"].Vector();
  575. updater->valPer20 = param[0].Integer();
  576. if(param.size() > 1)
  577. updater->stepSize = param[1].Integer();
  578. b->addUpdater(updater);
  579. }
  580. else
  581. logMod->warn("Unknown updater type \"%s\"", updaterJson["type"].String());
  582. }
  583. return true;
  584. }
  585. //returns first Key with value equal to given one
  586. template<class Key, class Val>
  587. Key reverseMapFirst(const Val & val, const std::map<Key, Val> & map)
  588. {
  589. for(auto it : map)
  590. {
  591. if(it.second == val)
  592. {
  593. return it.first;
  594. }
  595. }
  596. assert(0);
  597. return "";
  598. }
  599. void JsonUtils::unparseBonus( JsonNode &node, const std::shared_ptr<Bonus>& bonus )
  600. {
  601. node["type"].String() = reverseMapFirst<std::string, Bonus::BonusType>(bonus->type, bonusNameMap);
  602. node["subtype"].Float() = bonus->subtype;
  603. node["val"].Float() = bonus->val;
  604. node["valueType"].String() = reverseMapFirst<std::string, Bonus::ValueType>(bonus->valType, bonusValueMap);
  605. node["additionalInfo"].Float() = bonus->additionalInfo;
  606. node["turns"].Float() = bonus->turnsRemain;
  607. node["sourceID"].Float() = bonus->source;
  608. node["description"].String() = bonus->description;
  609. node["effectRange"].String() = reverseMapFirst<std::string, Bonus::LimitEffect>(bonus->effectRange, bonusLimitEffect);
  610. node["duration"].String() = reverseMapFirst<std::string, ui16>(bonus->duration, bonusDurationMap);
  611. node["source"].String() = reverseMapFirst<std::string, Bonus::BonusSource>(bonus->source, bonusSourceMap);
  612. if(bonus->limiter)
  613. {
  614. node["limiter"].String() = reverseMapFirst<std::string, TLimiterPtr>(bonus->limiter, bonusLimiterMap);
  615. }
  616. if(bonus->propagator)
  617. {
  618. node["propagator"].String() = reverseMapFirst<std::string, TPropagatorPtr>(bonus->propagator, bonusPropagatorMap);
  619. }
  620. }
  621. void minimizeNode(JsonNode & node, const JsonNode & schema)
  622. {
  623. if (schema["type"].String() == "object")
  624. {
  625. std::set<std::string> foundEntries;
  626. for(auto & entry : schema["required"].Vector())
  627. {
  628. std::string name = entry.String();
  629. foundEntries.insert(name);
  630. minimizeNode(node[name], schema["properties"][name]);
  631. if (vstd::contains(node.Struct(), name) &&
  632. node[name] == schema["properties"][name]["default"])
  633. {
  634. node.Struct().erase(name);
  635. }
  636. }
  637. // erase all unhandled entries
  638. for (auto it = node.Struct().begin(); it != node.Struct().end();)
  639. {
  640. if (!vstd::contains(foundEntries, it->first))
  641. it = node.Struct().erase(it);
  642. else
  643. it++;
  644. }
  645. }
  646. }
  647. void JsonUtils::minimize(JsonNode & node, std::string schemaName)
  648. {
  649. minimizeNode(node, getSchema(schemaName));
  650. }
  651. // FIXME: except for several lines function is identical to minimizeNode. Some way to reduce duplication?
  652. void maximizeNode(JsonNode & node, const JsonNode & schema)
  653. {
  654. // "required" entry can only be found in object/struct
  655. if (schema["type"].String() == "object")
  656. {
  657. std::set<std::string> foundEntries;
  658. // check all required entries that have default version
  659. for(auto & entry : schema["required"].Vector())
  660. {
  661. std::string name = entry.String();
  662. foundEntries.insert(name);
  663. if (node[name].isNull() &&
  664. !schema["properties"][name]["default"].isNull())
  665. {
  666. node[name] = schema["properties"][name]["default"];
  667. }
  668. maximizeNode(node[name], schema["properties"][name]);
  669. }
  670. // erase all unhandled entries
  671. for (auto it = node.Struct().begin(); it != node.Struct().end();)
  672. {
  673. if (!vstd::contains(foundEntries, it->first))
  674. it = node.Struct().erase(it);
  675. else
  676. it++;
  677. }
  678. }
  679. }
  680. void JsonUtils::maximize(JsonNode & node, std::string schemaName)
  681. {
  682. maximizeNode(node, getSchema(schemaName));
  683. }
  684. bool JsonUtils::validate(const JsonNode &node, std::string schemaName, std::string dataName)
  685. {
  686. std::string log = Validation::check(schemaName, node);
  687. if (!log.empty())
  688. {
  689. logMod->warn("Data in %s is invalid!", dataName);
  690. logMod->warn(log);
  691. }
  692. return log.empty();
  693. }
  694. const JsonNode & getSchemaByName(std::string name)
  695. {
  696. // cached schemas to avoid loading json data multiple times
  697. static std::map<std::string, JsonNode> loadedSchemas;
  698. if (vstd::contains(loadedSchemas, name))
  699. return loadedSchemas[name];
  700. std::string filename = "config/schemas/" + name + ".json";
  701. if (CResourceHandler::get()->existsResource(ResourceID(filename)))
  702. {
  703. loadedSchemas[name] = JsonNode(ResourceID(filename));
  704. return loadedSchemas[name];
  705. }
  706. logMod->error("Error: missing schema with name %s!", name);
  707. assert(0);
  708. return nullNode;
  709. }
  710. const JsonNode & JsonUtils::getSchema(std::string URI)
  711. {
  712. size_t posColon = URI.find(':');
  713. size_t posHash = URI.find('#');
  714. if(posColon == std::string::npos)
  715. {
  716. logMod->error("Invalid schema URI:%s", URI);
  717. return nullNode;
  718. }
  719. std::string protocolName = URI.substr(0, posColon);
  720. std::string filename = URI.substr(posColon + 1, posHash - posColon - 1);
  721. if(protocolName != "vcmi")
  722. {
  723. logMod->error("Error: unsupported URI protocol for schema: %s", URI);
  724. return nullNode;
  725. }
  726. // check if json pointer if present (section after hash in string)
  727. if(posHash == std::string::npos || posHash == URI.size() - 1)
  728. return getSchemaByName(filename);
  729. else
  730. return getSchemaByName(filename).resolvePointer(URI.substr(posHash + 1));
  731. }
  732. void JsonUtils::merge(JsonNode & dest, JsonNode & source)
  733. {
  734. if (dest.getType() == JsonNode::JsonType::DATA_NULL)
  735. {
  736. std::swap(dest, source);
  737. return;
  738. }
  739. switch (source.getType())
  740. {
  741. case JsonNode::JsonType::DATA_NULL:
  742. {
  743. dest.clear();
  744. break;
  745. }
  746. case JsonNode::JsonType::DATA_BOOL:
  747. case JsonNode::JsonType::DATA_FLOAT:
  748. case JsonNode::JsonType::DATA_INTEGER:
  749. case JsonNode::JsonType::DATA_STRING:
  750. case JsonNode::JsonType::DATA_VECTOR:
  751. {
  752. std::swap(dest, source);
  753. break;
  754. }
  755. case JsonNode::JsonType::DATA_STRUCT:
  756. {
  757. //recursively merge all entries from struct
  758. for(auto & node : source.Struct())
  759. merge(dest[node.first], node.second);
  760. }
  761. }
  762. }
  763. void JsonUtils::mergeCopy(JsonNode & dest, JsonNode source)
  764. {
  765. // uses copy created in stack to safely merge two nodes
  766. merge(dest, source);
  767. }
  768. void JsonUtils::inherit(JsonNode & descendant, const JsonNode & base)
  769. {
  770. JsonNode inheritedNode(base);
  771. merge(inheritedNode,descendant);
  772. descendant.swap(inheritedNode);
  773. }
  774. JsonNode JsonUtils::intersect(const std::vector<JsonNode> & nodes, bool pruneEmpty)
  775. {
  776. if(nodes.size() == 0)
  777. return nullNode;
  778. JsonNode result = nodes[0];
  779. for(int i = 1; i < nodes.size(); i++)
  780. {
  781. if(result.isNull())
  782. break;
  783. result = JsonUtils::intersect(result, nodes[i], pruneEmpty);
  784. }
  785. return result;
  786. }
  787. JsonNode JsonUtils::intersect(const JsonNode & a, const JsonNode & b, bool pruneEmpty)
  788. {
  789. if(a.getType() == JsonNode::JsonType::DATA_STRUCT && b.getType() == JsonNode::JsonType::DATA_STRUCT)
  790. {
  791. // intersect individual properties
  792. JsonNode result(JsonNode::JsonType::DATA_STRUCT);
  793. for(auto property : a.Struct())
  794. {
  795. if(vstd::contains(b.Struct(), property.first))
  796. {
  797. JsonNode propertyIntersect = JsonUtils::intersect(property.second, b.Struct().find(property.first)->second);
  798. if(pruneEmpty && !propertyIntersect.containsBaseData())
  799. continue;
  800. result[property.first] = propertyIntersect;
  801. }
  802. }
  803. return result;
  804. }
  805. else
  806. {
  807. // not a struct - same or different, no middle ground
  808. if(a == b)
  809. return a;
  810. }
  811. return nullNode;
  812. }
  813. JsonNode JsonUtils::difference(const JsonNode & node, const JsonNode & base)
  814. {
  815. if(node.getType() == JsonNode::JsonType::DATA_STRUCT && base.getType() == JsonNode::JsonType::DATA_STRUCT)
  816. {
  817. // subtract individual properties
  818. JsonNode result(JsonNode::JsonType::DATA_STRUCT);
  819. for(auto property : node.Struct())
  820. {
  821. if(vstd::contains(base.Struct(), property.first))
  822. {
  823. const JsonNode propertyDifference = JsonUtils::difference(property.second, base.Struct().find(property.first)->second);
  824. if(!propertyDifference.isNull())
  825. result[property.first] = propertyDifference;
  826. }
  827. else
  828. {
  829. result[property.first] = property.second;
  830. }
  831. }
  832. if(result.Struct().empty())
  833. return nullNode;
  834. return result;
  835. }
  836. else
  837. {
  838. if(node == base)
  839. return nullNode;
  840. }
  841. return node;
  842. }
  843. JsonNode JsonUtils::assembleFromFiles(std::vector<std::string> files)
  844. {
  845. bool isValid;
  846. return assembleFromFiles(files, isValid);
  847. }
  848. JsonNode JsonUtils::assembleFromFiles(std::vector<std::string> files, bool &isValid)
  849. {
  850. isValid = true;
  851. JsonNode result;
  852. for(std::string file : files)
  853. {
  854. bool isValidFile;
  855. JsonNode section(ResourceID(file, EResType::TEXT), isValidFile);
  856. merge(result, section);
  857. isValid |= isValidFile;
  858. }
  859. return result;
  860. }
  861. JsonNode JsonUtils::assembleFromFiles(std::string filename)
  862. {
  863. JsonNode result;
  864. ResourceID resID(filename, EResType::TEXT);
  865. for(auto & loader : CResourceHandler::get()->getResourcesWithName(resID))
  866. {
  867. // FIXME: some way to make this code more readable
  868. auto stream = loader->load(resID);
  869. std::unique_ptr<ui8[]> textData(new ui8[stream->getSize()]);
  870. stream->read(textData.get(), stream->getSize());
  871. JsonNode section((char*)textData.get(), stream->getSize());
  872. merge(result, section);
  873. }
  874. return result;
  875. }
  876. DLL_LINKAGE JsonNode JsonUtils::boolNode(bool value)
  877. {
  878. JsonNode node;
  879. node.Bool() = value;
  880. return node;
  881. }
  882. DLL_LINKAGE JsonNode JsonUtils::floatNode(double value)
  883. {
  884. JsonNode node;
  885. node.Float() = value;
  886. return node;
  887. }
  888. DLL_LINKAGE JsonNode JsonUtils::stringNode(std::string value)
  889. {
  890. JsonNode node;
  891. node.String() = value;
  892. return node;
  893. }
  894. DLL_LINKAGE JsonNode JsonUtils::intNode(si64 value)
  895. {
  896. JsonNode node;
  897. node.Integer() = value;
  898. return node;
  899. }