1 files changed, 140 insertions, 0 deletions
diff --git a/src/pybind/mgr/dashboard/plugins/feature_toggles.py b/src/pybind/mgr/dashboard/plugins/feature_toggles.py
new file mode 100644
index 00000000..7edbfa8f
--- /dev/null
+++ b/src/pybind/mgr/dashboard/plugins/feature_toggles.py
@@ -0,0 +1,140 @@
+# -*- coding: utf-8 -*-
+from __future__ import absolute_import
+
+from enum import Enum
+import cherrypy
+from mgr_module import CLICommand, Option
+
+from . import PLUGIN_MANAGER as PM
+from . import interfaces as I
+from .ttl_cache import ttl_cache
+
+from ..controllers.rbd import Rbd, RbdSnapshot, RbdTrash
+from ..controllers.rbd_mirroring import (
+    RbdMirroringSummary, RbdMirroringPoolMode, RbdMirroringPoolPeer)
+from ..controllers.iscsi import Iscsi, IscsiTarget
+from ..controllers.cephfs import CephFS
+from ..controllers.rgw import Rgw, RgwDaemon, RgwBucket, RgwUser
+
+
+class Features(Enum):
+    RBD = 'rbd'
+    MIRRORING = 'mirroring'
+    ISCSI = 'iscsi'
+    CEPHFS = 'cephfs'
+    RGW = 'rgw'
+
+
+PREDISABLED_FEATURES = set()
+
+
+Feature2Controller = {
+    Features.RBD: [Rbd, RbdSnapshot, RbdTrash],
+    Features.MIRRORING: [
+        RbdMirroringSummary, RbdMirroringPoolMode, RbdMirroringPoolPeer],
+    Features.ISCSI: [Iscsi, IscsiTarget],
+    Features.CEPHFS: [CephFS],
+    Features.RGW: [Rgw, RgwDaemon, RgwBucket, RgwUser],
+}
+
+
+class Actions(Enum):
+    ENABLE = 'enable'
+    DISABLE = 'disable'
+    STATUS = 'status'
+
+
+@PM.add_plugin
+class FeatureToggles(I.CanMgr, I.CanLog, I.Setupable, I.HasOptions,
+                     I.HasCommands, I.FilterRequest.BeforeHandler,
+                     I.HasControllers):
+    OPTION_FMT = 'FEATURE_TOGGLE_{}'
+    CACHE_MAX_SIZE = 128  # Optimum performance with 2^N sizes
+    CACHE_TTL = 10  # seconds
+
+    @PM.add_hook
+    def setup(self):
+        self.Controller2Feature = {
+            controller: feature
+            for feature, controllers in Feature2Controller.items()
+            for controller in controllers}
+
+    @PM.add_hook
+    def get_options(self):
+        return [Option(
+            name=self.OPTION_FMT.format(feature.value),
+            default=(feature not in PREDISABLED_FEATURES),
+            type='bool',) for feature in Features]
+
+    @PM.add_hook
+    def register_commands(self):
+        @CLICommand(
+            "dashboard feature",
+            "name=action,type=CephChoices,strings={} ".format(
+                "|".join(a.value for a in Actions))
+            + "name=features,type=CephChoices,strings={},req=false,n=N".format(
+                "|".join(f.value for f in Features)),
+            "Enable or disable features in Ceph-Mgr Dashboard")
+        def cmd(mgr, action, features=None):
+            ret = 0
+            msg = []
+            if action in [Actions.ENABLE.value, Actions.DISABLE.value]:
+                if features is None:
+                    ret = 1
+                    msg = ["At least one feature must be specified"]
+                else:
+                    for feature in features:
+                        mgr.set_module_option(
+                            self.OPTION_FMT.format(feature),
+                            action == Actions.ENABLE.value)
+                        msg += ["Feature '{}': {}".format(
+                            feature,
+                            'enabled' if action == Actions.ENABLE.value else
+                            'disabled')]
+            else:
+                for feature in features or [f.value for f in Features]:
+                    enabled = mgr.get_module_option(self.OPTION_FMT.format(feature))
+                    msg += ["Feature '{}': '{}'".format(
+                        feature,
+                        'enabled' if enabled else 'disabled')]
+            return ret, '\n'.join(msg), ''
+        return {'handle_command': cmd}
+
+    def _get_feature_from_request(self, request):
+        try:
+            return self.Controller2Feature[
+                request.handler.callable.__self__]
+        except (AttributeError, KeyError):
+            return None
+
+    @ttl_cache(ttl=CACHE_TTL, maxsize=CACHE_MAX_SIZE)
+    def _is_feature_enabled(self, feature):
+        return self.mgr.get_module_option(self.OPTION_FMT.format(feature.value))
+
+    @PM.add_hook
+    def filter_request_before_handler(self, request):
+        feature = self._get_feature_from_request(request)
+        if feature is None:
+            return
+
+        if not self._is_feature_enabled(feature):
+            raise cherrypy.HTTPError(
+                404, "Feature='{}' disabled by option '{}'".format(
+                    feature.value,
+                    self.OPTION_FMT.format(feature.value),
+                    )
+                )
+
+    @PM.add_hook
+    def get_controllers(self):
+        from ..controllers import ApiController, RESTController
+
+        @ApiController('/feature_toggles')
+        class FeatureTogglesEndpoint(RESTController):
+
+            def list(_):
+                return {
+                    feature.value: self._is_feature_enabled(feature)
+                    for feature in Features
+                }
+        return [FeatureTogglesEndpoint]