aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorKousik Kumar <kousikk@google.com>2021-07-12 13:55:13 -0400
committerKousik Kumar <kousikk@google.com>2021-07-12 14:05:13 -0400
commit9aa9e47b8c3921037f779453341ae1a822d454ce (patch)
tree4db0ebe47c469fbcd49e2dfd44a9ae17eb8d7d5d
parent37557784fde4deb3e263246a142fde2ec970dc60 (diff)
downloadbuild-9aa9e47b8c3921037f779453341ae1a822d454ce.tar.gz
Add the ability to source RBE related scripts from envsetup.sh
This is so that developers don't have to remember to source a separate script to source RBE related variables. I'll modify the docs once this change is submitted. Test: USE_RBE=true source build/envsetup.sh sources RBE variables. USE_RBE=false source build/envsetup.sh does NOT source RBE variables. source build/envsetup.sh does NOT source RBE variables. Bug: b/193407413 Change-Id: Ib8abf4c307b02fbb0c63b8a62d7aeda8be42d0fb
-rw-r--r--envsetup.sh10
-rw-r--r--rbesetup.sh7
2 files changed, 15 insertions, 2 deletions
diff --git a/envsetup.sh b/envsetup.sh
index 4f9440e9fb..f9c458b764 100644
--- a/envsetup.sh
+++ b/envsetup.sh
@@ -1871,6 +1871,16 @@ function showcommands() {
fi
}
+# Source necessary setup scripts needed to run the build with Remote Execution.
+function source_rbe() {
+ local T=$(gettop)
+
+ if [[ "x$USE_RBE" != "x" && "$USE_RBE" != "false" ]]; then
+ . $T/make/rbesetup.sh --skip-envsetup
+ fi
+}
+
validate_current_shell
source_vendorsetup
+source_rbe
addcompletions
diff --git a/rbesetup.sh b/rbesetup.sh
index ec39e6ef17..3b0e7cf7dc 100644
--- a/rbesetup.sh
+++ b/rbesetup.sh
@@ -24,8 +24,11 @@ function _source_env_setup_script() {
}
# This function needs to run first as the remaining defining functions may be
-# using the envsetup.sh defined functions.
-_source_env_setup_script || return
+# using the envsetup.sh defined functions. Skip this part if this script is already
+# being invoked from envsetup.sh.
+if [[ "$1" != "--skip-envsetup" ]]; then
+ _source_env_setup_script || return
+fi
# This function prefixes the given command with appropriate variables needed
# for the build to be executed with RBE.