#!/usr/bin/perl -w

use DBI;
use CGI ':all';
use strict;
use Time::Local;
use POSIX qw(locale_h);

my ($PWD) = `pwd`;
chomp $PWD;
my ($SRCPATH) = "$PWD/../dimrub/src";
my ($ZIP) = "/usr/bin/zip";
my ($SENDMAIL) = "/usr/sbin/sendmail";
my ($TMSECS) = 30*24*60*60;
my (%RevMonths) = 
	('Jan', '0', 'Feb', '1', 'Mar', '2', 'Apr', '3', 'May', '4', 'Jun', '5',
	'Jul', '6', 'Aug', '7', 'Sep', '8', 'Oct', '9', 'Nov', '10',
	'Dec', '11',
	 'Янв', '0', 'Фев', 1, 'Мар', 2, 'Апр', 3, 'Май', '4',
	 'Июн', '5', 'Июл', 6, 'Авг', '7', 'Сен', '8', 
	 'Окт', '9', 'Ноя', '19', 'Дек', '11');

# Determine whether the given time is within 2 months from now.
sub NewEnough {
	my ($a) = @_;
	my ($year, $month, $day) = split('-', $a);

	return (time - timelocal(0, 0, 0, $day, $month -1, $year) < $TMSECS);
}

# Reads one question from the DB. Gets DB handler and Question ID.
sub GetTournament {
	my ($dbh, $Id) = @_;
	my (%Tournament, $field, @arr);

	return %Tournament if ($Id == 0);

	my ($sth) = $dbh->prepare("SELECT * FROM Tournaments WHERE Id=$Id");
	$sth->execute;

	@arr = $sth->fetchrow;
	my($i, $name) = 0;
	foreach $name (@{$sth->{NAME}}) {
		$Tournament{$name} = $arr[$i++];
	}

	return %Tournament;
}

# Reads one question from the DB. Gets DB handler and Question ID.
sub GetQuestion {
	my ($dbh, $QuestionId) = @_;
	my (%Question, $field, @arr);

	my($sth) = $dbh->prepare("
		SELECT * FROM Questions WHERE QuestionId=$QuestionId
	");

	$sth->execute;

	@arr = $sth->fetchrow;
	my($i, $name) = 0;
	foreach $name (@{$sth->{NAME}}) {
		$Question{$name} = $arr[$i++];
	}

	return %Question;
}

# Gets numbers of all the questions from the given tour.
sub GetTourQuestions {
	my ($dbh, $ParentId) = @_;
	my (@arr, @Questions);

	my ($sth) = $dbh->prepare("SELECT QuestionId FROM Questions 
		WHERE ParentId=$ParentId ORDER BY QuestionId");

	$sth->execute;

	while (@arr = $sth->fetchrow) {
		push @Questions, $arr[0];
	}

	return @Questions;
}

# Returns list of children of the given tournament.
sub GetTours {
	my ($dbh, $ParentId) = @_;
	my (@arr, @Tours);

	my ($sth) = $dbh->prepare("SELECT Id FROM Tournaments
		WHERE ParentId=$ParentId ORDER BY Id");

	$sth->execute;

	while (@arr = $sth->fetchrow) {
		push @Tours, $arr[0];
	}

	return @Tours;
}


# Returns list of QuestionId's, that have the search string in them.
sub Search {
	my ($dbh, $sstr) = @_;
	my (@arr, @Questions, @fields);
	my (@sar, $i, $sth);

	push @fields, 'Question';
	foreach (qw/Answer Sources Authors Comments/) {
		if (param($_)) {
			push @fields, "IFNULL($_, '')";
		}
	}

	@sar = split " ", $sstr;
	for $i (0 .. $#sar) {
		$sar[$i] = $dbh->quote("%${sar[$i]}%");
	}

	my($f) = "CONCAT(" . join(',', @fields) . ")";
	if (param('all') eq 'yes') {
		$sstr = join " AND $f LIKE ", @sar;
	} else {
		$sstr = join " OR $f LIKE ", @sar;
	}
	
	$sth = $dbh->prepare("SELECT QuestionId FROM Questions
		WHERE $f LIKE $sstr ORDER BY QuestionId");

	$sth->execute;
	while (@arr = $sth->fetchrow) {
		push @Questions, $arr[0];
	}
	
	return @Questions;
}

 # Substitute every letter by a pair (for case insensitive search).
 my (@letters) = qw/аА бБ вВ гГ дД еЕ жЖ зЗ иИ йЙ кК лЛ мМ нН оО 
 пП рР сС тТ уУ фФ хХ цЦ чЧ шШ щЩ ьЬ ыЫ эЭ юЮ яЯ/;
 
sub NoCase {
	my ($sstr) = shift;
	my ($res);

	if (($res) = grep(/$sstr/, @letters)) {
		return "[$res]";
	} else {
		return $sstr;
	}
}

sub PrintSearch {
	my ($dbh, $sstr) = @_;
   my (@Questions) = &Search($dbh, $sstr);
	my ($output, $i, $suffix, $hits) = ('', 0, '', $#Questions + 1);

	if ($hits =~ /1.$/  || $hits =~ /[5-90]$/) {
		$suffix = 'й';
	} elsif ($hits =~ /1$/) {
		$suffix = 'е';
	} else {
		$suffix = 'я'; 
	}
	
	print p({align=>"center"}, "Результаты поиска на " . strong($sstr)
	. " : $hits попадани$suffix.");

	if (param('word')) {
		$sstr = '[ 	\.\,:;]' . $sstr . '[  \.\,:\;]';
	}

	$sstr =~ s/(.)/&NoCase($1)/ge;

	my(@sar) = split(/\s/, $sstr);
	for ($i = 0; $i <= $#Questions; $i++) {
		$output = &PrintQuestion($dbh, $Questions[$i], 1, $i + 1, 1);
		foreach  (@sar) {
			$output =~ s/$_/<strong>$&<\/strong>/gs;
		}
		print $output;
	}
}

sub PrintRandom {
   my ($dbh, $type, $num, $text) = @_;
   my (@Questions) = &Get12Random($dbh, $type, $num);
	my ($output, $i) = ('', 0);

	if ($text) {
		$output .= "	$num случайных вопросов.\n\n";
	} else {
		$output .=
			h2({align=>"center"}, "$num случайных вопросов.");
	}

	for ($i = 0; $i <= $#Questions; $i++) {
		# Passing DB handler, question ID, print answer, question
		# number, print title, print text/html
		$output .= 
			&PrintQuestion($dbh, $Questions[$i], 1, $i + 1, 0, $text);
	}
	return $output; 
}

sub PrintTournament {
   my ($dbh, $Id, $answer) = @_;
	my (%Tournament, @Tours, $i, $list, $qnum, $imgsrc, $alt,
		$SingleTour);
	my ($output) = '';

	%Tournament = &GetTournament($dbh, $Id) if ($Id);
	
	my ($URL) = $Tournament{'URL'};
	my ($Info) = $Tournament{'Info'};
	my ($Copyright) = $Tournament{'Copyright'};

	@Tours = &GetTours($dbh, $Id);

	if ($Id) {
		for ($Tournament{'Type'}) {
			/Г/ && do {
				$output .= h2({align=>"center"}, 
					"Группа: $Tournament{'Title'}") . p . "\n";
				last;
			};
			/Ч/ && do {
				return &PrintTour($dbh, $Tours[0], $answer)
					if ($#Tours == 0);

				$output .= h2({align=>"center"}, 
					"Чемпионат: $Tournament{'Title'}") . p . "\n";
				last;
			};
			/Т/ && do {
				return &PrintTour($dbh, $Id, $answer);
			};
		}
	} else {
		my ($qnum) = GetQNum($dbh);
		$output .= h2("Банк Вопросов: $qnum вопросов") . p . "\n";
	}

	for ($i = 0; $i <= $#Tours; $i++) { 
		%Tournament = &GetTournament($dbh, $Tours[$i]);
		
		if ($Tournament{'Type'} =~ /Ч/) {
			$SingleTour = 0;
			my (@Tours) = &GetTours($dbh, $Tournament{'Id'});
			$SingleTour = 1
				if ($#Tours == 0);
		}
		if ($Tournament{'QuestionsNum'} > 0) {
			$qnum = " ($Tournament{'QuestionsNum'} вопрос" .
				&Suffix($Tournament{'QuestionsNum'}) . ")\n";
		} else {
			$qnum = '';
		}
		if ($Tournament{'Type'} =~ /Г/) {
			$imgsrc = "/icons/folder.gif";
			$alt = "[*]";
		} else {
			$imgsrc = "/icons/folder.gif";
			$alt = "[-]";
		}

		if ($SingleTour or $Tournament{'Type'} =~ /Т/) {
			$list .= dd(img({src=>$imgsrc, alt=>$alt})
				. " " . $Tournament{'Title'} . $qnum) . 
				dl(
					dd("["
						. a({href=>url .  "?tour=$Tournament{'Id'}&answer=0"},
						"вопросы") . "] ["
                  . a({href=>url .  "?tour=$Tournament{'Id'}&answer=1"},
                  "вопросы + ответы") . "]")
				);
		} else {
			$list .= dd(a({href=>url . "?tour=$Tournament{'Id'}&comp=1"}, 
				img({src=>'/icons/compressed.gif', alt=>'[ZIP]', border=>1}))
				. " " . img({src=>$imgsrc, alt=>$alt}) 
				. " " . a({href=>url . "?tour=$Tournament{'Id'}&answer=0"}, 
				$Tournament{'Title'}) . $qnum);
		}
	}
	$output .= dl($list);

	if ($URL) {
		$output .=
		p("Дополнительная информация об этом турнире - по адресу " . 
			a({-'href'=>$URL}, $URL));
	}

	if ($Copyright) {
		$output .= p("Копирайт: " .   $Copyright);
	}

	if ($Info) {
		$output .= p($Info);
	}
	
	return $output;
}

sub Suffix {
	my ($qnum) = @_;
	my ($suffix) = 'а' if $qnum =~ /[234]$/;
   $suffix = '' if $qnum =~ /1$/;
   $suffix = 'ов' if $qnum =~ /[567890]$/ || $qnum =~ /1.$/;
	return $suffix;
}

sub IsTour {
	my ($dbh, $Id) = @_;
	my ($sth) = $dbh->prepare("SELECT Type FROM Tournaments 
		WHERE Id=$Id");
	$sth->execute;
	return ($sth->fetchrow)[0] =~ /Т/;
}

# Gets a DB handler (ofcourse) and a tour Id. Prints all the
# question of that tour, according to the options.
sub PrintTour {
	my ($dbh, $Id, $answer) = @_;
	my ($output, $q, $bottom, $field) = ('', 0, '', '');

	my (%Tour) = &GetTournament($dbh, $Id);
	my (@Tours) = &GetTours($dbh, $Tour{'ParentId'});
	my (%Tournament) = &GetTournament($dbh, $Tour{'ParentId'});

	return 0
		if ($Tour{'Type'} !~ /Т/);

	my ($qnum) = $Tour{'QuestionsNum'};
	my ($suffix) = &Suffix($qnum); 
	
	$output .= h2({align=>"center"}, $Tournament{"Title"}, 
		      "<br>", $Tour{"Title"} . 
		" ($qnum вопрос$suffix)\n") . p;

	my (@Questions) = &GetTourQuestions($dbh, $Id);
	for ($q = 0; $q <= $#Questions; $q++) {
		$output .= &PrintQuestion($dbh, $Questions[$q], $answer, 0);
	} 

	$output .= hr({-'align'=>'center', -'width'=>'80%'});

	if ($Tournament{'URL'}) {
		$output .=
		p("Дополнительная информация об этом турнире - по адресу " . 
			a({-'href'=>$Tournament{'URL'}}, $Tournament{'URL'}));
	}

	if ($Tournament{'Copyright'}) {
		$output .= p("Копирайт: " .   $Tournament{'Copyright'});
	}

	if ($Tournament{'Info'}) {
		$output .= p($Tournament{'Info'});
	}
	

	if ($answer == 0) {
		$bottom .= 
			"[" . a({href=>url . "?tour=$Id&answer=1"}, "ответы") .  "] " . br;
	}
	if (&IsTour($dbh, $Id - 1)) {
		$bottom .= 
			"[" . a({href=>url . "?tour=" . ($Id - 1) . "&answer=0"}, 
			"предыдущий тур") . "] ";
		$bottom .= 
			"[" . a({href=>url . "?tour=" . ($Id - 1) . "&answer=1"}, 
			"предыдущий тур с ответами") . "] " . br;
	}
	if (&IsTour($dbh, $Id + 1)) {
		$bottom .= 
			"[" . a({href=>url . "?tour=" . ($Id + 1) . "&answer=0"}, 
			"следующий тур") . "] ";
		$bottom .= 
			"[" . a({href=>url . "?tour=" . ($Id + 1) . "&answer=1"}, 
			"следующий тур с ответами") . "] ";
	}

	$output .=
		p({align=>"center"}, font({size=>-1}, $bottom));

	return $output;
}

sub PrintField {
	my ($header, $value, $text) = @_;
	if ($text) {
		$value =~ s/<[\/\w]*>//sg;
	}
	return $text ? "$header:\n$value\n\n" : 
		strong("$header: ") . $value . p . "\n";
}

# Gets a DB handler (ofcourse) and a question Id. Prints 
# that question, according to the options.
sub PrintQuestion {
	my ($dbh, $Id, $answer, $qnum, $title, $text) = @_;
	my ($output, $titles) = ('', '');

	my (%Question) = &GetQuestion($dbh, $Id);
	if (!$text) {
		$output .= hr({width=>"50%"});
		if ($title) {
			my (%Tour) = GetTournament($dbh, $Question{'ParentId'});
			my (%Tournament) = GetTournament($dbh, $Tour{'ParentId'});
			$titles .=
				dd(img({src=>"/icons/folder.open.gif"}) . " " .
					 a({href=>url . "?tour=$Tournament{'Id'}"}, $Tournament{'Title'}));
			$titles .=
				dl(dd(img({src=>"/icons/folder.open.gif"}) . " " .
					a({href=>url . "?tour=$Tour{'Id'}"}, $Tour{'Title'})));
		}
		$output .= dl(strong($titles));
	}
	
	$qnum = $Question{'Number'}
		if ($qnum == 0);

	$output .= 
		&PrintField("Вопрос $qnum", $Question{'Question'}, $text);

	if ($answer) {
		$output .= 
			&PrintField("Ответ", $Question{'Answer'}, $text);

		if ($Question{'Authors'}) {
			$output .= &PrintField("Автор(ы)", $Question{'Authors'}, $text);
		}

		if ($Question{'Sources'}) {
			$output .= &PrintField("Источник(и)", $Question{'Sources'}, $text);
		}

		if ($Question{'Comments'}) {
			$output .= &PrintField("Комментарии", $Question{'Comments'}, $text);
		}
	}
	return $output;
}

# Returns the total number of questions currently in the DB.
sub GetQNum {
	my ($dbh) = @_;
	my ($sth) = $dbh->prepare("SELECT COUNT(*) FROM Questions");
	$sth->execute;
 	return ($sth->fetchrow)[0];
}

# Returns Id's of 12 random questions
sub Get12Random {
   my ($dbh, $type, $num) = @_;
	my ($i, @questions, $q, $t, $sth);
	my ($qnum) = &GetQNum($dbh);
	my (%chosen);
	srand;
	
	for ($i = 0; $i < $num; $i++) {
		do {
			$q = int(rand($qnum));
			$sth = $dbh->prepare("SELECT Type FROM Questions
				WHERE QuestionId=$q");
			$sth->execute;
			$t = ($sth->fetchrow)[0];
		} until !$chosen{$q} && $t =~ /$type/;
		$chosen{$q} = 'y';
		push @questions, $q;
	}
	return @questions;
}

sub Include_virtual {
	my ($fn, $output) = (@_, '');

	open F , $fn
		or return; #die "Can't open the file $fn: $!\n";
	
	while (<F>) {
		if (/<!--#include/o) {
			s/<!--#include virtual="\/(.*)" -->/&Include_virtual($1)/e;
		}
		if (/<!--#exec/o) {
			s/<!--#exec.*cmd\s*=\s*"([^"]*)".*-->/`$1`/e;
		}
		$output .= $_;
	}
	return $output;
}

sub PrintArchive {
	my($dbh, $Id) = @_;
	my ($output, @list, $i);

	my (%Tournament) = &GetTournament($dbh, $Id);
	my (@Tours) = &GetTours($dbh, $Id);
	
	if ($Tournament{'Type'} =~ /Г/ || $Id == 0) {
		for ($i = 0; $i <= $#Tours; $i++) {
			push(@list ,&PrintArchive($dbh, $Tours[$i]));
		}
		return @list;
	}
	return "$SRCPATH/$Tournament{'FileName'} ";
}

sub PrintAll {
	my ($dbh, $Id) = @_;
	my ($output, $list, $i);

	my (%Tournament) = &GetTournament($dbh, $Id);
	my (@Tours) = &GetTours($dbh, $Id);
	my ($New) = ($Id and $Tournament{'Type'} eq 'Ч' and 
		&NewEnough($Tournament{"CreatedAt"})) ?
		img({src=>"/znatoki/dimrub/db/new-sml.gif", alt=>"NEW!"}) : "";

	if ($Id == 0) {
		$output = h3("Все турниры");
	} else {
		$output .= dd(img({src=>"/icons/folder.gif", alt=>"[*]"}) .
      " " . a({href=>url . "?tour=$Tournament{'Id'}&answer=0"},
      $Tournament{'Title'}) . " $New");
	}
	if ($Id == 0 or $Tournament{'Type'} =~ /Г/) {
		for ($i = 0; $i <= $#Tours; $i++) {
			$list .= &PrintAll($dbh, $Tours[$i]);
		}
		$output .= dl($list);
	}
	return $output;
}

sub PrintDates {
	my ($dbh) = @_;
	my ($from) = param('from_year') . "-" . param('from_month') . 
		"-" .  param('from_day');
	my ($to) = param('to_year') . "-" . param('to_month') . "-" .  param('to_day');
	$from = $dbh->quote($from);
	$to = $dbh->quote($to);
	my ($sth) = $dbh->prepare("
		SELECT DISTINCT Id
		FROM Tournaments
		WHERE PlayedAt >= $from AND PlayedAt <= $to
		AND Type = 'Ч'
	");
	$sth->execute;
	my (%Tournament, @array, $output, $list);

	$output = h3("Список турниров, проходивших между $from и $to.");
	while (@array = $sth->fetchrow) {
		next
			if (!$array[0]);
		%Tournament = &GetTournament($dbh, $array[0]);
      $list .= dd(img({src=>"/icons/folder.gif", alt=>"[*]"}) .
      " " . a({href=>url . "?tour=$Tournament{'Id'}&answer=0"},
      $Tournament{'Title'}));
	}
	$output .= dl($list);
	return $output;
}

MAIN:
{
	setlocale(LC_CTYPE,'russian');
	my($i, $tour);
	my($text) = (param('text')) ? 1 : 0;
	my($dbh) = DBI->connect("DBI:mysql:chgk", "piataev", "")
		or do {
			print h1("Временные проблемы") . "База данных временно не
			работает. Заходите попозже.";
			print &Include_virtual("../dimrub/db/reklama.html");
		    print end_html;
			die "Can't connect to DB chgk\n";
		};
	if (!param('comp') and !$text) {
	   print header;
	   print start_html(-"title"=>'Database of the questions',
	           -author=>'dimrub@icomverse.com',
	           -bgcolor=>'#fff0e0',
				  -vlink=>'#800020');
		print &Include_virtual("../dimrub/db/reklama.html");
	}

	if ($text) {
		print header('text/plain');
	}

	if (param('rand')) {
		my ($type, $qnum) = ('Ч', 12);
		$type = 'Б' if (param('brain'));
		$qnum = param('qnum') if (param('qnum') =~ /^\d+$/);	
		if (param('email') && -x $SENDMAIL && 
		open(F, "| $SENDMAIL -t -n")) {
			my ($Email) = param('email');
			my ($mime_type) = $text ? "plain" : "html";
			print F <<EOT;
To: $Email
From: dimrub\@icomverse.com
Subject: Sluchajnij Paket Voprosov "Chto? Gde? Kogda?"
MIME-Version: 1.0
Content-type: text/$mime_type; charset="koi8-r"

EOT
			print F &PrintRandom($dbh, $type, $qnum, $text);
			close F;
			print "Пакет случайно выбранных вопросов послан. Нажмите
			на <B>Reload</B> для получения еще одного пакета";
		} else {
			print &PrintRandom($dbh, $type, $qnum, $text);
		}
	} elsif (param('sstr')) {
		&PrintSearch($dbh, param('sstr'));
	} elsif (param('all')) {
		print &PrintAll($dbh, 0);
	} elsif (param('from_year') && param('to_year')) {
		print &PrintDates($dbh);	
	} elsif (param('comp')) {
		print header(
			-'Content-Type' => 'application/x-zip-compressed; name="db.zip"',
			-'Content-Disposition' => 'attachment; filename="db.zip"'
		);
		$tour = (param('tour')) ? param('tour') : 0;
		my (@files) = &PrintArchive($dbh, $tour);
		open F, "$ZIP -j - $SRCPATH/COPYRIGHT @files |";
		print (<F>);
		close F;
		$dbh->disconnect;
		exit;
	} else {
		$tour = (param('tour')) ? param('tour') : 0;
		if ($tour !~ /^[0-9]*$/) {
			my ($sth) = $dbh->prepare("SELECT Id FROM Tournaments
			WHERE FileName = '$tour.txt'");
			$sth->execute;
			$tour = ($sth->fetchrow)[0];
		}
		print &PrintTournament($dbh, $tour, param('answer'));
	}
	if (!$text) {
		print &Include_virtual("../dimrub/db/footer.html");
		print end_html;
	}
	$dbh->disconnect;
}