br.pl

   1 package TB_Temp_Packname;
   2
   3 use Web::Simple;
   4 use Module::Runtime qw(use_module);
   5 use Scalar::Util qw(blessed);
   6 use IO::All;
   7 use JSON;
   8 use URI::Escape;
   9
  10 has root => (is => 'lazy');
  11
  12 has json => (is => 'lazy');
  13
  14 sub _build_root {
  15   io->dir( $ENV{JTV_ROOT} || die "JTV_ROOT env var not set" )
  16 }
  17
  18 sub _build_json {
  19   JSON->new->relaxed->pretty
  20 }
  21
  22 sub dispatch_request {
  23   my ($self) = @_;
  24   sub (/raw/...) {
  25     use_module('Plack::App::Directory')->new({
  26       root => $self->root->name
  27     });
  28   },
  29   sub () {
  30     response_filter {
  31       ref($_[0][0]) eq 'HASH' ? $self->render_table($_[0][0]) : $_[0]
  32     }
  33   },
  34   sub (/) {
  35     sub (?@host=) {
  36       [ 302, [ 'Location', '/'.join('+', @{$_[1]}).'/' ], [] ];
  37     },
  38     sub () {
  39       $self->root_structure;
  40     },
  41   },
  42   sub (/**/) {
  43     $self->structure(split '/', $_[1]);
  44   },
  45 }
  46
  47 sub root_structure {
  48   my ($self) = @_;
  49   my $struct = $self->mangle_structure($self->root);
  50   push @{$struct->[0]{columns}}, 'select';
  51   foreach my $host (@{$struct->[0]{data}}) {
  52     use HTML::Tags;
  53     my $name = $host->{name};
  54     my $html = join '', HTML::Tags::to_html_string(
  55       <input type="checkbox" name="host" value="$name" />
  56     );
  57     $host->{select} = \$html;
  58   }
  59   $struct->[0]{wrapper} = sub {
  60     use HTML::Tags;
  61     '  ', <form>, "\n",
  62     (map /^\s*$/ ? "    $_" : $_, @_),
  63     '    ', <input type="submit" value="Now Multify" />, "\n",
  64     '  ', </form>, "\n"
  65   };
  66   return $struct;
  67 }
  68
  69 sub structure {
  70   my ($self, @parts) = @_;
  71   if ($parts[0] =~ /\+/) {
  72     ($parts[0], my @extra) = split /\+/, $parts[0];
  73     my $struct = $self->mangle_structure($self->descend($self->root, @parts));
  74     return $struct unless $struct->[0]{show_columns};
  75     my $first = shift @parts;
  76     my @rest = map [ $_, $self->mangle_structure(
  77                      $self->descend($self->root, $_, @parts)
  78                    )->[0] ], @extra;
  79     my %by_name;
  80     my %host_cols;
  81     my %complex_cols;
  82     foreach my $thing ([ $first, $struct->[0] ], @rest) {
  83       foreach my $el (@{$thing->[1]{data}}) {
  84         my $by = $by_name{$el->{name}} ||= { name => $el->{name} };
  85         foreach my $key (keys %$el) {
  86           next if $key eq 'name';
  87           if (ref($el->{$key}) eq 'HASH') {
  88             $complex_cols{$key} = 1;
  89             $by->{$key} = {};
  90           } else {
  91             my $full_key = $key.' ('.$thing->[0].')';
  92             $host_cols{$full_key} = 1;
  93             $by->{$full_key} = $el->{$key};
  94           }
  95         }
  96       }
  97     }
  98     return [{
  99       columns => [ 'name', sort(keys %host_cols), sort(keys %complex_cols) ],
 100       show_columns => 1,
 101       data => [
 102         map $by_name{$_}, sort keys %by_name
 103       ],
 104     }];
 105   }
 106   return $self->mangle_structure($self->descend($self->root, @parts));
 107 }
 108
 109 sub mangle_structure {
 110   my ($self, $data) = @_;
 111   return unless $data;
 112   if (ref($data) eq 'HASH') {
 113     if (keys %$data > 1
 114         and values %$data == grep ref($_) eq 'HASH', values %$data) {
 115       my %tmp;
 116       $tmp{join '|', keys %$_} = 1 for values %$data;
 117       if (keys %tmp == 1) {
 118         $data->{$_}->{name} ||= $_ for keys %$data;
 119         my @cols = grep $_ ne 'name', sort keys %{(values %$data)[0]};
 120         unshift @cols, 'name';
 121         return [{
 122           columns => \@cols,
 123           show_columns => 1,
 124           data => [ @{$data}{sort keys %$data} ],
 125         }]
 126       }
 127     }
 128     return [{
 129       columns => [ 'key', 'value' ],
 130       data => [ map +{ key => $_, value => $data->{$_} }, sort keys %$data ],
 131     }];
 132   } elsif (ref($data) eq 'ARRAY') {
 133     if (not grep { not ref($_) eq 'HASH' } @$data) {
 134       my %key;
 135       $key{$_} = 1
 136         for map { keys %$_ } @$data;
 137       return [{
 138         columns => [sort keys %key],
 139         show_columns => 1,
 140         data => $data,
 141       }];
 142     }
 143   } elsif (blessed($data) and $data->isa('IO::All::Dir')) {
 144     return [{
 145       columns => [ 'name', 'explore' ],
 146       data => [
 147         map +{ name => $_, explore => $self->link_to($_) }, keys %$data,
 148       ]
 149     }];
 150   } else {
 151     die "Confused by $data";
 152   }
 153 }
 154
 155 sub link_to {
 156   my ($self, @to) = @_;
 157   use HTML::Tags;
 158   my @link = map uri_escape(uri_escape($_)), @to;
 159   my $link = join('/', @link, '');
 160   my $to = $to[-1];
 161   my $html = join '', HTML::Tags::to_html_string(
 162     <a href="${link}">, "Explore $to", </a>
 163   );
 164   return \$html;
 165 }
 166
 167 sub descend {
 168   my ($self, $target, @path) = @_;
 169   return unless $target;
 170   if (blessed($target) and $target->isa('IO::All::File')) {
 171     $target = $self->json->decode(scalar $target->all);
 172   }
 173   return $target unless @path;
 174   my $step = uri_unescape(uri_unescape( shift @path));
 175   $self->descend($target->{$step}, @path);
 176 }
 177
 178 sub render_table {
 179   my ($self, $data) = @_;
 180   use HTML::Tags;
 181   my @rows = (
 182     $data->{show_columns} ? { map +($_ => $_), @{$data->{columns}} } : (),
 183     @{$data->{data}}
 184   );
 185   [ 200, [ 'Content-type' => 'text/html' ], [
 186     HTML::Tags::to_html_string(
 187       <html>, <body>, "\n",
 188       ($data->{wrapper}||sub{@_})->(
 189         '', <table>, "\n",
 190           (map { my $el = $_;
 191             '  ', ($el->{key} eq '__error__') ? <tr class="error"> : <tr>,
 192               (map {
 193                 <td>, $self->render_el($el, $_, $el->{$_}), </td>
 194               } @{$data->{columns}}),
 195             </tr>, "\n"
 196           } @rows),
 197         '', </table>, "\n",
 198       ),
 199       </body>, </html>, "\n",
 200     )
 201   ] ];
 202 }
 203
 204 sub render_el {
 205   my ($self, $whole, $key, $part) = @_;
 206   if (ref($part) eq 'ARRAY') {
 207     if (grep { ref($_) eq 'HASH' } @$part) {
 208       if ($whole->{key}) {
 209         return $self->link_to($whole->{key})
 210       } elsif ($whole->{name}) {
 211         return $self->link_to($whole->{name}, $key);
 212       }
 213     }
 214     return join ', ', @$part
 215       if @$part < 5;
 216     use HTML::Tags;
 217     return <ul>, (map { (<li>, $_, </li>) } @$part), </ul>;
 218   }
 219   if (ref($part) eq 'HASH') {
 220     if ($whole->{key}) {
 221       return $self->link_to($whole->{key})
 222     } elsif ($whole->{name}) {
 223       return $self->link_to($whole->{name}, $key);
 224     }
 225     $part = '(complex)';
 226   }
 227   use HTML::Tags;
 228   return $part =~ /\n/ ? (<pre>, $part, </pre>) : $part;
 229 }
 230
 231 __PACKAGE__->run_if_script;