summaryrefslogtreecommitdiff
path: root/httemplate/search/elements/search-xls.html
blob: a3a8226c55e9b53c730e69e0883aa82c368a7590 (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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
<% $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 %date_format;
xl_parse_date_init();

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;
#    warn "formatting $value as money\n";
    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};
  }
  elsif ( $value =~ /^([A-Z][a-z]{2}) (\d{2}) (\d{4})$/ ) {
    # Date: convert the value to an Excel date number and set 
    # the format
    $value = xl_parse_date($value);
#    warn "formatting $value as date\n";
    if ( !exists($date_format{$format}) ) {
      $date_format{$format} = $workbook->add_format();
      $date_format{$format}->copy($format);
      $date_format{$format}->set_num_format('mmm dd yyyy');
    }
    $format = $date_format{$format};
  }
  else {
    # String: replace line breaks with newlines
    $value =~ s/<BR>/\n/gi;
  }
  $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>