backup the schema for tables we don't need the data from. RT#85959
[freeside.git] / FS / FS / Cron / backup.pm
1 package FS::Cron::backup;
2
3 use strict;
4 use vars qw( @ISA @EXPORT_OK );
5 use Exporter;
6 use File::Copy;
7 use Date::Format;
8 use FS::UID qw(driver_name datasrc);
9 use FS::Log
10
11 @ISA = qw( Exporter );
12 @EXPORT_OK = qw( backup );
13
14 sub backup {
15   my $conf = new FS::Conf;
16   my $localdest = $conf->config('dump-localdest');
17   my $scpdest = $conf->config('dump-scpdest');
18   return unless $localdest || $scpdest;
19
20   my $filename = time2str('%Y%m%d%H%M%S',time);
21
22   datasrc =~ /dbname=([\w\.]+)$/
23     or backup_log_and_die($filename,"unparsable datasrc ". datasrc);
24   my $database = $1;
25
26   my $ext;
27   if ( driver_name eq 'Pg' ) {
28     system('pg_dump -Fc '. join(' ', map { "--exclude-table-data $_" }
29                                        qw( h_cdr h_queue h_queue_arg sessions )
30                                ).
31                            " $database >/var/tmp/$database.Pg"
32           );
33     $ext = 'Pg';
34   } elsif ( driver_name eq 'mysql' ) {
35     system("mysqldump $database >/var/tmp/$database.sql");
36     $ext = 'sql';
37   } else {
38     backup_log_and_die($filename,"database dumps not yet supported for ". driver_name);
39   }
40   chmod 0600, "/var/tmp/$database.$ext";
41
42   if ( $conf->config('dump-pgpid') ) {
43     eval 'use GnuPG;';
44     backup_log_and_die($filename,$@) if $@;
45     my $gpg = new GnuPG;
46     $gpg->encrypt( plaintext => "/var/tmp/$database.$ext",
47                    output    => "/var/tmp/$database.gpg",
48                    recipient => $conf->config('dump-pgpid'),
49                  );
50     unlink "/var/tmp/$database.$ext"
51       or backup_log_and_die($filename,$!);
52     chmod 0600, "/var/tmp/$database.gpg";
53     $ext = 'gpg';
54   }
55
56   if ( $localdest ) {
57     copy("/var/tmp/$database.$ext", "$localdest/$filename.$ext")
58       or backup_log_and_die($filename,$!);
59     chmod 0600, "$localdest/$filename.$ext";
60   }
61
62   if ( $scpdest ) {
63     eval "use Net::SCP qw(scp);";
64     backup_log_and_die($filename,$@) if $@;
65     my $scp = new Net::SCP;
66     $scp->scp("/var/tmp/$database.$ext", "$scpdest/$filename.$ext")
67       or backup_log_and_die($filename, $scp->{errstr});
68   }
69
70   unlink "/var/tmp/$database.$ext" or backup_log_and_die($filename,$!); #or just warn?
71
72   backup_log($filename);
73
74 }
75
76 #runs backup_log and dies with same error message
77 sub backup_log_and_die {
78   my ($filename,$error) = @_;
79   $error = "backup_log_and_die called without error message" unless $error;
80   backup_log($filename,$error);
81   die $error;
82 }
83
84 #logs result
85 sub backup_log {
86   my ($filename,$error) = @_;
87   my $result = $error ? "FAILED: $error" : 'succeeded';
88   my $message = "backup $filename $result\n";
89   my $log = FS::Log->new('Cron::backup');
90   if ($error) {
91     $log->error($message);
92   } else {
93     $log->info($message);
94   }
95   return;
96 }
97
98 1;