GoogleCloudPlatform · mctavish · Jul 26, 2022 · Jul 5, 2022 · Jul 6, 2022 · Jul 6, 2022
diff --git a/firebase-sample/app.py b/firebase-sample/app.py
@@ -0,0 +1,11 @@
+import googleclouddebugger
+googleclouddebugger.enable(use_firebase= True)
+
+from flask import Flask
+
+app = Flask(__name__)
+
+@app.route("/")
+def hello_world():
+  return "<p>Hello World!</p>"
+
diff --git a/firebase-sample/build-and-run.sh b/firebase-sample/build-and-run.sh
@@ -0,0 +1,20 @@
+#!/bin/bash -e
+
+SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
+cd "${SCRIPT_DIR}/.."
+
+cd src
+./build.sh
+cd ..
+
+python3 -m venv /tmp/cdbg-venv
+source /tmp/cdbg-venv/bin/activate
+pip3 install -r requirements.txt
+pip3 install src/dist/* --force-reinstall
+
+cd firebase-sample
+pip3 install -r requirements.txt
+python3 -m flask run
+cd ..
+
+deactivate
diff --git a/firebase-sample/requirements.txt b/firebase-sample/requirements.txt
@@ -0,0 +1 @@
+flask
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,5 @@
 google-auth-httplib2
 google-api-python-client
 google-api-core
+firebase_admin
 pyyaml
diff --git a/requirements_dev.txt b/requirements_dev.txt
@@ -1,3 +1,4 @@
 -r requirements.txt
 absl-py
 pytest
+requests-mock
diff --git a/src/googleclouddebugger/__init__.py b/src/googleclouddebugger/__init__.py
@@ -30,6 +30,7 @@
 from . import collector
 from . import error_data_visibility_policy
 from . import gcp_hub_client
+from . import firebase_client
 from . import glob_data_visibility_policy
 from . import yaml_data_visibility_config_reader
 from . import cdbg_native
@@ -38,42 +39,51 @@
 __version__ = version.__version__
 
 _flags = None
-_hub_client = None
+_backend_client = None
 _breakpoints_manager = None
 
 
 def _StartDebugger():
   """Configures and starts the debugger."""
-  global _hub_client
+  global _backend_client
   global _breakpoints_manager
 
   cdbg_native.InitializeModule(_flags)
   cdbg_native.LogInfo(
       f'Initializing Cloud Debugger Python agent version: {__version__}')
 
-  _hub_client = gcp_hub_client.GcpHubClient()
+  use_firebase = _flags.get('use_firebase')
+  if use_firebase:
+    _backend_client = firebase_client.FirebaseClient()
+    _backend_client.SetupAuth(
+        _flags.get('project_id'), _flags.get('service_account_json_file'),
+        _flags.get('firebase_db_url'))
+  else:
+    _backend_client = gcp_hub_client.GcpHubClient()
+    _backend_client.SetupAuth(
+        _flags.get('project_id'), _flags.get('project_number'),
+        _flags.get('service_account_json_file'))
+    _backend_client.SetupCanaryMode(
+        _flags.get('breakpoint_enable_canary'),
+        _flags.get('breakpoint_allow_canary_override'))
+
   visibility_policy = _GetVisibilityPolicy()
 
   _breakpoints_manager = breakpoints_manager.BreakpointsManager(
-      _hub_client, visibility_policy)
+      _backend_client, visibility_policy)
 
   # Set up loggers for logpoints.
   collector.SetLogger(logging.getLogger())
 
   collector.CaptureCollector.pretty_printers.append(
       appengine_pretty_printers.PrettyPrinter)
 
-  _hub_client.on_active_breakpoints_changed = (
+  _backend_client.on_active_breakpoints_changed = (
       _breakpoints_manager.SetActiveBreakpoints)
-  _hub_client.on_idle = _breakpoints_manager.CheckBreakpointsExpiration
-  _hub_client.SetupAuth(
-      _flags.get('project_id'), _flags.get('project_number'),
-      _flags.get('service_account_json_file'))
-  _hub_client.SetupCanaryMode(
-      _flags.get('breakpoint_enable_canary'),
-      _flags.get('breakpoint_allow_canary_override'))
-  _hub_client.InitializeDebuggeeLabels(_flags)
-  _hub_client.Start()
+  _backend_client.on_idle = _breakpoints_manager.CheckBreakpointsExpiration
+
+  _backend_client.InitializeDebuggeeLabels(_flags)
+  _backend_client.Start()
 
 
 def _GetVisibilityPolicy():