🚧 Adding storage pool support

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

Author: mhkarimi1383

rawfile/config/model.py

@@ -1,3 +1,4 @@
+import re
 import warnings
 from pydantic_settings import (
     BaseSettings,
@@ -18,11 +19,37 @@
     model_validator,
 )
 from pydantic.networks import IPvAnyAddress
-from typing import Annotated, Literal
+from typing import Annotated, Final, Literal
 import consts
 from utils.logs import LoggingFormats
 from datetime import timedelta
 
+NAME_REGEX: Final[re.Pattern] = re.compile(
+    r"^(?=.{1,253}$)(?!.*\.\.)([a-zA-Z0-9]([a-zA-Z0-9\-]{0,61}[a-zA-Z0-9])?)(\.([a-zA-Z0-9]([a-zA-Z0-9\-]{0,61}[a-zA-Z0-9])?))*$"
+)
+
+
+class StoragePool(BaseModel):
+    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",
+    )
+
 
 class CSIDriverCmd(BaseModel):
     endpoint: (
@@ -33,6 +60,14 @@ class CSIDriverCmd(BaseModel):
     ) = Field(
         description="Listen address for gRPC server",
     )
+    storage_pools: dict[str, StoragePool] | None = Field(
+        description="List of storage pools (Map of name to configuration), required when running node plugin",
+        default=None,
+    )
+    default_pool: str | None = Field(
+        description="Name of the default storage pool, used when no storage pool is defined in storage class",
+        default=None,
+    )
     internal_ip: IPvAnyAddress | None = Field(
         description="Listen ip for gRPC server (used for internal communication only)",
         default=None,
@@ -77,13 +112,31 @@ 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 or len(self.storage_pools) == 0:
+                raise ValueError(
+                    "Storage Pool list is required when running node plugin"
+                )
+
+            paths = []
+            for name, pool in self.storage_pools.items():
+                if len(name) > 63 or len(name) < 3:
+                    raise ValueError(
+                        "Name of the storage pool should be between 3 and 63 characters"
+                    )
+                if not NAME_REGEX.match(name):
+                    raise ValueError(
+                        "Name of the storage pool should be DNS compatible"
+                    )
+                if pool.path in paths:
+                    raise ValueError("Duplicate path in storage pool is not supported")
+                paths.append(pool.path)
         return self
 
 
@@ -110,24 +163,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",
     )