itemstackmetadata.cpp 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147
  1. /*
  2. Minetest
  3. Copyright (C) 2017-8 rubenwardy <rw@rubenwardy.com>
  4. This program is free software; you can redistribute it and/or modify
  5. it under the terms of the GNU Lesser General Public License as published by
  6. the Free Software Foundation; either version 2.1 of the License, or
  7. (at your option) any later version.
  8. This program is distributed in the hope that it will be useful,
  9. but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  11. GNU Lesser General Public License for more details.
  12. You should have received a copy of the GNU Lesser General Public License along
  13. with this program; if not, write to the Free Software Foundation, Inc.,
  14. 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  15. */
  16. #include "itemstackmetadata.h"
  17. #include "util/serialize.h"
  18. #include "util/strfnd.h"
  19. #include <algorithm>
  20. #include <optional>
  21. #define DESERIALIZE_START '\x01'
  22. #define DESERIALIZE_KV_DELIM '\x02'
  23. #define DESERIALIZE_PAIR_DELIM '\x03'
  24. #define DESERIALIZE_START_STR "\x01"
  25. #define DESERIALIZE_KV_DELIM_STR "\x02"
  26. #define DESERIALIZE_PAIR_DELIM_STR "\x03"
  27. #define TOOLCAP_KEY "tool_capabilities"
  28. #define WEAR_BAR_KEY "wear_color"
  29. void ItemStackMetadata::clear()
  30. {
  31. SimpleMetadata::clear();
  32. updateToolCapabilities();
  33. updateWearBarParams();
  34. }
  35. static void sanitize_string(std::string &str)
  36. {
  37. str.erase(std::remove(str.begin(), str.end(), DESERIALIZE_START), str.end());
  38. str.erase(std::remove(str.begin(), str.end(), DESERIALIZE_KV_DELIM), str.end());
  39. str.erase(std::remove(str.begin(), str.end(), DESERIALIZE_PAIR_DELIM), str.end());
  40. }
  41. bool ItemStackMetadata::setString(const std::string &name, std::string_view var)
  42. {
  43. std::string clean_name = name;
  44. std::string clean_var(var);
  45. sanitize_string(clean_name);
  46. sanitize_string(clean_var);
  47. bool result = SimpleMetadata::setString(clean_name, clean_var);
  48. if (clean_name == TOOLCAP_KEY)
  49. updateToolCapabilities();
  50. else if (clean_name == WEAR_BAR_KEY)
  51. updateWearBarParams();
  52. return result;
  53. }
  54. void ItemStackMetadata::serialize(std::ostream &os) const
  55. {
  56. std::ostringstream os2(std::ios_base::binary);
  57. os2 << DESERIALIZE_START;
  58. for (const auto &stringvar : m_stringvars) {
  59. if (!stringvar.first.empty() || !stringvar.second.empty())
  60. os2 << stringvar.first << DESERIALIZE_KV_DELIM
  61. << stringvar.second << DESERIALIZE_PAIR_DELIM;
  62. }
  63. os << serializeJsonStringIfNeeded(os2.str());
  64. }
  65. void ItemStackMetadata::deSerialize(std::istream &is)
  66. {
  67. std::string in = deSerializeJsonStringIfNeeded(is);
  68. m_stringvars.clear();
  69. if (!in.empty()) {
  70. if (in[0] == DESERIALIZE_START) {
  71. Strfnd fnd(in);
  72. fnd.to(1);
  73. while (!fnd.at_end()) {
  74. std::string name = fnd.next(DESERIALIZE_KV_DELIM_STR);
  75. std::string var = fnd.next(DESERIALIZE_PAIR_DELIM_STR);
  76. m_stringvars[name] = var;
  77. }
  78. } else {
  79. // BACKWARDS COMPATIBILITY
  80. m_stringvars[""] = in;
  81. }
  82. }
  83. updateToolCapabilities();
  84. updateWearBarParams();
  85. }
  86. void ItemStackMetadata::updateToolCapabilities()
  87. {
  88. if (contains(TOOLCAP_KEY)) {
  89. toolcaps_overridden = true;
  90. toolcaps_override = ToolCapabilities();
  91. std::istringstream is(getString(TOOLCAP_KEY));
  92. toolcaps_override.deserializeJson(is);
  93. } else {
  94. toolcaps_overridden = false;
  95. }
  96. }
  97. void ItemStackMetadata::setToolCapabilities(const ToolCapabilities &caps)
  98. {
  99. std::ostringstream os;
  100. caps.serializeJson(os);
  101. setString(TOOLCAP_KEY, os.str());
  102. }
  103. void ItemStackMetadata::clearToolCapabilities()
  104. {
  105. setString(TOOLCAP_KEY, "");
  106. }
  107. void ItemStackMetadata::updateWearBarParams()
  108. {
  109. if (contains(WEAR_BAR_KEY)) {
  110. std::istringstream is(getString(WEAR_BAR_KEY));
  111. wear_bar_override = WearBarParams::deserializeJson(is);
  112. } else {
  113. wear_bar_override.reset();
  114. }
  115. }
  116. void ItemStackMetadata::setWearBarParams(const WearBarParams &params)
  117. {
  118. std::ostringstream os;
  119. params.serializeJson(os);
  120. setString(WEAR_BAR_KEY, os.str());
  121. }
  122. void ItemStackMetadata::clearWearBarParams()
  123. {
  124. setString(WEAR_BAR_KEY, "");
  125. }