Mirrored_Repos/veloren
1
0
Fork 0
mirror of https://gitlab.com/veloren/veloren.git synced 2024-08-30 18:12:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed potential panic when adding dead pet

Browse Source
This commit is contained in:
Joshua Barretto 2023-10-16 09:53:15 +00:00
parent 912002bd0c
commit 285a0b2b01
1 changed files with 27 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
common/src/comp/group.rs
View File

@ -264,27 +264,35 @@ impl GroupManager {
uids: &Uids,
notifier: &mut impl FnMut(specs::Entity, ChangeNotification<specs::Entity>),
) {
let group = match groups.get(owner).copied() {
Some(group) => group,
None => {
let new_group = self.create_group(owner, 1);
groups.insert(owner, new_group).unwrap();
// Inform
notifier(owner, ChangeNotification::NewLeader(owner));
new_group
},
};
if !entities.is_alive(owner) {
warn!("Tried to create new pet for non-existent owner {owner:?}");
} else if !entities.is_alive(pet) {
warn!("Tried to create new pet for non-existent pet {pet:?}");
} else {
let group = match groups.get(owner).copied() {
Some(group) => group,
None => {
let new_group = self.create_group(owner, 1);
// Unwrap can't fail, we checked that `owner` is alive above
groups.insert(owner, new_group).unwrap();
// Inform
notifier(owner, ChangeNotification::NewLeader(owner));
new_group
},
};
// Inform
members(group, &*groups, entities, alignments, uids).for_each(|(e, role)| match role {
Role::Member => {
notifier(e, ChangeNotification::Added(pet, Role::Pet));
},
Role::Pet => {},
});
// Inform
members(group, &*groups, entities, alignments, uids).for_each(|(e, role)| match role {
Role::Member => {
notifier(e, ChangeNotification::Added(pet, Role::Pet));
},
Role::Pet => {},
});
// Add
groups.insert(pet, group).unwrap();
// Add
// Unwrap can't fail, we checked that `pet` is alive above
groups.insert(pet, group).unwrap();
}
}
pub fn leave_group(