🚧 Adding storage pool support

Signed-off-by: Muhammed Hussain Karimi <[email protected]>

Author: mhkarimi1383

rawfile/config/model.py

@@ -24,6 +24,34 @@
 from datetime import timedelta
 
 
+class StoragePool(BaseModel):
+    name: Annotated[
+        str, StringConstraints(strip_whitespace=True, min_length=3, max_length=63)
+    ] = Field(description="Name of the pool, Should be unique")
+    path: DirectoryPath = Field(description="Path of the pool, Should be unique")
+    reserved_capacity: (
+        ByteSize
+        | Annotated[
+            str,
+            StringConstraints(strip_whitespace=True, pattern=r"^\d+%$"),
+        ]
+    ) = Field(
+        default=ByteSize(0),
+        description="Reserves capacity of storage pool",
+    )
+    capacity_override: ByteSize | None = Field(
+        default=None,
+        description="Overrides total capacity of storage pool",
+    )
+    default_fs: FileSystemName = Field(
+        default=FileSystemName.EXT4,
+        description="Default filesystem used where creating volumes and fsType is not specified in storage class parameters",
+    )
+    is_default: bool = Field(
+        description="Makes this storage pool to be default, used when no storage pool name has been specified in storage class parameters"
+    )
+
+
 class CSIDriverCmd(BaseModel):
     endpoint: (
         AnyUrl
@@ -33,6 +61,10 @@ class CSIDriverCmd(BaseModel):
     ) = Field(
         description="Listen address for gRPC server",
     )
+    storage_pools: list[StoragePool] | None = Field(
+        description="List of storage pools, required when running node plugin",
+        default=None,
+    )
     internal_ip: IPvAnyAddress | None = Field(
         description="Listen ip for gRPC server (used for internal communication only)",
         default=None,
@@ -77,13 +109,30 @@ class CSIDriverCmd(BaseModel):
     def validate_node_plugin(
         self,
     ):
-        if self.plugin_type == "node":
+        if self.plugin_type != "node":
             if not self.internal_ip:
                 raise ValueError(
                     "Internal Communication IP/PORT is required on node plugin"
                 )
             if not self.metadata_dir:
                 raise ValueError("Metadata Dir is required when running node plugin")
+            if not self.storage_pools:
+                raise ValueError(
+                    "Storage Pool list is required when running node plugin"
+                )
+
+            names = []
+            paths = []
+            has_default = False
+            for pool in self.storage_pools:
+                if pool.name in names:
+                    raise ValueError("Duplicate name in storage pool is not supported")
+                names.append(pool.name)
+                if pool.path in paths:
+                    raise ValueError("Duplicate path in storage pool is not supported")
+                paths.append(pool.path)
+                if pool.is_default and has_default:
+                    raise ValueError("Only one default pool is allowed")
         return self
 
 
@@ -110,24 +159,6 @@ def settings_customise_sources(
     ) -> tuple[PydanticBaseSettingsSource, ...]:
         return CliSettingsSource(settings_cls, cli_parse_args=True), env_settings
 
-    reserved_capacity: (
-        ByteSize
-        | Annotated[
-            str,
-            StringConstraints(strip_whitespace=True, pattern=r"^\d+%$"),
-        ]
-    ) = Field(
-        default=ByteSize(0),
-        description="Reserves capacity of data dir",
-    )
-    capacity_override: ByteSize | None = Field(
-        default=None,
-        description="Overrides total capacity of data dir",
-    )
-    default_fs: FileSystemName = Field(
-        default=FileSystemName.EXT4,
-        description="Default filesystem used where creating volumes and fsType is not specified in storage class parameters",
-    )
     namespace: str = Field(
         description="K8s Namespace of the driver",
     )