I would reach for class-variance-authority (aka cva) instead for stuff like a size prop, it declaratively solves that.

Interesting, I didn't know that project yet! I'll give it a thorough read.