From b8228a036016fa58cab4b00a2e62298df299d41f Mon Sep 17 00:00:00 2001
From: mat <27899617+mat-1@users.noreply.github.com>
Date: Sat, 27 Aug 2022 20:31:21 -0500
Subject: Azalea registry (#20)

* make azalea-registry crate

* add trait feature to az-block

* registr

* registry macro

* impl Display for registry things

* registries
---
 codegen/migrate.py | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

(limited to 'codegen/migrate.py')

diff --git a/codegen/migrate.py b/codegen/migrate.py
index 2dacc208..50bdb354 100644
--- a/codegen/migrate.py
+++ b/codegen/migrate.py
@@ -1,8 +1,9 @@
 from lib.code.packet import fix_state
 from lib.utils import PacketIdentifier, group_packets
-import lib.code.utils
 import lib.code.version
+import lib.code.blocks
 import lib.code.packet
+import lib.code.utils
 import lib.download
 import lib.extract
 import sys
@@ -102,6 +103,20 @@ lib.code.version.set_protocol_version(
     new_burger_data[0]['version']['protocol'])
 lib.code.version.set_version_id(new_version_id)
 
+print('Updated protocol!')
+
+
+old_ordered_blocks = lib.extract.get_ordered_blocks_burger(old_version_id)
+new_ordered_blocks = lib.extract.get_ordered_blocks_burger(new_version_id)
+if old_ordered_blocks != new_ordered_blocks:
+    print('Blocks changed, updating...')
+    block_states_burger = lib.extract.get_block_states_burger(new_version_id)
+    block_states_report = lib.extract.get_block_states_report(new_version_id)
+
+    lib.code.blocks.generate_blocks(
+        block_states_burger, block_states_report, old_ordered_blocks, new_mappings)
+
+
 lib.code.utils.fmt()
 
 print('Done!')
-- 
cgit v1.2.3