I guess you need to figure out what you want. I don't think there's anything "right" for you to do. If you still love him and want to be with, keep that goal in mind. If not, don't. It's up to you, there are no right or wrong answers.
I think you should maintain your distance right now. He's got a lot of things to work through and he's going to want to run away from that work. Don't let him.