This is a great idea because bases on worlds, on servers and realms it will make beds more personal, it also is a great way to add a small fun achievement. It could also be a way to show off a cape.
For 1.17 instead of doing a visible biome mechanic or mob update. Instead, clean up Minecraft's code so that way it is more lag friendly. For example clean up redstone so it updates behind the scen...