<% my(%opt) = @_; #warn join(' / ', map { "$_ => $opt{$_}" } keys %opt ). "\n"; my %align = ( 'l' => 'left', 'r' => 'right', 'c' => 'center', ' ' => '', '.' => '', ); $opt{align} = [ map $align{$_}, split(//, $opt{align}) ], unless !$opt{align} || ref($opt{align}); my $type = ''; my $limit = ''; my($maxrecords, $total, $offset, $count_arrayref); if ( $cgi->param('_type') =~ /^(csv|\w*\.xls)$/ ) { $type = $1; } else { #setup some pagination things if we're in html mode unless (exists($opt{'count_query'}) && length($opt{'count_query'})) { ( $opt{'count_query'} = $opt{'query'} ) =~ s/^\s*SELECT\s*(.*?)\s+FROM\s/SELECT COUNT(*) FROM /i; } my $conf = new FS::Conf; $maxrecords = $conf->config('maxsearchrecordsperpage'); $limit = $maxrecords ? "LIMIT $maxrecords" : ''; $offset = $cgi->param('offset') || 0; $limit .= " OFFSET $offset" if $offset; my $count_sth = dbh->prepare($opt{'count_query'}) or die "Error preparing $opt{'count_query'}: ". dbh->errstr; $count_sth->execute or die "Error executing $opt{'count_query'}: ". $count_sth->errstr; $count_arrayref = $count_sth->fetchrow_arrayref; $total = $count_arrayref->[0]; } # run the query my $header = $opt{'header'}; my $rows; if ( ref($opt{'query'}) ) { #eval "use FS::$opt{'query'};"; $rows = [ qsearch( $opt{'query'}->{'table'}, $opt{'query'}->{'hashref'} || {}, $opt{'query'}->{'select'}, $opt{'query'}->{'extra_sql'}. " $limit", '', (exists($opt{'query'}->{'addl_from'}) ? $opt{'query'}->{'addl_from'} : '') ) ]; } else { my $sth = dbh->prepare("$opt{'query'} $limit") or die "Error preparing $opt{'query'}: ". dbh->errstr; $sth->execute or die "Error executing $opt{'query'}: ". $sth->errstr; #can get # of rows without fetching them all? $rows = $sth->fetchall_arrayref; $header ||= $sth->{NAME}; } if ( $type eq 'csv' ) { #http_header('Content-Type' => 'text/comma-separated-values' ); #IE chokes http_header('Content-Type' => 'text/plain' ); my $csv = new Text::CSV_XS { 'always_quote' => 1, 'eol' => "\n", #"\015\012", #"\012" }; $csv->combine(@$header); #or die $csv->status; %><%= $csv->string %><% foreach my $row ( @$rows ) { if ( $opt{'fields'} ) { my @line = (); foreach my $field ( @{$opt{'fields'}} ) { if ( ref($field) eq 'CODE' ) { push @line, &{$field}($row); } else { push @line, $row->$field(); } } $csv->combine(@line); #or die $csv->status; } else { $csv->combine(@$row); #or die $csv->status; } %><%= $csv->string %><% } #} elsif ( $type eq 'excel' ) { } elsif ( $type =~ /\.xls$/ ) { #http_header('Content-Type' => 'application/excel' ); #eww http_header('Content-Type' => 'application/vnd.ms-excel' ); #http_header('Content-Type' => 'application/msexcel' ); #alas my $data = ''; my $XLS = new IO::Scalar \$data; my $workbook = Spreadsheet::WriteExcel->new($XLS) or die "Error opening .xls file: $!"; my $worksheet = $workbook->add_worksheet(substr($opt{'title'},0,31)); my($r,$c) = (0,0); $worksheet->write($r, $c++, $_) foreach @$header; foreach my $row ( @$rows ) { $r++; $c = 0; if ( $opt{'fields'} ) { #my $links = $opt{'links'} ? [ @{$opt{'links'}} ] : ''; #my $aligns = $opt{'align'} ? [ @{$opt{'align'}} ] : ''; foreach my $field ( @{$opt{'fields'}} ) { #my $align = $aligns ? shift @$aligns : ''; #$align = " ALIGN=$align" if $align; #my $a = ''; #if ( $links ) { # my $link = shift @$links; # $link = &{$link}($row) if ref($link) eq 'CODE'; # if ( $link ) { # my( $url, $method ) = @{$link}; # if ( ref($method) eq 'CODE' ) { # $a = $url. &{$method}($row); # } else { # $a = $url. $row->$method(); # } # $a = qq(); # } #} if ( ref($field) eq 'CODE' ) { $worksheet->write($r, $c++, &{$field}($row) ); } else { $worksheet->write($r, $c++, $row->$field() ); } } } else { $worksheet->write($r, $c++, $_) foreach @$row; } } $workbook->close();# or die "Error creating .xls file: $!"; http_header('Content-Length' => length($data) ); %><%= $data %><% } else { # regular HTML if ( exists($opt{'redirect'}) && scalar(@$rows) == 1 && $total == 1 ) { my( $url, $method ) = @{$opt{'redirect'}}; redirect( $url. $rows->[0]->$method() ); } else { ( my $xlsname = $opt{'name'} ) =~ s/\W//g; $opt{'name'} =~ s/s$// if $total == 1; my @menubar = (); if ( $opt{'menubar'} ) { @menubar = @{ $opt{'menubar'} }; } else { @menubar = ( 'Main menu' => $p ); } %> <%= include( '/elements/header.html', $opt{'title'}, include( '/elements/menubar.html', @menubar ) ) %> <% my $pager = include ( '/elements/pager.html', 'offset' => $offset, 'num_rows' => scalar(@$rows), 'total' => $total, 'maxrecords' => $maxrecords, ); %> <% unless ( $total ) { %> No matching <%= $opt{'name'} %> found.
<% } else { %>
<%= $total %> total <%= $opt{'name'} %>
<% if ( $opt{'count_addl'} ) { %> <% my $n=0; foreach my $count ( @{$opt{'count_addl'}} ) { %> <%= sprintf( $count, $count_arrayref->[++$n] ) %>
<% } %> <% } %>
<% $cgi->param('_type', "$xlsname.xls" ); %> Download full results
as Excel spreadsheet
<% $cgi->param('_type', 'csv'); %> as CSV file
<%= $pager %> <% foreach my $header ( @$header ) { %> <% } %> <% my $bgcolor1 = '#eeeeee'; my $bgcolor2 = '#ffffff'; my $bgcolor; foreach my $row ( @$rows ) { if ( $bgcolor eq $bgcolor1 ) { $bgcolor = $bgcolor2; } else { $bgcolor = $bgcolor1; } %> <% if ( $opt{'fields'} ) { my $links = $opt{'links'} ? [ @{$opt{'links'}} ] : ''; my $aligns = $opt{'align'} ? [ @{$opt{'align'}} ] : ''; foreach my $field ( @{$opt{'fields'}} ) { my $align = $aligns ? shift @$aligns : ''; $align = " ALIGN=$align" if $align; my $a = ''; if ( $links ) { my $link = shift @$links; $link = &{$link}($row) if ref($link) eq 'CODE'; if ( $link ) { my( $url, $method ) = @{$link}; if ( ref($method) eq 'CODE' ) { $a = $url. &{$method}($row); } else { $a = $url. $row->$method(); } $a = qq(); } } %> <% if ( ref($field) eq 'CODE' ) { %> <% } else { %> <% } %> <% } %> <% } else { %> <% foreach ( @$row ) { %> <% } %> <% } %> <% } %>
<%= $header %>
><%= $a %><%= &{$field}($row) %><%= $a ? '' : '' %>><%= $a %><%= $row->$field() %><%= $a ? '' : '' %><%= $_ %>
<%= $pager %>
<% } %> <% } %> <% } %>