I haven't seen a way to do this in-shader yet.
But I know that there's at least one tutorial out there that just had a couple of references faces/planes facing the directions they wanted, then just copy/pasting the normal from the reference planes onto the appropriate mesh sections. The reference faces were erased afterwards.