#!/usr/local/bin/perl
# Save slice label (GPT label or glabel)

use strict;
use warnings;
no warnings 'redefine';
no warnings 'uninitialized';
require './bsdfdisk-lib.pl';
our (%in, %text, $module_name);
&ReadParse();
&error_setup($text{'slice_label_err'});

# Validate input parameters to prevent command injection
$in{'device'} =~ /^[a-zA-Z0-9_\/.-]+$/ or &error("Invalid device name");
$in{'device'} !~ /\.\./ or &error("Invalid device name");
$in{'slice'} =~ /^\d+$/ or &error("Invalid slice number");

my $label = defined $in{'label'} ? $in{'label'} : '';
$label =~ s/^\s+|\s+$//g;
$label ne '' || &error($text{'slice_label_empty'});

# Get the disk and slice
my @disks = &list_disks_partitions();
my ($disk) = grep { $_->{'device'} eq $in{'device'} } @disks;
$disk || &error($text{'disk_egone'});
my ($slice) = grep { $_->{'number'} eq $in{'slice'} } @{$disk->{'slices'}};
$slice || &error($text{'slice_egone'});

# Ensure we can set labels on this scheme
my $base_device = $disk->{'device'}; $base_device =~ s{^/dev/}{};
my $ds = get_disk_structure($base_device);
if (!$ds || !$ds->{'scheme'} || $ds->{'scheme'} !~ /GPT/i) {
    if (!has_command('glabel')) {
        &error($text{'slice_label_noglabel'});
    }
}

my $err = set_partition_label(
    disk  => $disk,
    slice => $slice,
    label => $label,
);
&error($err) if ($err);

&webmin_log("label", "slice", $slice->{'device'}, { 'label' => $label });
&redirect("edit_slice.cgi?device=$in{'device'}&slice=$in{'slice'}");