Yay, versioning!
Jess Robinson [Sun, 7 May 2006 21:13:17 +0000 (21:13 +0000)]
lib/DBIx/Class/Versioning.pm [new file with mode: 0644]

diff --git a/lib/DBIx/Class/Versioning.pm b/lib/DBIx/Class/Versioning.pm
new file mode 100644 (file)
index 0000000..864f06f
--- /dev/null
@@ -0,0 +1,175 @@
+package DBIx::Class::Version::Table;
+use base 'DBIx::Class';
+use strict;
+use warnings;
+
+__PACKAGE__->load_components(qw/ Core/);
+__PACKAGE__->table('SchemaVersions');
+
+__PACKAGE__->add_columns
+    ( 'Version' => {
+        'data_type' => 'VARCHAR',
+        'is_auto_increment' => 0,
+        'default_value' => undef,
+        'is_foreign_key' => 0,
+        'name' => 'Version',
+        'is_nullable' => 0,
+        'size' => '10'
+        },
+      'Installed' => {
+          'data_type' => 'VARCHAR',
+          'is_auto_increment' => 0,
+          'default_value' => undef,
+          'is_foreign_key' => 0,
+          'name' => 'Installed',
+          'is_nullable' => 0,
+          'size' => '20'
+          },
+      );
+__PACKAGE__->set_primary_key('Version');
+
+package DBIx::Class::Version;
+use base 'DBIx::Class::Schema';
+use strict;
+use warnings;
+
+__PACKAGE__->register_class('Table', 'DBIx::Class::Version::Table');
+
+
+# ---------------------------------------------------------------------------
+package DBIx::Class::Versioning;
+
+use strict;
+use warnings;
+use base 'DBIx::Class';
+use POSIX 'strftime';
+use Data::Dumper;
+# use DBIx::Class::Version;
+
+__PACKAGE__->mk_classdata('_filedata');
+__PACKAGE__->mk_classdata('upgrade_directory');
+
+sub on_connect
+{
+    my ($self) = @_;
+    print "on_connect\n";
+    my $vschema = DBIx::Class::Version->connect(@{$self->storage->connect_info()});
+    my $vtable = $vschema->resultset('Table');
+    my $pversion;
+    if(!$self->exists($vtable))
+    {
+        print "deploying.. \n";
+        $vschema->storage->debug(1);
+        print "Debugging is: ", $vschema->storage->debug, "\n";
+        $vschema->deploy();
+        $pversion = 0;
+    }
+    else
+    {
+        $pversion = $vtable->search(undef, 
+                                    { select => [
+                                             'Version',
+                                             { 'max' => 'Installed' },
+                                             ],
+                                      group_by => [ 'Version' ],
+                                      })->first;
+        $pversion = $pversion->Version if($pversion);
+    }
+    if($pversion eq $self->VERSION)
+    {
+        print "This version is already installed\n";
+        return 1;
+    }
+
+    
+    $vtable->create({ Version => $self->VERSION,
+                      Installed => strftime("%Y-%m-%d %H:%M:%S", gmtime())
+                      });
+
+    if(!$pversion)
+    {
+        print "No previous version found, skipping upgrade\n";
+        return 1;
+    }
+
+    my $file = $self->ddl_filename($self->upgrade_directory,
+                                   $self->storage->sqlt_type,
+                                   $self->VERSION
+                                   );
+    $file =~ s/@{[ $self->VERSION ]}/"${pversion}-" . $self->VERSION/e;
+    if(!-f $file)
+    {
+        warn "Upgrade not possible, no upgrade file found ($file)\n";
+        return;
+    }
+    print "Found Upgrade file: $file\n";
+    my $fh;
+    open $fh, "<$file" or warn("Can't open upgrade file, $file ($!)");
+    my @data = split(/;\n/, join('', <$fh>));
+    close($fh);
+    @data = grep { $_ && $_ !~ /^-- / } @data;
+    @data = grep { $_ !~ /^(BEGIN TRANACTION|COMMIT)/m } @data;
+    print "Commands: ", join("\n", @data), "\n";
+    $self->_filedata(\@data);
+
+    $self->backup();
+    $self->upgrade();
+
+# X Create version table if not exists?
+# Make backup
+# Run create statements
+# Run post-create callback
+# Run alter/drop statement
+# Run post-alter callback
+}
+
+sub exists
+{
+    my ($self, $rs) = @_;
+
+    eval {
+        $rs->search({ 1, 0 })->count;
+    };
+
+    return 0 if $@;
+
+    return 1;
+}
+
+sub backup
+{
+    my ($self) = @_;
+}
+
+sub upgrade
+{
+    my ($self) = @_;
+
+    ## overridable sub, per default just run all the commands.
+
+    $self->run_upgrade(qr/create/i);
+    $self->run_upgrade(qr/alter table .*? add/i);
+    $self->run_upgrade(qr/alter table .*? (?!drop)/i);
+    $self->run_upgrade(qr/alter table .*? drop/i);
+    $self->run_upgrade(qr/drop/i);
+    $self->run_upgrade(qr//i);
+}
+
+
+sub run_upgrade
+{
+    my ($self, $stm) = @_;
+    print "Reg: $stm\n";
+    my @statements = grep { $_ =~ $stm } @{$self->_filedata};
+#    print "Statements: ", join("\n", @statements), "\n";
+    $self->_filedata([ grep { $_ !~ /$stm/i } @{$self->_filedata} ]);
+
+    for (@statements)
+    {
+        $self->storage->debugfh->print("$_\n") if $self->storage->debug;
+        print "Running \n>>$_<<\n";
+        $self->storage->dbh->do($_) or warn "SQL was:\n $_";
+    }
+
+    return 1;
+}