/[BackupPC]/trunk/bin/BackupPC_burnArchiveCLI
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Contents of /trunk/bin/BackupPC_burnArchiveCLI

Parent Directory Parent Directory | Revision Log Revision Log


Revision 319 - (show annotations)
Mon Jan 30 14:58:46 2006 UTC (18 years, 3 months ago) by dpavlin
File size: 13294 byte(s)
finish all sths, spelling fix

1 #!/usr/bin/perl
2
3 use strict;
4 use lib "__INSTALLDIR__/lib";
5
6 use DBI;
7 use BackupPC::Lib;
8 use BackupPC::SearchLib;
9 use Time::HiRes qw/time/;
10 use POSIX qw/strftime/;
11 use Term::Menus;
12 use File::Which;
13 use File::Path;
14
15 use Data::Dumper;
16
17 my $debug = 0;
18 # set this to 1 to prompt for DVD removal for each selected item.
19 my $prompt_for_delete = shift @ARGV;
20 $|=1;
21
22 # don't check for user
23 my $bpc = BackupPC::Lib->new(undef, undef, 1) || die;
24 my %Conf = $bpc->Conf();
25 %BackupPC::SearchLib::Conf = %Conf;
26
27 my $conf_bin;
28
29 $conf_bin->{'cdrecord'} = $Conf{CDRecordBin} || die "Need CDRecordBin in config.pl\n";
30 my $cdr_opts = $Conf{CDRecordOpts} || die "Need CDRecordOpts in config.pl\n";
31 $conf_bin->{'eject'} = $Conf{ejectBin} || die "Need ejectBin in config.pl\n";
32 my $eject_opts = $Conf{ejectOpts} || die "Need ejectOpts in config.pl\n";
33 $conf_bin->{'mkisofs'} = $Conf{mkisofsBin} || die "Need mkisofsBin in config.pl\n";
34
35
36 my $bin;
37 foreach my $c (qw/cdrecord eject mkisofs/) {
38 $bin->{$c} = which($conf_bin->{$c}) || die "$0 needs $c ($conf_bin->{$c}), install it\n";
39 }
40
41 my $start_t = time();
42
43 my $t_fmt = '%Y-%m-%d %H:%M:%S';
44
45
46 my $dsn = $Conf{SearchDSN} || die "Need SearchDSN in config.pl\n";
47 my $user = $Conf{SearchUser} || '';
48
49 my $dbh = DBI->connect($dsn, $user, "", { RaiseError => 1, AutoCommit => 1 });
50
51 my $tar_dir = $Conf{InstallDir}.'/';
52 $tar_dir .= $Conf{GzipTempDir} || die "GzipTempDir isn't defined in configuration";
53
54 die "problem with $tar_dir, check GzipTempDir in configuration\n" unless (-d $tar_dir);
55
56 my $iso_dir = $Conf{InstallDir}.'/';
57 $iso_dir .= $Conf{ISOTempDir} || die "ISOTempDir isn't defined in configuration";
58 die "problem with $iso_dir, check ISOTempDir in configuration\n" unless (-d $iso_dir && -w $iso_dir);
59
60 #---- subs ----
61
62 sub fmt_time {
63 my $t = shift || return;
64 my $out = "";
65 my ($ss,$mm,$hh) = gmtime($t);
66 $out .= "${hh}h" if ($hh);
67 $out .= sprintf("%02d:%02d", $mm,$ss);
68 return $out;
69 }
70
71 sub curr_time {
72 return strftime($t_fmt,localtime());
73 }
74
75 sub dumpArchive2XML($$$)
76 {
77 my ($dbh, $dvd_nr, $filename) = @_;
78 my %archive;
79
80 my $output = new IO::File('> '.$filename.'.tmp');
81 my $writer = new XML::Writer(OUTPUT=>$output, NEWLINES => 1);
82
83 print "Dumping file list for DVD $dvd_nr, countdown:";
84
85 $writer->pi('xml-stylesheet', 'href="archive.css" type="text/css"');
86
87 my $files_sql = q{
88 SELECT
89 files.path AS path,
90 files.date AS date,
91 files.type AS type,
92 files.size AS size
93 FROM files, backups, shares
94 WHERE files.backupnum=backups.num AND
95 files.shareid=shares.id AND
96 shares.hostid=backups.hostid AND
97 backups.id=?
98 };
99
100 my $backups_sql = q{
101 SELECT
102 backups.id AS backup_id,
103 hosts.name AS host,
104 backups.num AS num,
105 backups.date AS date,
106 shares.name AS share,
107 backups.size AS backup_size,
108 backups.inc_size AS compress_size
109 FROM backups, archive_backup, hosts, shares
110 WHERE archive_backup.backup_id = backups.id
111 AND hosts.id=backups.hostid
112 AND shares.id=backups.shareid
113 AND archive_backup.archive_id = ?
114 ORDER BY
115 hosts.name, shares.name, backups.num
116 };
117
118 my $sth = $dbh->prepare("SELECT dvd_nr, total_size, note, username, date,id FROM archive WHERE dvd_nr=?");
119 $sth->execute($dvd_nr);
120 my $row = $sth->fetchrow_hashref();
121
122 my $row_h;
123 foreach my $hide (qw/id note/) {
124 $row_h->{$hide} = $row->{$hide} || '';
125 delete $row->{$hide};
126 }
127
128 $writer->startTag("archive", %{$row});
129
130 $writer->startTag("note");
131 $writer->characters( $row_h->{'note'} );
132 $writer->endTag("note");
133
134 my $sth_backups = $dbh->prepare( $backups_sql );
135 $sth_backups->execute( $row_h->{'id'} );
136
137 my $total_backups = $sth_backups->rows;
138 my $curr_backup = 0;
139 my $last_pcnt = 0;
140
141 while (my $row = $sth_backups->fetchrow_hashref()) {
142
143 $curr_backup++;
144 my $pcnt = int(($curr_backup * 10) / $total_backups);
145 if ($pcnt > $last_pcnt) {
146 print " ",(10 - $pcnt);
147 $last_pcnt = $pcnt;
148 }
149
150 my $sth_files = $dbh->prepare( $files_sql);
151 $sth_files->execute($row->{'backup_id'});
152
153 delete($row->{'backup_id'});
154 $row->{'date'} = strftime($t_fmt,gmtime($row->{'date'}));
155
156 $writer->startTag( "backup", %{$row} );
157
158 while (my $row = $sth_files->fetchrow_hashref()) {
159 # convert filetype to string
160 $row->{'type'} = BackupPC::Attrib::fileType2Text(undef, $row->{'type'});
161 $row->{'date'} = strftime($t_fmt,gmtime($row->{'date'}));
162 my $path = $row->{'path'}; delete $row->{'path'};
163
164 $writer->startTag("file", %{$row});
165 $writer->characters( $path );
166 $writer->endTag("file");
167 }
168 $writer->endTag("backup");
169 }
170
171 $writer->endTag("archive");
172 $writer->end();
173
174 rename $filename.'.tmp', $filename || die "can't rename $filename: $!";
175
176 print "\n";
177 }
178
179
180
181 #----- main
182
183 my $sth = $dbh->prepare( qq{
184 select
185 id,dvd_nr,total_size,note,username,
186 archive.date as date,
187 count(archive_burned.archive_id) as copies
188 from archive
189 left outer join archive_burned on archive.id=archive_burned.archive_id
190 group by id,dvd_nr,total_size,note,username,archive.date
191 order by date asc
192 } );
193
194 $sth->execute();
195
196 sub fmt_archive($) {
197 my $row = shift || die;
198
199 $row->{'date'} =~ s/\.\d+$//;
200 $row->{'copies'} =~ s/^\s*0+\s*$/no/;
201 $row->{'total_size'} /= (1024*1024);
202
203 my $copies = 'copies';
204 $copies = 'copy' if ($row->{'copies'} == 1);
205
206 return
207 sprintf("%d by %s on %s, %s %s [%.2f Mb]",
208 $row->{'dvd_nr'},
209 $row->{'username'},
210 $row->{'date'},
211 $row->{'copies'}, $copies,
212 $row->{'total_size'},
213 );
214 }
215
216 my @burned;
217 my @not_burned;
218
219 while (my $row = $sth->fetchrow_hashref) {
220 if ($row->{'copies'}) {
221 push @burned, fmt_archive($row);
222 } else {
223 push @not_burned, fmt_archive($row);
224 }
225 }
226
227 my %Menu_archive = (
228 Label => 'Menu_archive',
229
230 Item_1 => {
231 Text => 'DVD #]Convey[',
232 Convey => [ @not_burned ],
233 Default => '*',
234 },
235 Item_2 => {
236 Text => 'DVD #]Convey[',
237 Convey => [ @burned ],
238 },
239
240 Select => 'Many',
241
242 Banner => "Select one or more DVD media for burning",
243 );
244
245 my @archives_to_burn = Menu(\%Menu_archive);
246
247 print "\n";
248
249 sub skip($) {
250 my $msg = shift;
251 print "WARNING: $msg, skipping...\n";
252 goto SKIP;
253 }
254
255 sub add_symlink($$) {
256 my ($from, $to) = @_;
257 symlink $from, $to || skip("can't symlink $from to $to: $!");
258 }
259
260 sub delete_dvd($) {
261 my $dvd_nr = shift || return;
262
263 print "Do you want to delete DVD #$dvd_nr now? [NO/yes]: ";
264 my $ok = <STDIN>;
265 chomp($ok);
266 if (lc($ok) eq 'yes') {
267 print "Deleting DVD #$dvd_nr from database...\n";
268
269 $dbh->begin_work;
270
271 my $sth_delete_dvd = $dbh->prepare( qq{
272 delete from archive where dvd_nr = ?
273 } );
274 $sth_delete_dvd->execute( $dvd_nr );
275 $dbh->do( qq{
276 select setval('dvd_nr', (select max(dvd_nr) from archive), true)
277 } );
278
279 # remove files for this DVD
280 map {
281 print "\tremoving $_\n";
282 unlink($_) || die "can't rm $_: $!";
283 } glob ( "/$iso_dir/$dvd_nr.*" );
284
285 $dbh->commit;
286
287 return 1;
288 } else {
289 return 0;
290 }
291 }
292
293 my $sth_archive_backup_parts = $dbh->prepare( qq{
294 select
295 archive_backup.backup_id,
296 archive_id,
297 hosts.name as host,
298 shares.name as share,
299 backups.num as num,
300 backups.parts as parts,
301 backup_parts.part_nr as part_nr,
302 backup_parts.size as part_size,
303 backup_parts.md5 as md5,
304 backup_parts.items
305 from archive_backup
306 join archive on archive_id = archive.id
307 join backups on archive_backup.backup_id = backups.id
308 join hosts on hostid = hosts.id
309 join shares on shareid = shares.id
310 join backup_parts on archive_backup.backup_id = backup_parts.backup_id
311 where archive.dvd_nr = ?
312 order by archive_backup.backup_id, backup_parts.part_nr
313 });
314
315 my $sth_archive_backup_check = $dbh->prepare( qq{
316 SELECT
317 count(backups.id)
318 FROM backups
319 JOIN archive_backup on archive_backup.backup_id = backups.id
320 JOIN archive on archive_id = archive.id
321 WHERE dvd_nr = ?
322 });
323
324 my $sth_archive_burned = $dbh->prepare( qq{
325 insert into archive_burned
326 (archive_id, iso_size, part, copy)
327 values ( (select id from archive where dvd_nr =?), ?, ?, ?)
328 });
329
330 foreach my $arc (@archives_to_burn) {
331 exit if ($arc eq ']quit[');
332
333 my $dvd_nr = $1 if ($arc =~ m/DVD #(\d+)/);
334 die "BUG: can't find dvd_nr in $arc\n" unless ($dvd_nr);
335
336 my $tmp_dir = "/$iso_dir/$dvd_nr";
337
338 my $t = time();
339
340 print "Working on DVD #$dvd_nr in $tmp_dir\n";
341
342 $sth_archive_backup_parts->execute($dvd_nr);
343
344 $sth_archive_backup_check->execute($dvd_nr);
345
346 my ($parts_nr, $check_nr) = ($sth_archive_backup_parts->rows, $sth_archive_backup_check->fetchrow_array);
347
348 if ($parts_nr != $check_nr) {
349 warn "ERROR: DVD #$dvd_nr is still not inconsistent state. Some backup parts are ",
350 ($parts_nr < $check_nr) ? "missing ($parts_nr < $check_nr)" : "extra ($parts_nr > $check_nr)",
351 " you should re-create this DVD after BackupPC_incPartsUpdate finish\n";
352 delete_dvd( $dvd_nr );
353 next;
354 }
355
356 if ($sth_archive_backup_parts->rows == 0) {
357 warn "ERROR: no backup parts found for $dvd_nr. You should re-create that DVD.\n";
358 next if delete_dvd( $dvd_nr );
359 }
360
361 if ($prompt_for_delete) {
362 next if delete_dvd( $dvd_nr );
363 }
364
365 my @volumes;
366 my $v; # emtpy volume
367 my $v_size = 0;
368
369 my $max_archive_size = $Conf{MaxArchiveSize} || die "no MaxArchiveSize";
370 while (my $row = $sth_archive_backup_parts->fetchrow_hashref) {
371 if (($v->{size} || 0) + $row->{part_size} > $max_archive_size) {
372 push @volumes, $v;
373 $v = {};
374 }
375 $v->{size} += $row->{part_size};
376 # this part
377 my $p = {
378 filename => BackupPC::SearchLib::getGzipName($row->{'host'}, $row->{'share'}, $row->{'num'}),
379 };
380 foreach my $fld (qw/part_nr md5/) {
381 $p->{$fld} = $row->{$fld} || die "missing $fld in row!";
382 }
383 push @{ $v->{parts} }, $p;
384 }
385 push @volumes, $v if ($v);
386
387 #warn "# volumes: ",Dumper(\@volumes),"\n";
388
389 my $volumes = $#volumes + 1;
390 my $volume_nr = 1;
391
392 foreach my $v (@volumes) {
393
394 #print Dumper($v);
395
396 my $iso_size = 0;
397 my $disk_name = $dvd_nr;
398 # suffix added to multi-volume archives
399 my $volume_suffix = '';
400
401 if ($volumes > 1) {
402 $volume_suffix = '_' . $volume_nr;
403 $disk_name .= ' ' . $volume_nr . '/' . $volumes;
404 }
405
406 print "Processing DVD #$dvd_nr, volume $volume_nr/$volumes\n";
407
408 my $iso_file = my $xml_file = my $stage =
409 "${iso_dir}/${dvd_nr}";
410
411 $iso_file .= $volume_suffix . '.iso';
412 $xml_file .= '.xml';
413
414 $stage .= $volume_suffix . '.stage';
415
416 my $md5_file = "${stage}/${dvd_nr}${volume_suffix}.md5";
417
418 #
419 # check if ISO file exists
420 #
421
422 if (! -e $iso_file) {
423
424 # create stage directory
425 if (-e $stage) {
426 rmtree($stage) || die "can't remove $stage: $!";
427 }
428 mkpath($stage);
429
430 # open file for md5sums
431 open(my $md5, "> $md5_file") || skip "can't open $md5_file: $!";
432
433 my $parts_on_this_volume = 0;
434
435 foreach my $p (@{ $v->{parts} }) {
436 my $tar_file = $p->{filename} || die "no filename in part", Dumper($p);
437 my $rel_path = $tar_file;
438
439 if (-d "$tar_dir/$rel_path") {
440 mkpath("$stage/$rel_path") unless (-d "$stage/$rel_path");
441 $rel_path .= '/' . $p->{part_nr};
442 }
443 $rel_path .= '.tar.gz';
444
445 skip "can't find increment $rel_path: $!" unless (-r "$tar_dir/$rel_path");
446
447 add_symlink("$tar_dir/$rel_path", "$stage/$rel_path");
448
449 my $md5sum = $p->{md5} || die "no md5 in part ", Dumper($p);
450 chomp($md5sum);
451 print $md5 "$md5sum $rel_path\n" || die "can't write md5sum: $!";
452
453 $parts_on_this_volume++;
454 }
455
456 # add file list and note in xml
457 dumpArchive2XML($dbh, $dvd_nr, $xml_file) unless (-f $xml_file);
458
459 add_symlink($xml_file, "$stage/${dvd_nr}.xml");
460
461 # add css file for archive
462 my $css_file = $Conf{CgiImageDir} . '/archive.css';
463 if (-r $css_file) {
464 add_symlink($css_file, "$stage/archive.css");
465 } else {
466 print "WARNING: missing $css_file, not added to iso image!\n";
467 }
468
469 print "Running mkisofs now for $parts_on_this_volume increments, disk $disk_name\n";
470
471 my $cmd = $bin->{'mkisofs'} . qq{ -A BackupPC -gui -J -r -T --input-charset ISO-8859-2 -V "$disk_name" -o ${iso_file}.tmp -f $stage };
472
473 system($cmd) == 0 or skip "can't run $cmd: $?";
474
475 rename $iso_file.'.tmp', $iso_file || skip "can't rename $iso_file: $!";
476
477 $iso_size = (stat($iso_file))[7];
478
479 print "Created $iso_file [$iso_size bytes] in ", fmt_time(time() - $t), "\n";
480
481 rmtree($stage) || warn "can't remove stage directory $stage: $!";
482
483 } else {
484 print "ISO $iso_file allready exists\n";
485 }
486
487 my $copies = $Conf{BurnMultipleCopies} || 1;
488
489 foreach my $copy_nr ( 1 .. $copies ) {
490
491 print "\nREADY TO BURN MEDIA $disk_name copy $copy_nr\n\nPlease insert blank media and press ENTER\n\n";
492
493 system($bin->{'eject'}.' '.$eject_opts) == 0 or skip "can't run eject: $?";
494
495 my $wait = <STDIN>;
496
497 my $cmd = $bin->{'cdrecord'} . ' ' . $cdr_opts . ' ' . $iso_file;
498
499 # FIXME
500 print "## $cmd\n";
501 system($cmd) == 0 or skip "can't run $cmd: $?";
502
503 print "\n\nPLEASE REMOVE DVD MEDIA AND LABEL IT WITH $disk_name\n\n";
504
505 $sth_archive_burned->execute($dvd_nr, $iso_size, $volume_nr, $copy_nr);
506
507 print "Media burn for $disk_name copy $copy_nr recorded\n";
508 }
509
510 $volume_nr++;
511 }
512
513 SKIP:
514
515 }
516
517 #my $tar_file = BackupPC::SearchLib::getGzipName($row->{'host'}, $row->{'share'}, $row->{'num'});
518 #print curr_time, sprintf(" %s:%s %-3d ", $row->{'host'}, $row->{'share'}, $row->{'num'}), " -> $tar_file ";
519
520 print "Recoding finished, exiting...\n";
521
522 $sth->finish;
523 $sth_archive_backup_parts->finish;
524 $sth_archive_burned->finish;
525 $sth_archive_backup_check->finish;
526 $sth_archive_burned->finish;
527
528 $dbh->disconnect;
529

Properties

Name Value
svn:executable *

  ViewVC Help
Powered by ViewVC 1.1.26