Rock vs Roman main eventing is, and always has been the right move. And if they do it right, they won't hurt Cody.
Cody facing Seth and winning seems like the wrong move though. I'd do something else with Cody. Have him beat Gunter or Logan Paul at Mania, but it's too late for that.