Free app called Stay Awake. Used it for over two years without a hitch. Does exactly what you are asking.Is there a way to keep the screen on during charging? I have searched through the settings but so far have not found anything.
I'm surprised the One can actually gain a charge while the screen is lit due to the slow charging.