Move cvs_keyword_subst_args() down with the other internal support routines.
[ikiwiki] / IkiWiki / Plugin / cvs.pm
1 #!/usr/bin/perl
2 package IkiWiki::Plugin::cvs;
3
4 # Copyright (c) 2009 Amitai Schlair
5 # All rights reserved.
6 #
7 # This code is derived from software contributed to ikiwiki
8 # by Amitai Schlair.
9 #
10 # Redistribution and use in source and binary forms, with or without
11 # modification, are permitted provided that the following conditions
12 # are met:
13 # 1. Redistributions of source code must retain the above copyright
14 #    notice, this list of conditions and the following disclaimer.
15 # 2. Redistributions in binary form must reproduce the above copyright
16 #    notice, this list of conditions and the following disclaimer in the
17 #    documentation and/or other materials provided with the distribution.
18 #
19 # THIS SOFTWARE IS PROVIDED BY IKIWIKI AND CONTRIBUTORS ``AS IS''
20 # AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 # TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
22 # PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION
23 # OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
26 # USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
27 # ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
28 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
29 # OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30 # SUCH DAMAGE.
31
32 use warnings;
33 use strict;
34 use IkiWiki;
35
36 use File::chdir;
37
38
39 # GENERAL PLUGIN API CALLS
40
41 sub import {
42         hook(type => "checkconfig", id => "cvs", call => \&checkconfig);
43         hook(type => "getsetup", id => "cvs", call => \&getsetup);
44         hook(type => "genwrapper", id => "cvs", call => \&genwrapper);
45
46         hook(type => "rcs", id => "rcs_update", call => \&rcs_update);
47         hook(type => "rcs", id => "rcs_prepedit", call => \&rcs_prepedit);
48         hook(type => "rcs", id => "rcs_commit", call => \&rcs_commit);
49         hook(type => "rcs", id => "rcs_commit_staged", call => \&rcs_commit_staged);
50         hook(type => "rcs", id => "rcs_add", call => \&rcs_add);
51         hook(type => "rcs", id => "rcs_remove", call => \&rcs_remove);
52         hook(type => "rcs", id => "rcs_rename", call => \&rcs_rename);
53         hook(type => "rcs", id => "rcs_recentchanges", call => \&rcs_recentchanges);
54         hook(type => "rcs", id => "rcs_diff", call => \&rcs_diff);
55         hook(type => "rcs", id => "rcs_getctime", call => \&rcs_getctime);
56         hook(type => "rcs", id => "rcs_getmtime", call => \&rcs_getmtime);
57 }
58
59 sub checkconfig () {
60         if (! defined $config{cvspath}) {
61                 $config{cvspath}="ikiwiki";
62         }
63         if (exists $config{cvspath}) {
64                 # code depends on the path not having extraneous slashes
65                 $config{cvspath}=~tr#/#/#s;
66                 $config{cvspath}=~s/\/$//;
67                 $config{cvspath}=~s/^\///;
68         }
69         if (defined $config{cvs_wrapper} && length $config{cvs_wrapper}) {
70                 push @{$config{wrappers}}, {
71                         wrapper => $config{cvs_wrapper},
72                         wrappermode => (defined $config{cvs_wrappermode} ? $config{cvs_wrappermode} : "04755"),
73                 };
74         }
75 }
76
77 sub getsetup () {
78         return
79                 plugin => {
80                         safe => 0, # rcs plugin
81                         rebuild => undef,
82                         section => "rcs",
83                 },
84                 cvsrepo => {
85                         type => "string",
86                         example => "/cvs/wikirepo",
87                         description => "cvs repository location",
88                         safe => 0, # path
89                         rebuild => 0,
90                 },
91                 cvspath => {
92                         type => "string",
93                         example => "ikiwiki",
94                         description => "path inside repository where the wiki is located",
95                         safe => 0, # paranoia
96                         rebuild => 0,
97                 },
98                 cvs_wrapper => {
99                         type => "string",
100                         example => "/cvs/wikirepo/CVSROOT/post-commit",
101                         description => "cvs post-commit hook to generate (triggered by CVSROOT/loginfo entry)",
102                         safe => 0, # file
103                         rebuild => 0,
104                 },
105                 cvs_wrappermode => {
106                         type => "string",
107                         example => '04755',
108                         description => "mode for cvs_wrapper (can safely be made suid)",
109                         safe => 0,
110                         rebuild => 0,
111                 },
112                 historyurl => {
113                         type => "string",
114                         example => "http://cvs.example.org/cvsweb.cgi/ikiwiki/[[file]]",
115                         description => "cvsweb url to show file history ([[file]] substituted)",
116                         safe => 1,
117                         rebuild => 1,
118                 },
119                 diffurl => {
120                         type => "string",
121                         example => "http://cvs.example.org/cvsweb.cgi/ikiwiki/[[file]].diff?r1=text&tr1=[[r1]]&r2=text&tr2=[[r2]]",
122                         description => "cvsweb url to show a diff ([[file]], [[r1]], and [[r2]] substituted)",
123                         safe => 1,
124                         rebuild => 1,
125                 },
126 }
127
128 sub genwrapper () {
129         return <<EOF;
130         {
131                 int j;
132                 for (j = 1; j < argc; j++)
133                         if (strstr(argv[j], "New directory") != NULL)
134                                 exit(0);
135         }
136 EOF
137 }
138
139
140 # VCS PLUGIN API CALLS
141
142 sub rcs_update () {
143         return unless cvs_is_controlling();
144         cvs_runcvs('update', '-dP');
145 }
146
147 sub rcs_prepedit ($) {
148         # Prepares to edit a file under revision control. Returns a token
149         # that must be passed into rcs_commit when the file is ready
150         # for committing.
151         # The file is relative to the srcdir.
152         my $file=shift;
153
154         return unless cvs_is_controlling();
155
156         # For cvs, return the revision of the file when
157         # editing begins.
158         my $rev=cvs_info("Repository revision", "$file");
159         return defined $rev ? $rev : "";
160 }
161
162 sub rcs_commit (@) {
163         # Tries to commit the page; returns undef on _success_ and
164         # a version of the page with the rcs's conflict markers on failure.
165         # The file is relative to the srcdir.
166         my %params=@_;
167
168         return unless cvs_is_controlling();
169
170         # Check to see if the page has been changed by someone
171         # else since rcs_prepedit was called.
172         my ($oldrev)=$params{token}=~/^([0-9]+)$/; # untaint
173         my $rev=cvs_info("Repository revision", "$config{srcdir}/$params{file}");
174         if (defined $rev && defined $oldrev && $rev != $oldrev) {
175                 # Merge their changes into the file that we've
176                 # changed.
177                 cvs_runcvs('update', $params{file}) ||
178                         warn("cvs merge from $oldrev to $rev failed\n");
179         }
180
181         if (! cvs_runcvs('commit', '-m',
182                          IkiWiki::possibly_foolish_untaint(commitmessage(%params)))) {
183                 my $conflict=readfile("$config{srcdir}/$params{file}");
184                 cvs_runcvs('update', '-C', $params{file}) ||
185                         warn("cvs revert failed\n");
186                 return $conflict;
187         }
188
189         return undef # success
190 }
191
192 sub rcs_commit_staged (@) {
193         # Commits all staged changes. Changes can be staged using rcs_add,
194         # rcs_remove, and rcs_rename.
195         my %params=@_;
196
197         if (! cvs_runcvs('commit', '-m',
198                          IkiWiki::possibly_foolish_untaint(commitmessage(%params)))) {
199                 warn "cvs staged commit failed\n";
200                 return 1; # failure
201         }
202         return undef # success
203 }
204
205 sub rcs_add ($) {
206         # filename is relative to the root of the srcdir
207         my $file=shift;
208         my $parent=IkiWiki::dirname($file);
209         my @files_to_add = ($file);
210
211         until ((length($parent) == 0) || cvs_is_controlling("$config{srcdir}/$parent")){
212                 push @files_to_add, $parent;
213                 $parent = IkiWiki::dirname($parent);
214         }
215
216         while ($file = pop @files_to_add) {
217                 if (@files_to_add == 0) {
218                         # file
219                         cvs_runcvs('add', cvs_keyword_subst_args($file)) ||
220                                 warn("cvs add $file failed\n");
221                 }
222                 else {
223                         # directory
224                         cvs_runcvs('add', $file) ||
225                                 warn("cvs add $file failed\n");
226                 }
227         }
228 }
229
230 sub rcs_remove ($) {
231         # filename is relative to the root of the srcdir
232         my $file=shift;
233
234         return unless cvs_is_controlling();
235
236         cvs_runcvs('rm', '-f', $file) ||
237                 warn("cvs rm $file failed\n");
238 }
239
240 sub rcs_rename ($$) {
241         # filenames relative to the root of the srcdir
242         my ($src, $dest)=@_;
243
244         return unless cvs_is_controlling();
245
246         local $CWD = $config{srcdir};
247
248         if (system("mv", "$src", "$dest") != 0) {
249                 warn("filesystem rename failed\n");
250         }
251
252         rcs_add($dest);
253         rcs_remove($src);
254 }
255
256 sub rcs_recentchanges ($) {
257         my $num = shift;
258         my @ret;
259
260         return unless cvs_is_controlling();
261
262         eval q{use Date::Parse};
263         error($@) if $@;
264
265         local $CWD = $config{srcdir};
266
267         # There's no cvsps option to get the last N changesets.
268         # Write full output to a temp file and read backwards.
269
270         eval q{use File::Temp qw/tempfile/};
271         error($@) if $@;
272         eval q{use File::ReadBackwards};
273         error($@) if $@;
274
275         my ($tmphandle, $tmpfile) = tempfile();
276         system("env TZ=UTC cvsps -q --cvs-direct -z 30 -x >$tmpfile");
277         if ($? == -1) {
278                 error "couldn't run cvsps: $!\n";
279         }
280         elsif (($? >> 8) != 0) {
281                 error "cvsps exited " . ($? >> 8) . ": $!\n";
282         }
283
284         tie(*SPSVC, 'File::ReadBackwards', $tmpfile)
285                 || error "couldn't open $tmpfile for read: $!\n";
286
287         while (my $line = <SPSVC>) {
288                 $line =~ /^$/ || error "expected blank line, got $line";
289
290                 my ($rev, $user, $committype, $when);
291                 my (@message, @pages);
292
293                 # We're reading backwards.
294                 # Forwards, an entry looks like so:
295                 # ---------------------
296                 # PatchSet $rev
297                 # Date: $when
298                 # Author: $user (or user CGI runs as, for web commits)
299                 # Branch: branch
300                 # Tag: tag
301                 # Log:
302                 # @message_lines
303                 # Members:
304                 #       @pages (and revisions)
305                 #
306
307                 while ($line = <SPSVC>) {
308                         last if ($line =~ /^Members:/);
309                         for ($line) {
310                                 s/^\s+//;
311                                 s/\s+$//;
312                         }
313                         my ($page, $revs) = split(/:/, $line);
314                         my ($oldrev, $newrev) = split(/->/, $revs);
315                         $oldrev =~ s/INITIAL/0/;
316                         $newrev =~ s/\(DEAD\)//;
317                         my $diffurl = defined $config{diffurl} ? $config{diffurl} : "";
318                         $diffurl=~s/\[\[file\]\]/$page/g;
319                         $diffurl=~s/\[\[r1\]\]/$oldrev/g;
320                         $diffurl=~s/\[\[r2\]\]/$newrev/g;
321                         unshift @pages, {
322                                 page => pagename($page),
323                                 diffurl => $diffurl,
324                         } if length $page;
325                 }
326
327                 while ($line = <SPSVC>) {
328                         last if ($line =~ /^Log:$/);
329                         chomp $line;
330                         unshift @message, { line => $line };
331                 }
332                 $committype = "web";
333                 if (defined $message[0] &&
334                     $message[0]->{line}=~/$config{web_commit_regexp}/) {
335                         $user=defined $2 ? "$2" : "$3";
336                         $message[0]->{line}=$4;
337                 }
338                 else {
339                         $committype="cvs";
340                 }
341
342                 $line = <SPSVC>;        # Tag
343                 $line = <SPSVC>;        # Branch
344
345                 $line = <SPSVC>;
346                 if ($line =~ /^Author: (.*)$/) {
347                         $user = $1 unless defined $user && length $user;
348                 }
349                 else {
350                         error "expected Author, got $line";
351                 }
352
353                 $line = <SPSVC>;
354                 if ($line =~ /^Date: (.*)$/) {
355                         $when = str2time($1, 'UTC');
356                 }
357                 else {
358                         error "expected Date, got $line";
359                 }
360
361                 $line = <SPSVC>;
362                 if ($line =~ /^PatchSet (.*)$/) {
363                         $rev = $1;
364                 }
365                 else {
366                         error "expected PatchSet, got $line";
367                 }
368
369                 $line = <SPSVC>;        # ---------------------
370
371                 push @ret, {
372                         rev => $rev,
373                         user => $user,
374                         committype => $committype,
375                         when => $when,
376                         message => [@message],
377                         pages => [@pages],
378                 } if @pages;
379                 last if @ret >= $num;
380         }
381
382         unlink($tmpfile) || error "couldn't unlink $tmpfile: $!\n";
383
384         return @ret;
385 }
386
387 sub rcs_diff ($;$) {
388         my $rev=IkiWiki::possibly_foolish_untaint(int(shift));
389         my $maxlines=shift;
390
391         local $CWD = $config{srcdir};
392
393         # diff output is unavoidably preceded by the cvsps PatchSet entry
394         my @cvsps = `env TZ=UTC cvsps -q --cvs-direct -z 30 -g -s $rev`;
395         my $blank_lines_seen = 0;
396
397         while (my $line = shift @cvsps) {
398                 $blank_lines_seen++ if ($line =~ /^$/);
399                 last if $blank_lines_seen == 2;
400         }
401
402         if (wantarray) {
403                 return @cvsps;
404         }
405         else {
406                 return join("", @cvsps);
407         }
408 }
409
410 sub rcs_getctime ($) {
411         my $file=shift;
412
413         local $CWD = $config{srcdir};
414
415         my $cvs_log_infoline=qr/^date: (.+);\s+author/;
416
417         open CVSLOG, "cvs -Q log -r1.1 '$file' |"
418                 || error "couldn't get cvs log output: $!\n";
419
420         my $date;
421         while (<CVSLOG>) {
422                 if (/$cvs_log_infoline/) {
423                         $date=$1;
424                 }
425         }
426         close CVSLOG || warn "cvs log $file exited $?";
427
428         if (! defined $date) {
429                 warn "failed to parse cvs log for $file\n";
430                 return 0;
431         }
432
433         eval q{use Date::Parse};
434         error($@) if $@;
435         $date=str2time($date, 'UTC');
436         debug("found ctime ".localtime($date)." for $file");
437         return $date;
438 }
439
440 sub rcs_getmtime ($) {
441         error "rcs_getmtime is not implemented for cvs\n"; # TODO
442 }
443
444
445 # INTERNAL SUPPORT ROUTINES
446
447 sub commitmessage (@) {
448         my %params=@_;
449
450         if (defined $params{session}) {
451                 if (defined $params{session}->param("name")) {
452                         return "web commit by ".
453                                 $params{session}->param("name").
454                                 (length $params{message} ? ": $params{message}" : "");
455                 }
456                 elsif (defined $params{session}->remote_addr()) {
457                         return "web commit from ".
458                                 $params{session}->remote_addr().
459                                 (length $params{message} ? ": $params{message}" : "");
460                 }
461         }
462         return $params{message};
463 }
464
465 sub cvs_info ($$) {
466         my $field=shift;
467         my $file=shift;
468
469         local $CWD = $config{srcdir};
470
471         my $info=`cvs status $file`;
472         my ($ret)=$info=~/^\s*$field:\s*(\S+)/m;
473         return $ret;
474 }
475
476 sub cvs_is_controlling {
477         my $dir=shift;
478         $dir=$config{srcdir} unless defined($dir);
479         return (-d "$dir/CVS") ? 1 : 0;
480 }
481
482 sub cvs_keyword_subst_args ($) {
483         my $file = shift;
484
485         local $CWD = $config{srcdir};
486
487         eval q{use File::MimeInfo};
488         error($@) if $@;
489         my $filemime = File::MimeInfo::default($file);
490         # if (-T $file) {
491
492         if (defined($filemime) && $filemime eq 'text/plain') {
493                 return ($file);
494         }
495         else {
496                 return ('-kb', $file);
497         }
498 }
499
500 sub cvs_runcvs(@) {
501         my @cmd = @_;
502         unshift @cmd, 'cvs', '-Q';
503
504         local $CWD = $config{srcdir};
505
506         open(my $savedout, ">&STDOUT");
507         open(STDOUT, ">", "/dev/null");
508         my $ret = system(@cmd);
509         open(STDOUT, ">&", $savedout);
510
511         return ($ret == 0) ? 1 : 0;
512 }
513
514 1