Part: allow to change label depending on purpose

7 files changed, 47 insertions, 0 deletions

diff --git a/lib/ManaTools/Shared/disk_backend/Plugin/Btrfs.pm b/lib/ManaTools/Shared/disk_backend/Plugin/Btrfs.pm
index 0683f669..ecf6dd03 100644
--- a/lib/ManaTools/Shared/disk_backend/Plugin/Btrfs.pm
+++ b/lib/ManaTools/Shared/disk_backend/Plugin/Btrfs.pm
@@ -196,6 +196,8 @@ use Moose;
 
 extends 'ManaTools::Shared::disk_backend::Part';
 
+with 'ManaTools::Shared::disk_backend::PurposeLabelRole';
+
 use MooseX::ClassAttribute;
 
 class_has '+type' => (
@@ -229,6 +231,7 @@ extends 'ManaTools::Shared::disk_backend::Part';
 use MooseX::ClassAttribute;
 
 with 'ManaTools::Shared::disk_backend::Mountable';
+with 'ManaTools::Shared::disk_backend::PurposeLabelRole';
 
 sub _get_mount_source {
     my $self = shift;
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;
diff --git a/lib/ManaTools/Shared/disk_backend/Plugin/Extfs.pm b/lib/ManaTools/Shared/disk_backend/Plugin/Extfs.pm
index 84f8aca4..1f004d73 100644
--- a/lib/ManaTools/Shared/disk_backend/Plugin/Extfs.pm
+++ b/lib/ManaTools/Shared/disk_backend/Plugin/Extfs.pm
@@ -180,6 +180,7 @@ extends 'ManaTools::Shared::disk_backend::Part';
 use MooseX::ClassAttribute;
 
 with 'ManaTools::Shared::disk_backend::Mountable';
+with 'ManaTools::Shared::disk_backend::PurposeLabelRole';
 
 sub _get_mount_source {
     my $self = shift;
diff --git a/lib/ManaTools/Shared/disk_backend/Plugin/Loop.pm b/lib/ManaTools/Shared/disk_backend/Plugin/Loop.pm
index 10e37fc0..2e7b7a5c 100644
--- a/lib/ManaTools/Shared/disk_backend/Plugin/Loop.pm
+++ b/lib/ManaTools/Shared/disk_backend/Plugin/Loop.pm
@@ -186,6 +186,7 @@ use Moose;
 extends 'ManaTools::Shared::disk_backend::Part';
 
 with 'ManaTools::Shared::disk_backend::BlockDevice';
+with 'ManaTools::Shared::disk_backend::PurposeLabelRole';
 
 use MooseX::ClassAttribute;
 
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'
diff --git a/lib/ManaTools/Shared/disk_backend/Plugin/Partition.pm b/lib/ManaTools/Shared/disk_backend/Plugin/Partition.pm
index e9a67ab8..534beda0 100644
--- a/lib/ManaTools/Shared/disk_backend/Plugin/Partition.pm
+++ b/lib/ManaTools/Shared/disk_backend/Plugin/Partition.pm
@@ -245,6 +245,7 @@ use Moose;
 extends 'ManaTools::Shared::disk_backend::Part';
 
 with 'ManaTools::Shared::disk_backend::BlockDevice';
+with 'ManaTools::Shared::disk_backend::PurposeLabelRole';
 
 use MooseX::ClassAttribute;
 
@@ -305,4 +306,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;
diff --git a/lib/ManaTools/Shared/disk_backend/Plugin/Swap.pm b/lib/ManaTools/Shared/disk_backend/Plugin/Swap.pm
index 3a96036f..fc3e51cb 100644
--- a/lib/ManaTools/Shared/disk_backend/Plugin/Swap.pm
+++ b/lib/ManaTools/Shared/disk_backend/Plugin/Swap.pm
@@ -164,6 +164,8 @@ use Moose;
 
 extends 'ManaTools::Shared::disk_backend::Part';
 
+with 'ManaTools::Shared::disk_backend::PurposeLabelRole';
+
 use MooseX::ClassAttribute;
 
 class_has '+type' => (
@@ -219,4 +221,12 @@ class_has '+restrictions' => (
     }
 );
 
+around('purpose_label', sub {
+    my $orig = shift;
+    my $self = shift;
+    my $partstate = shift;
+
+    return 'SWAP';
+});
+
 1;