From 5c33431ce132ef7fc5db72d0c06cb89c5524c3a1 Mon Sep 17 00:00:00 2001
From: Sage Weil <sage@newdream.net>
Date: Tue, 1 Jun 2021 12:23:24 -0400
Subject: [PATCH] mgr/nfs: abstract FSAL; add RGWFSAL

Signed-off-by: Sage Weil <sage@newdream.net>
---
 src/pybind/mgr/nfs/export_utils.py | 58 ++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)

diff --git a/src/pybind/mgr/nfs/export_utils.py b/src/pybind/mgr/nfs/export_utils.py
index cfbafa39d0bdc..e4842393245ee 100644
--- a/src/pybind/mgr/nfs/export_utils.py
+++ b/src/pybind/mgr/nfs/export_utils.py
@@ -164,12 +164,16 @@ class FSAL(object):
     def from_dict(cls, fsal_dict: Dict[str, Any]) -> 'FSAL':
         if fsal_dict.get('name') == 'CEPH':
             return CephFSFSAL.from_dict(fsal_dict)
+        if fsal_dict.get('name') == 'RGW':
+            return RGWFSAL.from_dict(fsal_dict)
         raise NFSInvalidOperation(f'Unknown FSAL {fsal_dict.get("name")}')
 
     @classmethod
     def from_fsal_block(cls, fsal_block: Dict[str, Any]) -> 'FSAL':
         if fsal_block.get('name') == 'CEPH':
             return CephFSFSAL.from_fsal_block(fsal_block)
+        if fsal_block.get('name') == 'RGW':
+            return RGWFSAL.from_fsal_block(fsal_block)
         raise NFSInvalidOperation(f'Unknown FSAL {fsal_block.get("name")}')
 
     def to_fsal_block(self) -> Dict[str, Any]:
@@ -235,6 +239,57 @@ class CephFSFSAL(FSAL):
         return r
 
 
+class RGWFSAL(FSAL):
+    def __init__(self,
+                 name: str,
+                 user_id: Optional[str] = None,
+                 access_key_id: Optional[str] = None,
+                 secret_access_key: Optional[str] = None
+                 ):
+        super().__init__(name)
+        assert name == 'RGW'
+        self.user_id = user_id
+        self.access_key_id = access_key_id
+        self.secret_access_key = secret_access_key
+
+    @classmethod
+    def from_fsal_block(cls, fsal_block: Dict[str, str]) -> 'RGWFSAL':
+        return cls(fsal_block['name'],
+                   fsal_block.get('user_id'),
+                   fsal_block.get('access_key'),
+                   fsal_block.get('secret_access_key'))
+
+    def to_fsal_block(self) -> Dict[str, str]:
+        result = {
+            'block_name': 'FSAL',
+            'name': self.name,
+        }
+        if self.user_id:
+            result['user_id'] = self.user_id
+        if self.fs_name:
+            result['access_key_id'] = self.access_key_id
+        if self.secret_access_key:
+            result['secret_access_key'] = self.secret_access_key
+        return result
+
+    @classmethod
+    def from_dict(cls, fsal_dict: Dict[str, str]) -> 'RGWFSAL':
+        return cls(fsal_dict['name'],
+                   fsal_dict.get('user_id'),
+                   fsal_dict.get('access_key_id'),
+                   fsal_dict.get('secret_access_key'))
+
+    def to_dict(self) -> Dict[str, str]:
+        r = {'name': self.name}
+        if self.user_id:
+            r['user_id'] = self.user_id
+        if self.access_key_id:
+            r['access_key_id'] = self.access_key_id
+        if self.secret_access_key:
+            r['secret_access_key'] = self.secret_access_key
+        return r
+
+
 class Client:
     def __init__(self,
                  addresses: List[str],
@@ -434,5 +489,8 @@ class Export:
             fs = cast(CephFSFSAL, self.fsal)
             if not check_fs(mgr, fs.fs_name):
                 raise FSNotFound(fs.fs_name)
+        elif self.fsal.name == 'RGW':
+            rgw = cast(RGWFSAL, self.fsal)
+            pass
         else:
             raise NFSInvalidOperation('FSAL {self.fsal.name} not supported')
-- 
2.39.5