Fix availability zones and topology configuration

config.yaml

@@ -37,3 +37,9 @@ options:
     description: |
       Availability zone to use with Cinder CSI. This is passed through to the
       parameters.availability field of the csi-cinder-default StorageClass.
+
+  topology:
+    type: boolean
+    default: True
+    description: |
+      Whether to enable the Cinder CSI topology awareness

src/storage_manifests.py

@@ -66,9 +66,8 @@ def __call__(self) -> Optional[AnyResource]:
                 volumeBindingMode="WaitForFirstConsumer",
             )
         )
-
         if az := self.manifests.config.get("availability-zone"):
-            sc.parameters.availability = az
+            sc.parameters = dict(availability=az)
         return sc
 
 
@@ -77,12 +76,7 @@ class UpdateSecrets(Patch):
 
     def __call__(self, obj):
         """Update the secret volume spec in daemonsets and deployments."""
-        if not any(
-            [
-                (obj.kind == "DaemonSet" and obj.metadata.name == "csi-cinder-nodeplugin"),
-                (obj.kind == "Deployment" and obj.metadata.name == "csi-cinder-controllerplugin"),
-            ]
-        ):
+        if not (obj.kind == "Deployment" and obj.metadata.name == "csi-cinder-controllerplugin"):
             return
 
         for volume in obj.spec.template.spec.volumes:
@@ -105,6 +99,7 @@ def __init__(self, charm, charm_config, kube_control, integrator):
                 ConfigRegistry(self),
                 CreateStorageClass(self, "default"),  # creates csi-cinder-default
                 UpdateSecrets(self),  # update secrets
+                UpdateControllerPlugin(self),
             ],
         )
         self.integrator = integrator
@@ -142,3 +137,24 @@ def evaluate(self) -> Optional[str]:
             if not self.config.get(prop):
                 return f"Storage manifests waiting for definition of {prop}"
         return None
+
+
+class UpdateControllerPlugin(Patch):
+    """Update the controller args in Deployments."""
+
+    def __call__(self, obj):
+        """Update the controller args in Deployments."""
+        if not any(
+            [
+                (obj.kind == "Deployment" and obj.metadata.name == "csi-cinder-controllerplugin"),
+            ]
+        ):
+            return
+
+        for container in obj.spec.template.spec.containers:
+            if container.name == "csi-provisioner":
+                for i, val in enumerate(container.args):
+                    if "feature-gates" in val.lower():
+                        topology = str(self.manifests.config.get("topology")).lower()
+                        container.args[i] = f"feature-gates=Topology={topology}"
+                        log.info("Configuring cinder topology awareness=%s", topology)