Part: allow to change label depending on purpose

1 files changed, 10 insertions, 0 deletions

diff --git a/lib/ManaTools/Shared/disk_backend/Plugin/Disk.pm b/lib/ManaTools/Shared/disk_backend/Plugin/Disk.pm
index 6f58d1d7..75bcd1f9 100644
--- a/lib/ManaTools/Shared/disk_backend/Plugin/Disk.pm
+++ b/lib/ManaTools/Shared/disk_backend/Plugin/Disk.pm
@@ -151,6 +151,7 @@ extends 'ManaTools::Shared::disk_backend::Part';
 use MooseX::ClassAttribute;
 
 with 'ManaTools::Shared::disk_backend::BlockDevice';
+with 'ManaTools::Shared::disk_backend::PurposeLabelRole';
 
 class_has '+type' => (
     default => 'Disk'
@@ -207,4 +208,13 @@ class_has '+restrictions' => (
     }
 );
 
+override('label', sub {
+    my $self = shift;
+    my $label = super;
+    return $label if $label ne $self->type();
+
+    # get the name from the devicepath
+    return $self->devicepath() =~ s'^.+/''r;
+});
+
 1;