%
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' => "\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;
%>
<%= include( '/elements/header.html', $opt{'title'},
include( '/elements/menubar.html', 'Main menu' => $p )
)
%>
<% 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 %>
|