From Redis to Postgres - Part 2 (Better routes)

[gruntmaster-page.git] / lib / Gruntmaster / Page / Generic.pm

package Gruntmaster::Page::Generic;

use 5.014000;
use strict;
use warnings;
our $VERSION = '5999.000_001';

use Gruntmaster::Page::Base;
use JSON qw/encode_json decode_json/;
use Plack::Request;
use Sub::Name qw/subname/;

use constant PAGE_SIZE => 10;

sub putsym {
	my ($key, $value) = @_;
	no strict 'refs';
	subname $key => $value if ref $value eq 'CODE';
	*{"$key"} = $value;
}

sub makepkg {
	my ($pkg, $id, $title) = @_;
	my $fn = $pkg =~ s,::,/,gr;
	return if $INC{"$fn.pm"};
	$INC{"$fn.pm"} = 1;
	Gruntmaster::Page::Base->import_to($pkg, $id, $title);
	1
}

sub list {
	my ($thing, $lang, $env) = @_;
	my %thing = %$thing;
	#debug $env => "Contest is $ct";
	$thing{makers} //= sub { db(shift)->resultset($thing{rsname}) };
	my $rs = $thing{makers}->($env);
	$rs = $rs->search(undef, {order_by => 'me.id'}) unless $rs->is_ordered;
	if (my $page = $env->{'gruntmaster.page'}) {
		my $pages = $rs->count / PAGE_SIZE;
		$rs = $rs->search(undef, {offset => ($page - 1) * PAGE_SIZE, ($page == $pages ? () : (rows => PAGE_SIZE))});
	}
	$rs = $rs->search(undef, {prefetch => $thing{prefetch}}) if exists $thing{prefetch};
	my @thing = map +{rs => $_, $_->get_columns}, $rs->all;
	@thing = map  { $thing{mangle}->(); $_ } @thing if exists $thing{mangle};
	@thing = grep { $thing{choose}->() } @thing if exists $thing{choose};
	@thing = sort { $thing{sortby}->() } @thing if exists $thing{sortby};
	my %params;
	$thing{group} //= sub { $thing{id} };
	for (@thing) {
		my $group = $thing{group}->();
		$params{$group} //= [];
		push $params{$group}, $_
	}
	wantarray ? %params : \%params
}

sub entry {
	my ($thing, $lang, $env, $id) = @_;
	my %thing = %$thing;
	debug $env => "Rsname is $thing{rsname} and id is $id";
	$thing{makers} //= sub { db(shift)->resultset($thing{rsname}) };
	my %params = map {+ rs => $_, $_->get_columns } $thing{makers}->($env)->find($id);
	$thing{mangle}->(local $_ = \%params) if exists $thing{mangle};
	wantarray ? %params : \%params
}

sub headers ($) { ['Content-Type' => 'application/json', 'Cache-Control' => 'max-age=' . $_[0]->max_age] }

sub create_thing {
	my %thing = @_;
	my $ucid = ucfirst $thing{id};
	my $pkg = "Gruntmaster::Page::$ucid";

	putsym "${pkg}::_generate", sub { $_[1]->param(list \%thing, @_[2..$#_]) } if makepkg $pkg, @thing{qw/id title/};
	putsym "${pkg}::Entry::_generate",  sub { $_[1]->param(entry \%thing, @_[2..$#_]) } if makepkg "${pkg}::Entry", "$thing{id}_entry", '<tmpl_var name>';
	putsym "${pkg}::Read::generate", sub { [200, headers shift, [encode_json list \%thing, @_]] } if makepkg "${pkg}::Read";
	putsym "${pkg}::Entry::Read::generate", sub { [200, headers shift, [encode_json entry \%thing, @_]] } if makepkg "${pkg}::Entry::Read";
}

sub params;
sub makers (&);
sub choose (&);
sub sortby (&);
sub group  (&);
sub mangle (&);
sub prefetch;

sub thing (&){
	my %thing;
	no strict 'refs';
	local *{"params"} = sub { @thing{qw/id rsname title/} = @_ };
	local *{"choose"} = sub { $thing{choose} = shift };
	local *{"sortby"} = sub { $thing{sortby} = shift };
	local *{"mangle"} = sub { $thing{mangle} = shift };
	local *{"group"} = sub { $thing{group} = shift };
	local *{"makers"} = sub { $thing{makers} = shift };
	local *{"prefetch"} = sub { $thing{prefetch} = \@_ };
	use strict 'refs';

	shift->();
	create_thing %thing
}

##################################################

thing {
	params qw/us User Users/;
	choose { $_->{name} =~ /\w/ };
	sortby { lc $a->{name} cmp lc $b->{name} };
};

thing {
	params qw/pb Problem Problems/;
	prefetch 'owner';
	makers {
		my $env = $_[0];
		my $db = db $env;
		return $db->contest($env->{'gruntmaster.contest'})->problems->search(undef, {order_by => 'problem.id'}) if exists $env->{'gruntmaster.contest'};
		return $db->problems->search({owner => $env->{'gruntmaster.user'}}) if exists $env->{'gruntmaster.user'};
		$db->problems->search({private => 0});
	};
	sortby { $a->{name} cmp $b->{name}};
	group { $_->{level} };
	mangle {
		my $env = shift;
		$_->{owner_name} = $_->{rs}->owner->name;
		$_->{cansubmit} = $_->{contest} ? time < $_->{rs}->contest->stop : 1;
		eval {
			db($env)->open->create({
				contest => $_->{contest},
				problem => $_->{id},
				owner   => $env->{REMOTE_USER},
			})
		} if $_->{contest} && time >= $_->{rs}->contest->start;
	};
};

thing {
	params qw/ct Contest Contests/;
	prefetch 'owner';
	sortby { $b->{start} <=> $a->{start} };
	group { time < $_->{start} ? 'pending' : time > $_->{stop} ? 'finished' : 'running' };
	mangle { $_->{started} = time >= $_->{start}; $_->{owner_name} =  $_->{rs}->owner->name };
};

thing {
	params qw/log Job/, 'Job log';
	prefetch 'owner', 'problem';
	makers {
		my $env = $_[0];
		my $db = db $env;
		return $db->jobs->search({'me.owner' => $env->{'gruntmaster.user'}}) if exists $env->{'gruntmaster.user'};
		return $db->jobs->search({problem => $env->{'gruntmaster.problem'}}) if exists $env->{'gruntmaster.problem'};
		$db->jobs->search({contest => $env->{'gruntmaster.contest'}})
	};
	sortby { $b->{id} <=> $a->{id}};
	mangle {
		$_->{results} &&= decode_json $_->{results};
		$_->{owner_name} = $_->{rs}->owner->name;
		$_->{problem_name} = $_->{rs}->problem->name;
		$_->{size} = length $_->{source};
		delete $_->{source};
	}
};

putsym 'Gruntmaster::Page::Pb::Entry::vary',    sub { 'Authorization' };
putsym 'Gruntmaster::Page::Pb::Entry::max_age', sub { 600 };

1