Add support for vpcConnector and vpcConnectorEgressSettings in firebase.json

Author: pcboy

src/deploy/functions/release.js

@@ -125,6 +125,9 @@ module.exports = function(context, options, payload) {
     return Promise.resolve();
   }
 
+  var vpcConnector = options.config.get("functions.vpcConnector");
+  var vpcConnectorEgressSettings = options.config.get("functions.vpcConnectorEgressSettings");
+
   var projectId = context.projectId;
   var sourceUrl = context.uploadUrl;
   var appEngineLocation = getAppEngineLocation(context.firebaseConfig);
@@ -217,6 +220,8 @@ module.exports = function(context, options, payload) {
                   timeout: functionInfo.timeout,
                   maxInstances: functionInfo.maxInstances,
                   environmentVariables: defaultEnvVariables,
+                  vpcConnector: vpcConnector,
+                  vpcConnectorEgressSettings: vpcConnectorEgressSettings,
                 })
                 .then((createRes) => {
                   if (_.has(functionTrigger, "httpsTrigger")) {
@@ -297,6 +302,8 @@ module.exports = function(context, options, payload) {
                 existingFunction.environmentVariables,
                 defaultEnvVariables
               ),
+              vpcConnector: vpcConnector,
+              vpcConnectorEgressSettings: vpcConnectorEgressSettings,
             };
             utils.logBullet(
               clc.bold.cyan("functions: ") +

src/gcp/cloudfunctions.js

@@ -65,13 +65,19 @@ function _createFunction(options) {
   const location = "projects/" + options.projectId + "/locations/" + options.region;
   const func = location + "/functions/" + options.functionName;
   const endpoint = "/" + API_VERSION + "/" + location + "/functions";
+
   const data = {
     sourceUploadUrl: options.sourceUploadUrl,
     name: func,
     entryPoint: options.entryPoint,
     labels: options.labels,
     runtime: options.runtime,
   };
+
+  if (options.vpcConnector) {
+    data.vpcConnector = options.vpcConnector;
+    data.vpcConnectorEgressSettings = options.vpcConnectorEgressSettings || "ALL_TRAFFIC";
+  }
   if (options.availableMemoryMb) {
     data.availableMemoryMb = options.availableMemoryMb;
   }
@@ -143,6 +149,7 @@ function _updateFunction(options) {
   const location = "projects/" + options.projectId + "/locations/" + options.region;
   const func = location + "/functions/" + options.functionName;
   const endpoint = "/" + API_VERSION + "/" + func;
+
   const data = _.assign(
     {
       sourceUploadUrl: options.sourceUploadUrl,
@@ -153,6 +160,13 @@ function _updateFunction(options) {
   );
   let masks = ["sourceUploadUrl", "name", "labels"];
 
+  if (options.vpcConnector) {
+    data.vpcConnector = options.vpcConnector;
+    data.vpcConnectorEgressSettings = options.vpcConnectorEgressSettings || "ALL_TRAFFIC";
+    masks.push("vpcConnector");
+    masks.push("vpcConnectorEgressSettings");
+  }
+
   if (options.runtime) {
     data.runtime = options.runtime;
     masks = _.concat(masks, "runtime");