Of course, in the vertex shader I just write a general equation so we don't have to deal with each special case individually like this.

Using vertex and fragment programs, it is possible to replace the traditional vertex projection and lighting equation used by real-time computer graphics hardware.