Yeah! There's specializations[1] to make this even simpler.

[1] https://nixos.wiki/wiki/Specialisation