use FactRef objects to sanify action handling

[scpubgit/DKit.git] / lib / DX / SetOver.pm

diff --git a/lib/DX/SetOver.pm b/lib/DX/SetOver.pm
index 88a964d..c72accc 100644 (file)
--- a/lib/DX/SetOver.pm
+++ b/lib/DX/SetOver.pm
@@ -7,6 +7,13 @@ has over => (is => 'ro', required => 1);
 
 has values => (is => 'ro', default => sub { {} });
 
+sub key_list { sort keys %{$_[0]->values} }
+
+sub get {
+  my ($self, $key) = @_;
+  return $self->values->{$key};
+}
+
 sub to_stream {
   my ($self) = @_;
   my $values = $self->values;
@@ -15,10 +22,20 @@ sub to_stream {
 
 sub with_value {
   my ($self, $value) = @_;
-  return ref($self)->new(
-    %$self,
-    values => { %{$self->values}, $value->${\$self->over} => $value }
-  );
+  return ref($self)->new(%$self, values => { %{$self->values} })
+                   ->set_value($value);
+}
+
+sub set_value {
+  my ($self, $value) = @_;
+  $self->values->{$value->${\$self->over}} = $value;
+  return $self;
+}
+
+sub remove_value {
+  my ($self, $value) = @_;
+  delete $self->values->{$value->${\$self->over}};
+  return $self;
 }
 
 1;