diff options
Diffstat (limited to 'codegen')
| -rwxr-xr-x[-rw-r--r--] | codegen/.gitignore | 0 | ||||
| -rwxr-xr-x[-rw-r--r--] | codegen/README.md | 0 | ||||
| -rwxr-xr-x[-rw-r--r--] | codegen/genblocks.py | 0 | ||||
| -rwxr-xr-x[-rw-r--r--] | codegen/genregistries.py | 0 | ||||
| -rwxr-xr-x[-rw-r--r--] | codegen/lib/code/blocks.py | 0 | ||||
| -rw-r--r-- | codegen/lib/code/entity.py | 36 | ||||
| -rwxr-xr-x[-rw-r--r--] | codegen/lib/code/packet.py | 0 | ||||
| -rwxr-xr-x[-rw-r--r--] | codegen/lib/code/registry.py | 0 | ||||
| -rwxr-xr-x[-rw-r--r--] | codegen/lib/code/shapes.py | 0 | ||||
| -rwxr-xr-x[-rw-r--r--] | codegen/lib/code/utils.py | 0 | ||||
| -rwxr-xr-x[-rw-r--r--] | codegen/lib/code/version.py | 0 | ||||
| -rwxr-xr-x[-rw-r--r--] | codegen/lib/download.py | 0 | ||||
| -rwxr-xr-x[-rw-r--r--] | codegen/lib/extract.py | 0 | ||||
| -rwxr-xr-x[-rw-r--r--] | codegen/lib/mappings.py | 0 | ||||
| -rwxr-xr-x[-rw-r--r--] | codegen/lib/utils.py | 0 | ||||
| -rwxr-xr-x[-rw-r--r--] | codegen/migrate.py | 0 | ||||
| -rwxr-xr-x[-rw-r--r--] | codegen/newpacket.py | 0 |
17 files changed, 35 insertions, 1 deletions
diff --git a/codegen/.gitignore b/codegen/.gitignore index 2ef6e1be..2ef6e1be 100644..100755 --- a/codegen/.gitignore +++ b/codegen/.gitignore diff --git a/codegen/README.md b/codegen/README.md index e2cb0fcd..e2cb0fcd 100644..100755 --- a/codegen/README.md +++ b/codegen/README.md diff --git a/codegen/genblocks.py b/codegen/genblocks.py index 6b802771..6b802771 100644..100755 --- a/codegen/genblocks.py +++ b/codegen/genblocks.py diff --git a/codegen/genregistries.py b/codegen/genregistries.py index 43591d59..43591d59 100644..100755 --- a/codegen/genregistries.py +++ b/codegen/genregistries.py diff --git a/codegen/lib/code/blocks.py b/codegen/lib/code/blocks.py index c32a3bbc..c32a3bbc 100644..100755 --- a/codegen/lib/code/blocks.py +++ b/codegen/lib/code/blocks.py diff --git a/codegen/lib/code/entity.py b/codegen/lib/code/entity.py index 13f99022..9b976ef6 100644 --- a/codegen/lib/code/entity.py +++ b/codegen/lib/code/entity.py @@ -44,7 +44,7 @@ def generate_entity_metadata(burger_entity_data: dict, mappings: Mappings): code.append('use azalea_block::BlockState;') code.append('use azalea_chat::Component;') code.append('use azalea_core::{BlockPos, Direction, Particle, Slot};') - code.append('use std::{collections::VecDeque, ops::Deref};') + code.append('use std::{collections::VecDeque, ops::{Deref, DerefMut}};') code.append('use uuid::Uuid;') code.append('') @@ -271,6 +271,10 @@ def generate_entity_metadata(burger_entity_data: dict, mappings: Mappings): code.append( f'fn deref(&self) -> &Self::Target {{ &self.{parent_field_name} }}') code.append('}') + code.append(f'impl DerefMut for {struct_name} {{') + code.append( + f'fn deref_mut(&mut self) -> &mut Self::Target {{ &mut self.{parent_field_name} }}') + code.append('}') code.append('') # make the EntityMetadata enum from entity_structs @@ -309,6 +313,36 @@ def generate_entity_metadata(burger_entity_data: dict, mappings: Mappings): code.append('}') code.append('') + # impl Deref for EntityMetadata { + # type Target = AbstractEntity; + # fn deref(&self) -> &Self::Target { + # match self { + # EntityMetadata::Allay(entity) => entity, + # _ => {} + # } + # } + # } + code.append('impl Deref for EntityMetadata {') + code.append('type Target = AbstractEntity;') + code.append('fn deref(&self) -> &Self::Target {') + code.append('match self {') + for struct_name in entity_structs: + code.append( + f'EntityMetadata::{struct_name}(entity) => entity,') + code.append('}') + code.append('}') + code.append('}') + code.append('impl DerefMut for EntityMetadata {') + code.append('fn deref_mut(&mut self) -> &mut Self::Target {') + code.append('match self {') + for struct_name in entity_structs: + code.append( + f'EntityMetadata::{struct_name}(entity) => entity,') + code.append('}') + code.append('}') + code.append('}') + code.append('') + with open(METADATA_RS_DIR, 'w') as f: f.write('\n'.join(code)) diff --git a/codegen/lib/code/packet.py b/codegen/lib/code/packet.py index 4beccd35..4beccd35 100644..100755 --- a/codegen/lib/code/packet.py +++ b/codegen/lib/code/packet.py diff --git a/codegen/lib/code/registry.py b/codegen/lib/code/registry.py index c22eefe9..c22eefe9 100644..100755 --- a/codegen/lib/code/registry.py +++ b/codegen/lib/code/registry.py diff --git a/codegen/lib/code/shapes.py b/codegen/lib/code/shapes.py index 9cf3093a..9cf3093a 100644..100755 --- a/codegen/lib/code/shapes.py +++ b/codegen/lib/code/shapes.py diff --git a/codegen/lib/code/utils.py b/codegen/lib/code/utils.py index 66b18eed..66b18eed 100644..100755 --- a/codegen/lib/code/utils.py +++ b/codegen/lib/code/utils.py diff --git a/codegen/lib/code/version.py b/codegen/lib/code/version.py index 35779af6..35779af6 100644..100755 --- a/codegen/lib/code/version.py +++ b/codegen/lib/code/version.py diff --git a/codegen/lib/download.py b/codegen/lib/download.py index 5988f8ec..5988f8ec 100644..100755 --- a/codegen/lib/download.py +++ b/codegen/lib/download.py diff --git a/codegen/lib/extract.py b/codegen/lib/extract.py index e66b9400..e66b9400 100644..100755 --- a/codegen/lib/extract.py +++ b/codegen/lib/extract.py diff --git a/codegen/lib/mappings.py b/codegen/lib/mappings.py index 7cbb863a..7cbb863a 100644..100755 --- a/codegen/lib/mappings.py +++ b/codegen/lib/mappings.py diff --git a/codegen/lib/utils.py b/codegen/lib/utils.py index efabbb38..efabbb38 100644..100755 --- a/codegen/lib/utils.py +++ b/codegen/lib/utils.py diff --git a/codegen/migrate.py b/codegen/migrate.py index 50bdb354..50bdb354 100644..100755 --- a/codegen/migrate.py +++ b/codegen/migrate.py diff --git a/codegen/newpacket.py b/codegen/newpacket.py index 48d97640..48d97640 100644..100755 --- a/codegen/newpacket.py +++ b/codegen/newpacket.py |