add ModifyAction op

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

diff --git a/lib/DX/RuleSet.pm b/lib/DX/RuleSet.pm
index 09a2ff5..baa9586 100644 (file)
--- a/lib/DX/RuleSet.pm
+++ b/lib/DX/RuleSet.pm
@@ -11,13 +11,23 @@ use DX::Op::Backtrack;
 use DX::Op::Observe;
 use DX::Op::Not;
 use DX::Op::ProposeAction;
+use DX::Op::ModifyAction;
 use DX::Op::Materialize;
 use DX::Op::Prop;
 use DX::Op::Exists;
+use DX::Op::Predicate;
 use List::Util qw(reduce);
 
 has rules => (is => 'ro', default => sub { {} });
 
+sub add_predicate {
+  my ($self, $name, $vars, @cases) = @_;
+  my $full_name = join('/', $name, scalar @$vars);
+  push @{$self->rules->{$full_name}}, DX::Op::Predicate->new(
+    arg_names => $vars, arg_cases => \@cases
+  );
+}
+
 sub add_rule {
   my ($self, $name, $vars, @body) = @_;
   my $full_name = join('/', $name, scalar @$vars);
@@ -99,6 +109,14 @@ sub _expand_op_act {
   );
 }
 
+sub _expand_op_react {
+  my ($self, $vars, $builder) = @_;
+  DX::Op::ModifyAction->new(
+    vars => $vars,
+    builder => $builder,
+  );
+}
+
 sub _expand_op_materialize {
   my ($self, $var_name) = @_;
   DX::Op::Materialize->new(var_name => $var_name);