Move badge hooks to the Yaga class for ease of executing via a plugin.