* Numerous tla fixes from Clint.
[ikiwiki.git] / IkiWiki / Rcs / tla.pm
1 #!/usr/bin/perl
2
3 use warnings;
4 use strict;
5 use IkiWiki;
6 use POSIX qw(setlocale LC_CTYPE);
7
8 package IkiWiki;
9
10 my $tla_webcommit=qr/^web commit (by (\w+)|from (\d+\.\d+\.\d+\.\d+)):?(.*)/;
11
12 sub quiet_system (@) {
13         # See Debian bug #385939.
14         open (SAVEOUT, ">&STDOUT");
15         close STDOUT;
16         open (STDOUT, ">/dev/null");
17         my $ret=system(@_);
18         open (STDOUT, ">&SAVEOUT");
19         close SAVEOUT;
20         return $ret;
21 }
22
23 sub rcs_update () { #{{{
24         if (-d "$config{srcdir}/{arch}") {
25                 if (quiet_system("tla", "replay", "-d", $config{srcdir}) != 0) {
26                         warn("tla replay failed\n");
27                 }
28         }
29 } #}}}
30
31 sub rcs_prepedit ($) { #{{{
32         my $file=shift;
33
34         if (-d "$config{srcdir}/{arch}") {
35                 # For Arch, return the tree-id of archive when
36                 # editing begins.
37                 my $rev=`tla tree-id $config{srcdir}`;
38                 return defined $rev ? $rev : "";
39         }
40 } #}}}
41
42 sub rcs_commit ($$$) { #{{{
43         my $file=shift;
44         my $message=shift;
45         my $rcstoken=shift;
46
47         if (-d "$config{srcdir}/{arch}") {
48                 # Check to see if the page has been changed by someone
49                 # else since rcs_prepedit was called.
50                 my ($oldrev)=$rcstoken=~/^([A-Za-z0-9@\/._-]+)$/; # untaint
51                 my $rev=`tla tree-id $config{srcdir}`;
52                 if (defined $rev && defined $oldrev && $rev ne $oldrev) {
53                         # Merge their changes into the file that we've
54                         # changed.
55                         if (quiet_system("tla", "update", "-d",
56                                    "$config{srcdir}") != 0) {
57                                 warn("tla update failed\n");
58                         }
59                 }
60
61                 if (quiet_system("tla", "commit",
62                            "-L".possibly_foolish_untaint($message),
63                            '-d', $config{srcdir}) != 0) {
64                         my $conflict=readfile("$config{srcdir}/$file");
65                         if (system("tla", "undo", "-n", "--quiet", "-d", "$config{srcdir}") != 0) {
66                                 warn("tla undo failed\n");
67                         }
68                         return $conflict;
69                 }
70         }
71         return undef # success
72 } #}}}
73
74 sub rcs_add ($) { #{{{
75         my $file=shift;
76
77         if (-d "$config{srcdir}/{arch}") {
78                 if (quiet_system("tla", "add", "$config{srcdir}/$file") != 0) {
79                         warn("tla add failed\n");
80                 }
81         }
82 } #}}}
83
84 sub rcs_recentchanges ($) {
85         my $num=shift;
86         my @ret;
87
88         return unless -d "$config{srcdir}/{arch}";
89
90         eval q{use Date::Parse};
91         eval q{use Mail::Header};
92
93         my $logs = `tla logs -d $config{srcdir}`;
94         my @changesets = reverse split(/\n/, $logs);
95
96         for (my $i=0; $i<$num && $i<$#changesets; $i++) {
97                 my ($change)=$changesets[$i]=~/^([A-Za-z0-9@\/._-]+)$/; # untaint
98
99                 open(LOG, "tla cat-log -d $config{srcdir} $change|");
100                 my $head = Mail::Header->new(\*LOG);
101                 close(LOG);
102
103                 my $rev = $head->get("Revision");
104                 my $summ = $head->get("Summary");
105                 my $newfiles = $head->get("New-files");
106                 my $modfiles = $head->get("Modified-files");
107                 my $user = $head->get("Creator");
108
109                 my @paths = grep {!/^(.*\/\)?.arch-ids\/.*\.id$/} split(/ /,
110                         "$newfiles $modfiles");
111
112                 my $sdate = $head->get("Standard-date");
113                 my $when = time - str2time($sdate, 'UTC');
114
115                 my $committype = "web";
116                 if (defined $summ && $summ =~ /$tla_webcommit/) {
117                         $user = defined $2 ? "$2" : "$3";
118                         $summ = $4;
119                 }
120                 else {
121                         $committype="tla";
122                 }
123
124                 my @message;
125                 push @message, { line => escapeHTML($summ) };
126
127                 my @pages;
128
129                 foreach my $file (@paths) {
130                         my $diffurl=$config{diffurl};
131                         $diffurl=~s/\[\[file\]\]/$file/g;
132                         $diffurl=~s/\[\[rev\]\]/$change/g;
133                         push @pages, {
134                                 page => pagename($file),
135                                 diffurl => $diffurl,
136                         } if length $file;
137                 }
138                 push @ret, { rev => $change,
139                         user => $user,
140                         committype => $committype,
141                         when => $when,
142                         message => [@message],
143                         pages => [@pages],
144                 } if @pages;
145
146                 last if $i == $num;
147         }
148
149         return @ret;
150 }
151
152 sub rcs_notify () { #{{{
153         # FIXME: Not set
154         if (! exists $ENV{ARCH_VERSION}) {
155                 error("ARCH_VERSION is not set, not running from tla post-commit hook, cannot send notifications");
156         }
157         my $rev=int(possibly_foolish_untaint($ENV{REV}));
158
159         eval q{use Mail::Header};
160         open(LOG, $ENV{"ARCH_LOG"});
161         my $head = Mail::Header->new(\*LOG);
162         close(LOG);
163
164         my $message = $head->get("Summary");
165         my $user = $head->get("Creator");
166
167         my $newfiles = $head->get("New-files");
168         my $modfiles = $head->get("Modified-files");
169
170         my @changed_pages = grep {!/(^.*\/)?\.arch-ids\/.*\.id$/} split(/ /,
171                 "$newfiles $modfiles");
172
173         if ($message =~ /$tla_webcommit/) {
174                 $user=defined $2 ? "$2" : "$3";
175                 $message=$4;
176         }
177
178         require IkiWiki::UserInfo;
179         my @email_recipients=commit_notify_list($user, @changed_pages);
180         if (@email_recipients) {
181                 # TODO: if a commit spans multiple pages, this will send
182                 # subscribers a diff that might contain pages they did not
183                 # sign up for. Should separate the diff per page and
184                 # reassemble into one mail with just the pages subscribed to.
185                 my $logs = `tla logs -d $config{srcdir}`;
186                 my @changesets = reverse split(/\n/, $logs);
187                 my $i;
188
189                 for($i=0;$i<$#changesets;$i++) {
190                         last if $changesets[$i] eq $rev;
191                 }
192
193                 my $revminusone = $changesets[$i+1];
194                 my $diff=`tla diff -d $ENV{ARCH_TREE_ROOT} $revminusone`;
195
196                 my $subject="$config{wikiname} update of ";
197                 if (@changed_pages > 2) {
198                         $subject.="$changed_pages[0] $changed_pages[1] etc";
199                 }
200                 else {
201                         $subject.=join(" ", @changed_pages);
202                 }
203                 $subject.=" by $user";
204
205                 my $template=template("notifymail.tmpl");
206                 $template->param(
207                         wikiname => $config{wikiname},
208                         diff => $diff,
209                         user => $user,
210                         message => $message,
211                 );
212
213                 eval q{use Mail::Sendmail};
214                 foreach my $email (@email_recipients) {
215                         sendmail(
216                                 To => $email,
217                                 From => "$config{wikiname} <$config{adminemail}>",
218                                 Subject => $subject,
219                                 Message => $template->output,
220                         ) or error("Failed to send update notification mail");
221                 }
222         }
223 } #}}}
224
225 sub rcs_getctime ($) { #{{{
226         my $file=shift;
227         eval q{use Date::Parse};
228         eval q{use Mail::Header};
229
230         my $logs = `tla logs -d $config{srcdir}`;
231         my @changesets = reverse split(/\n/, $logs);
232         my $sdate;
233
234         for (my $i=0; $i<$#changesets; $i++) {
235                 my $change = $changesets[$i];
236
237                 open(LOG, "tla cat-log -d $config{srcdir} $change|");
238                 my $head = Mail::Header->new(\*LOG);
239                 close(LOG);
240
241                 $sdate = $head->get("Standard-date");
242                 my $newfiles = $head->get("New-files");
243
244                 my ($lastcreation) = grep {/^$file$/} split(/ /, "$newfiles");
245                 last if defined($lastcreation);
246         }
247
248         my $date=str2time($sdate, 'UTC');
249         debug("found ctime ".localtime($date)." for $file");
250         return $date;
251 } #}}}
252
253 1