Add some missing `WithContext` wrappers in `peer_handler.rs`