With states still brawling over how to share dwindling flows, the Trump administration is about to take control.