feat: adding single language build support

.trampolinerc

@@ -22,6 +22,7 @@ pass_down_envvars+=(
     "SOURCE_BLOB"        # Set to force regeneration of a single source blob.
     "FORCE_GENERATE_ALL" # Set to force regeneration of all blobs.
     "TEST_BUCKET"        # Must set when running tests.
+    "LANGUAGE"           # For specifying specific language builds.
 )
 
 # Prevent unintentional override on the default image.

docpipeline/__main__.py

@@ -103,6 +103,22 @@ def build_one_doc(bucket_name, object_name, credentials):
         log.error(e)
         sys.exit(1)
 
+@main.command()
+@click.argument("bucket_name")
+@click.argument("language")
+@click.option(
+    "--credentials",
+    default=credentials.find(),
+    help="Path to the credentials file to use for Google Cloud Storage.",
+)
+def build_language_docs(bucket_name, language, credentials):
+    verify(credentials)
+
+    try:
+      generate.build_language_docs(bucket_name, language, credentials)
+    except Exception as e:
+      log.error(e)
+      sys.exit(1)
 
 if __name__ == "__main__":
     main()

docpipeline/generate.py

@@ -223,3 +223,9 @@ def build_new_docs(bucket_name, credentials):
         if new_name not in other_names:
             new_blobs.append(blob)
     build_blobs(new_blobs, credentials)
+
+def build_language_docs(bucket_name, language, credentials):
+    all_blobs = storage_client(credentials).list_blobs(bucket_name)
+    language_prefix = DOCFX_PREFIX+language+"-"
+    docfx_blobs = [blob for blob in all_blobs if blob.name.startswith(language_prefix)]
+    build_blobs(docfx_blobs, credentials)

generate.sh

@@ -27,6 +27,8 @@ python3 -m pip install .
 
 if [ "$FORCE_GENERATE_ALL" == "true" ]; then
     python3 docpipeline/__main__.py build-all-docs $SOURCE_BUCKET
+elif [ -n "$LANGUAGE" ]; then
+    python3 docpipeline/__main__.py build-language-docs $SOURCE_BUCKET $LANGUAGE
 elif [ -n "$SOURCE_BLOB" ]; then
     python3 docpipeline/__main__.py build-one-doc $SOURCE_BUCKET $SOURCE_BLOB
 else