From 249c4d4634e211ccfe24dab7c0fc18707c6ed4ce Mon Sep 17 00:00:00 2001
From: Luke Murphy <lukewm@riseup.net>
Date: Sun, 9 Aug 2020 02:52:58 +0200
Subject: [PATCH] Add generaton cli entrypoint

---
 example/client.py    |  4 ++++
 example/schema.proto |  9 +++++++++
 example/server.py    |  4 ++++
 hrpc/hrpc.py         | 24 ++++++++++++++++++++++++
 4 files changed, 41 insertions(+)
 create mode 100644 example/client.py
 create mode 100644 example/schema.proto
 create mode 100644 example/server.py
 create mode 100644 hrpc/hrpc.py

diff --git a/example/client.py b/example/client.py
new file mode 100644
index 0000000..1e7dc84
--- /dev/null
+++ b/example/client.py
@@ -0,0 +1,4 @@
+"""Echo client."""
+
+if __name__ == "__main__":
+    pass
diff --git a/example/schema.proto b/example/schema.proto
new file mode 100644
index 0000000..3247e86
--- /dev/null
+++ b/example/schema.proto
@@ -0,0 +1,9 @@
+syntax = "proto2";
+
+message EchoMsg {
+  required string value = 1;
+}
+
+service Example {
+  rpc Echo (EchoMsg) returns (EchoMsg) {}
+}
diff --git a/example/server.py b/example/server.py
new file mode 100644
index 0000000..1921bbe
--- /dev/null
+++ b/example/server.py
@@ -0,0 +1,4 @@
+"""Echo server."""
+
+if __name__ == "__main__":
+    pass
diff --git a/hrpc/hrpc.py b/hrpc/hrpc.py
new file mode 100644
index 0000000..b8eb2fe
--- /dev/null
+++ b/hrpc/hrpc.py
@@ -0,0 +1,24 @@
+"""Simple RPC with Protobuf Services."""
+from argparse import ArgumentParser, FileType
+from shlex import split
+from subprocess import run
+
+
+def main():
+    """Command-line entrypoint."""
+    parser = ArgumentParser(description="Generate hrpc service and stubs")
+    parser.add_argument("protobuf", type=FileType("r"), help="protobuf file")
+    args = parser.parse_args()
+    generate(args)
+
+
+def generate(args):
+    """Generate services and stubs."""
+    cmd = (
+        "python -m grpc_tools.protoc "
+        "--purerpc_out=. "
+        "--python_out=. "
+        "-I. "
+        f"{args.protobuf.name}"
+    )
+    run(split(cmd))