basic code cleanup
[dbsrgits/DBIx-Class-Journal.git] / lib / DBIx / Class / Schema / Journal / DB.pm
1 package DBIx::Class::Schema::Journal::DB;
2
3 use base 'DBIx::Class::Schema';
4
5 __PACKAGE__->mk_classdata('nested_changesets');
6 __PACKAGE__->mk_group_accessors( simple => 'current_user' );
7 __PACKAGE__->mk_group_accessors( simple => 'current_session' );
8 __PACKAGE__->mk_group_accessors( simple => '_current_changeset_container' );
9
10 DBIx::Class::Schema::Journal::DB->load_classes(qw(ChangeSet ChangeLog));
11
12 require DBIx::Class::Schema::Journal::DB::AuditLog;
13 require DBIx::Class::Schema::Journal::DB::AuditHistory;
14
15 sub _current_changeset {
16     my $self = shift;
17     my $ref = $self->_current_changeset_container;
18     $ref && $ref->{changeset};
19 }
20
21 # this is for localization of the current changeset
22 sub current_changeset {
23     my ( $self, @args ) = @_;
24
25     $self->throw_exception(
26        'setting current_changeset is not supported, use txn_do to create a new changeset'
27     ) if @args;
28
29     my $id = $self->_current_changeset;
30
31     $self->throw_exception(
32        q{Can't call current_changeset outside of a transaction}
33     ) unless $id;
34
35     return $id;
36 }
37
38 sub journal_create_changeset {
39     my ( $self, @args ) = @_;
40
41     my %changesetdata = ( @args, ID => undef );
42
43     delete $changesetdata{parent_id} unless $self->nested_changesets;
44
45     if( defined( my $user = $self->current_user() ) ) {
46         $changesetdata{user_id} = $user;
47     }
48     if( defined( my $session = $self->current_session() ) ) {
49         $changesetdata{session_id} = $session;
50     }
51
52     ## Create a new changeset, then run $code as a transaction
53     my $cs = $self->resultset('ChangeSet');
54
55     $cs->create({ %changesetdata });
56 }
57
58 sub journal_create_change {
59     my $self = shift;
60     $self->resultset('ChangeLog')->create({
61        changeset_id => $self->current_changeset
62     });
63 }
64
65 sub journal_update_or_create_log_entry {
66     my ($self, $row, @cols) = @_;
67
68     my $s_name = $row->result_source->source_name;
69
70     my %id = map { $_ => $row->get_column($_)} $row->primary_columns;
71
72     $self->resultset("${s_name}AuditLog")->update_or_create({ @cols, %id });
73 }
74
75 sub journal_record_in_history {
76     my ($self, $row, @cols) = @_;
77
78     my $s_name = $row->result_source->source_name;
79
80     $self->resultset("${s_name}AuditHistory")->create({ $row->get_columns, @cols });
81 }
82
83
84 1;