summaryrefslogtreecommitdiff
path: root/httemplate/search/elements/search-xls.html
blob: 8334497d2ec00a59973257ed5dedc2e0d4717048 (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
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
<%init>

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

my $style  = $opt{'style'};

my $override = scalar(@$rows) >= 65536 ? 'XLSX' : '';

my $format = $FS::CurrentUser::CurrentUser->spreadsheet_format($override);

my $filename = $opt{'name'} || PL($opt{'name_singular'});
$filename .= $format->{extension};

#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' => $format->{mime_type} );
http_header('Content-Disposition' => qq!attachment;filename="$filename"! );
 
#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 = $format->{class}->new($XLS)
  or die "Error opening Excel file: $!";

my $title = $opt{'title'};
$title =~ s/[\[\]\:\*\?\/\/]//g;
$title = substr($title, 0, 31);
my $worksheet = $workbook->add_worksheet($title);

$worksheet->protect();

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

my $header_format = $workbook->add_format(
  bold     => 1,
  locked   => 1,
  bg_color => 55, #22,
  bottom   => 3,
);
my $footer_format = $workbook->add_format(
  italic   => 1,
  locked   => 1,
  bg_color => 55,
  top      => 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 %bold_format;

my $writer = sub {
  # Wrapper for $worksheet->write.
  # Do any massaging of the value/format here.
  my ($r, $c, $value, $format) = @_;
  #warn "writer called with format $format\n";

  if ( $style->[$c] eq 'b' or $value =~ /<b>/i ) { # the only one in common use
    $value =~ s[</?b>][]ig;
    if ( !exists($bold_format{$format}) ) {
      $bold_format{$format} = $workbook->add_format();
      $bold_format{$format}->copy($format);
      $bold_format{$format}->set_bold();
    }
    $format = $bold_format{$format};
  }

  # convert HTML entities
  # both Spreadsheet::WriteExcel and Excel::Writer::XLSX accept UTF-8 strings
  $value = decode_entities($value);

  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;
  }
  #warn "writing with format $format\n";
  $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;
  }

}

if ( $opt{'footer'} ) {
  $r++;
  $c = 0;
  foreach my $item (@{ $opt{'footer'} }) {
    if ( ref($item) eq 'CODE' ) {
      $item = &{$item}();
    }
    $writer->( $r, $c++, $item, $footer_format );
  }
}

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

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

</%init>