Part: allow to change label depending on purpose

1 files changed, 12 insertions, 0 deletions

diff --git a/lib/ManaTools/Shared/disk_backend/Plugin/Mount.pm b/lib/ManaTools/Shared/disk_backend/Plugin/Mount.pm
index 635fc505..1ff2b5d1 100644
--- a/lib/ManaTools/Shared/disk_backend/Plugin/Mount.pm
+++ b/lib/ManaTools/Shared/disk_backend/Plugin/Mount.pm
@@ -463,6 +463,8 @@ use Moose;
 
 extends 'ManaTools::Shared::disk_backend::Part';
 
+with 'ManaTools::Shared::disk_backend::PurposeLabelRole';
+
 use MooseX::ClassAttribute;
 
 class_has '+type' => (
@@ -526,6 +528,14 @@ override('_reverse_tag', sub {
     return super;
 });
 
+around('purpose_label', sub {
+    my $orig = shift;
+    my $self = shift;
+    my $partstate = shift;
+
+    return $self->path()
+});
+
 package ManaTools::Shared::disk_backend::Part::UnknownBlockDevice;
 
 use Moose;
@@ -535,6 +545,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 => 'UnknownBlockDevice'
@@ -567,6 +578,7 @@ use MooseX::ClassAttribute;
 
 with 'ManaTools::Shared::disk_backend::FileSystem';
 with 'ManaTools::Shared::disk_backend::Mountable';
+with 'ManaTools::Shared::disk_backend::PurposeLabelRole';
 
 class_has '+type' => (
     default => 'UnknownFS'