0c5720130a3e8611282ff3dd58b779bc72aeb40c
[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                 my @filter_values = @{ $filters->{$n} };
168                 my $include_missing = grep { /^_missing/ } @filter_values;
169                 if ( ! exists $i->{$n} ) {
170                         if ( $include_missing ) {
171                                 $pass = 1;
172                                 next;
173                         } else {
174                                 $pass = 0;
175                                 last;
176                         }
177                 }
178                 # and match any of values in element
179                 my $have_values = 0;
180                 foreach my $v ( @{ $i->{$n} } ) { # FIXME not array?
181                         $have_values ||= 1 if grep { m/^\Q$v\E$/ } @filter_values;
182                 }
183                 if ( ! $have_values ) {
184                         $pass = 0;
185                         last;
186                 }
187         }
188         return $pass;
189 }
190
191 sub _data_items {
192         my $self = shift;
193         my $filters = $self->session('filters');
194         grep {
195                 $filters ? $self->_filter_item( $filters, $_ ) : 1;
196         } @{ $data->{items} };
197 }
198
199 sub table {
200     my $self = shift;
201
202         $self->redirect_to('/data/index') unless $data->{items};
203
204         my @columns = $self->_perm_array('columns');
205         $self->redirect_to('/data/columns') unless @columns;
206         my $order   = $self->_perm_scalar('order', $columns[0]);
207         my $sort    = $self->_perm_scalar('sort', 'a');
208         my $offset  = $self->_perm_scalar('offset', 0);
209         my $limit   = $self->_perm_scalar('limit', 20);
210
211         # fix offset when changing limit
212         $offset = int( $offset / $limit ) * $limit;
213
214         # FIXME - multi-level sort
215         my $numeric = $self->_is_numeric($order);
216         my @sorted = sort {
217                 my $v1 = exists $a->{$order} ? join('', @{$a->{$order}}) : '';
218                 my $v2 = exists $b->{$order} ? join('', @{$b->{$order}}) : '';
219                 ($v1,$v2) = ($v2,$v1) if $sort eq 'd';
220                 $numeric ? $v1 <=> $v2 : $v1 cmp $v2 ;
221         } $self->_data_items;
222
223 #       warn "# sorted ", dump @sorted;
224
225         my $rows = $#sorted + 1;
226
227         $self->render(
228                 order => $order,
229                 offset => $offset,
230                 limit => $limit,
231                 sorted => [ splice @sorted, $offset, $limit ],
232                 columns => [ @columns ],
233                 rows => $rows,
234                 numeric => { map { $_, $self->_is_numeric($_) } @columns },
235         );
236
237 }
238
239
240 sub order {
241         my $self = shift;
242         $self->session('order', $self->param('order'));
243         $self->session('sort', $self->param('sort'));
244         $self->redirect_to('/data/table');
245 }
246
247 sub _is_numeric {
248         my ( $self, $name ) = @_;
249
250         # sort facet numerically if more >50% elements are numeric
251         defined $stats->{$name}->{numeric} &&
252                 $stats->{$name}->{numeric} > $stats->{$name}->{count} / 2;
253 }
254
255 sub facet {
256         my $self = shift;
257
258         if ( my $remove = $self->param('remove') ) {
259                 my $f = $self->session('filters');
260                 delete $f->{$remove};
261                 $self->session( 'filters' => $f );
262                 $self->redirect_to( '/data/table' );
263         }
264
265         my $facet;
266         my $name = $self->param('name') || die "no name";
267
268         foreach my $i ( $self->_data_items ) {
269                 if ( ! exists $i->{$name} ) {
270                         $facet->{ _missing }++;
271                 } elsif ( ref $i->{$name} eq 'ARRAY' ) {
272                         $facet->{$_}++ foreach @{ $i->{$name} };
273                 } else {
274                         $facet->{ $i->{$name} }++;
275                 }
276         }
277
278 #       warn "# facet $name ",dump $facet;
279
280         my $checked;
281         if ( my $f = $self->session('filters') ) {
282                 if ( defined $f->{$name} ) {
283                         $checked = $self->_checked( @{ $f->{$name} } );
284                 }
285         }
286
287         my $sort = $self->param('sort') || 'c';
288
289         # sort facet numerically if more >50% elements are numeric
290         my $numeric = $self->_is_numeric($name);
291
292         my @facet_names = sort {
293                 if ( $sort =~ m/a/i ) {
294                         $numeric ? $a <=> $b : lc $a cmp lc $b;
295                 } elsif ( $sort =~ m/d/i ) {
296                         $numeric ? $b <=> $a : lc $b cmp lc $a;
297                 } elsif ( $sort =~ m/c/i ) {
298                         $facet->{$b} <=> $facet->{$a};
299                 } else {
300                         warn "unknown sort: $sort";
301                         $a cmp $b;
302                 }
303         } keys %$facet;
304
305         $self->render( name => $name, facet => $facet, checked => $checked,
306                 facet_names => \@facet_names, sort => $sort, numeric => $numeric,
307         );
308 }
309
310 sub _checked {
311         my $self = shift;
312         my $checked;
313         $checked->{$_}++ foreach @_;
314         warn "# _checked ",dump($checked);
315         return $checked;
316 }
317
318 1;