While working on the replication recovery code during PM-2209, we noticed it was difficult to extend the logic since much of the code is written in large functions, with many branching conditionals. We had to write a separate function specific for magic restore that duplicated a little bit of some of the recovery logic.

We should try to refactor the replication recovery code to be more modular, allowing us to reuse code blocks for different functions. We should also try to clean up some of the branching behavior (example) to make the code structure easier to understand.

This is just a code cleanup ticket, so we can aim to do this after the release.