From bb4946a6e60c7c987daa1dda30912591d0279f1c Mon Sep 17 00:00:00 2001
From: Christian Decker <decker.christian@gmail.com>
Date: Thu, 27 Jan 2022 14:19:48 +0100
Subject: [PATCH] msggen: Support renaming methods in GRPC

There is at least one clash with a built-in for the grpc server trait,
namely `connect` so we add support for renaming a method when
generating the scaffolding
---
 contrib/msggen/msggen/grpc.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/contrib/msggen/msggen/grpc.py b/contrib/msggen/msggen/grpc.py
index d79a1ea2e..c622c5368 100644
--- a/contrib/msggen/msggen/grpc.py
+++ b/contrib/msggen/msggen/grpc.py
@@ -35,6 +35,10 @@ overrides = {
     'ListFunds.channels[].state': 'ChannelState',
 }
 
+method_name_overrides = {
+    "Connect": "ConnectPeer",
+}
+
 
 class GrpcGenerator:
     """A generator that generates protobuf files.
@@ -81,8 +85,9 @@ class GrpcGenerator:
         """)
 
         for method in service.methods:
+            mname = method_name_overrides.get(method.name, method.name)
             self.write(
-                f"	rpc {method.name}({method.request.typename}) returns ({method.response.typename}) {{}}\n",
+                f"	rpc {mname}({method.request.typename}) returns ({method.response.typename}) {{}}\n",
                 cleanup=False,
             )
 
@@ -348,8 +353,9 @@ class GrpcServerGenerator(GrpcConverterGenerator):
         """)
 
         for method in service.methods:
+            mname = method_name_overrides.get(method.name, method.name)
             # Tonic will convert to snake-case, so we have to do it here too
-            name = re.sub(r'(?<!^)(?=[A-Z])', '_', method.name).lower()
+            name = re.sub(r'(?<!^)(?=[A-Z])', '_', mname).lower()
             self.write(f"""\
             async fn {name}(
                 &self,