FS/bin/freeside-upgrade

   1 #!/usr/bin/perl -w
   2
   3 use strict;
   4 use vars qw($DEBUG $DRY_RUN);
   5 use Term::ReadKey;
   6 use DBIx::DBSchema 0.27;
   7 use FS::UID qw(adminsuidsetup checkeuid datasrc );  #getsecrets);
   8 use FS::Schema qw( dbdef dbdef_dist reload_dbdef );
   9
  10
  11 $DEBUG = 1;
  12 $DRY_RUN = 0;
  13
  14
  15 die "Not running uid freeside!" unless checkeuid();
  16
  17 my $user = shift or die &usage;
  18 my $dbh = adminsuidsetup($user);
  19
  20 #needs to match FS::Schema...
  21 my $dbdef_file = "/usr/local/etc/freeside/dbdef.". datasrc;
  22
  23 dbdef_create($dbh, $dbdef_file);
  24
  25 delete $FS::Schema::dbdef_cache{$dbdef_file}; #force an actual reload
  26 reload_dbdef($dbdef_file);
  27
  28
  29 foreach my $table ( dbdef_dist->tables ) {
  30
  31   if ( dbdef->table($table) ) {
  32
  33     warn "$table exists\n" if $DEBUG > 1;
  34
  35     foreach my $column ( dbdef_dist->table($table)->columns ) {
  36       if ( dbdef->table($table)->column($column) )  {
  37         warn "  $table.$column exists\n" if $DEBUG > 2;
  38       } else {
  39
  40         if ( $DEBUG ) {
  41           print STDERR "column $table.$column does not exist.  create?";
  42           next unless yesno();
  43         }
  44
  45         foreach my $statement (
  46           dbdef_dist->table($table)->column($column)->sql_add_column( $dbh )
  47         ) {
  48           warn "$statement\n" if $DEBUG || $DRY_RUN;
  49           unless ( $DRY_RUN ) {
  50             $dbh->do( $statement)
  51               or die "CREATE error: ". $dbh->errstr. "\nexecuting: $statement";
  52           }
  53         }
  54
  55       }
  56
  57     }
  58
  59     #should eventually check & create missing indices
  60
  61     #should eventually drop columns not in dbdef_dist...
  62
  63   } else {
  64
  65     if ( $DEBUG ) {
  66       print STDERR "table $table does not exist.  create?";
  67       next unless yesno();
  68     }
  69
  70     foreach my $statement (
  71       dbdef_dist->table($table)->sql_create_table( $dbh )
  72     ) {
  73       warn "$statement\n" if $DEBUG || $DRY_RUN;
  74       unless ( $DRY_RUN ) {
  75         $dbh->do( $statement)
  76           or die "CREATE error: ". $dbh->errstr. "\nexecuting: $statement";
  77       }
  78     }
  79
  80   }
  81
  82 }
  83
  84 # should eventually drop tables not in dbdef_dist too i guess...
  85
  86 $dbh->commit or die $dbh->errstr;
  87
  88 dbdef_create($dbh, $dbdef_file);
  89
  90 $dbh->disconnect or die $dbh->errstr;
  91
  92 ###
  93
  94 my $all = 0;
  95 sub yesno {
  96   print STDERR ' [yes/no/all] ';
  97   if ( $all ) {
  98     warn "yes\n";
  99     return 1;
 100   } else {
 101     while ( 1 ) {
 102       ReadMode 4;
 103       my $x = lc(ReadKey);
 104       ReadMode 0;
 105       if ( $x eq 'n' ) {
 106         warn "no\n";
 107         return 0;
 108       } elsif ( $x eq 'y' ) {
 109         warn "yes\n";
 110         return 1;
 111       } elsif ( $x eq 'a' ) {
 112         warn "yes\n";
 113         $all = 1;
 114         return 1;
 115       }
 116     }
 117   }
 118 }
 119
 120 sub dbdef_create { # reverse engineer the schema from the DB and save to file
 121   my( $dbh, $file ) = @_;
 122   my $dbdef = new_native DBIx::DBSchema $dbh;
 123   $dbdef->save($file);
 124 }
 125
 126 sub usage {
 127   die "Usage:\n  freeside-upgrade user\n";
 128 }
 129
 130 1;
 131