summaryrefslogtreecommitdiff
path: root/httemplate/search/elements/search-xls.html
blob: 0b5636c0ed4e5996316c4231763699b1a752870d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
<% $data %>
<%init>

my %args = @_;
my $type   = $args{'type'};
my $header = $args{'header'};
my $rows   = $args{'rows'};
my %opt    = %{ $args{'opt'} };    

#http_header('Content-Type' => 'application/excel' ); #eww
#http_header('Content-Type' => 'application/msexcel' ); #alas
#http_header('Content-Type' => 'application/x-msexcel' ); #?

#http://support.microsoft.com/kb/199841
http_header('Content-Type' => 'application/vnd.ms-excel' );
http_header('Content-Disposition' => 
  'attachment;filename="'.($opt{'name'} || PL($opt{'name_singular'}) ).'.xls"');
 
#http://support.microsoft.com/kb/812935
#http://support.microsoft.com/kb/323308
$HTML::Mason::Commands::r->headers_out->{'Cache-control'} = 'max-age=0';

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));

$worksheet->protect();

my($r,$c) = (0,0);

my $header_format = $workbook->add_format(
  bold     => 1,
  locked   => 1,
  bg_color => 55, #22,
  bottom   => 3,
);
my $default_format = $workbook->add_format(locked => 0);

my %money_format;
my $money_char = FS::Conf->new->config('money_char') || '$';

my $writer = sub {
  # Wrapper for $worksheet->write.
  # Do any massaging of the value/format here.
  my ($r, $c, $value, $format) = @_;
  if ( $value =~ /^\Q$money_char\E(\d+\.?\d*)$/ ) {
    # Currency: strip the symbol, clone the requested format,
    # and format it for currency
    $value = $1;
    if ( !exists($money_format{$format}) ) {
      $money_format{$format} = $workbook->add_format();
      $money_format{$format}->copy($format);
      $money_format{$format}->set_num_format($money_char.'#0.00#');
    }
    $format = $money_format{$format};
  }
  $worksheet->write($r, $c, $value, $format);
};

$writer->( $r, $c++, $_, $header_format ) foreach @$header;

foreach my $row ( @$rows ) {
  $r++;
  $c = 0;

  if ( $opt{'fields'} ) {

    #my $links = $opt{'links'} ? [ @{$opt{'links'}} ] : '';
    #my $aligns = $opt{'align'} ? [ @{$opt{'align'}} ] : '';
    #could also translate color, size, style into xls equivalents?
    my $formats = $opt{'xls_format'} ? [ @{$opt{'xls_format'}} ] : [];

    foreach my $field ( @{$opt{'fields'}} ) {

      my $xls_format = $default_format;

      if ( my $format = shift @$formats ) {
        $format = &{$format}($row) if ref($format) eq 'CODE';
        $format ||= {};
        $xls_format = $workbook->add_format(locked=>0, %$format);
      }

      if ( ref($field) eq 'CODE' ) {
        foreach my $value ( &{$field}($row) ) {
          if ( ref($value) eq 'ARRAY' ) { 
            $writer->($r, $c++, '(N/A)' ); #unimplemented
          } else {
            $writer->($r, $c++, $value, $xls_format );
          }
        }
      } else {
        $writer->( $r, $c++, $row->$field(), $xls_format );
      }
    }

  } else {
    # no need for each row to need a new format
    #my $xls_format = $workbook->add_format(locked=>0);
    $writer->( $r, $c++, $_, $default_format ) foreach @$row;
  }

}

$workbook->close();# or die "Error creating .xls file: $!";

http_header('Content-Length' => length($data) );

</%init>