Commit | Line | Data |
51bec050 |
1 | package DBIx::Class::FilterColumn; |
51bec050 |
2 | use strict; |
3 | use warnings; |
4 | |
5 | use base qw/DBIx::Class::Row/; |
6 | |
7 | sub filter_column { |
8 | my ($self, $col, $attrs) = @_; |
9 | |
10 | $self->throw_exception("No such column $col to filter") |
11 | unless $self->has_column($col); |
12 | |
13 | $self->throw_exception("filter_column needs attr hashref") |
14 | unless ref $attrs eq 'HASH'; |
15 | |
16 | $self->column_info($col)->{_filter_info} = $attrs; |
17 | my $acc = $self->column_info($col)->{accessor}; |
d7d38bef |
18 | $self->mk_group_accessors(filtered_column => [ (defined $acc ? $acc : $col), $col]); |
51bec050 |
19 | return 1; |
20 | } |
21 | |
4c0c3038 |
22 | sub _column_from_storage { |
9b2c0de6 |
23 | my ($self, $col, $value) = @_; |
51bec050 |
24 | |
25 | return $value unless defined $value; |
26 | |
27 | my $info = $self->column_info($col) |
28 | or $self->throw_exception("No column info for $col"); |
29 | |
30 | return $value unless exists $info->{_filter_info}; |
31 | |
d7d38bef |
32 | my $filter = $info->{_filter_info}{filter_from_storage}; |
51bec050 |
33 | $self->throw_exception("No inflator for $col") unless defined $filter; |
34 | |
9b2c0de6 |
35 | return $self->$filter($value); |
51bec050 |
36 | } |
37 | |
4c0c3038 |
38 | sub _column_to_storage { |
9b2c0de6 |
39 | my ($self, $col, $value) = @_; |
51bec050 |
40 | |
41 | my $info = $self->column_info($col) or |
42 | $self->throw_exception("No column info for $col"); |
43 | |
44 | return $value unless exists $info->{_filter_info}; |
45 | |
d7d38bef |
46 | my $unfilter = $info->{_filter_info}{filter_to_storage}; |
51bec050 |
47 | $self->throw_exception("No unfilter for $col") unless defined $unfilter; |
9b2c0de6 |
48 | return $self->$unfilter($value); |
51bec050 |
49 | } |
50 | |
d7d38bef |
51 | sub get_filtered_column { |
51bec050 |
52 | my ($self, $col) = @_; |
53 | |
54 | $self->throw_exception("$col is not a filtered column") |
55 | unless exists $self->column_info($col)->{_filter_info}; |
56 | |
57 | return $self->{_filtered_column}{$col} |
58 | if exists $self->{_filtered_column}{$col}; |
59 | |
60 | my $val = $self->get_column($col); |
61 | |
9b2c0de6 |
62 | return $self->{_filtered_column}{$col} = $self->_column_from_storage($col, $val); |
51bec050 |
63 | } |
64 | |
85439e0c |
65 | sub set_column { |
66 | my ($self, $col) = (shift, @_); |
67 | |
68 | # blow cache |
69 | delete $self->{_filtered_column}{$col}; |
70 | |
71 | $self->next::method(@_); |
72 | } |
73 | |
d7d38bef |
74 | sub set_filtered_column { |
51bec050 |
75 | my ($self, $col, $filtered) = @_; |
76 | |
9b2c0de6 |
77 | $self->set_column($col, $self->_column_to_storage($col, $filtered)); |
51bec050 |
78 | |
79 | delete $self->{_filtered_column}{$col}; |
80 | |
81 | return $filtered; |
82 | } |
83 | |
7b461f8a |
84 | sub update { |
85 | my ($self, $attrs, @rest) = @_; |
86 | foreach my $key (keys %{$attrs||{}}) { |
87 | if ($self->has_column($key) && |
88 | exists $self->column_info($key)->{_filter_info}) { |
89 | my $val = delete $attrs->{$key}; |
d7d38bef |
90 | $self->set_filtered_column($key, $val); |
9b2c0de6 |
91 | $attrs->{$key} = $self->_column_to_storage($key, $val) |
7b461f8a |
92 | } |
93 | } |
94 | return $self->next::method($attrs, @rest); |
95 | } |
96 | |
7b461f8a |
97 | sub new { |
98 | my ($class, $attrs, @rest) = @_; |
9b2c0de6 |
99 | my $source = $attrs->{-result_source} |
d9ea6d6d |
100 | or $class->throw_exception('Sourceless rows are not supported with DBIx::Class::FilterColumn'); |
101 | |
9b2c0de6 |
102 | my $obj = $class->next::method($attrs, @rest); |
7b461f8a |
103 | foreach my $key (keys %{$attrs||{}}) { |
9b2c0de6 |
104 | if ($obj->has_column($key) && |
105 | exists $obj->column_info($key)->{_filter_info} ) { |
106 | my $val = delete $attrs->{$key}; |
107 | $obj->set_filtered_column($key, $val); |
7b461f8a |
108 | } |
109 | } |
7b461f8a |
110 | return $obj; |
111 | } |
112 | |
51bec050 |
113 | 1; |
22d9e05a |
114 | |
9b2c0de6 |
115 | =head1 NAME |
22d9e05a |
116 | |
9b2c0de6 |
117 | DBIx::Class::FilterColumn - Automatically convert column data |
118 | |
119 | =head1 SYNOPSIS |
120 | |
121 | # In your result classes |
122 | __PACKAGE__->filter_column( money => { |
123 | filter_to_storage => 'to_pennies', |
124 | filter_from_storage => 'from_pennies', |
125 | }); |
22d9e05a |
126 | |
127 | sub to_pennies { $_[1] * 100 } |
9b2c0de6 |
128 | |
22d9e05a |
129 | sub from_pennies { $_[1] / 100 } |
130 | |
131 | 1; |
132 | |
9b2c0de6 |
133 | =head1 DESCRIPTION |
22d9e05a |
134 | |
9b2c0de6 |
135 | This component is meant to be a more powerful, but less DWIM-y, |
136 | L<DBIx::Class::InflateColumn>. One of the major issues with said component is |
137 | that it B<only> works with references. Generally speaking anything that can |
138 | be done with L<DBIx::Class::InflateColumn> can be done with this component. |
22d9e05a |
139 | |
9b2c0de6 |
140 | =head1 METHODS |
22d9e05a |
141 | |
9b2c0de6 |
142 | =head2 filter_column |
22d9e05a |
143 | |
9b2c0de6 |
144 | __PACKAGE__->filter_column( colname => { |
145 | filter_from_storage => 'method', |
146 | filter_to_storage => 'method', |
147 | }) |
22d9e05a |
148 | |
9b2c0de6 |
149 | This is the method that you need to call to set up a filtered column. It takes |
150 | exactly two arguments; the first being the column name the second being a |
151 | C<HashRef> with C<filter_from_storage> and C<filter_to_storage> having |
152 | something that can be called as a method. The method will be called with |
153 | the value of the column as the first non-C<$self> argument. |
22d9e05a |
154 | |
9b2c0de6 |
155 | =head2 get_filtered_column |
22d9e05a |
156 | |
9b2c0de6 |
157 | $obj->get_filtered_column('colname') |
158 | |
159 | Returns the filtered value of the column |
160 | |
161 | =head2 set_filtered_column |
162 | |
163 | $obj->set_filtered_column(colname => 'new_value') |
164 | |
165 | Sets the filtered value of the column |
166 | |
167 | =head2 update |
168 | |
169 | Just overridden to filter changed columns through this component |
170 | |
171 | =head2 new |
22d9e05a |
172 | |
9b2c0de6 |
173 | Just overridden to filter columns through this component |