feat: dump special tools required to mine each block type

EndstoneMC · Jul 1, 2024 · 5225116 · 5225116
1 parent 511841d
commit 5225116
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/include/bedrock/world/level/block/block_legacy.h b/include/bedrock/world/level/block/block_legacy.h
@@ -289,6 +289,11 @@ class BlockLegacy : public BlockComponentStorage {
         return name_info_.namespace_name;
     }
 
+    [[nodiscard]] const Block *getDefaultState() const
+    {
+        return default_state_;
+    }
+
     void forEachBlockPermutation(std::function<bool(Block const &)> callback) const
     {
         for (const auto &block_permutation : block_permutations_) {

diff --git a/src/endstone_devtools/devtools.cpp b/src/endstone_devtools/devtools.cpp
@@ -242,7 +242,7 @@ void render()
                                  [&](CompoundTag *arg) {
                                      BigEndianStringByteOutput output;
                                      NbtIo::writeNamedTag("", *arg, output);
-                                     zstr::ofstream file(path.string(), std::ios::binary);
+                                     zstr::ofstream file(path.string(), std::ios::out | std::ios::binary);
                                      file << output.buffer;
                                  }},
                 file_to_save);

diff --git a/src/endstone_devtools/vanilla_data.cpp b/src/endstone_devtools/vanilla_data.cpp
@@ -111,6 +111,7 @@ void dumpBlockData(VanillaData &data, ::Level &level)
 {
     auto overworld = level.getDimension(VanillaDimensions::Overworld);
     auto &region = overworld->getBlockSourceFromMainChunkSource();
+    auto item_registry = level.getItemRegistry().weak_registry.lock();
 
     ::ListTag blocks;
     BlockTypeRegistry::forEachBlock([&](const BlockLegacy &block_legacy) {
@@ -146,6 +147,16 @@ void dumpBlockData(VanillaData &data, ::Level &level)
             data.block_types[name]["tags"] = tags;
         }
 
+        nlohmann::json special_tools;
+        for (const auto &[key, item] : item_registry->getNameToItemMap()) {
+            if (item->canDestroySpecial(*block_legacy.getDefaultState())) {
+                special_tools.push_back(item->getFullItemName());
+            }
+        }
+        if (!special_tools.is_null()) {
+            data.block_types[name]["specialTools"] = special_tools;
+        }
+
         block_legacy.forEachBlockPermutation([&](const Block &block) {
             AABB collision_shape = {0};
             block.getCollisionShape(collision_shape, region, {0, 0, 0}, nullptr);