From f554e46faba13b90fa2c8a20715ec32ccc48ef0f Mon Sep 17 00:00:00 2001
From: Marc Plano-Lesay <mplano@google.com>
Date: Tue, 19 Jul 2022 09:09:55 +1000
Subject: [PATCH] Support generating lite protos with Bazel

While the Kotlin layer itself doesn't change depending on the flavor
(normal or lite), the underlying Java generated library does. This
effectively just selects either a `java_lite_proto_library` or a
`java_proto_library` depending on the flavor, everything else remains
the same.
---
 kt_jvm_grpc.bzl | 38 +++++++++++++++++++++++++++-----------
 1 file changed, 27 insertions(+), 11 deletions(-)

diff --git a/kt_jvm_grpc.bzl b/kt_jvm_grpc.bzl
index f7a62a89..5266b134 100644
--- a/kt_jvm_grpc.bzl
+++ b/kt_jvm_grpc.bzl
@@ -255,6 +255,7 @@ def kt_jvm_proto_library(
         compatible_with = None,
         restricted_to = None,
         visibility = None,
+        flavor = None,
         deprecation = None,
         features = []):
     """
@@ -278,23 +279,38 @@ def kt_jvm_proto_library(
       compatible_with: Standard attribute
       restricted_to: Standard attribute
       visibility: Standard attribute
+      flavor: "normal" (default) for normal proto runtime, or "lite" for the lite runtime
+        (for Android usage)
       deprecation: Standard attribute
       features: Standard attribute
     """
     java_proto_target = ":%s_DO_NOT_DEPEND_java_proto" % name
     helper_target = ":%s_DO_NOT_DEPEND_kt_proto" % name
 
-    native.java_proto_library(
-        name = java_proto_target[1:],
-        deps = deps,
-        testonly = testonly,
-        compatible_with = compatible_with,
-        visibility = ["//visibility:private"],
-        restricted_to = restricted_to,
-        tags = tags,
-        deprecation = deprecation,
-        features = features,
-    )
+    if flavor == "lite":
+        native.java_lite_proto_library(
+            name = java_proto_target[1:],
+            deps = deps,
+            testonly = testonly,
+            compatible_with = compatible_with,
+            visibility = ["//visibility:private"],
+            restricted_to = restricted_to,
+            tags = tags,
+            deprecation = deprecation,
+            features = features,
+        )
+    else:
+        native.java_proto_library(
+            name = java_proto_target[1:],
+            deps = deps,
+            testonly = testonly,
+            compatible_with = compatible_with,
+            visibility = ["//visibility:private"],
+            restricted_to = restricted_to,
+            tags = tags,
+            deprecation = deprecation,
+            features = features,
+        )
 
     _kt_jvm_proto_library_helper(
         name = helper_target[1:],