Graves auto attack suggestion

I just watched the new Graves re-work and I thought it would be cool if his ammo recharges like Eponas carrots From the Legend of Zelda. Change it so instead of a measly 2 shots he can hold 6 (or 5?), and he has to be out of combat for 1 second (or whatever works without interrupting last hit cadence) to start reloading, and like the carrots he loads 1 at a time. But what happens if you use up all your ammo? If you don't wait to reload? Or what if you have to 1v1 someone? Your not gonna have time to reload, so my solution is to make it so graves starts beating his enemies with the butt of his gun (melee range). He already has a dash in his kit so it can double as a gap closer, and graves is meant to be an "in your face" ADC anyways. If he has to reload an empty gun he has to wait a little longer (3-4 seconds?) but he gets all his shots back. And if we want to get fancy, his ultimate gives him back full ammo when used. Maybe do some buffs/debuffs/balancing for when hes in melee range so that its not better or worse to his regular autos if he has to brawl it out for a long time (extended team fights).

TL;DR Graves should pistol whip people when hes out of ammo.