CG Demo: enabling/disabling certain functionality

chaos_genius/app.py

@@ -4,11 +4,14 @@
 import sys
 import json
 
-from flask import Flask, render_template
+from flask import Flask, render_template, request
+from flask.json import jsonify
 from flask_cors import CORS
 
 from chaos_genius import commands
 from chaos_genius.logger import configure_logger
+from chaos_genius.settings import REACT_APP_IS_DEMO
+from chaos_genius.utils.utils import DEMO_ENDPOINT_WHITELIST
 from chaos_genius.views import (
     data_source_view,
     download_view,
@@ -47,6 +50,7 @@ def create_app(config_object="chaos_genius.settings"):
     app.config.from_object(config_object)
     register_extensions(app)
     register_blueprints(app)
+    register_hooks(app)
     register_errorhandlers(app)
     register_shellcontext(app)
     register_commands(app)
@@ -125,6 +129,32 @@ def shell_context():
     app.shell_context_processor(shell_context)
 
 
+def register_hooks(app):
+    """Register hooks for Flask requests."""
+
+    @app.before_request
+    def demo_endpoints_blocker():
+        """Hook to allow access to only whitelisted APIs in demo."""
+        if REACT_APP_IS_DEMO:
+            if request.endpoint not in DEMO_ENDPOINT_WHITELIST:
+                return jsonify(
+                    {
+                        "status": "failure",
+                        "message": "Endpoint not accessible in demo version",
+                    }
+                )
+            else:
+                if request.method not in DEMO_ENDPOINT_WHITELIST[request.endpoint]:
+                    return jsonify(
+                        {
+                            "status": "failure",
+                            "message": f"{request.method} method is not allowed for this endpoint in demo version",
+                        }
+                    )
+                else:
+                    pass
+
+
 def register_commands(app):
     """Register Click commands."""
     app.cli.add_command(commands.test)

chaos_genius/settings.py

@@ -169,6 +169,7 @@ def _make_bool(val: Union[str, bool]) -> bool:
     "CHAOSGENIUS_ENTERPRISE_EDITION_KEY"
 )
 
+REACT_APP_IS_DEMO = _make_bool(os.getenv("REACT_APP_IS_DEMO", default=True))
 """Dynamic Third Party Data Sources"""
 SOURCE_GOOGLE_ANALYTICS = _make_bool(
     os.getenv("SOURCE_GOOGLE_ANALYTICS", default=True)

chaos_genius/utils/utils.py

@@ -96,3 +96,72 @@ def make_path_safe(s: str) -> str:
     Does not replace spaces.
     """
     return re.sub(_invalid_chars_reg, "_", s)
+
+
+DEMO_ENDPOINT_WHITELIST = {
+    "alert.alert_meta_info": ["GET"],
+    "alert.get_alert_info_by_id": ["GET"],
+    "alert.get_used_channels": ["GET"],
+    "alert.get_used_statuses": ["GET"],
+    "alert.disable_alert": ["GET"],
+    "alert.enable_alert": ["GET"],
+    "alert.list_alert": ["GET"],
+    "anomaly_data.anomaly_settings_status": ["GET"],
+    "anomaly_data.kpi_anomaly_data_quality": ["GET"],
+    "anomaly_data.kpi_anomaly_detection": ["GET"],
+    "anomaly_data.kpi_anomaly_drilldown": ["GET"],
+    "anomaly_data.kpi_anomaly_params": ["GET"],
+    "anomaly_data.kpi_anomaly_params_meta": ["GET"],
+    "anomaly_data.kpi_subdim_anomaly": ["GET"],
+    "anomaly_data.list_anomaly_data": ["GET"],
+    "api_data_source.data_source": ["GET"],
+    "api_data_source.data_source_meta_info": ["GET"],
+    "api_data_source.get_data_source_info": ["GET"],
+    "api_data_source.list_data_source_type": ["GET"],
+    # "api_data_source.log_data_source",
+    "api_data_source.metadata_data_source": ["POST"],
+    "api_data_source.test_data_source_connection": ["GET"],
+    "api_data_source.check_views_availability": ["POST"],
+    "api_data_source.get_schema_list": ["POST"],
+    "api_data_source.get_schema_tables": ["POST"],
+    "api_data_source.get_table_info": ["POST"],
+    "api_data_source.used_data_source_types_endpoint": ["GET"],
+    "api_kpi.kpi": ["GET"],
+    "api_kpi.get_all_kpis": ["GET"],
+    "api_kpi.get_kpi_info": ["GET"],
+    "api_kpi.kpi_get_dimensions": ["GET"],
+    "api_kpi.get_timecuts_list": ["GET"],
+    "api_kpi.kpi_meta_info": ["GET"],
+    "api_rca.kpi_get_aggregation": ["GET"],
+    "api_rca.kpi_get_line_data": ["GET"],
+    "api_rca.kpi_rca_analysis": ["GET"],
+    "api_rca.kpi_rca_hierarchical_data": ["GET"],
+    "api_summary.kpi_get_aggregation": ["GET"],
+    "api_summary.kpi_get_line_data": ["GET"],
+    "config_settings.get_all_config": ["GET"],
+    "config_settings.get_config": ["POST"],
+    "config_settings.get_config_meta_data": ["GET"],
+    "config_settings.get_onboarding_status": ["GET"],
+    "config_settings.global_config": ["GET"],
+    # "config_settings.global_settings",
+    "config_settings.multidim_status": ["GET"],
+    "dashboard.get_all_dashboard_names": ["GET"],
+    "dashboard.get_dashboard_list": ["GET"],
+    "dashboard.get_dashboard": ["GET"],
+    "digest.task_monitor_view": ["GET"],
+    "digest.static": ["GET"],
+    "status.task_monitor_view": ["GET"],
+    "downloads.download_anomaly_data": ["GET"],
+    "downloads.download_hierarchical_data": ["GET"],
+    "downloads.download_multidim_analysis_data": ["GET"],
+    "downloads.kpi_download_line_data": ["GET"],
+    # "meta.static",
+    # "public.api_view",
+    # "public.home",
+    # "public.static",
+    # "static",
+    # "status.static",
+    # "status.task_monitor_view",
+    "meta.support_timezones": ["GET"],
+    "meta.version_view": ["GET"],
+}

docker-compose.dev.yml

@@ -27,6 +27,7 @@ services:
       - *chaosgenius-version
       - IN_DOCKER=True
       - AIRBYTE_ENABLED=${AIRBYTE_ENABLED}
+      - REACT_APP_IS_DEMO=${REACT_APP_IS_DEMO}
       - FLASK_APP=${FLASK_APP}
       - FLASK_ENV=${FLASK_ENV}
       - FLASK_DEBUG=${FLASK_DEBUG}
@@ -89,6 +90,7 @@ services:
       - REACT_APP_BASE_URL=${REACT_APP_BASE_URL}
       - REACT_APP_DISABLE_TELEMETRY=${REACT_APP_DISABLE_TELEMETRY}
       - REACT_APP_EVENT_ALERT=${REACT_APP_EVENT_ALERT}
+      - REACT_APP_IS_DEMO=${REACT_APP_IS_DEMO}
     depends_on:
       - chaosgenius-server
 

docker-compose.fluentd.yml

@@ -37,6 +37,7 @@ services:
       - *chaosgenius-version
       - IN_DOCKER=True
       - AIRBYTE_ENABLED=${AIRBYTE_ENABLED}
+      - REACT_APP_IS_DEMO=${REACT_APP_IS_DEMO}
       - FLASK_APP=${FLASK_APP}
       - FLASK_ENV=${FLASK_ENV}
       - FLASK_DEBUG=${FLASK_DEBUG}
@@ -98,6 +99,7 @@ services:
       - REACT_APP_BASE_URL=${REACT_APP_BASE_URL}
       - REACT_APP_DISABLE_TELEMETRY=${REACT_APP_DISABLE_TELEMETRY}
       - REACT_APP_EVENT_ALERT=${REACT_APP_EVENT_ALERT}
+      - REACT_APP_IS_DEMO=${REACT_APP_IS_DEMO}
     depends_on:
       - chaosgenius-server
       - chaosgenius-fluentd

docker-compose.latest-thirdparty.yml

@@ -50,6 +50,7 @@ services:
       - *chaosgenius-version
       - IN_DOCKER=True
       - AIRBYTE_ENABLED=True
+      - REACT_APP_IS_DEMO=${REACT_APP_IS_DEMO}
       - FLASK_APP=${FLASK_APP}
       - FLASK_ENV=${FLASK_ENV}
       - FLASK_DEBUG=${FLASK_DEBUG}
@@ -110,6 +111,7 @@ services:
       - REACT_APP_BASE_URL=${REACT_APP_BASE_URL}
       - REACT_APP_DISABLE_TELEMETRY=${REACT_APP_DISABLE_TELEMETRY}
       - REACT_APP_EVENT_ALERT=${REACT_APP_EVENT_ALERT}
+      - REACT_APP_IS_DEMO=${REACT_APP_IS_DEMO}
     depends_on:
       - chaosgenius-server
 

docker-compose.latest.yml

@@ -23,6 +23,7 @@ services:
       - *chaosgenius-version
       - IN_DOCKER=True
       - AIRBYTE_ENABLED=${AIRBYTE_ENABLED}
+      - REACT_APP_IS_DEMO=${REACT_APP_IS_DEMO}
       - FLASK_APP=${FLASK_APP}
       - FLASK_ENV=${FLASK_ENV}
       - FLASK_DEBUG=${FLASK_DEBUG}
@@ -83,6 +84,7 @@ services:
       - REACT_APP_BASE_URL=${REACT_APP_BASE_URL}
       - REACT_APP_DISABLE_TELEMETRY=${REACT_APP_DISABLE_TELEMETRY}
       - REACT_APP_EVENT_ALERT=${REACT_APP_EVENT_ALERT}
+      - REACT_APP_IS_DEMO=${REACT_APP_IS_DEMO}
     depends_on:
       - chaosgenius-server
 

docker-compose.thirdparty.yml

@@ -50,6 +50,7 @@ services:
       - *chaosgenius-version
       - IN_DOCKER=True
       - AIRBYTE_ENABLED=True
+      - REACT_APP_IS_DEMO=${REACT_APP_IS_DEMO}
       - FLASK_APP=${FLASK_APP}
       - FLASK_ENV=${FLASK_ENV}
       - FLASK_DEBUG=${FLASK_DEBUG}
@@ -110,6 +111,7 @@ services:
       - REACT_APP_BASE_URL=${REACT_APP_BASE_URL}
       - REACT_APP_DISABLE_TELEMETRY=${REACT_APP_DISABLE_TELEMETRY}
       - REACT_APP_EVENT_ALERT=${REACT_APP_EVENT_ALERT}
+      - REACT_APP_IS_DEMO=${REACT_APP_IS_DEMO}
     depends_on:
       - chaosgenius-server
 

docker-compose.yml

@@ -23,6 +23,7 @@ services:
       - *chaosgenius-version
       - IN_DOCKER=True
       - AIRBYTE_ENABLED=${AIRBYTE_ENABLED}
+      - REACT_APP_IS_DEMO=${REACT_APP_IS_DEMO}
       - FLASK_APP=${FLASK_APP}
       - FLASK_ENV=${FLASK_ENV}
       - FLASK_DEBUG=${FLASK_DEBUG}
@@ -83,6 +84,7 @@ services:
       - REACT_APP_BASE_URL=${REACT_APP_BASE_URL}
       - REACT_APP_DISABLE_TELEMETRY=${REACT_APP_DISABLE_TELEMETRY}
       - REACT_APP_EVENT_ALERT=${REACT_APP_EVENT_ALERT}
+      - REACT_APP_IS_DEMO=${REACT_APP_IS_DEMO}
     depends_on:
       - chaosgenius-server
 

frontend/public/index.html

@@ -2,6 +2,19 @@
 <html lang="en">
   <head>
     <meta charset="utf-8" />
+    <!-- Global site tag (gtag.js) - Google Analytics -->
+    <script
+      async
+      src="https://www.googletagmanager.com/gtag/js?id=G-ZTDCWNG9GQ"
+    ></script>
+    <script>
+      window.dataLayer = window.dataLayer || [];
+      function gtag() {
+        dataLayer.push(arguments);
+      }
+      gtag('js', new Date());
+      gtag('config', 'G-ZTDCWNG9GQ');
+    </script>
     <script src="%PUBLIC_URL%/env.js"></script>
     <link rel="icon" href="%PUBLIC_URL%/favicon.ico" />
     <meta name="viewport" content="width=device-width, initial-scale=1" />

frontend/src/components/AlertTable/index.jsx

@@ -41,7 +41,8 @@ const AlertTable = ({
   alertSearch,
   setPgInfo,
   pagination,
-  pgInfo
+  pgInfo,
+  customToast
 }) => {
   const dispatch = useDispatch();
 
@@ -59,7 +60,7 @@ const AlertTable = ({
 
   const onDeleteConfirmation = (id) => {
     store.dispatch(RESET_DELETE_DATA);
-    dispatch(kpiAlertDeleteById(id));
+    dispatch(kpiAlertDeleteById(id, customToast));
     setIsOpen(false);
   };
 

frontend/src/components/AlertsForm/index.jsx

@@ -394,7 +394,7 @@ const AlertsForm = () => {
   };
 
   const dispatchGetAllAlertEmail = (data) => {
-    dispatch(getAllAlertEmail(data));
+    dispatch(getAllAlertEmail(data, customToast));
   };
 
   const editableStatus = (type) => {

frontend/src/components/Analystics/index.js

@@ -331,7 +331,7 @@ const Analystics = ({ kpi, setAnalystics, onboarding }) => {
       if (edit && Object.keys(needForCleanup)?.length) {
         setModalOpen(true);
       } else {
-        dispatch(kpiSettingSetup(kpi, data));
+        dispatch(kpiSettingSetup(kpi, data, customToast));
       }
     }
   };
@@ -913,7 +913,7 @@ const Analystics = ({ kpi, setAnalystics, onboarding }) => {
                     const data = {
                       anomaly_params: { ...editForm }
                     };
-                    dispatch(kpiSettingSetup(kpi, data));
+                    dispatch(kpiSettingSetup(kpi, data, customToast));
                     setModalOpen(false);
                   }}>
                   <span>Save Changes</span>