Merge branch 'master' of git.freeside.biz:/home/git/freeside
[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::Misc qw( send_email );
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_email_and_die($conf,$filename,"unparsable datasrc ". datasrc);
24   my $database = $1;
25
26   my $ext;
27   if ( driver_name eq 'Pg' ) {
28     system("pg_dump -Fc $database >/var/tmp/$database.Pg");
29     $ext = 'Pg';
30   } elsif ( driver_name eq 'mysql' ) {
31     system("mysqldump $database >/var/tmp/$database.sql");
32     $ext = 'sql';
33   } else {
34     backup_email_and_die($conf,$filename,"database dumps not yet supported for ". driver_name);
35   }
36   chmod 0600, "/var/tmp/$database.$ext";
37
38   if ( $conf->config('dump-pgpid') ) {
39     eval 'use GnuPG;';
40     backup_email_and_die($conf,$filename,$@) if $@;
41     my $gpg = new GnuPG;
42     $gpg->encrypt( plaintext => "/var/tmp/$database.$ext",
43                    output    => "/var/tmp/$database.gpg",
44                    recipient => $conf->config('dump-pgpid'),
45                  );
46     unlink "/var/tmp/$database.$ext"
47       or backup_email_and_die($conf,$filename,$!);
48     chmod 0600, "/var/tmp/$database.gpg";
49     $ext = 'gpg';
50   }
51
52   if ( $localdest ) {
53     copy("/var/tmp/$database.$ext", "$localdest/$filename.$ext")
54       or backup_email_and_die($conf,$filename,$!);
55     chmod 0600, "$localdest/$filename.$ext";
56   }
57
58   if ( $scpdest ) {
59     eval "use Net::SCP qw(scp);";
60     backup_email_and_die($conf,$filename,$@) if $@;
61     scp("/var/tmp/$database.$ext", "$scpdest/$filename.$ext");
62   }
63
64   unlink "/var/tmp/$database.$ext" or backup_email_and_die($conf,$filename,$!); #or just warn?
65
66   backup_email($conf,$filename);
67
68 }
69
70 #runs backup_email and dies with same error message
71 sub backup_email_and_die {
72   my ($conf,$filename,$error) = @_;
73   backup_email($conf,$filename,$error);
74   warn "backup_email_and_die called without error message" unless $error;
75   die $error;
76 }
77
78 #checks if email should be sent, sends it
79 sub backup_email {
80   my ($conf,$filename,$error) = @_;
81   my $to = $conf->config('dump-email_to');
82   return unless $to;
83   my $result = $error ? 'FAILED' : 'succeeded';
84   my $email_error = send_email(
85     'from'    => $conf->config('invoice_from'), #or whatever, don't think it matters
86     'to'      => $to,
87     'subject' => 'FREESIDE NOTIFICATION: Backup ' . $result,
88     'body'    => [ 
89       "This is an automatic message from your Freeside installation.\n",
90       "Freeside backup $filename $result",
91       ($error ? " with the following error:\n\n" : "\n"),
92       ($error || ''),      
93       "\n",
94     ],
95     'msgtype' => 'admin',
96   );
97   warn $email_error if $email_error;
98   return;
99 }
100
101 1;