### IPTVPortal Middleware module
### Version 0.1.0001, date 2015-07-25
### Author: Alibek Bolatov <alibek@mail.ru>;

# Кодировка UTF-8, формат файла Unix

=encoding UTF-8

=pod

История изменений

=over

=item 0.1.0001 2015-07-25
  Первый релиз

=back

=cut

use utf8;
use strict;
use warnings;

package IPTVPortal;

use boolean;
use Encode;
use JSON;
use Tie::IxHash;
use Digest::MD5 qw(md5_hex);
use URI::Escape;
use LWP::UserAgent;
use HTTP::Request;
use Data::Dumper;
$Data::Dumper::Useperl = 1;
{
	no warnings 'redefine';
	sub Data::Dumper::qquote {return "'$_[0]'";}
}

my $cfg = {};

sub _init(;$)
{
	my $domain = (shift||"provider");
	$cfg->{'rpc'} = {};
	$cfg->{'rpc'}->{'domain'} = $domain;
	$cfg->{'rpc'}->{'address'} = "https://admin.$cfg->{'rpc'}->{'domain'}.iptvportal.ru";
	$cfg->{'rpc'}->{'username'} = "admin";
	$cfg->{'rpc'}->{'password'} = "password";
	$cfg->{'rpc'}->{'api'} = "/api/jsonrpc/";
	$cfg->{'rpc'}->{'sql'} = "/api/jsonsql/";
	$cfg->{'rpc'}->{'expire'} = 86400;
	$cfg->{'rpc'}->{'version'} = "2.0";
}

sub new
{
	{
		my ($type,$section,$name);
		my $map = sub
		{
			my $type = shift;
			my $data = shift;
			my %map = 
			(
				'json' => sub 
					{
						my $data = Encode::encode_utf8(shift);
						return JSON->new->utf8->allow_nonref->allow_blessed->convert_blessed->relaxed->decode($data);
					}
			);
			if (exists($map{$type}))
			{
				if (ref($map{$type}) eq 'CODE')
				{
					$data = &{$map{$type}}($data);
				}
			}
			return $data;
		};
		$cfg->{'json'} = {};
		while (my $line = <DATA>)
		{
			chomp $line;
			last if ($line eq '__END__');
			if ($line =~ /^#(\w+)\s*@@\s*(\w+)\s*@\s*(\w+)\s*$/)
			{
				$cfg->{$type}->{$section}->{$name} = &$map($type, $cfg->{$type}->{$section}->{$name}) if (defined($type) && defined($section) && defined($name) && !(ref($cfg->{$type}->{$section}->{$name}) eq 'ARRAY'));
				$type = $1;
				$section = $2;
				$name = $3;
				$cfg->{$type}->{$section}->{$name} = (exists($cfg->{$type}) ? '' : []);
			}
			else
			{
				if (defined($type) && defined($section) && defined($name))
				{
					if (ref($cfg->{$type}->{$section}->{$name}) eq 'ARRAY')
					{
						$line = &$map($type, $line);
						push @{$cfg->{$type}->{$section}->{$name}}, $line if (length($line) > 0);
					}
					else
					{
						$cfg->{$type}->{$section}->{$name} .= $line . "\n";
					}
				}
			}
		}
		$cfg->{$type}->{$section}->{$name} = &$map($type, $cfg->{$type}->{$section}->{$name}) if (defined($type) && defined($section) && defined($name) && !(ref($cfg->{$type}->{$section}->{$name}) eq 'ARRAY'));
	}
	my $this = shift;
	my $domain = shift;
	&_init($domain);
	my $self = bless {}, (ref($this) || $this);
	$self->{'ua'}  = LWP::UserAgent->new('agent' => "JSON::RPC::Client", 'default_headers'=>HTTP::Headers->new('Accept'=>"application/json"), 'ssl_opts'=>{'verify_hostname'=>0}, 'timeout'=>10);
	$self->{'json'} = JSON->new->utf8->allow_nonref->allow_blessed->convert_blessed;
	$self->auth();
	return $self;
}

sub DESTROY
{
	my $self = shift;
}

=head2 cmd
Выполнение запроса JSON-RPC и получение ответа
=cut
sub cmd($;$$)
{
	my $self = shift;
	my $obj = shift;
	my $url = shift;
	$url = $cfg->{'rpc'}->{'address'} . $cfg->{'rpc'}->{'sql'} unless (defined($url));
	my $res;
	if ($url =~ /\?/)
	{
		$res = $self->{'ua'}->get($url);
	}
	else
	{
		my $data = $self->{'json'}->encode($obj);
		$res = $self->{'ua'}->post($url, 'Content_Type'=>"application/json", 'Content'=>$data);
	}
	return unless ($res);
	return "#".$res->status_line if ($res->is_error);
	return {} unless ($res->content);
	my $data = $self->{'json'}->decode($res->decoded_content);
	if (ref($data) eq 'ARRAY')
	{
		return {map {$_->{'id'}=>($_->{'error'} ? $_->{'error'} : $_->{'result'})} @$data};
	}
	else
	{
		return (ref($data->{'error'}) ? $data->{'error'}->{'message'} : $data->{'error'}) if ($data->{'error'});
		return $data->{'result'};
	}
}

=head2 obj
Сформировать тело запроса для JSON-RPC
=cut
sub obj($;$$)
{
	my $self = shift;
	my $method = shift;
	my $params = shift;
	my $id = shift;
	my $auto = (defined($id) ? 0 : 1);
	$id = 1+($self->{'last_id'}||0) if ($auto);
	my $json; tie %$json, 'Tie::IxHash';
	$json->{'jsonrpc'} = $cfg->{'rpc'}->{'version'} if (defined($cfg->{'rpc'}->{'version'}));
	$json->{'method'} = $method;
	$json->{'params'} = $params if (defined($params));
	$json->{'id'} = $id if (defined($id));
	$self->{'last_id'} = $id if ($auto);
	return $json;
}

=head2 init
Инициализация сеанса JSON-RPC
=cut
sub auth
{
	my $self = shift;
	my $tmp = "/tmp/iptvportal-$cfg->{'rpc'}->{'domain'}-session.tmp";
	my $sid;
	if (-e $tmp)
	{
		open TMP, '<', $tmp;
		my $t = <TMP>;
		close TMP;
		if ($t)
		{
			my ($t1,$t2) = split(' ', $t);
			if ((time - $t1) < $cfg->{'rpc'}->{'expire'})
			{
				$sid = $t2;
			}
		}
	}
	unless (defined($sid))
	{
		my $auth = $self->obj('authorize_user', {'username'=>$cfg->{'rpc'}->{'username'}, 'password'=>$cfg->{'rpc'}->{'password'}});
		my $url = $cfg->{'rpc'}->{'address'} . $cfg->{'rpc'}->{'api'};
		my $res = $self->cmd($auth, $url);
		return undef unless (ref($res) eq 'HASH');
		$sid = $res->{'session_id'};
		open TMP, '>', $tmp;
		print TMP time . " " . $sid;
		close TMP;
	}
	$self->{'sid'} = $sid;
	$self->{'ua'}->default_header('Iptvportal-Authorization' => "sessionid=$sid");
	return $sid;
}

=head2 retrieve
Получить списки основных объектов БД.
=cut
sub retrieve
{
	my $self = shift;
	my @list = @_;
	my $fldmap = {};
	my $batch = [];
	foreach my $item (@list)
	{
		if (exists($cfg->{'json'}->{'list'}->{$item}))
		{
			my $obj = $self->obj('select', $cfg->{'json'}->{'list'}->{$item}, $item);
			$fldmap->{$item} = $obj->{'params'}->{'data'};
			push @$batch, $obj;
		}
	}
	my $ret = $self->cmd($batch);
	my $res = {};
	foreach my $item (keys(%$ret))
	{
		my $rows = $ret->{$item};
		return undef unless ($rows && ref($rows) eq 'ARRAY');
		my $fld = $fldmap->{$item};
		$res->{$item} = undef;
		if ($item ~~ [qw(domain)])
		{
			$rows = $rows->[0];
			$res->{$item} = {'id'=>$rows->[0], 'name'=>$rows->[1]};
			next;
		}
		$res->{$item} = [];
		foreach my $row (@$rows)
		{
			push @{$res->{$item}}, {map {$fld->[$_]=>$row->[$_]} (0..$#{$fld})};
		}
	}
	return $res;
}


1;


__DATA__

#json @@ list @ domain
{
"data": ["id", "name"],
"from": "domain",
}

#json @@ list @ playlist
{
"data": ["id", "name", "title"],
"from": "playlist",
"where": {"eq": ["is_tv", true]},
}

#json @@ list @ package
{
"data": ["id", "name", "paid"],
"from": "package",
}

#json @@ list @ profile
{
"data": ["id", "name"],
"from": "profile",
}

#json @@ list @ sprofile
{
"data": ["id", "name"],
"from": "sprofile",
}

#json @@ list @ pprofile
{
"data": ["id", "name"],
"from": "pprofile",
}

#json @@ list @ media
{
"data": ["id", "index", "name", "protocol","inet_addr","port", "disabled", "channel_id", "encrypted", "key_id", "is_tv"],
"from": "media",
}

#json @@ list @ playlist_media
{
"data": ["media_id", "playlist_id", "index", "disabled"],
"from": "playlist_media",
}

#json @@ list @ package_media
{
"data": ["media_id", "package_id", "disabled"],
"from": "package_media",
}

#json @@ list @ subscriber
{
"data": ["id", "username", "password", "max_terminal", "sprofile_id", "language_id", "first_name", "middle_name", "surname", "email", "address", "phone", "disabled", "comment"],
"from": "subscriber",
}

#json @@ list @ subscriber_package
{
"data": ["subscriber_id", "package_id", "enabled"],
"from": "subscriber_package",
}

#json @@ list @ subscriber_media
{
"data": ["subscriber_id", "media_id", "enabled"],
"from": "subscriber_media",
}

#json @@ list @ profile_media
{
"data": ["domain_id", "media_id", "profile_id", "disabled"],
"from": "profile_media",
}

#json @@ list @ sprofile_media
{
"data": ["media_id", "sprofile_id", "disabled"],
"from": "sprofile_media",
}

#json @@ list @ pprofile_media
{
"data": ["media_id", "pprofile_id", "disabled"],
"from": "pprofile_media",
}


__END__

$log->run("# Формирование пакетного задания");
my $batch = {'insert'=>[], 'delete'=>[], 'update'=>[], 'packages'=>[], 'channels'=>[]};
my $cnt = {'all'=>0, 'ena'=>0, 'dis'=>0, 'ins'=>0, 'del'=>0, 'upd'=>0};
$key->{'package'}->{$_}->{'count'} = {} foreach (keys(%{$key->{'package'}}));
foreach my $sub (@{$cache->{'subscriber'}})
{
	my $id = $sub->{'id'};
	my $login = $sub->{'username'};
	next if ($login =~ /^!/);
	unless (exists($list->{$login}))
	{
		$log->act("- $login - удален");
		push @{$batch->{'delete'}}, $id;
		$cnt->{'del'}++;
		next;
	}
	$cnt->{'all'}++;
	my $usr = $list->{$login};
	delete $list->{$login};
	my $update = {};
	if ($sub->{'password'} ne lc(md5_hex($usr->{'PASSWORD'}||'')))
	{
		$update->{'password'} = {'value'=>lc(md5_hex($usr->{'PASSWORD'}||'')), 'label'=>"пароль"};
	}
	if ($sub->{'surname'} ne $usr->{'TARIFF'})
	{
		$update->{'surname'} = {'value'=>$usr->{'TARIFF'}, 'label'=>"тариф"};
	}
	if (($sub->{'max_terminal'}||0) != ($usr->{'TERMINALS'}||0))
	{
		$update->{'max_terminal'} = {'value'=>$usr->{'TERMINALS'}, 'label'=>"терминалы"};
	}
	if (($sub->{'disabled'} ? 0 : 1) != ($usr->{'STATE'} ? 1 : 0))
	{
		$update->{'disabled'} = {'value'=>($usr->{'STATE'} ? undef : true), 'label'=>"статус"};
		$cnt->{($usr->{'STATE'} ? "ena" : "dis")}++;
	}
	$key->{'package'}->{$_}->{'count'}->{$login} = $usr->{'STATE'} foreach (keys %{$usr->{'_PACKAGE'}});
	$usr->{'_PACKAGE_LIST'} = [sort map {$key->{'package'}->{$_}->{'id'}} grep {exists($key->{'package'}->{$_}) && exists($key->{'package'}->{$_}->{'id'})} (keys %{$usr->{'_PACKAGE'}})];
	$usr->{'_CHANNEL_LIST'} = [sort map {$key->{'channel'}->{$_}} grep {exists($key->{'channel'}->{$_})} (keys %{$usr->{'_CHANNEL'}})];
	my @pa = sort grep {$cache->{'subscriber_package'}->{$id}->{$_}} (keys %{$cache->{'subscriber_package'}->{$id}});
	if (!(\@pa ~~ $usr->{'_PACKAGE_LIST'}))
	{
		$update->{'tariff'} = {'label'=>"пакеты"};
		push @{$batch->{'packages'}}, {'key'=>$login, 'list'=>$usr->{'_PACKAGE_LIST'}};
	}
	my @ca = sort grep {$cache->{'subscriber_media'}->{$id}->{$_}} (keys %{$cache->{'subscriber_media'}->{$id}});
	if (!(\@ca ~~ $usr->{'_CHANNEL_LIST'}))
	{
		$update->{'channel'} = {'label'=>"каналы"};
		push @{$batch->{'channels'}}, {'key'=>$login, 'list'=>$usr->{'_CHANNEL_LIST'}};
	}
	if (%$update)
	{
		$update->{'comment'} = {'value'=>"Обновлено: " . $log->_timestamp()};
		$log->act("- $login - обновлен (" . join(', ', grep {defined($_)} map {$_->{'label'}} values(%$update)) . ")");
		delete $update->{$_} foreach grep {!exists($update->{$_}->{'value'})} (keys %$update);
		push @{$batch->{'update'}}, {'id'=>$id, 'update'=>$update};
		$cnt->{'upd'}++;
	}
}
foreach my $usr (values(%$list))
{
	my $login = $usr->{'USERNAME'} ;
	$log->act("- $login - добавлен" . ($usr->{'STATE'} ? " (активный)" : " (неактивный)"));
	my $update = {'username'=>$login};
	$update->{'password'} = lc(md5_hex($usr->{'PASSWORD'}||''));
	$update->{'surname'} = $usr->{'TARIFF'};
	$update->{'max_terminal'} = $usr->{'TERMINALS'};
	$update->{'disabled'} = ($usr->{'STATE'} ? undef : true);
	$update->{'comment'} = "Добавлено: " . $log->_timestamp();
	$cnt->{'all'}++;
	$cnt->{'ins'}++;
	$cnt->{($usr->{'STATE'} ? "ena" : "dis")}++;
	$key->{'package'}->{$_}->{'count'}->{$login} = $usr->{'STATE'} foreach (keys %{$usr->{'_PACKAGE'}});
	$usr->{'_PACKAGE_LIST'} = [sort map {$key->{'package'}->{$_}->{'id'}} grep {exists($key->{'package'}->{$_}) && exists($key->{'package'}->{$_}->{'id'})} (keys %{$usr->{'_PACKAGE'}})];
	$usr->{'_CHANNEL_LIST'} = [sort map {$key->{'channel'}->{$_}} grep {exists($key->{'channel'}->{$_})} (keys %{$usr->{'_CHANNEL'}})];
	push @{$batch->{'insert'}}, $update;
	push @{$batch->{'packages'}}, {'key'=>$login, 'list'=>$usr->{'_PACKAGE_LIST'}} if (@{$usr->{'_PACKAGE_LIST'}});
	push @{$batch->{'channels'}}, {'key'=>$login, 'list'=>$usr->{'_CHANNEL_LIST'}} if (@{$usr->{'_CHANNEL_LIST'}});
}

$log->run("# Выполнение пакетного задания...");
{
	my $job = [];
	my $qs = ($meta->{'config'}->{'batch'}||1);
	if (@{$batch->{'delete'}})
	{
		$log->print("- Удаление - " . scalar(@{$batch->{'delete'}}) . "...");
		while (my @batch = splice(@{$batch->{'delete'}}, 0, $qs))
		{
			my $req = &json('batch','delete');
			$req->{'where'}->{'in'} = [@{$req->{'where'}->{'in'}}, @batch];
			my $cmd = $tv->obj('delete', $req);
			push @$job, $cmd;
		}
	}
	my $new = {};
	if (@{$batch->{'insert'}})
	{
		$log->print("- Добавление - " . scalar(@{$batch->{'insert'}}) . "...");
		my $i = 0;
		while (my @batch = splice(@{$batch->{'insert'}}, 0, $qs))
		{
			my $req = &json('batch','insert');
			foreach my $insert (@batch)
			{
				push @{$req->{'values'}}, [map {$insert->{$_}} @{$req->{'columns'}}];
			}
			my $cmd = $tv->obj('insert', $req, "new#$i");
			push @$job, $cmd;
			$new->{"new#$i"} = [map {$_->{'username'}} @batch];
			$i += scalar(@batch);
		}
	}
	if (@{$batch->{'update'}})
	{
		$log->print("- Обновление - " . scalar(@{$batch->{'update'}}) . "...");
		while (my @batch = splice(@{$batch->{'update'}}, 0, $qs))
		{
			foreach my $update (@batch)
			{
				my $req = &json('batch','update');
				push @{$req->{'where'}->{'eq'}}, $update->{'id'};
				$req->{'set'} = {map {$_=>$update->{'update'}->{$_}->{'value'}}keys(%{$update->{'update'}})};
				my $cmd = $tv->obj('update', $req);
				push @$job, $cmd;
			}
		}
	}
	my $res;
	if ($job && @$job)
	{
		$res = $tv->cmd($job);
		foreach my $label (keys(%$new))
		{
			my @rows = $res->{$label};
			my @user = @{$new->{$label}};
			foreach my $i (0..$#rows)
			{
				$key->{'client'}->{$user[$i]} = $rows[$i];
				$idx->{'client'}->{$rows[$i]} = $user[$i];
			}
		}
		$job = undef;
		$res = undef;
	}
	if (@{$batch->{'packages'}})
	{
		$log->print("- Обновление пакетов - " . scalar(@{$batch->{'packages'}}) . "...");
		$job = [];
		while (my @batch = splice(@{$batch->{'packages'}}, 0, $qs))
		{
			foreach my $item (@batch)
			{
				my $lst = $item->{'list'};
				my $id = $key->{'client'}->{$item->{'key'}} || next;
				my $req = &json('batch','package_clean');
				push @{$req->{'where'}->{'eq'}}, $id;
				push @$job, $tv->obj('delete', $req);
				if (@$lst)
				{
					$req = &json('batch','package_insert');
					push @{$req->{'values'}}, map {[$id, $_, true]} @$lst;
					push @$job, $tv->obj('insert', $req);
				}
			}
		}
		$res = $tv->cmd($job) if (@$job);
		$job = undef;
		$res = undef;
	}
	if (@{$batch->{'channels'}})
	{
		$log->print("- Обновление каналов - " . scalar(@{$batch->{'channels'}}) . "...");
		$job = [];
		while (my @batch = splice(@{$batch->{'channels'}}, 0, $qs))
		{
			foreach my $item (@batch)
			{
				my $lst = $item->{'list'};
				my $id = $key->{'client'}->{$item->{'key'}} || next;
				my $req = &json('batch','channel_clean');
				push @{$req->{'where'}->{'eq'}}, $id;
				push @$job, $tv->obj('delete', $req);
				if (@$lst)
				{
					$req = &json('batch','channel_insert');
					push @{$req->{'values'}}, map {[$id, $_, true]} @$lst;
					push @$job, $tv->obj('insert', $req);
				}
			}
		}
		$res = $tv->cmd($job) if (@$job);
		$job = undef;
		$res = undef;
	}
}