#!/usr/bin/perl


use strict;
use Inline::Files;
use Data::Dumper;
use Switch;
use XML::Simple;
use Getopt::Long;

our $XMLClass;
our $XMLCourse;
our @Labs;

our %Machines;			# Machines list from class.xml
our @SelectedMachines;		# Machines list given as the command line argument

our $Config_File = "labmaker.conf";
our %Config = (
	"show_host" 	=> "no",

	# Вспомогательные программы
	"l3-report"	=> "./lm-report",

	# Каталоги
	"path_labmaker" => "/var/labmaker/",
	"path_classes"	=> "/var/labmaker/classes/",
	"path_lablogs"	=> "/var/labmaker/lablogs/",
	"courses_path"	=> "/var/labmaker/courses/",
	"outpath"	=> "/var/labmaker/out/",
	"path_web"	=> "/var/www/l3",		# Путь к web-отчётам
	"path_share"	=> "./share/",		# Путь к web-отчётам

	# Файлы
	"runfile"	=> "lm.run", 
	"logfile"	=> "lm.log", 

	"class" 	=> "class", 				# Имя файла класса
	"class_suffix" 	=> ".xml", 				# Cуффикс файла класса

	"sshkey"	=> "$ENV{HOME}/.ssh/id_dsa.pub",
	"lmssh"		=> "./lm-ssh",
	"lminstall"	=> "./lm-install",
	"ssh_user"	=> "r",
);

our %Run = (
	"lab" => ""
);

our %Scripts;

sub load_class;
sub load_config;
sub load_course;
sub load_scripts;

sub lm_next;
sub lm_prev;
sub lm_start;
sub lm_stop;
sub lm_set;
sub lm_do;
sub lm_report;
sub lm_show_hosts;
sub lm_show_labs;

sub load_run;
sub save_run;
sub print_log;
sub print_usage_info;
sub main();

main();

sub main()
{
	binmode STDOUT, ":utf8";

	if (! @ARGV) {
		print_usage_info();
		exit(0);
	}

	load_config;
	load_run;
	load_scripts;
	load_class;
	load_course;
	
	my $arg = join " ", @ARGV;

	# Getting @SelectedMachines if any
	if ($arg =~ s/@(.*?)\s//) {
		my $machines = $1;
		my @list = split /,/, $machines;
		for my $interval (@list) {
			my ($first, $last) = split /-/, $interval;

			push @SelectedMachines, $first;
			while ($first < $last) {
				push @SelectedMachines, ++$first;
			}	
		}
	}

	# Choose command to do
	switch ($arg) {
		case "next"	{ lm_next }
		case "prev"	{ lm_prev }
		case /set /	{ $arg =~ /set (.*)/; lm_set $1 }
		case "report"	{ lm_report }
		case "start"	{ lm_start }
		case "stop"	{ lm_stop }
		case "show hosts" { lm_show_hosts }
		case "show labs" { lm_show_labs }
		case /do /  { $arg =~ /do (.*)/;  lm_do "$1" }
		else 		{ print_usage_info() }
	}
	save_run;
	exit(0);
}

sub load_scripts
{
	local $/;
	$_=<SCRIPTS>;
	%Scripts = ("empty-element", split (/###(.*)\n/));
	delete($Scripts{"empty-element"});
}

sub load_config
{
	my %file_config;
	my %argv_config;
	#read_config_file(\%file_config, $Config_File);
	GetOptions(\%argv_config, map "$_=s", keys %Config);
	%Config = (%Config, %file_config, %argv_config);
}

sub load_course
{
	$XMLCourse = XMLin($Config{"courses_path"}.$XMLClass->{"course"}.".xml", ForceArray => 1 )  
		or die "Can't open file of the course ",$XMLClass->{"course"}," [with .xml extension]\n";
#	print Dumper($XMLCourse);
	for my $lab (@{$XMLCourse->{"module"}}) {
		push @Labs, $lab->{"code"};
	}
}

sub load_class
{
	my $classfile =
	$Config{"path_classes"}."/".$Config{"class"}.$Config{"class_suffix"};
	$XMLClass = XMLin($classfile , ForceArray => [ 'student' ] )  
		or die "Can't open file of the class ",$classfile,"\n";

	for my $student (@{$XMLClass->{"student"}}) {
		$Machines{$student->{"host"}} = {
			"name"	=> "$student->{firstname} $student->{surname}",
			"user"	=> "$student->{user}",
			"student" => $student,
		}	
	}
#	print Dumper($XMLClass);
#	print Dumper(\%Machines);
}


sub lm_next
{
	for(my $i=0; $i<=$#Labs; $i++){
		if ( $Labs[$i] eq $Run{"lab"} ) {
			if ($i < $#Labs) {
				lm_set($Labs[$i+1]);
				return ;
			} else {
				die "Lab ", $Run{"lab"}, " is the last. Switch to next lab is impossible"
			}
		}
		
	}
	die "Lab ", $Run{"lab"}, " not found. Don't know which is next"
}

sub lm_prev
# Switch to previous lab
{
	for(my $i=0; $i<=$#Labs; $i++){
		if ( $Labs[$i] eq $Run{"lab"} ) {
			if ($i > 0) {
				lm_set($Labs[$i-1]);
				return ;
			} else {
				die "Lab ", $Run{"lab"}, " is the first. Switch to previous lab is impossible"
			}
		}
		
	}
	die "Lab ", $Run{"lab"}, " not found. Don't know which is previous"
}

sub lm_set
# Switch to $_[0] lab
# FIXME
{
	my $lab = shift;
	print "Current lab is $lab\n";
	$Run{"lab"} = "$lab";
	lm_do "setlab", $lab;
}


sub lm_start
# Start new training day
{
	print_log(`date`." STARTED\n");
	if ($Run{"lab"}) {
		lm_next;
	}
	else
	{
		# First lab in the course
		lm_set($Labs[0]);
	}
}

sub lm_stop
# Stop this training day
{
	print_log(`date`." STOPPED\n");
}


sub lm_show_hosts
# Show hosts used to run a commands
{
	my $i=1;
	for my $m (sort keys %Machines) {
		if (!@SelectedMachines || grep /^$i$/, @SelectedMachines) {
			print "($i)","\t",$m,"\t",$Machines{$m}->{"name"},"\n";
		}	
		$i++;
	}
}

sub lm_show_labs
# Show hosts used to run a commands
{
	my $i=1;
	for my $lab (@Labs) {
		print $lab;
		print "*" if $lab eq $Run{"lab"};
		print "\n";
	}
}

sub lm_do
# Do the $_[0] command on all of the hosts 
{
	my $command = shift;
	my $arg = join " ", @_;
	my $i=1;
	for my $m (sort keys %Machines) {
		if (!@SelectedMachines || grep $_ eq $i, @SelectedMachines) {
			print "$m:\n" if $Config{"show_host"} =~ /y/i;

			my %myenv = ( %Config, 
				host 	=>	$m,
				dirs	=>	"/root /home/".$Machines{$m}->{"user"},
				lablogs =>	$Config{"path_lablogs"}."/".
						$XMLClass->{"course"}."/".
						$XMLClass->{"date"}."/".
						"$m",
				lab	=> 	$arg,			
				
				email	=>	$Machines{$m}->{"student"}->{"email"},
				company	=>	$Machines{$m}->{"student"}->{"company"},
				center 	=>	$XMLClass->{"center"},
				course 	=>	$XMLClass->{"course"},
				date 	=>	$XMLClass->{"date"},
				name	=> 	$Machines{$m}->{"name"},
				coursepath =>	$XMLCourse->{"path"},

			);
			if (grep { $_ eq $command} keys %Scripts) {
				$_=$Scripts{"$command"};
				s/\$(\w+)/$myenv{$1}/ge;
				open(SHELL, "|/bin/sh -s");
				binmode SHELL, ":utf8";
				print SHELL $_;
				close (SHELL);
			}
			else {
				my $res = `ssh $Config{"ssh_user"}\@$m $command`;
				if ($res) {
					my $count = ($res =~ s/(^)/$m: /mg);
					print $res;
					print "\n" if ($count > 1);
				}
			}	
		}	
		$i++;
	}
}



=cut comment

lm report

Построить html представление для журналов текущего класса.
Для построения используется скрипт l3-report.

=cut

sub lm_report
{

	my $webdir = $Config{"path_web"};
	my $course=$XMLClass->{"course"};
	my $date=$XMLClass->{"date"};
	my $encoding=$XMLClass->{"charset"};

	print "date=$date\n";

	for my $student (@{$XMLClass->{"student"}}) {
		my $user = $student->{"user"};
		my $hostname = $student->{"host"};
		my $encoding = $student->{"encoding"};

		system("mkdir -p $webdir/$date/$hostname");
		system("cp ".$Config{"path_share"}."/*.{ico,css} $webdir/$date/$hostname");
		system($Config{"l3-report"}.
			" --input ".$Config{"path_lablogs"}."/$course/$date/$hostname/$user".
			" --diffs ".$Config{"path_lablogs"}."/$course/$date/$hostname/$user ".
				   $Config{"path_lablogs"}."/$course/$date/$hostname/root".
			" --output $webdir/$date/$hostname/$user.html".
			" --encoding $encoding"
		);
		system($Config{"l3-report"}.
			" --input ".$Config{"path_lablogs"}."/$course/$date/$hostname/root".
			" --diffs ".$Config{"path_lablogs"}."/$course/$date/$hostname/root ".
			" --output $webdir/$date/$hostname/root.html".
			" --encoding $encoding"
		);
	}
}

sub load_run
{
	my $runfile = $Config{"path_labmaker"}."/".$Config{"path_runfile"};
	open (RUN, $runfile)
		or return;
	while (<RUN>) {
		chomp;
		my ($var, $val) = split /\s+/,$_,2;
		$Run{$var}=$val;
	}
	close (RUN);	
}

sub save_run
{
	my $runfile = $Config{"path_labmaker"}."/".$Config{"path_runfile"};
	open (RN, "$runfile")
		or die "Can't save running state to $runfile";
	for my $var (keys %Run) {
		print RN $var,"\t",$Run{$var},"\n";
	}
	close (RN);	
}

sub print_log
{
	my $logfile = $Config{"path_labmaker"}."/".$Config{"path_logfile"};
	open (LOG, ">>$logfile")
		or die "Can't open logfile $logfile for writing";
	print LOG  @_;
	close (LOG);	
}


sub print_usage_info
{
	print "Usage:\n\n\t$0 [host-list] command\n";
	while (<USAGE>) {
		print $_;
	}
}

__USAGE__

Commands:

	next		-- next lab
	prev		-- prev lab
	set LAB		-- set current lab to LAB
	start		-- start this day training
	stop		-- stop this day training
	show hosts	-- show available hosts in the class
	show labs	-- show available labs in the course
	do COMMAND	-- do specified command on the hosts of hostlist
	report		-- generate XML/HTML reports

	
do commands:
	
	install [PROFILE] -- install profile 
	
Host list:	

	@N		-- machine N
	@N1-N2		-- all of the machines from N1 to N2
	@N1,N2,N3	-- machine N1, N2 and N3

	N* is numbers or domain names of the machines.
	
	If host list is not specified, 
	command is executed on all of the machines



__SCRIPTS__
###install
cat $sshkey | $lmssh $ssh_user@$host /bin/sh -c '"mkdir -p ~/.ssh; cat >>~/.ssh/authorized_keys; chmod 600 ~/.ssh/authorized_keys"'

###install-lm
cat $lminstall | ssh $ssh_user@$host /bin/sh -s $dirs

###copy-lablogs
for i in $dirs
do
	mkdir -p $lablogs/${i##*/}
	scp -q $ssh_user@$host:${i}/.labmaker/* $lablogs/${i##*/}
done

###setlab
for i in $dirs
do
	echo $lab | ssh $ssh_user@$host "cat > "${i}"/.labmaker/lab"
done

###makeout
common=$course-$date
personal=$course-$date-$email
mkdir -p $outpath/${common}/{Lablogs,Docs}
mkdir -p $outpath/${personal}/{Course,Files}
cd $outpath/${personal}
ln -s ../${common}/Lablogs .
ln -s ../${common}/Docs .
cd ~-
export UG_PERSONAL=${PWD}/$outpath/${personal}/Course
export UG_CENTER="$center"
export UG_COURSE="$course"
export UG_DATE="$date"
export UG_STUDENT="$name"
export UG_COMPANY="$company"
cd $coursepath; make personal; cd ~-

###watch
cat taillast.pl | ssh $ssh_user@$host perl - /root/.labmaker