#!/usr/bin/perl

##
## Simple example of how to implement a '[add|delete] share command' for
## use with the Windows NT Server Manager. See smb.conf(5) for details
## on the '[add|delete] share command'
##
## Author : Gerald (Jerry) Carter <jerry@samba.org>
##
## Changed for sambatools: Michiel Broek <mbroek@mbse.eu>
##

use POSIX qw(tmpnam);

##
## local variables
##
my $delete_mode = undef;
my $add_mode = undef;
my $tmp_file_name = undef;


## check for correct parameters
if ($#ARGV == 1) {
	$delete_mode = 1;
}
elsif ($#ARGV == 3) {
	$add_mode = 1;
}
else {
	print "Usage: $0 configfile share [path] [comment]\n"; 
	exit -1;
}
if ( ("$ARGV[1]" eq "global")   || ("$ARGV[1]" eq "homes") || 
     ("$ARGV[1]" eq "printers") || ("$ARGV[1]" eq "netlogon") || 
     ("$ARGV[1]" eq "print\$")  || ("$ARGV[1]" eq "profiles") || 
     ("$ARGV[1]" eq "IPC\$") ) {
	print "share name '$ARGV[1]' not allowed\n";
	exit -1;
}


## first param is always the config file, never mind, we use a fixed
## included configfile from the OS.
##
open (CONFIGFILE, "/etc/samba/smb.conf") || die "Unable to open /etc/samba/smb.conf for reading!\n";

## FIXME!!  Right now we throw away all comments in the file.
while (<CONFIGFILE>) {

	chomp($_);
	
	## eat leading whitespace
	$_ =~ s/^\s*//;
	
	## eat trailing whitespace
	$_ =~ s/\s*$//;
	

	## throw away comments
	next if (($_ =~ /^#/) || ($_ =~ /^;/));

	## set the current section name for storing the hash
	if ($_ =~ /^\[.*\]$/) {
	
		$_ = substr($_, 1, length($_)-2);
		
		if ( length($_) ) {
			$section = $_;
		}
		else {
			print "Bad Section Name - no closing ]\n";
			exit -1;
		}

		next;
	}	
	
	## check for a param = value
	if ($_ =~ /=/) {
		($param, $value) = split (/=/, $_,2);
		$param =~ s/./\l$&/g;
		$param =~ s/\s+//g;
		$value =~ s/^\s+//;
		
		$config{$section}{$param} = $value;
		
		next;
	}

	## should have a hash of hashes indexed by section name
}
close (CONFIGFILE);

##
## We have the smb.conf in our hash of hashes now.
## Add or delete 
##
if ($add_mode) {
    $config{$ARGV[1]}{'comment'} = $ARGV[3];
    $config{$ARGV[1]}{'path'} = $ARGV[2];
    $config{$ARGV[1]}{'read only'} = "No";
    $config{$ARGV[1]}{'create mask'} = "0664";
    $config{$ARGV[1]}{'directory mask'} = "0775";
    system ("mkdir -p $ARGV[2]");
    system ("chmod og+rwxt $ARGV[2]");
} elsif ($delete_mode) {
    delete $config{$ARGV[1]};
}

##
## Print the resulting configuration
##
$tmp_file_name = tmpnam();
open (TMP, ">$tmp_file_name") || die "Unable to open temporary file for writing!\n";

PrintConfigFile(TMP);

## now overwrite the original config file
close (TMP);
system ("cp -pf /etc/samba/smb.conf /etc/samba/smb.conf.bak");
system ("cp -pf $tmp_file_name /etc/samba/smb.conf");
unlink $tmp_file_name; 


exit 0;





#######################################################################################
## PrintConfigFile()
##
sub PrintConfigFile {
	my ($output) = @_;

	## print the file back out, beginning with the global section
	print $output "#\n# Generated by $0\n#\n";
	PrintSection ($output, 'global', $config{'global'});

	foreach $section (keys %config) {

		if ("$section" ne "global") {
			print $output "## Section - [$section]\n";
			PrintSection ($output, $section, $config{$section});
		}
	}

	print $output "#\n# end of generated smb.conf\n#\n";
}

#######################################################################################
## PrintSection()
##
sub PrintSection {
	my ($outfile, $name, $section) = @_;

	print $outfile "[$name]\n";
	foreach $param (keys %$section) {
		print $outfile "\t$param".' 'x(25-length($param)). " = $$section{$param}\n";
	}
	print $outfile "\n";

}