Merge pull request #500 from silvergasp/master

Add bazel rules for nanopb generation

1 files changed, 6 insertions, 8 deletions

diff --git a/generator/protoc-gen-nanopb b/generator/protoc-gen-nanopb
index 471a620..20a36c7 100755
--- a/generator/protoc-gen-nanopb
+++ b/generator/protoc-gen-nanopb
@@ -1,13 +1,11 @@
-#!/bin/sh
-
+#!/usr/bin/env python3
 # This file is used to invoke nanopb_generator.py as a plugin
 # to protoc on Linux and other *nix-style systems.
 # Use it like this:
 # protoc --plugin=protoc-gen-nanopb=..../protoc-gen-nanopb --nanopb_out=dir foo.proto
-#
-# Note that if you use the binary package of nanopb, the protoc
-# path is already set up properly and there is no need to give
-# --plugin= on the command line.
 
-MYPATH=$(dirname "$0")
-exec "$MYPATH/nanopb_generator.py" --protoc-plugin
+from nanopb_generator import *
+
+if __name__ == '__main__':
+    # Assume we are running as a plugin under protoc.
+    main_plugin()