root/trunk/midgard/tools/midrepository/database-upgrade.in

#!/usr/bin/perl

use DBI;
use Getopt::Std;

getopts('td:u:p:U:P:l');

$database = $opt_d ? $opt_d : "@STAGING_DB@";
$user = $opt_u ? $opt_u : "@DB_USER@";
$password = $opt_p ? $opt_p : "@DB_USER_PASSWORD@";
$root = $opt_U ? $opt_U : '@DB_ADMIN_USER@';
$rootpw = $opt_P ? $opt_P : '@DB_ADMIN_PASSWORD@';

######

$driver=0;
foreach (DBI->available_drivers) {
   ($_ eq 'mysql') && ($driver = 1) && last;
}
if (!$driver) { fail("The MySQL DBD driver is not installed"); }

$dsn = "DBI:mysql:database=$database";
($dbh = DBI->connect($dsn, $user, $password)) || fail("Failed to connect as $user");
($rdbh = DBI->connect($dsn, $root, $rootpw)) || fail("Failed to connect as $root");


############ UPGRADE ##############################

if (!tableexists('article_i')) {
  `mysql $database < i18n.sql`;
  `repligard -m -c convertdb.conf`;
}

if (!columnexists('repligard', 'locked')) {
  `mysql $database < vc.sql`;
}

exit;



################ UTIL ################################

sub tableexists
{
my $table = shift;

   foreach ($dbh->tables) {
      if ($_ eq $table) {
         return 1;
      }
   }

   return 0;
}

sub columnexists
{
my $table = shift;
my $column = shift;
my $colname;

my $sth = $rdbh->prepare("describe $table");
   ($sth && $sth->execute()) || return 0;

   while (($colname) = $sth->fetchrow_array) {
      if ($colname eq $column) { return 1; }
   }

   return 0;
}

sub fail
{
   print STDERR $_[0], "\n";
   exit($_[1] ? $_[1] : 1);
}