tmpl_process3.pl did not know how to handle absolute pathnames in -i.
[koha.git] / misc / translator / tmpl_process3.pl
1 #!/usr/bin/perl
2 # This file is part of Koha
3 # Parts copyright 2003-2004 Paul Poulain
4 # Parts copyright 2003-2004 Jerome Vizcaino
5 # Parts copyright 2004 Ambrose Li
6
7 =head1 NAME
8
9 tmpl_process3.pl - Experimental version of tmpl_process.pl
10 using gettext-compatible translation files
11
12 =cut
13
14 use strict;
15 use Getopt::Long;
16 use Locale::PO;
17 use File::Temp qw( :POSIX );
18 use TmplTokenizer;
19 use VerboseWarnings qw( error_normal warn_normal );
20
21 ###############################################################################
22
23 use vars qw( @in_files $in_dir $str_file $out_dir );
24 use vars qw( @excludes $exclude_regex );
25 use vars qw( $recursive_p );
26 use vars qw( $pedantic_p );
27 use vars qw( $href );
28 use vars qw( $type );   # file extension (DOS form without the dot) to match
29 use vars qw( $charset_in $charset_out );
30
31 ###############################################################################
32
33 sub find_translation ($) {
34     my($s) = @_;
35     my $key = $s;
36     if ($s =~ /\S/s) {
37         $key = TmplTokenizer::string_canon($key);
38         $key = TmplTokenizer::charset_convert($key, $charset_in, $charset_out);
39         $key = TmplTokenizer::quote_po($key);
40     }
41     return defined $href->{$key}
42                 && !$href->{$key}->fuzzy
43                 && length Locale::PO->dequote($href->{$key}->msgstr)?
44            Locale::PO->dequote($href->{$key}->msgstr): $s;
45 }
46
47 sub text_replace_tag ($$) {
48     my($t, $attr) = @_;
49     my $it;
50     # value [tag=input], meta
51     my $tag = lc($1) if $t =~ /^<(\S+)/s;
52     my $translated_p = 0;
53     for my $a ('alt', 'content', 'title', 'value') {
54         if ($attr->{$a}) {
55             next if $a eq 'content' && $tag ne 'meta';
56             next if $a eq 'value' && ($tag ne 'input'
57                 || (ref $attr->{'type'} && $attr->{'type'}->[1] =~ /^(?:hidden|radio)$/)); # FIXME
58             my($key, $val, $val_orig, $order) = @{$attr->{$a}}; #FIXME
59             if ($val =~ /\S/s) {
60                 my $s = find_translation($val);
61                 if ($attr->{$a}->[1] ne $s) { #FIXME
62                     $attr->{$a}->[1] = $s; # FIXME
63                     $attr->{$a}->[2] = ($s =~ /"/s)? "'$s'": "\"$s\""; #FIXME
64                     $translated_p = 1;
65                 }
66             }
67         }
68     }
69     if ($translated_p) {
70         $it = "<$tag"
71             . join('', map {
72                     sprintf(' %s=%s', $_, $attr->{$_}->[2]) #FIXME
73                 } sort {
74                     $attr->{$a}->[3] <=> $attr->{$b}->[3] #FIXME
75                 } keys %$attr)
76             . '>';
77     } else {
78         $it = $t;
79     }
80     return $it;
81 }
82
83 sub text_replace (**) {
84     my($h, $output) = @_;
85     for (;;) {
86         my $s = TmplTokenizer::next_token $h;
87     last unless defined $s;
88         my($kind, $t, $attr) = ($s->type, $s->string, $s->attributes);
89         if ($kind eq TmplTokenType::TEXT) {
90             print $output find_translation($t);
91         } elsif ($kind eq TmplTokenType::TEXT_PARAMETRIZED) {
92             my $fmt = find_translation($s->form);
93             print $output TmplTokenizer::parametrize($fmt, [ map {
94                 my($kind, $t, $attr) = ($_->type, $_->string, $_->attributes);
95                 $kind == TmplTokenType::TAG && %$attr?
96                     text_replace_tag($t, $attr): $t } $s->parameters ], [ $s->anchors ]);
97         } elsif ($kind eq TmplTokenType::TAG && %$attr) {
98             print $output text_replace_tag($t, $attr);
99         } elsif (defined $t) {
100             print $output $t;
101         }
102     }
103 }
104
105 sub listfiles ($$) {
106     my($dir, $type) = @_;
107     my @it = ();
108     if (opendir(DIR, $dir)) {
109         my @dirent = readdir DIR;       # because DIR is shared when recursing
110         closedir DIR;
111         for my $dirent (@dirent) {
112             my $path = "$dir/$dirent";
113             if ($dirent =~ /^\./ || $dirent eq 'CVS' || $dirent eq 'RCS'
114             || (defined $exclude_regex && $dirent =~ /^(?:$exclude_regex)$/)) {
115                 ;
116             } elsif (-f $path) {
117                 push @it, $path if !defined $type || $dirent =~ /\.(?:$type)$/;
118             } elsif (-d $path && $recursive_p) {
119                 push @it, listfiles($path, $type);
120             }
121         }
122     } else {
123         warn_normal "$dir: $!", undef;
124     }
125     return @it;
126 }
127
128 ###############################################################################
129
130 sub usage ($) {
131     my($exitcode) = @_;
132     my $h = $exitcode? *STDERR: *STDOUT;
133     print $h <<EOF;
134 Usage: $0 create [OPTION]
135   or:  $0 update [OPTION]
136   or:  $0 install [OPTION]
137   or:  $0 --help
138 Create or update PO files from templates, or install translated templates.
139
140   -i, --input=SOURCE          Get or update strings from SOURCE file.
141                               SOURCE is a directory if -r is also specified.
142   -o, --outputdir=DIRECTORY   Install translation(s) to specified DIRECTORY
143       --pedantic-warnings     Issue warnings even for detected problems
144                               which are likely to be harmless
145   -r, --recursive             SOURCE in the -i option is a directory
146   -s, --str-file=FILE         Specify FILE as the translation (po) file
147                               for input (install) or output (create, update)
148   -x, --exclude=REGEXP        Exclude files matching the given REGEXP
149       --help                  Display this help and exit
150
151 The -o option is ignored for the "create" and "update" actions.
152 EOF
153     exit($exitcode);
154 }
155
156 ###############################################################################
157 sub usage_error (;$) {
158     for my $msg (split(/\n/, $_[0])) {
159         print STDERR "$msg\n";
160     }
161     print STDERR "Try `$0 --help' for more information.\n";
162     exit(-1);
163 }
164
165 ###############################################################################
166
167 GetOptions(
168     'input|i=s'                         => \@in_files,
169     'outputdir|o=s'                     => \$out_dir,
170     'recursive|r'                       => \$recursive_p,
171     'str-file|s=s'                      => \$str_file,
172     'exclude|x=s'                       => \@excludes,
173     'pedantic-warnings|pedantic'        => sub { $pedantic_p = 1 },
174     'help'                              => \&usage,
175 ) || usage_error;
176
177 VerboseWarnings::set_application_name $0;
178 VerboseWarnings::set_pedantic_mode $pedantic_p;
179
180 # keep the buggy Locale::PO quiet if it says stupid things
181 $SIG{__WARN__} = sub {
182         my($s) = @_;
183         print STDERR $s unless $s =~ /^Strange line in [^:]+: #~/s
184     };
185
186 my $action = shift or usage_error('You must specify an ACTION.');
187 usage_error('You must at least specify input and string list filenames.')
188     if !@in_files || !defined $str_file;
189
190 # Type match defaults to *.tmpl plus *.inc if not specified
191 $type = "tmpl|inc" if !defined($type);
192
193 # Check the inputs for being files or directories
194 for my $input (@in_files) {
195     usage_error("$input: Input must be a file or directory.\n"
196             . "(Symbolic links are not supported at the moment)")
197         unless -d $input || -f $input;;
198 }
199
200 # Generates the global exclude regular expression
201 $exclude_regex =  '(?:'.join('|', @excludes).')' if @excludes;
202
203 # Generate the list of input files if a directory is specified
204 if (-d $in_files[0]) {
205     die "If you specify a directory as input, you must specify only it.\n"
206             if @in_files > 1;
207
208     # input is a directory, generates list of files to process
209     $in_dir = $in_files[0];
210     $in_dir =~ s/\/$//; # strips the trailing / if any
211     @in_files = listfiles($in_dir, $type);
212 } else {
213     for my $input (@in_files) {
214         die "You cannot specify input files and directories at the same time.\n"
215                 unless -f $input;
216     }
217 }
218
219 # restores the string list from file
220 $href = Locale::PO->load_file_ashash($str_file);
221
222 # guess the charsets. HTML::Templates defaults to iso-8859-1
223 if (defined $href) {
224     die "$str_file: PO file is corrupted, or not a PO file\n"
225             unless defined $href->{'""'};
226     $charset_out = TmplTokenizer::charset_canon $2
227             if $href->{'""'}->msgstr =~ /\bcharset=(["']?)([^;\s"'\\]+)\1/;
228     for my $msgid (keys %$href) {
229         if ($msgid =~ /\bcharset=(["']?)([^;\s"'\\]+)\1/) {
230             my $candidate = TmplTokenizer::charset_canon $2;
231             die "Conflicting charsets in msgid: $charset_in vs $candidate\n"
232                     if defined $charset_in && $charset_in ne $candidate;
233             $charset_in = $candidate;
234         }
235     }
236 }
237 if (!defined $charset_in) {
238     $charset_in = TmplTokenizer::charset_canon 'iso8859-1';
239     warn "Warning: Can't determine original templates' charset, defaulting to $charset_in\n";
240 }
241
242 my $xgettext = './xgettext.pl'; # actual text extractor script
243
244 if ($action eq 'create')  {
245     # updates the list. As the list is empty, every entry will be added
246     if (!-s $str_file) {
247         warn "Removing empty file $str_file\n";
248         unlink $str_file || die "$str_file: $!\n";
249     }
250     die "$str_file: Output file already exists\n" if -f $str_file;
251     my($tmph, $tmpfile) = tmpnam();
252     # Generate the temporary file that acts as <MODULE>/POTFILES.in
253     for my $input (@in_files) {
254         print $tmph "$input\n";
255     }
256     close $tmph;
257     # Generate the specified po file ($str_file)
258     my $st = system ($xgettext, '-s', '-f', $tmpfile, '-o', $str_file);
259     warn_normal "Text extraction failed: $xgettext: $!\n", undef if $st != 0;
260     unlink $tmpfile || warn_normal "$tmpfile: unlink failed: $!\n", undef;
261
262 } elsif ($action eq 'update') {
263     my($tmph1, $tmpfile1) = tmpnam();
264     my($tmph2, $tmpfile2) = tmpnam();
265     close $tmph2; # We just want a name
266     # Generate the temporary file that acts as <MODULE>/POTFILES.in
267     for my $input (@in_files) {
268         print $tmph1 "$input\n";
269     }
270     close $tmph1;
271     # Generate the temporary file that acts as <MODULE>/<LANG>.pot
272     my $st = system($xgettext, '-s', '-f', $tmpfile1, '-o', $tmpfile2,
273             '--po-mode',
274             (defined $charset_in? ('-I', $charset_in): ()),
275             (defined $charset_out? ('-O', $charset_out): ()));
276     if ($st == 0) {
277         # Merge the temporary "pot file" with the specified po file ($str_file)
278         # FIXME: msgmerge(1) is a Unix dependency
279         # FIXME: need to check the return value
280         system('msgmerge', '-U', '-s', $str_file, $tmpfile2);
281     } else {
282         warn_normal "Text extraction failed: $xgettext: $!\n", undef;
283         warn_normal "Will not run msgmerge\n", undef;
284     }
285     unlink $tmpfile1 || warn_normal "$tmpfile1: unlink failed: $!\n", undef;
286     unlink $tmpfile2 || warn_normal "$tmpfile2: unlink failed: $!\n", undef;
287
288 } elsif ($action eq 'install') {
289     if(!defined($out_dir)) {
290         usage_error("You must specify an output directory when using the install method.");
291     }
292         
293     if ($in_dir eq $out_dir) {
294         warn "You must specify a different input and output directory.\n";
295         exit -1;
296     }
297
298     # Make sure the output directory exists
299     # (It will auto-create it, but for compatibility we should not)
300     -d $out_dir || die "$out_dir: The directory does not exist\n";
301
302     # Try to open the file, because Locale::PO doesn't check :-/
303     open(INPUT, "<$str_file") || die "$str_file: $!\n";
304     close INPUT;
305
306     # creates the new tmpl file using the new translation
307     for my $input (@in_files) {
308         die "Assertion failed"
309                 unless substr($input, 0, length($in_dir) + 1) eq "$in_dir/";
310
311         my $h = TmplTokenizer->new( $input );
312         $h->set_allow_cformat( 1 );
313         VerboseWarnings::set_input_file_name $input;
314
315         my $target = $out_dir . substr($input, length($in_dir));
316         my $targetdir = $` if $target =~ /[^\/]+$/s;
317         if (!-d $targetdir) {
318             print STDERR "Making directory $targetdir...";
319             # creates with rwxrwxr-x permissions
320             mkdir($targetdir, 0775) || warn_normal "$targetdir: $!", undef;
321         }
322         print STDERR "Creating $target...\n";
323         open( OUTPUT, ">$target" ) || die "$target: $!\n";
324         text_replace( $h, *OUTPUT );
325         close OUTPUT;
326     }
327
328 } else {
329     usage_error('Unknown action specified.');
330 }
331 exit 0;
332
333 ###############################################################################
334
335 =head1 SYNOPSIS
336
337 ./tmpl_process3.pl [ I<tmpl_process.pl options> ]
338
339 =head1 DESCRIPTION
340
341 This is an experimental version of the tmpl_process.pl script,
342 using standard gettext-style PO files.  Note that the behaviour
343 of this script should still be considered unstable.
344
345 Currently, the create, update, and install actions have all been
346 reimplemented and seem to work.
347
348 The create action calls xgettext.pl to do the actual work;
349 the update action calls xgettext.pl and msgmerge(1) to do the
350 actual work.
351
352 The script can detect <TMPL_VAR> directives embedded inside what
353 appears to be a full sentence (this actual work being done by
354 TmplTokenizer(3)); these larger patterns appear in the translation
355 file as c-format strings with %s.
356
357 Whitespace in extracted strings are folded to single blanks, in
358 order to prevent new strings from appearing when minor changes in
359 the original templates occur, and to prevent overly difficult to
360 read strings in the PO file.
361
362 =head1 BUGS
363
364 xgettext.pl must be present in the current directory; the
365 msgmerge(1) command must also be present in the search path.
366 The script currently does not check carefully whether these
367 dependent commands are present.
368
369 Locale::PO(3) has a lot of bugs. It can neither parse nor
370 generate GNU PO files properly; a couple of workarounds have
371 been written in TmplTokenizer and more is likely to be needed
372 (e.g., to get rid of the "Strange line" warning for #~).
373
374 There are probably some other bugs too, since this has not been
375 tested very much.
376
377 =head1 SEE ALSO
378
379 xgettext.pl,
380 msgmerge(1),
381 Locale::PO(3),
382 translator_doc.txt
383
384 =cut