multi column PK support

[dbsrgits/DBIx-Class-Journal.git] / lib / DBIx / Class / Schema / Journal / DB.pm

diff --git a/lib/DBIx/Class/Schema/Journal/DB.pm b/lib/DBIx/Class/Schema/Journal/DB.pm
index 89e7387..d41ff5a 100644 (file)
--- a/lib/DBIx/Class/Schema/Journal/DB.pm
+++ b/lib/DBIx/Class/Schema/Journal/DB.pm
@@ -2,6 +2,58 @@ package DBIx::Class::Schema::Journal::DB;
 
 use base 'DBIx::Class::Schema';
 
-DBIx::Class::Schema::Journal::DB->load_classes();
+__PACKAGE__->mk_classdata('nested_changesets');
+__PACKAGE__->mk_group_accessors( simple => 'current_user' );
+__PACKAGE__->mk_group_accessors( simple => 'current_session' );
+__PACKAGE__->mk_group_accessors( simple => '_current_changeset_container' );
+
+DBIx::Class::Schema::Journal::DB->load_classes(qw/
+                                               ChangeSet
+                                               ChangeLog
+                                               AuditHistory
+                                               /);
+
+require DBIx::Class::Schema::Journal::DB::AuditLog;
+
+sub _current_changeset {
+    my $self = shift;
+    my $ref = $self->_current_changeset_container;
+    $ref && $ref->{changeset};
+}
+
+# this is for localization of the current changeset
+sub current_changeset {
+    my ( $self, @args ) = @_;
+
+    $self->throw_exception("setting current_changeset is not supported, use txn_do to create a new changeset") if @args;
+
+    my $id = $self->_current_changeset;
+
+    $self->throw_exception("Can't call current_changeset outside of a transaction") unless $id;
+
+    return $id;
+}
+
+sub journal_create_changeset {
+    my ( $self, @args ) = @_;
+
+    my %changesetdata = ( @args, ID => undef );
+
+    delete $changesetdata{parent_id} unless $self->nested_changesets;
+
+    if( defined( my $user = $self->current_user() ) )
+    {
+        $changesetdata{user_id} = $user;
+    }
+    if( defined( my $session = $self->current_session() ) )
+    {
+        $changesetdata{session_id} = $session;
+    }
+
+    ## Create a new changeset, then run $code as a transaction
+    my $cs = $self->resultset('ChangeSet');
+
+    $cs->create({ %changesetdata });
+}
 
 1;