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