#!/usr/bin/perl

# drakupdate_fstab
# Copyright (C) 2002-2008 Mandriva (pixel@mandriva.com)
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2, or (at your option)
# any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

use lib qw(/usr/lib/libDrakX);

use detect_devices;
use security::level;
use common;
use fsedit;
use lang;
use any;
use fs;

$::isStandalone = 1; #- not using standalone.pm which generates too many logs for drakupdate_fstab purpose

log::l("drakupdate_fstab called with @ARGV\n");

my ($debug, $verbose, $removed);

my %args = (
            '--auto' => \$::auto,
            '--debug' => \$debug,
            '--test' => \$::testing,
            '--verbose' => \$verbose,
           );

each_index {
    if ($args{$_}) {
        ${$args{$_}} = 1;
        splice @ARGV, $::i - $removed, 1;
        $removed++;
    }
} @ARGV;

my ($raw_action, $device_name) = @ARGV;
my ($action) = $raw_action =~ /^--(add|del)/;

@ARGV == 2 && $action or die "usage: drakupdate_fstab [--test] [--verbose] [--auto] [--add | --del] <device>\n";

main($action, $device_name);

sub device_name_to_entry {
    my ($name) = @_;
    $name =~ s|/dev/||;
    $name =~ /fd[01]/ && !$::auto and return { device => $name };
    my @l = detect_devices::get();
    if ($debug) {
	require Data::Dumper;
	output("/tmp/drakdump_devices-$action", Data::Dumper->Dump([ \@l ], [ qw($l) ]));
    }

    my $e;
    {
	unless ($e = find { $name eq $_->{device} } @l) {
	    my ($prefix) = $name =~ m/^(.*?)\d*$/;
	    $e = find { $prefix eq ($_->{prefix} || $_->{device}) } @l;
	    $e->{device} = $name;
	}
    }

    $e->{media_type} = 'fd' if $name =~ /fd[01]/;
    $e;
}

sub set_options {
    my ($part) = @_;
    $part->{is_removable} = 1;
}

sub set_mount_point {
    my ($part, $fstab, $is_hd) = @_;

    my $mntpoint = $is_hd ? 'hd' : detect_devices::suggest_mount_point($part);
    $mntpoint = "/media/$mntpoint";

    foreach ('', 2 .. 10) {
	next if fs::get::mntpoint2part("$mntpoint$_", $fstab);
	$part->{mntpoint} = "$mntpoint$_";
	return 1;
    }
    0;
}

sub main {
    my ($action, $device_name) = @_;	
    my $device = device_name_to_entry($device_name);
    if ($device->{media_type} eq 'hd') {
	my $all_hds = fsedit::get_hds();
	my $hd = find { fs::get::is_same_hd($_, $device) } fs::get::hds($all_hds);
	$hd or die "unable to find hd for $device->{device}";
	my @parts = grep { !fsedit::isSwap($_) && fs::type::maybeFormatted($_) } partition_table::get_normal_parts($hd);
	configure_part($action, $_, 'hd') foreach @parts;
    } else {
        configure_part($action, $device, undef);
    }
}

sub configure_part {
    my ($action, $part, $is_hd) = @_;
    my $fstab_file = '/etc/fstab';
    if (!$part) {
	print STDERR "Can not find device $device_name\n" if $::testing;
	return;
    } elsif ($::testing) {
	cp_af('/etc/fstab', $fstab_file = '/tmp/fstab');
    }

    my $fstab = [ fs::read_fstab('', '/etc/fstab', 'keep_freq_passno', 'verbatim_credentials') ];
    my ($existing_fstab_entries, $fstab_) = partition { fs::get::is_same_hd($_, $part) } @$fstab;

    if ($debug) {
	require Data::Dumper;
	output("/tmp/drakdump_entries-$action", Data::Dumper->Dump([ \@ARGV, $part, $fstab, $fstab_, $existing_fstab_entries ],
								   [ qw($ARGV $part $fstab $fstab_ $existing_fstab_entries) ]));
    }
    if ($action eq 'add') {
	if (@$existing_fstab_entries) {
	    print STDERR "Already in fstab\n" if $::testing;
	    return;
	}
	set_mount_point($part, $fstab, $is_hd) or return;
	set_options($part);

	my ($line) = fs::prepare_write_fstab([$part]);
	if ($line) {
	    append_to_file($fstab_file, $line);
	    system("mount $part->{mntpoint}") if !$::testing && ($device_name =~ /^fd\d+/ || $is_hd);
	}

	if ($verbose) {
	    print $part->{mntpoint}, " user\n";
	}
    } else {
	if (!@$existing_fstab_entries) {
	    print STDERR "Not found in fstab\n" if $::testing;
	    return;
	}

	my ($s) = fs::prepare_write_fstab($fstab_, '', 'keep_smb_credentials');
	output($fstab_file, $s);

	if ($verbose) {
	    print "$_->{mntpoint}\n" foreach @$existing_fstab_entries;
	}
    }

    if ($::testing) {
	print "fstab would have changed:\n";
	system("diff -u /etc/fstab $fstab_file");
    }
}