added .pairs format NAME1="VAL1" NAME2="VAL2" ...
[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 Encode;
11 use locale;
12 use File::Find;
13 use Storable;
14 use Time::HiRes qw(time);
15 use File::Path qw(mkpath);
16 use Text::Unaccent;
17 use Digest::MD5;
18 use Statistics::Descriptive;
19
20 use MojoFacets::Import::File;
21 use MojoFacets::Import::HTMLTable;
22 use MojoFacets::Import::CSV;
23 use MojoFacets::Import::CouchDB;
24 use MojoFacets::Import::SQL;
25 use MojoFacets::Import::Pairs;
26
27 our $loaded;
28 our $filters;
29
30 sub index {
31         my $self = shift;
32
33         my $data_dir = $self->app->home->rel_dir('data');
34         die "no data dir $data_dir" unless -d $data_dir;
35
36         my @files;
37         my $changes;
38         find( sub {
39                 my $file = $File::Find::name;
40                 if ( -f $file && $file =~ m/\.(js(on)?|txt)$/ ) {
41                         $file =~ s/$data_dir\/*//;
42                         push @files, $file;
43                 } elsif ( -f $file && $file =~ m/([^\/]+)\.changes\/(\d+\.\d+.+)/ ) {
44                         push @{ $changes->{$1} }, $2
45                 } elsif ( -d $file && $file =~ m/\.html$/ ) {
46                         $file =~ s/$data_dir\/*//;
47                         push @files, $file;
48                 } elsif ( -f $file && $file =~ m/\.(csv|storable|couchdb|sql|pairs)$/i ) {
49                         $file =~ s/$data_dir\/*//;
50                         push @files, $file;
51                 } else {
52                         #warn "IGNORE: $file\n";
53                 }
54         }, $data_dir);
55
56         no warnings qw(uninitialized); # mtime
57         @files = sort { $loaded->{$b}->{mtime} <=> $loaded->{$a}->{mtime} || lc $a cmp lc $b } @files,
58                         grep { defined $loaded->{$_}->{generated} } keys %$loaded;
59
60         my $size;
61         $size->{$_} = -s "$data_dir/$_" foreach @files;
62
63         $self->render(
64                 files => [ @files ],
65                 size => $size,
66                 loaded => $loaded,
67                 filters => $filters,
68                 dump_path => { map { $_ => $self->_dump_path($_) } @files },
69                 changes => $changes,
70         );
71 }
72
73 sub _dump_path {
74         my ( $self, $name ) = @_;
75         my $dir = $self->app->home->rel_dir('data');
76         $name =~ s/^$dir//;
77         $name =~ s/\/+/_/g;
78         return '/tmp/mojo_facets.' . $name . '.storable';
79 }
80
81 sub _save {
82         my ( $self, $path ) = @_;
83
84         my $dump_path = $self->_dump_path( $path );
85         my $first_load = ! -e $dump_path;
86         warn "save loaded to $dump_path";
87         my $info = $loaded->{$path};
88         store $info, $dump_path;
89
90         if ( $first_load ) {
91                 my $mtime = $loaded->{$path}->{mtime};
92                 utime $mtime, $mtime, $dump_path;
93                 warn "sync time to $path at $mtime\n";
94         }
95
96         warn $dump_path, ' ', -s $dump_path, " bytes\n";
97         return $dump_path;
98 }
99
100
101 sub __stats {
102
103         my $stats;
104
105         my $nr_items = $#{ $_[0] } + 1;
106
107         warn "__stats $nr_items\n";
108
109 #       foreach my $e ( @{ $_[0] } ) {
110         foreach my $i ( 0 .. $#{$_[0]} ) {
111                 print STDERR " $i" if $i % 5000;
112                 my $e = $_[0]->[$i];
113                 foreach my $n ( keys %$e ) {
114                         $stats->{$n}->{count}++;
115                         my @v;
116                         if ( ref $e->{$n} eq 'ARRAY' ) {
117                                 $stats->{$n}->{array} += $#{ $e->{$n} } + 1;
118                                 @v = @{ $e->{$n} };
119                         } else {
120                                 @v = ( $e->{$n} );
121                         }
122
123                         foreach my $x ( @v ) {
124                                 if ( ! defined $x ) { # FIXME really null
125                                         $stats->{$n}->{empty}++;
126                                         next;
127                                 }
128
129                                 $stats->{$n}->{numeric}++
130                                         if $x =~ m/^[-+]?([0-9]*\.[0-9]+|[0-9]+)$/;
131                                 $stats->{$n}->{empty}++
132                                         if length $x == 0; # faster than $x =~ m/^\s*$/;
133                         }
134
135                 }
136         }
137
138         foreach my $n ( keys %$stats ) {
139                 my $s = $stats->{$n};
140                 next unless defined $s->{array};
141                 if ( $s->{array} == $s->{count} ) {
142                         delete $s->{array};
143                         if ( $s->{count} == $nr_items ) {
144                                 warn "check $n for uniqeness\n";
145                                 my $unique;
146                                 foreach my $e ( @{ $_[0] } ) {
147                                         if ( ++$unique->{ $e->{$n}->[0] } == 2 ) {
148                                                 $unique = 0;
149                                                 last;
150                                         }
151                                 }
152                                 if ( $unique ) {
153                                         $stats->{$n}->{unique} = 1;
154                                         #warn "# $n unique ",dump( $unique );
155                                 }
156                         }
157                 }
158         }
159
160         warn "# __stats ",dump($stats);
161
162         return $stats;
163 }
164
165 sub _param_or_session {
166         $_[0]->param( $_[1] ) || $_[0]->session( $_[1] )
167 }
168
169 sub stats {
170         my $self = shift;
171         my $path = $self->_param_or_session('path');
172         warn "stats $path\n";
173         delete $loaded->{$path}->{stats};
174         return $self->redirect_to( '/data/columns' );
175 }
176
177
178 sub _load_path {
179         my ( $self, $path ) = @_;
180
181         return if defined $loaded->{$path}->{generated};
182
183         my $full_path = $self->app->home->rel_file( 'data/' . $path );
184         return $self->redirect_to('/data/index') unless -r $full_path;
185
186         my $dump_path = $self->_dump_path( $path );
187
188         if ( defined $loaded->{$path}->{data} ) {
189                 my $mtime = (stat($full_path))[9];
190                 return if $loaded->{$path}->{mtime} == $mtime;
191                 warn "reload $full_path, modified ", time() - $mtime, " seconds ago\n";
192         } elsif ( -e $dump_path ) {
193                 warn "dump_path $dump_path ", -s $dump_path, " bytes loading...\n";
194                 my $info = retrieve $dump_path;
195                 $loaded->{ $path } = $info;
196                 return;
197         }
198
199         my $data;
200         if ( -f $full_path ) {
201                 if ( $full_path =~ m/.storable$/ ) { # check storable first to catch files copied from /tmp/
202                         $data->{generated}++;
203                         warn "open $full_path ", -s $full_path, " bytes";
204                         open(my $pipe, "<", $full_path) || die $!;
205                         while ( my $o = eval { Storable::fd_retrieve $pipe } ) {
206                                 if ( exists $o->{item} ) {
207                                         # stream of storable objects
208                                         push @{ $data->{items} }, $o->{item};
209                                 } elsif ( exists $o->{data}->{items} ) {
210                                         # /tmp/mojofacets.*.storable
211                                         $data->{items} = $o->{data}->{items};
212                                         $data->{header} = $o->{header};
213                                         delete $data->{generated};
214                                 } else {
215                                         warn "SKIP ",dump($o);
216                                 }
217                         }
218                         close($pipe);
219                         warn "loaded ", $#{ $data->{items} } + 1, " items from $full_path\n";
220                 } elsif ( $full_path =~ m/.csv/i ) {
221                         $data = MojoFacets::Import::CSV->new( full_path => $full_path )->data;
222                 } elsif ( $full_path =~ m/.sql/i ) {
223                         $data = MojoFacets::Import::SQL->new( full_path => $full_path )->data;
224                 } elsif ( $full_path =~ m/.couchdb/i ) {
225                         $data = MojoFacets::Import::CouchDB->new( full_path => $full_path )->data;
226                 } elsif ( $full_path =~ m/.pairs/i ) {
227                         $data = MojoFacets::Import::Pairs->new( full_path => $full_path )->data;
228                 } else {
229                         $data = MojoFacets::Import::File->new( full_path => $full_path, path => $path )->data;
230                 }
231         } elsif ( -d $full_path && $full_path =~ m/.html/ ) {
232                 $data = MojoFacets::Import::HTMLTable->new( dir => $full_path )->data;
233         } else {
234                 die "can't load $full_path";
235         }
236
237         my @header;
238
239         if ( defined $data->{header} ) {
240                 if ( ref $data->{header} eq 'ARRAY' ) {
241                         @header = @{ $data->{header} };
242                 } else {
243                         warn "header not array ", dump( $data->{header} );
244                 }
245         }
246
247         my $stats = __stats( $data->{items} );
248
249         @header =
250                 sort { $stats->{$b}->{count} <=> $stats->{$a}->{count} }
251                 grep { defined $stats->{$_}->{count} } keys %$stats
252                 unless @header;
253
254         my $info = {
255                 header => [ @header ],
256                 stats  => $stats,
257                 full_path => $full_path,
258                 size => -s $full_path,
259                 mtime => (stat($full_path))[9],
260                 data => $data,
261                 defined $data->{generated} ? ( generated => 1 ) : (),
262         };
263
264         $loaded->{ $path } = $info;
265         $self->_save( $path ) unless $info->{generated};
266
267 }
268
269
270 sub load {
271         my $self = shift;
272
273         my @paths = $self->param('paths');
274         warn "# paths ", dump @paths;
275
276         foreach my $p ( keys %$loaded ) {
277                 next if grep { /^\Q$p\E$/ } @paths;
278                 warn "remove $p from memory\n";
279                 delete $loaded->{$p};
280         }
281
282         $self->_load_path( $_ ) foreach @paths;
283
284         my $path = $self->param('path') || $self->session('path') || @paths[0] || $self->redirect_to('/data/index');
285
286         warn "# path $path\n";
287         $self->_load_path( $path );
288
289         $self->session( 'path' => $path );
290
291         my $redirect_to = '/data/items';
292
293         $self->session( 'header' => $loaded->{$path}->{header} );
294         if ( ! defined $loaded->{$path}->{columns} ) {
295                 my $columns_path = $self->_permanent_path( 'columns' );
296                 if ( -e $columns_path ) {
297                         my @columns = map { s/[\r\n]+$//; $_ } read_file $columns_path, binmode => ':utf8';
298                         $loaded->{$path}->{columns} = [ @columns ];
299                         warn "# columns_path $columns_path ",dump(@columns);
300                 } else {
301                         $loaded->{$path}->{columns} = $loaded->{$path}->{header}
302                 }
303
304                 $redirect_to = '/data/columns';
305         }
306         $self->session( 'columns' => $loaded->{$path}->{columns} );
307         $self->session( 'order'   => $loaded->{$path}->{columns}->[0] );
308         return $self->redirect_to( $redirect_to );
309 }
310
311
312 sub _loaded {
313         my ( $self, $name ) = @_;
314         my $path = $self->session('path') || $self->param('path');
315         return $self->redirect_to('/data/index') unless $path;
316
317         if ( defined $loaded->{$path}->{modified} && $loaded->{$path}->{modified} > 1 ) {
318                 my $caller = (caller(1))[3];
319                 if ( $caller =~ m/::edit/ ) {
320                         warn "rebuild stats for $path ignored caller $caller\n";
321                 } else {
322                         warn "rebuild stats for $path FORCED by modified caller $caller\n";
323 #                       $loaded->{$path}->{stats} = __stats( $loaded->{$path}->{data}->{items} );
324                         $loaded->{$path}->{rebuild_stats} = 1;
325                         $loaded->{$path}->{modified} = 1;
326                 }
327         }
328
329         if ( defined $loaded->{$path}->{rebuild_stats} ) {
330                 warn "rebuild_stats $path";
331                 $loaded->{$path}->{stats} = __stats( $loaded->{$path}->{data}->{items} );
332                 delete $loaded->{$path}->{rebuild_stats};
333         }
334
335         if ( ! defined $loaded->{$path}->{$name} ) {
336                 warn "$path $name isn't loaded\n";
337                 $self->_load_path( $path );
338                 if ( ! defined $loaded->{$path}->{stats} ) {
339                         warn "rebuild stats for $path\n";
340                         $loaded->{$path}->{stats} = __stats( $loaded->{$path}->{data}->{items} );
341                 }
342                 if ( ! defined $loaded->{$path}->{$name} ) {
343                         warn "MISSING $name for $path\n";
344                         return $self->redirect_to('/data/index')
345                 }
346         }
347
348         $self->session( 'modified' => $loaded->{$path}->{modified} );
349
350         return $loaded->{$path}->{$name};
351 }
352
353
354 sub _checked {
355         my $self = shift;
356         my $checked;
357         $checked->{$_}++ foreach @_;
358 #       warn "# _checked ",dump($checked);
359         return $checked;
360 }
361
362 sub _permanent_path {
363         my $self = shift;
364         my $path = $self->_param_or_session('path');
365         $self->app->home->rel_dir('data') . '/' . join('.', $path, @_);
366 }
367
368 sub __unac {
369         my $n = shift;
370         $n = unac_string('utf-8',$n);
371         $n =~ s/\W+/_/g;
372         return $n;
373 }
374
375 sub _column_from_unac {
376         my ($self,$name) = @_;
377
378         my $stats = $self->_loaded('stats');
379         my $cols_norm = { map { __unac( $_ ) => $_ } keys %$stats };
380
381         $cols_norm->{$name} || die "can't find column $name in ", dump($cols_norm);
382 }
383
384 sub _export_path {
385         my $self = shift;
386         my $path = $self->_param_or_session('path');
387         if ( ! $path ) {
388                 warn "no path in param or session";
389                 return;
390         }
391         my $dir = $self->app->home->rel_dir('public') . "/export/$path";
392         mkpath $dir unless -e $dir;
393         my $name = __export_path_name( $path, @_ );
394         my $full = $dir . '/' . $name;
395         $full =~ s/\/+$// if -d $full; # strip trailing slash for dirs
396         return $full;
397 }
398
399 sub __export_path_name {
400         my $max_length = 80;
401
402         my $path = shift;
403         my $name = join('.', map { __unac($_) } @_ );
404         if ( length($name) > $max_length ) {
405                 $name = substr($name,0,$max_length) . Digest::MD5::md5_hex substr($name,$max_length);
406         }
407         return $name;
408 }
409
410 sub columns {
411     my $self = shift;
412
413         my $view_path = $self->_permanent_path( 'views' );
414
415         if ( $self->param('columns') ) {
416                 my @columns = $self->_param_array('columns');
417                 write_file( $self->_permanent_path( 'columns' ), { binmode => ':utf8' }, map { "$_\n" } @columns );
418                 if ( my $view = $self->param('view') ) {
419                         mkdir $view_path unless -e $view_path;
420                         write_file( "$view_path/$view", { binmode => ':utf8' }, map { "$_\n" } @columns );
421                         warn "view $view_path/$view ", -s "$view_path/$view", " bytes\n";
422                 }
423
424                 return $self->redirect_to('/data/items');
425
426         } elsif ( ! $self->session('header') ) {
427                 return $self->redirect_to('/data/load');
428         }
429
430         if ( my $id = $self->param('id') ) {
431                 my $view_full = "$view_path/$id";
432                 if ( -f $view_full ) {
433                         my @columns = map { chomp; $_ } read_file $view_full, binmode => ':utf8';
434                         warn "view $view_full loaded ", dump @columns;
435                         $self->session( 'columns' => [ @columns ] );
436                         return $self->redirect_to('/data/items');
437                 }
438         }
439
440         my $stats = $self->_loaded( 'stats' );
441
442         my @columns;
443         @columns = grep { defined $stats->{$_}->{count} } @{ $self->session('columns') } if $self->session('columns');
444
445         foreach my $c ( @{ $self->session( 'header' ) } ) {
446                 push @columns, $c unless grep { /^\Q$c\E$/ } @columns;
447         }
448
449         my @views;
450         if ( -d $view_path ) {
451                 @views = map { s{^\Q$view_path\E/*}{}; $_ } glob "$view_path/*";
452                 warn "# views ",dump @views;
453         }
454
455     $self->render(
456                 message => 'Select columns to display',
457                 stats => $stats,
458                 columns => \@columns,
459                 checked => $self->_checked( $self->_param_array('columns') ),
460                 views => \@views,
461         );
462 }
463
464 sub _param_array {
465     my ($self,$name) = @_;
466
467         my @array = $self->param($name);
468         my $path  = $self->session('path');
469
470         if ( @array ) {
471                 $self->session($name => [ @array ]);
472                 $loaded->{$path}->{$name} = [ @array ];
473         } elsif ( my $session = $self->session($name) ) {
474                 if ( ref $session eq 'ARRAY' ) {
475                         @array = @$session;
476                 } else {
477                         die "$name not array ",dump($session);
478                 }
479         }
480         #warn "# $name ",dump @array;
481         return @array;
482 }
483
484 sub _param_scalar {
485     my ($self,$name,$default) = @_;
486
487         my $scalar = $self->param($name);
488
489         if ( defined $scalar ) {
490                 $self->session($name => $scalar);
491         } else {
492                 $scalar = $self->session($name);
493         }
494
495         if ( ! defined $scalar ) {
496                 $scalar = $default;
497                 if ( defined $scalar ) {
498                         $self->session($name => $scalar);
499                 } else {
500                         warn "no default for $name";
501                 }
502         }
503
504         warn "# _param_scalar $name ",dump $scalar;
505         return $scalar;
506 }
507
508 sub filter {
509         my $self = shift;
510
511         my $name = $self->param('filter_name') || die "name?";
512         my @vals = $self->param('filter_vals');
513
514         $self->_remove_filter( $name );
515         if ( @vals ) {
516                 $self->_filter_on_data( $name, @vals );
517                 if ( my $permanent = $self->param('_permanent') ) {
518                         my $permanent_path = $self->_export_path( 'filter', $name, $permanent );
519                         write_file $permanent_path, { binmode => ':utf8' }, map { "$_\n" } @vals;
520                         warn "permanent filter $permanent_path ", -s $permanent_path;
521                 }
522         }
523
524         $self->session( 'offset' => 0 );
525
526         return $self->redirect_to('/data/items');
527 }
528
529 sub _filter_on_data {
530         my ( $self, $name, @vals ) = @_;
531
532         my $path = $self->session('path');
533
534         if ( ! defined $loaded->{$path}->{stats}->{ $name } ) {
535                 warn "filter $name not found in data set";
536                 return;
537         }
538
539         $filters->{$name} = [ @vals ];
540         warn "_filter_on_data $name ", $#vals + 1, " values on $path\n";
541
542         my $filter_hash;
543         $filter_hash->{$_}++ foreach @vals;
544
545         #warn "# filter_hash ",dump( $filter_hash );
546
547         my $items = $self->_loaded('data')->{items};
548
549         my $include_missing = defined $filter_hash->{_missing};
550         my $filtered_items;
551
552         foreach my $i ( 0 .. $#$items ) {
553
554                 if ( defined $items->[$i]->{$name} ) {
555                         my $row = $items->[$i]->{$name};
556                         $row = [ $row ] unless ref $row eq 'ARRAY'; # FIXME probably wrong place
557                         foreach my $v ( @$row ) {
558                                 if ( defined $filter_hash->{ $v } ) {
559                                         $filtered_items->{$i}++;
560                                 }
561                         }
562                 } elsif ( $include_missing ) {
563                         $filtered_items->{$i}++;
564                 }
565         }
566
567         #warn "# filter $name ",dump($filtered_items);
568
569         # invalidate filters on other datasets
570         foreach my $dataset ( grep { exists $loaded->{$_}->{filters}->{$name} } keys %$loaded ) {
571                 delete $loaded->{$dataset}->{filters}->{$name};
572                 delete $loaded->{$dataset}->{filtered};
573         }
574
575         $loaded->{$path}->{filters}->{$name} = $filtered_items;
576         warn "filter $name with ", scalar keys %$filtered_items, " items created\n";
577 }
578
579
580 sub _current_filters {
581         my $self = shift;
582         my $current_filters;
583         my $stats = $self->_loaded('stats');
584
585         $current_filters->{ $_ } = $filters->{ $_ }
586                 foreach ( grep { defined $filters->{ $_ } } keys %$stats )
587         ;
588         warn "# _current_filters ",dump( keys %$current_filters );
589         return $current_filters;
590 }
591
592 sub _data_sorted_by {
593         my ( $self, $order ) = @_;
594
595         my $path = $self->session('path');
596
597         warn "_data_sorted_by $order from $path";
598
599         if ( defined $loaded->{$path}->{sorted}->{$order} ) {
600                 return $loaded->{$path}->{sorted}->{$order};
601         }
602
603         my $data = $self->_loaded( 'data' );
604         my $numeric = $self->_is_numeric($order);
605         my $missing = $numeric ? 0 : '';
606         no warnings qw(numeric);
607         my $nr = 0;
608         my @sorted = map {
609                 $_->[0]
610         } sort {
611                 if ( $numeric ) {
612                         $a->[1] <=> $b->[1]
613                 } else {
614                         $a->[1] cmp $b->[1]
615                 }
616         } map {
617                 my $v;
618                 if ( ! exists $_->{$order} ) {
619                         $v = $missing;
620                 } elsif ( ref $_->{$order} eq 'ARRAY' ) {
621                         $v = join('', @{$_->{$order}});
622                 } else {
623                         $v = $_->{$order};
624                 }
625                 [ $nr++, $v ]
626         } @{ $data->{items} }
627         ;
628
629         warn "sorted: $order numeric: $numeric items: ", $#sorted + 1, "\n";
630         #warn "# sorted ",dump( @sorted );
631
632         $loaded->{$path}->{sorted}->{$order} = [ @sorted ];
633 }
634
635
636 sub __all_filters {
637         my $order = pop @_;
638         join(',', sort(@_), 'order', $order);
639 }
640
641 our ($out, $key,$value);
642
643 our $lookup_path_col;
644 our $on;
645
646 sub __commit_begin {
647         warn "__commit_begin";
648         $lookup_path_col = undef;
649         $on = undef;
650 }
651
652 sub __commit_end {
653         warn "__commit_end";
654         $lookup_path_col = undef; # cleanup memory
655         $on = undef;
656 }
657
658 sub lookup {
659         warn "# lookup ",dump @_;
660         my ( $vals, $on_path, $on_col, $code, $stat_code ) = @_;
661         die "code is not sub{ ... } but ", dump $code unless ref $code eq 'CODE';
662
663         if ( ! exists $loaded->{$on_path} ) {
664                 my @possible_paths = grep { /\Q$on_path\E/ } keys %$loaded;
665                 die "more than one dataset available for '$on_path' ",dump @possible_paths if $#possible_paths > 0;
666                 $on_path = shift @possible_paths;
667                 warn "## fuzzy selected path $on_path";
668         }
669
670         my $items = $loaded->{$on_path}->{data}->{items} || die "no items for $on_path";
671
672         if ( ! exists $lookup_path_col->{$on_path}->{$on_col} ) {
673                 warn "create lookup_path_col $on_path $on_col";
674                 foreach my $i ( 0 .. $#$items ) {
675                         my $item = $items->[$i];
676                         if ( exists $item->{$on_col} ) {
677                                 if ( ref $item->{$on_col} eq 'ARRAY' ) {
678                                         foreach my $v ( @{ $item->{$on_col} } ) {
679                                                 push @{ $lookup_path_col->{$on_path}->{$on_col}->{$v} }, $i;
680                                         }
681                                 } elsif ( ! ref $item->{$on_col} ) { # scalar
682                                         my $v = $item->{$on_col};
683                                         push @{ $lookup_path_col->{$on_path}->{$on_col}->{$v} }, $i;
684                                 } else {
685                                         die "unknown type of ",dump $item->{$on_col};
686                                 }
687                         }
688                 }
689                 warn "XXX ",dump $lookup_path_col->{$on_path}->{$on_col} if $ENV{DEBUG};
690         }
691
692         my $stat;
693         $stat = Statistics::Descriptive::Full->new() if $stat_code;
694
695         foreach my $v ( ref $vals eq 'ARRAY' ? @$vals : ( $vals ) ) {
696                 foreach my $i ( @{ $lookup_path_col->{$on_path}->{$on_col}->{$v} } ) {
697                         $on = $items->[$i];
698                         warn "XXX lookup code $v $i ",dump $on if $ENV{DEBUG};
699                         $code->($stat);
700                 }
701                 $stat_code->( $stat ) if $stat_code;
702         }
703 }
704
705 sub __commit_path_code {
706         my ( $path, $i, $code, $commit_changed ) = @_;
707
708         my $items = $loaded->{$path}->{data}->{items} || die "no items for $path";
709         my $row = $items->[$i];
710         my $update;
711         eval $code;
712         foreach ( keys %$update ) {
713                 $$commit_changed->{$_}++;
714                 $loaded->{$path}->{data}->{items}->[$i]->{$_} = $update->{$_};
715         }
716         #warn "__commit_path_code $path $i ",dump( $update );
717 }
718
719 # uses templates/admin.html.ep
720 sub _switch_dataset {
721         my $self = shift;
722
723         my $datasets;
724
725         foreach my $path ( keys %$loaded ) {
726                 next unless exists $loaded->{$path}->{data};
727                 push @$datasets, $path;
728         }
729
730         warn "# datasets ",dump($datasets);
731
732         $self->stash( 'datasets' => $datasets );
733 }
734
735 sub items {
736         my $self = shift;
737
738         $self->_switch_dataset;
739
740         if ( my $show = $self->param('id') ) {
741                 $self->param('show', $show);
742                 warn "show $show\n";
743         }
744
745         my $path = $self->_param_scalar('path');
746
747         my @columns = $self->_param_array('columns');
748         return $self->redirect_to('/data/columns') unless @columns;
749         my $order   = $self->_param_scalar('order', $columns[0]);
750         my $sort    = $self->_param_scalar('sort', 'a');
751         my $offset  = $self->_param_scalar('offset', 0);
752         my $limit   = $self->_param_scalar('limit', 20);
753         $self->_param_scalar('show', 'table');
754
755         # fix offset when changing limit
756         $offset = int( $offset / $limit ) * $limit;
757
758         if ( ! grep { /^\Q$order\E$/ } @columns ) {
759                 $order = $columns[0];
760                 $self->session( order => $order );
761         }
762         my $sorted = $self->_data_sorted_by( $order );
763
764         my @filter_names;
765         if ( $filters ) {
766                 foreach my $name ( keys %$filters ) {
767                         if ( ! defined $loaded->{$path}->{stats}->{ $name } ) {
768                                 warn "skip filter $name not found in $path\n";
769                                 next;
770                         }
771                         push @filter_names, $name;
772                 }
773                 warn "filter_names ",dump( @filter_names );
774                 foreach my $name ( @filter_names ) {
775                         next if ref $loaded->{$path}->{filters}->{$name} eq 'ARRAY';
776                         $self->_filter_on_data( $name, @{ $filters->{$name} } );
777                 }
778         }
779
780         my $all_filters = __all_filters( @filter_names,$order );
781
782 #       warn "# all_filters $all_filters ", dump( $loaded->{$path}->{filtered}->{$all_filters} );
783
784         if ( ! defined $loaded->{$path}->{filtered}->{$all_filters} ) {
785
786                 my $path_filters = $loaded->{$path}->{filters};
787
788                 warn "create combined filter for $all_filters from ", $#$sorted + 1, " items\n";
789
790                 my @filtered;
791                 foreach my $i ( 0 .. $#$sorted ) {
792                         my $pos = $sorted->[$i];
793
794                         if ( $#filter_names == -1 ) {
795                                 push @filtered, $pos;
796                                 next;
797                         }
798
799                         my $skip = 0;
800                         foreach ( @filter_names ) {
801                                 $skip ||= 1 if ! defined $path_filters->{$_}->{$pos};
802                         }
803                         next if $skip;
804
805                         push @filtered, $pos;
806                 }
807
808                 $loaded->{$path}->{filtered}->{$all_filters} = [ @filtered ];
809         }
810
811         my $filtered = $loaded->{$path}->{filtered}->{$all_filters}
812                 if defined $loaded->{$path}->{filtered}->{$all_filters};
813
814         warn "all_filters $all_filters produced ", $#$filtered + 1, " items\n" if $filtered;
815
816         my $data = $self->_loaded('data');
817
818         my $code = $self->_param_scalar('code','');
819         $code =~ s{\r}{}gs;
820         $code =~ s{\n+$}{\n}s;
821
822         # XXX convert @row->{foo} into @{$row->{foo}}
823         $code =~ s|\@(row->{[^}]+})|\@{\$$1}|gs;
824
825         my $commit = $self->param('commit');
826         my $test = $self->param('test');
827
828         my $commit_changed;
829         __commit_begin;
830
831         if ( $code && ( $test || $commit ) ) {
832                 # XXX find columns used in code snippet and show them to user
833                 my $order = 0;
834                 foreach my $column ( $code =~ m/\$row->{([^}]+)}/g ) {
835                         if ( $column =~ s/^(['"])// ) {
836                                 $column =~ s/$1$//;
837                         }
838                         next if $column =~ m/\$/; # hide columns with vars in them
839                         $commit_changed->{$column} = 0;
840                 }
841         }
842
843         my $code_path = $self->app->home->rel_dir('public') . "/code";
844         if ( $commit ) {
845
846                 __path_modified( $path, 'commit' );
847
848                 warn "# commit on ", $#$filtered + 1, " items:\n$code\n";
849                 ( $key, $value, $out ) = ( 'key', 'value' );
850                 foreach ( 0 .. $#$filtered ) {
851                         my $i = $filtered->[$_];
852                         __commit_path_code( $path, $i, $code, \$commit_changed );
853                 }
854
855                 # this might move before $out to recalculate stats on source dataset?
856                 __path_rebuild_stats( $path );
857                 my $c = { map { $_ => 1 } @columns };
858                 my @added_columns = sort grep { ! $c->{$_} } keys %$commit_changed;
859                 warn "# added_columns ",dump( @added_columns );
860                 unshift @columns, @added_columns;
861
862                 $loaded->{$path}->{columns} = [ @columns ];
863                 warn "# new columns ",dump( @columns );
864
865                 __invalidate_path_column( $path, $_ ) foreach keys %$commit_changed;
866
867                 $self->_save_change({
868                         path => $path,
869                         time => $self->param('time') || time(),
870                         user => $self->param('user') || $ENV{'LOGNAME'},
871                         code => $code,
872                         commit_changed => $commit_changed,
873                 });
874
875                 if ( my $description = $self->param('code_description') ) {
876                         my $depends = $self->param('code_depends') || die "no code_depends?";
877                         my $path = "$code_path/$depends.$description.pl";
878                         if ( -e $path && ! $self->param('overwrite') ) {
879                                 warn "# code $path not saved\n";
880                         } else {
881                                 write_file(  $path, { binmode => ':utf8' }, "$code\n" );
882                                 warn "code $path ", -s $path, " bytes saved\n";
883                         }
884                 }
885
886                 # remove console
887                 $code = '';
888                 if ( $out ) {
889                         my $commit_dataset = join('.'
890                                 , $self->param('code_depends')
891                                 , $self->param('code_description')
892                                 , time()
893                         );
894                         $key ||= 'key';
895                         $value ||= 'value';
896                         warn "key $key value $value";
897                         my $items;
898                         foreach my $n ( keys %$out ) {
899                                 my $i = { $key => [ $n ] };
900                                 my $ref = ref $out->{$n};
901                                 if ( $ref eq 'HASH' ) {
902                                         $i->{$_} = [ $out->{$n}->{$_} ] foreach keys %{ $out->{$n} };
903                                 } elsif ( $ref eq 'ARRAY' ) {
904                                         $i->{$_} = $out->{$n};
905                                 } elsif ( ! $ref ) {
906                                         $i->{$value} = [ $out->{$n} ];
907                                 } else {
908                                         $i->{_error} = [ dump($out->{$n}) ];
909                                 }
910                                 push @$items, $i;
911                         };
912                         undef $out;
913                         my $stats = __stats( $items );
914                         my @columns = grep { ! m/^\Q$key\E$/ } sort keys %$stats;
915                         unshift @columns, $key;
916
917                         $loaded->{$commit_dataset} = {
918                                 header => [ @columns ],
919                                 columns => [ @columns ],
920                                 mtime => time(),
921                                 data => { items => $items },
922                                 stats => $stats,
923                                 generated => 1,
924                         };
925                         warn "# loaded out ", dump( $loaded->{$commit_dataset} );
926                         $self->session('path', $commit_dataset);
927                         $self->session('columns', [ @columns ]);
928                         $self->session('order', $key);
929                         return $self->redirect_to('/data/items');
930                 }
931
932                 $self->session('columns', [ @columns ]);
933         }
934
935         my $sorted_items;
936         my $from_end = $sort eq 'd' ? $#$filtered : 0;
937         my $test_changed;
938         my ( $key, $value, $out ) = ( 'key', 'value' ); # XXX make local
939         foreach ( 0 .. $limit ) {
940                 my $i = $_ + $offset;
941                 last unless defined $filtered->[$i];
942                 $i = $from_end - $i if $from_end;
943                 my $id = $filtered->[$i];
944                 my $row = Storable::dclone $data->{items}->[ $id ];
945                 if ( $code && $test ) {
946                         my $update;
947                         eval $code;
948                         if ( $@ ) {
949                                 warn "ERROR evaling $@", dump($code);
950                                 $self->stash('eval_error', $@) if $@;
951                         } else {
952                                 warn "EVAL ",dump($update);
953                                 foreach ( keys %$update ) {
954                                         $test_changed->{$_}++;
955                                         $row->{$_} = $update->{$_};
956                                 }
957                         }
958                 }
959                 $row->{_row_id} ||= $id;
960                 push @$sorted_items, $row;
961         }
962
963         if ( $self->param('export') ) {
964                 my $export_path = $self->_export_path( 'items', @columns);
965                 open(my $fh, '>', $export_path) || die "ERROR: can't open $export_path: $!";
966                 print $fh "#",join("\t",@columns),"\n";
967                 foreach my $f ( 0 .. $#$filtered ) {
968                         print $fh join("\t", map {
969                                 my $i = $data->{items}->[ $filtered->[$f] ];
970                                 my $v = '\N';
971                                 if ( ! defined $i->{$_} ) {
972                                         # nop
973                                 } elsif ( ref $i->{$_} eq 'ARRAY' ) {
974                                         $v =join(',', @{ $i->{$_} });
975                                         $v = '\N' if length($v) == 0;
976                                 } elsif ( ! ref $i->{$_} ) {
977                                         $v = $i->{$_};
978                                 } else {
979                                         $v = dump $i->{$_};
980                                 }
981                                 $v;
982                         } @columns),"\n";
983                 }
984                 close($fh);
985                 warn "export $export_path ", -s $export_path, " bytes\n";
986         }
987
988         my ( $code_depends, $code_description );
989
990         if ( $test ) {
991
992                 warn "# test_changed ",dump( $test_changed );
993                 my $c = { map { $_ => 1 } @columns };
994                 my @added_columns = sort grep { ! $c->{$_} } keys %$test_changed;
995                 unshift @columns, @added_columns;
996
997                 warn "# sorted_items ", $#$sorted_items + 1, " offset $offset limit $limit order $sort";
998
999                 my $depends_on;
1000                 my $tmp = $code; $tmp =~ s/\$row->{(['"]?)([\w\s]+)\1/$depends_on->{$2}++/gse;
1001                 warn "# depends_on ",dump $depends_on;
1002
1003                 my $test_added = Storable::dclone $test_changed;
1004                 delete $test_added->{$_} foreach keys %$depends_on;
1005
1006                 $code_depends = $self->param('code_depends')
1007                 || join(',', keys %$depends_on);
1008
1009                 $code_description = $self->param('code_description') ||
1010                 join(',', keys %$test_added);
1011
1012                 $code_depends ||= $code_description; # self-modifing
1013                 if ( ! $code_depends && $out ) {
1014                         $code_depends = $key;
1015                         $code_description = $value;
1016                 }
1017
1018                 warn "# test_changed ",dump( $test_changed, $code_depends, $code_description );
1019
1020         } # test?
1021
1022         __commit_end;
1023
1024         $self->render(
1025                 order => $order,
1026                 offset => $offset,
1027                 limit => $limit,
1028                 sorted => $sorted_items,
1029                 columns => [ @columns ],
1030                 rows => $#$filtered + 1,
1031                 numeric => { map { $_, $self->_is_numeric($_) } @columns },
1032                 unique  => { map { $_, $self->_is_unique( $_) } @columns },
1033                 filters => $self->_current_filters,
1034                 code => $code,
1035                 cols_changed => $commit ? $commit_changed : $test_changed,
1036                 code_depends => $code_depends,
1037                 code_description => $code_description,
1038                 code_path => $code_path,
1039                 out => $out,
1040         );
1041
1042 }
1043
1044
1045 sub order {
1046         my $self = shift;
1047         $self->session('order', $self->param('order'));
1048         $self->session('sort', $self->param('sort'));
1049         return $self->redirect_to('/data/items');
1050 }
1051
1052 sub _is_numeric {
1053         my ( $self, $name ) = @_;
1054
1055         my $stats = $self->_loaded( 'stats' );
1056
1057         # sort facet numerically if more >50% elements are numeric
1058         my $count = $stats->{$name}->{count};
1059         $count   -= $stats->{$name}->{empty} if defined $stats->{$name}->{empty};
1060         defined $stats->{$name}->{numeric} &&
1061                 $stats->{$name}->{numeric} > $count / 2;
1062 }
1063
1064 sub _is_unique {
1065         my ( $self, $name ) = @_;
1066         my $stats = $self->_loaded( 'stats' );
1067         defined $stats->{$name}->{unique};
1068 }
1069
1070 sub _remove_filter {
1071         my ($self,$name) = @_;
1072         warn "_remove_filter $name\n";
1073
1074         my $path = $self->session('path');
1075
1076         delete $filters->{$name};
1077         delete $loaded->{$path}->{filters}->{$name};
1078         warn "filters left: ", keys %{ $loaded->{$path}->{filters} };
1079
1080         foreach (
1081                         grep { /\Q$name\E/ }
1082                         keys %{ $loaded->{$path}->{filtered} }
1083         ) {
1084                 delete $loaded->{$path}->{filtered}->{$_};
1085                 warn "remove filtered cache $_";
1086         }
1087 }
1088
1089 sub facet {
1090         my $self = shift;
1091
1092         my $path = $self->session('path') || return $self->redirect_to( '/data/index' );
1093
1094         if ( my $name = $self->param('remove') ) {
1095                 $self->_remove_filter( $name );
1096                 return $self->redirect_to( '/data/items' );
1097         }
1098
1099         my $facet;
1100         my $name = $self->param('name') || die "no name";
1101
1102         my $all = $self->_param_scalar('all', 1);
1103         my $data = $self->_loaded('data');
1104
1105         my $filters = $self->_current_filters;
1106         my $all_filters = __all_filters( keys %$filters,$self->session('order') );
1107         my $filtered = $loaded->{$path}->{filtered}->{$all_filters}
1108                 if defined $loaded->{$path}->{filtered}->{$all_filters};
1109
1110         if ( ! $filtered || $all ) {
1111                 $filtered = [ 0 .. $#{ $data->{items} } ];
1112                 warn "filter all values\n";
1113         } else {
1114                 warn "filter using $all_filters\n";
1115         }
1116
1117         foreach my $i ( @$filtered ) {
1118                 my $item = $data->{items}->[$i];
1119                 if ( ! exists $item->{$name} || ! defined $item->{$name} ) {
1120                         $facet->{ _missing }++;
1121                 } elsif ( ref $item->{$name} eq 'ARRAY' ) {
1122                         $facet->{$_}++ foreach @{ $item->{$name} };
1123                 } else {
1124                         $facet->{ $item->{$name} }++;
1125                 }
1126         }
1127
1128         my $checked_values = $self->_checked( @{ $filters->{$name} } ) if defined $filters->{$name};
1129
1130         if ( my $code = $self->param('code') ) {
1131                 my $out;
1132                 foreach my $value ( keys %$facet ) {
1133                         my $count = $facet->{$value};
1134                         my $checked = $checked_values->{$value};
1135                         eval $code;
1136                         if ( $@ ) {
1137                                 $out = $@;
1138                                 warn "ERROR: $@\n$code\n";
1139                                 last;
1140                         } elsif ( $checked != $checked_values->{$value} ) {
1141                                 warn "checked $value $count -> $checked\n";
1142                                 $checked_values->{$value} = $checked;
1143                         }
1144                 }
1145                 warn "out ",dump( $out );
1146                 $self->stash( out => $out );
1147         }
1148
1149 #       warn "# facet $name ",dump $facet;
1150
1151         my @facet_names =
1152                   $all                      ? keys %$facet
1153                 : defined $filters->{$name} ? @{ $filters->{$name} }
1154                 : keys %$facet;
1155
1156         my $numeric = $self->_is_numeric($name);
1157
1158         my $sort = $self->param('sort');
1159         # sort numeric facets with more than 5 values ascending
1160         $sort ||= $numeric && $#facet_names > 4 ? 'a' : 'c';
1161
1162         @facet_names = sort {
1163                 my $result;
1164                 if ( $sort eq 'a' ) {
1165                         $result = $numeric ? $a <=> $b : lc $a cmp lc $b;
1166                 } elsif ( $sort eq 'd' ) {
1167                         $result = $numeric ? $b <=> $a : lc $b cmp lc $a;
1168                 } elsif ( $sort eq 'c' ) {
1169                         $result = ( $facet->{$b} || -1 ) <=> ( $facet->{$a} || -1 )
1170                 } else {
1171                         warn "unknown sort: $sort";
1172                         $result = $a cmp $b;
1173                 }
1174                 $result = $a cmp $b unless defined $result; # FIXME cludge for numeric facets with invalid data
1175                 $result;
1176         } @facet_names;
1177
1178         $self->render( name => $name, facet => $facet, checked => $checked_values,
1179                 facet_names => \@facet_names, sort => $sort, numeric => $numeric,
1180         );
1181 }
1182
1183
1184 sub __invalidate_path_column {
1185         my ( $path, $name ) = @_;
1186
1187         if ( defined $loaded->{$path}->{sorted}->{$name} ) {
1188                 delete $loaded->{$path}->{sorted}->{$name};
1189                 warn "# invalidate $path sorted $name\n";
1190         }
1191
1192         foreach ( grep { m/\Q$name\E/ } keys %{ $loaded->{$path}->{filtered} } ) {
1193                 delete $loaded->{$path}->{filtered}->{$_};
1194                 warn "# invalidate $path filtered $_\n";
1195         }
1196 }
1197
1198 sub __path_modified {
1199         my ( $path, $value ) = @_;
1200         $value = 1 unless defined $value;
1201         
1202         $loaded->{$path}->{modified}  = $value;
1203
1204         warn "# __path_modified $path $value\n";
1205 }
1206
1207 sub __path_rebuild_stats { $loaded->{ $_[0] }->{rebuild_stats} = 1 };
1208
1209 sub _save_change {
1210         my ($self,$change) = @_;
1211
1212         my $change_path = $self->_permanent_path( 'changes' );
1213         mkdir $change_path unless -d $change_path;
1214         $change_path .= '/' . $change->{time};
1215         store $change, $change_path;
1216         utime $change->{time}, $change->{time}, $change_path;
1217         warn "_save_change $change_path ", dump($change);
1218 }
1219
1220 sub edit {
1221         my $self = shift;
1222         my $new_content = $self->param('new_content');
1223         $new_content  ||= $self->param('content'); # backward compatibility with old actions
1224
1225         my $i = $self->param('_row_id');
1226         die "invalid _row_id ",dump($i) unless $i =~ m/^\d+$/;
1227         my $path = $self->param('path') || die "no path";
1228         my $name = $self->param('name') || die "no name";
1229         my $status = 200; # 200 = OK, 201 = Created
1230
1231         my $data = $self->_loaded('data');
1232
1233         if ( defined $loaded->{$path}->{data}->{items}->[$i] ) {
1234                 $new_content =~ s/^\s+//s;
1235                 $new_content =~ s/\s+$//s;
1236                 my $v;
1237                 if ( $new_content =~ /\xB6/ ) { # para
1238                         $v = [ split(/\s*\xB6\s*/, $new_content) ];
1239                 } else {
1240                         $v = [ $new_content ];
1241                 }
1242
1243                 my $old = dump $loaded->{$path}->{data}->{items}->[$i]->{$name};
1244                 my $new = dump $v;
1245                 if ( $old ne $new
1246                         && ! ( $old eq 'undef' && length($new_content) == 0 ) # new value empty, previous undef
1247                 ) {
1248                         $self->_save_change({
1249                                 path => $path,
1250                                 column => $name,
1251                                 pos => $i,
1252                                 old => $loaded->{$path}->{data}->{items}->[$i]->{$name},
1253                                 new => $v,
1254                                 time => $self->param('time') || time(),
1255                                 user => $self->param('user') || $ENV{'LOGNAME'},
1256                                 unique => {
1257                                         map { $_ => $loaded->{$path}->{data}->{items}->[$i]->{$_}->[0] }
1258                                         grep { defined $loaded->{$path}->{stats}->{$_}->{unique} }
1259                                         keys %{ $loaded->{$path}->{stats} }
1260                                 },
1261                         });
1262
1263                         warn "# change $path $i $old -> $new\n";
1264                         $loaded->{$path}->{data}->{items}->[$i]->{$name} = $v;
1265
1266                         __invalidate_path_column( $path, $name );
1267
1268                         $status = 201; # created
1269                         __path_rebuild_stats( $path );
1270         
1271                         $new_content = join("\xB6",@$v);
1272
1273                 } else {
1274                         warn "# unchanged $path $i $old\n";
1275                         $status = 304;
1276                 }
1277         } else {
1278                 $new_content = "$path $i $name doesn't exist\n";
1279                 $status = 404;
1280         }
1281
1282         warn "# edit $status ", dump $new_content;
1283
1284         $self->render(
1285                 status => $status,
1286                 new_content => scalar $new_content,
1287         );
1288 }
1289
1290
1291 sub save {
1292         my $self = shift;
1293         my $path = $self->_param_or_session('path');
1294         my $dump_path = $self->_save( $path );
1295         __path_modified( $path, 0 );
1296
1297         return $self->redirect_to( '/data/items' );
1298 }
1299
1300 sub export {
1301         my $self = shift;
1302
1303         my $dir = $self->app->home->rel_dir('public');
1304
1305         if ( my $import = $self->param('import') ) {
1306
1307                 if ( $import =~ m{/filter\.(.+?)\..+} ) {
1308                         my $name = $self->_column_from_unac( $1 );
1309
1310                         my @vals = map { chomp; $_ }
1311                                 read_file "$dir/export/$import", binmode => ':utf8';
1312
1313                         $self->_remove_filter( $name );
1314                         $self->_filter_on_data( $name, @vals );
1315                         $self->session( 'offset' => 0 );
1316                         return $self->redirect_to('/data/items');
1317                 } else {
1318                         warn "UNKNOWN IMPORT $import";
1319                 }
1320         }
1321
1322         if ( my $remove = $self->param('remove') ) {
1323                 my $path = "$dir/export/$remove";
1324                 unlink $path if -e $path;
1325                 $path .= '.png';
1326                 unlink $path if -e $path;
1327         }
1328
1329         my $path = $self->_export_path || return $self->redirect_to('/data/index');
1330
1331         my @files = grep { ! /\.png$/ } glob "$path/*";
1332         my $mtime = { map { $_ => (stat($_))[9] } @files };
1333         @files = sort { $mtime->{$b} <=> $mtime->{$a} } @files;
1334         $self->render( export => [ @files ] );
1335 }
1336
1337 sub __loaded_paths {
1338         return
1339                 grep { defined $loaded->{$_}->{data} }
1340                 keys %$loaded;
1341 }
1342
1343 sub reload {
1344         my $self = shift;
1345         $self->stash( reload => 1 );
1346         $self->remove;
1347 #       $self->_load_path( $self->param('path') );
1348         $self->redirect_to('/data/load?path=' . $self->param('path') );
1349 }
1350
1351 sub remove {
1352         my $self = shift;
1353         my $path = $self->param('path');
1354         if ( $path =~ m{^/tmp/mojo_facets\.} ) {
1355                 unlink $path;
1356                 warn "# unlink $path";
1357                 if ( my $name = $self->param('name') ) {
1358                         delete $loaded->{$name};
1359                         warn "# remove $name from memory";
1360                 }
1361         } else {
1362                 warn "WARNING: $path unlink ignored";
1363         }
1364         return if $self->stash('reload');
1365         return $self->redirect_to( '/data/load' );
1366 }
1367
1368 1;