use locale to help sorting
[MojoFacets.git] / lib / MojoFacets / Data.pm
1 package MojoFacets::Data;
2
3 use strict;
4 use warnings;
5
6 use base 'Mojolicious::Controller';
7
8 use Data::Dump qw(dump);
9 use File::Slurp;
10 use JSON;
11 use Encode;
12 use locale;
13
14 sub index {
15         my $self = shift;
16
17         my $path = $self->app->home->rel_dir('data');
18         die "no data dir $path" unless -d $path;
19
20         opendir(my $dir, $path) || die $!;
21         my @files = 
22                 grep { -f "$path/$_" && $_ =~ m/\.js(on)?$/ }
23                 readdir $dir;
24         close($dir);
25
26         $self->render( files => [ @files ] );
27 }
28
29 our $data;
30 our $stats;
31
32 sub load {
33         my $self = shift;
34
35         my $path = $self->app->home->rel_file( 'data/' . $self->param('path') );
36         die "$path $!" unless -r $path;
37
38         $self->session('path' => $self->param('path'));
39
40         # we could use Mojo::JSON here, but it's too slow
41 #       $data = from_json read_file $path;
42         $data = read_file $path;
43         Encode::_utf8_on($data);
44         warn "# json snippet: ", substr($data,0,200);
45         $data = from_json $data;
46
47         $stats = {};
48
49         foreach my $e ( @{ $data->{items} } ) {
50                 foreach my $n ( keys %$e ) {
51                         $stats->{$n}->{count}++;
52                         if ( ref $e->{$n} eq 'ARRAY' ) {
53
54                                 $stats->{$n}->{array} += $#{ $e->{$n} } + 1;
55
56                                 foreach my $x ( @{$e->{$n}} ) {
57                                         $stats->{$n}->{numeric}++
58                                                 if $x =~ m/^[-+]?([0-9]*\.[0-9]+|[0-9]+)$/;
59                                 }
60
61                         } else {
62                                 $stats->{$n}->{numeric}++
63                                         if $e->{$n} =~ m/^[-+]?([0-9]*\.[0-9]+|[0-9]+)$/;
64                         }
65                 }
66         }
67
68         foreach my $n ( keys %$stats ) {
69                 next unless defined $stats->{$n}->{array};
70                 delete $stats->{$n}->{array}
71                         if $stats->{$n}->{array} == $stats->{$n}->{count};
72         }
73
74         warn dump($stats);
75
76         $self->redirect_to( '/data/columns' );
77 }
78
79
80 sub columns {
81     my $self = shift;
82
83
84         my @columns;
85         @columns = @{ $self->session('columns') } if $self->session('columns');
86
87         foreach my $c ( sort { $stats->{$b}->{count} <=> $stats->{$a}->{count} } keys %$stats ) {
88                 push @columns, $c unless grep { /^\Q$c\E$/ } @columns;
89         }
90
91         $self->redirect_to( '/data/index' ) unless @columns;
92
93     $self->render(
94                 message => 'Select columns to display',
95                 stats => $stats,
96                 columns => \@columns,
97                 checked => $self->_checked( $self->_perm_array('columns') ),
98         );
99 }
100
101 sub _perm_array {
102     my ($self,$name) = @_;
103
104         my @array = $self->param($name);
105
106         if ( @array ) {
107                 $self->session($name => [ @array ]);
108         } elsif ( my $session = $self->session($name) ) {
109                 if ( ref $session eq 'ARRAY' ) {
110                         @array = @$session;
111                 } else {
112                         die "$name not array ",dump($session);
113                 }
114         }
115         warn "# $name ",dump @array;
116         return @array;
117 }
118
119 sub _perm_scalar {
120     my ($self,$name,$default) = @_;
121
122         my $scalar = $self->param($name);
123
124         if ( defined $scalar ) {
125                 $self->session($name => $scalar);
126         } else {
127                 $scalar = $self->session($name);
128         }
129
130         if ( ! defined $scalar ) {
131                 $scalar = $default;
132                 die "no default for $name" unless defined $scalar;
133                 $self->session($name => $scalar);
134         }
135
136         warn "# $name ",dump $scalar;
137         return $scalar;
138 }
139
140 sub filter {
141         my $self = shift;
142
143         my $name = $self->param('filter_name') || die "name?";
144         my @vals = $self->param('filter_vals');
145
146         warn "# filter $name vals ",dump(@vals);
147
148         my $filters = $self->session('filters');
149         if ( @vals ) {
150                 $filters->{$name} = [ @vals ];
151         } else {
152                 delete $filters->{$name};
153         }
154         $self->session( 'filters' => $filters );
155
156         warn "# filters ",dump($self->session('filters'));
157
158         $self->session( 'offset' => 0 );
159
160         $self->redirect_to('/data/table');
161 }
162
163 sub _filter_item {
164         my ( $self, $filters, $i ) = @_;
165         my $pass = 1;
166         foreach my $n ( keys %$filters ) {
167                 # filter must exists in element
168                 if ( ! defined $i->{$n} ) {
169                         $pass = 0;
170                         last;
171                 }
172                 # and match any of values in element
173                 my $have_values = 0;
174                 foreach my $v ( @{ $i->{$n} } ) { # FIXME not array?
175                         $have_values ||= 1 if grep { m/^\Q$v\E$/ } @{ $filters->{$n} };
176                 }
177                 if ( ! $have_values ) {
178                         $pass = 0;
179                         last;
180                 }
181         }
182         return $pass;
183 }
184
185 sub _data_items {
186         my $self = shift;
187         my $filters = $self->session('filters');
188         grep {
189                 $filters ? $self->_filter_item( $filters, $_ ) : 1;
190         } @{ $data->{items} };
191 }
192
193 sub table {
194     my $self = shift;
195
196         $self->redirect_to('/data/index') unless $data->{items};
197
198         my @columns = $self->_perm_array('columns');
199         $self->redirect_to('/data/columns') unless @columns;
200         my $order   = $self->_perm_scalar('order', $columns[0]);
201         my $sort    = $self->_perm_scalar('sort', 'a');
202         my $offset  = $self->_perm_scalar('offset', 0);
203         my $limit   = $self->_perm_scalar('limit', 20);
204
205         # FIXME - multi-level sort
206         my $numeric = $self->_is_numeric($order);
207         my @sorted = sort {
208                 my ($v1,$v2) = ( $a->{$order}->[0], $b->{$order}->[0] );
209                 ($v1,$v2) = ($v2,$v1) if $sort eq 'd';
210                 $numeric
211                         ? $v1 <=> $v2
212                         : ( $v1 || '' ) cmp ( $v2 || '' )
213                         ;
214         } $self->_data_items;
215
216 #       warn "# sorted ", dump @sorted;
217
218         my $rows = $#sorted + 1;
219
220         $self->render(
221                 order => $order,
222                 offset => $offset,
223                 limit => $limit,
224                 sorted => [ splice @sorted, $offset, $limit ],
225                 columns => [ @columns ],
226                 rows => $rows,
227                 numeric => { map { $_, $self->_is_numeric($_) } @columns },
228         );
229
230 }
231
232
233 sub order {
234         my $self = shift;
235         $self->session('order', $self->param('order'));
236         $self->session('sort', $self->param('sort'));
237         $self->redirect_to('/data/table');
238 }
239
240 sub _is_numeric {
241         my ( $self, $name ) = @_;
242
243         # sort facet numerically if more >50% elements are numeric
244         defined $stats->{$name}->{numeric} &&
245                 $stats->{$name}->{numeric} > $stats->{$name}->{count} / 2;
246 }
247
248 sub facet {
249         my $self = shift;
250
251         if ( my $remove = $self->param('remove') ) {
252                 my $f = $self->session('filters');
253                 delete $f->{$remove};
254                 $self->session( 'filters' => $f );
255                 $self->redirect_to( '/data/table' );
256         }
257
258         my $facet;
259         my $name = $self->param('name') || die "no name";
260
261         foreach my $i ( $self->_data_items ) {
262                 next unless exists $i->{$name};
263                 if ( ref $i->{$name} eq 'ARRAY' ) {
264                         $facet->{$_}++ foreach @{ $i->{$name} };
265                 } else {
266                         $facet->{ $i->{$name} }++;
267                 }
268         }
269
270 #       warn "# facet $name ",dump $facet;
271
272         my $checked;
273         if ( my $f = $self->session('filters') ) {
274                 if ( defined $f->{$name} ) {
275                         $checked = $self->_checked( @{ $f->{$name} } );
276                 }
277         }
278
279         my $sort = $self->param('sort') || 'c';
280
281         # sort facet numerically if more >50% elements are numeric
282         my $numeric = $self->_is_numeric($name);
283
284         my @facet_names = sort {
285                 if ( $sort =~ m/a/i ) {
286                         $numeric ? $a <=> $b : lc $a cmp lc $b;
287                 } elsif ( $sort =~ m/d/i ) {
288                         $numeric ? $b <=> $a : lc $b cmp lc $a;
289                 } elsif ( $sort =~ m/c/i ) {
290                         $facet->{$b} <=> $facet->{$a};
291                 } else {
292                         warn "unknown sort: $sort";
293                         $a cmp $b;
294                 }
295         } keys %$facet;
296
297         $self->render( name => $name, facet => $facet, checked => $checked,
298                 facet_names => \@facet_names, sort => $sort, numeric => $numeric,
299         );
300 }
301
302 sub _checked {
303         my $self = shift;
304         my $checked;
305         $checked->{$_}++ foreach @_;
306         warn "# _checked ",dump($checked);
307         return $checked;
308 }
309
310 1;