api: disks: delete: add flag for cleaning up storage config

[pve-storage.git] / PVE / API2 / Storage / Config.pm

diff --git a/PVE/API2/Storage/Config.pm b/PVE/API2/Storage/Config.pm
index bf38df3f1926e0c98372b71d55a1710745f21e36..6bd770edcff39d6aa4c779467ad65f1188b8cf54 100755 (executable)
--- a/PVE/API2/Storage/Config.pm
+++ b/PVE/API2/Storage/Config.pm
@@ -38,6 +38,33 @@ my $api_storage_config = sub {
     return $scfg;
 };
 
+# For storages that $match->($scfg), update node restrictions to not include $node anymore and
+# in case no node remains, remove the storage altogether.
+sub cleanup_storages_for_node {
+    my ($self, $match, $node) = @_;
+
+    my $config = PVE::Storage::config();
+    my $cluster_nodes = PVE::Cluster::get_nodelist();
+
+    for my $storeid (keys $config->{ids}->%*) {
+       my $scfg = PVE::Storage::storage_config($config, $storeid);
+       next if !$match->($scfg);
+
+       my $nodes = $scfg->{nodes} || { map { $_ => 1 } $cluster_nodes->@* };
+       next if !$nodes->{$node}; # not configured on $node, so nothing to do
+       delete $nodes->{$node};
+
+       if (scalar(keys $nodes->%*) > 0) {
+           $self->update({
+               nodes => join(',', sort keys $nodes->%*),
+               storage => $storeid,
+           });
+       } else {
+           $self->delete({storage => $storeid});
+       }
+    }
+}
+
 __PACKAGE__->register_method ({
     name => 'index',
     path => '',