1 files changed, 65 insertions, 0 deletions
diff --git a/src/pybind/mgr/dashboard/controllers/crush_rule.py b/src/pybind/mgr/dashboard/controllers/crush_rule.py
new file mode 100644
index 000000000..0310ab1c3
--- /dev/null
+++ b/src/pybind/mgr/dashboard/controllers/crush_rule.py
@@ -0,0 +1,65 @@
+
+# -*- coding: utf-8 -*-
+from __future__ import absolute_import
+
+from cherrypy import NotFound
+
+from .. import mgr
+from ..security import Scope
+from ..services.ceph_service import CephService
+from . import APIDoc, APIRouter, Endpoint, EndpointDoc, ReadPermission, RESTController, UIRouter
+
+LIST_SCHEMA = {
+    "rule_id": (int, 'Rule ID'),
+    "rule_name": (str, 'Rule Name'),
+    "ruleset": (int, 'RuleSet related to the rule'),
+    "type": (int, 'Type of Rule'),
+    "min_size": (int, 'Minimum size of Rule'),
+    "max_size": (int, 'Maximum size of Rule'),
+    'steps': ([{str}], 'Steps included in the rule')
+}
+
+
+@APIRouter('/crush_rule', Scope.POOL)
+@APIDoc("Crush Rule Management API", "CrushRule")
+class CrushRule(RESTController):
+    @EndpointDoc("List Crush Rule Configuration",
+                 responses={200: LIST_SCHEMA})
+    def list(self):
+        return mgr.get('osd_map_crush')['rules']
+
+    def get(self, name):
+        rules = mgr.get('osd_map_crush')['rules']
+        for r in rules:
+            if r['rule_name'] == name:
+                return r
+        raise NotFound('No such crush rule')
+
+    def create(self, name, root, failure_domain, device_class=None):
+        rule = {
+            'name': name,
+            'root': root,
+            'type': failure_domain,
+            'class': device_class
+        }
+        CephService.send_command('mon', 'osd crush rule create-replicated', **rule)
+
+    def delete(self, name):
+        CephService.send_command('mon', 'osd crush rule rm', name=name)
+
+
+@UIRouter('/crush_rule', Scope.POOL)
+@APIDoc("Dashboard UI helper function; not part of the public API", "CrushRuleUi")
+class CrushRuleUi(CrushRule):
+    @Endpoint()
+    @ReadPermission
+    def info(self):
+        '''Used for crush rule creation modal'''
+        osd_map = mgr.get_osdmap()
+        crush = osd_map.get_crush()
+        crush.dump()
+        return {
+            'names': [r['rule_name'] for r in mgr.get('osd_map_crush')['rules']],
+            'nodes': mgr.get('osd_map_tree')['nodes'],
+            'roots': crush.find_roots()
+        }