Data Driven Elytra boosting

Currently, as far as I can tell, Elytra boosting is exclusive to using a rocket in flight. It would be nice if we could define anything being used to boost the elytra, or at least anything with the 'fireworks' tag or something.

Also, there doesn't seem to be a Datapack Topic here, only 'add-ons' which are bedrock-only and work differently.