findall

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

diff --git a/lib/DX/RuleSet.pm b/lib/DX/RuleSet.pm
index 328eb85..1fbb924 100644 (file)
--- a/lib/DX/RuleSet.pm
+++ b/lib/DX/RuleSet.pm
@@ -1,15 +1,35 @@
 package DX::RuleSet;
 
 use Moo;
+use DX::Op::SetupScope;
 use DX::Op::CallRule;
 use DX::Op::MemberOf;
-use DX::Op::MemberLookup;
 use DX::Op::ApplyConstraint;
 use DX::Op::Return;
+use DX::Op::Cut;
+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 DX::Op::HasAction;
+use DX::Op::FindAll;
 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);
@@ -19,18 +39,27 @@ sub add_rule {
 
 sub _make_rule {
   my ($self, $vars, @body) = @_;
-  my $head = reduce { $b->but(next => $a) }
-               DX::Op::Return->new,
-               reverse map $self->expand(@$_), @body;
-  [ $vars, $head ];
+  my $head = $self->expand_and_link(DX::Op::Return->new, @body);
+  DX::Op::SetupScope->new(arg_names => $vars, next => $head);
+}
+
+sub expand_and_link {
+  my ($self, $last, @body) = @_;
+  return reduce { $b->but(next => $a) }
+           $last,
+           reverse map $self->expand($_), @body;
 }
 
 sub expand {
-  my ($self, $type, @rest) = @_;
-  if ($self->can(my $expand_meth = "_expand_${type}")) {
-    return $self->$expand_meth(@rest);
+  my ($self, $thing) = @_;
+  if (ref($thing) eq 'ARRAY') {
+    my ($type, @rest) = @$thing;
+    if ($self->can(my $expand_meth = "_expand_op_${type}")) {
+      return $self->$expand_meth(@rest);
+    }
+    return $self->_expand_call(@$thing);
   }
-  return $self->_expand_call($type, @rest);
+  return $thing;
 }
 
 sub _expand_call {
@@ -38,25 +67,36 @@ sub _expand_call {
   DX::Op::CallRule->new(rule_name => $name, rule_args => \@args);
 }
 
-sub _expand_member_of {
-  my ($self, $member_var, $coll_name) = @_;
-  DX::Op::MemberOf->new(
-    member_var => $member_var,
+sub _expand_op_cut { return DX::Op::Cut->new }
+
+sub _expand_op_fail { return DX::Op::Backtrack->new }
+
+sub _expand_op_not {
+  my ($self, @contents) = @_;
+  my $cut = DX::Op::Cut->new(next => DX::Op::Backtrack->new);
+  DX::Op::Not->new(
+    body => $self->expand_and_link($cut, @contents)
+  );
+}
+
+sub _expand_op_findall {
+  my ($self, $coll_name, $var_name, @contents) = @_;
+  my $findall = DX::Op::FindAll->new(
     coll_name => $coll_name,
+    var_name => $var_name,
+    body => $self->expand_and_link(DX::Op::Return->new, @contents),
   );
 }
 
-sub _expand_member_lookup {
-  my ($self, $member_var, $coll_name, $key_name, $key_var) = @_;
-  DX::Op::MemberLookup->new(
+sub _expand_op_member_of {
+  my ($self, $member_var, $coll_var) = @_;
+  DX::Op::MemberOf->new(
     member_var => $member_var,
-    coll_name => $coll_name,
-    key_name => $key_name,
-    key_var => $key_var
+    coll_var => $coll_var,
   );
 }
 
-sub _expand_constrain {
+sub _expand_op_constrain {
   my ($self, $vars, $constraint) = @_;
   DX::Op::ApplyConstraint->new(
     vars => $vars,
@@ -64,4 +104,52 @@ sub _expand_constrain {
   );
 }
 
+sub _expand_op_observe {
+  my ($self, $vars, $builder) = @_;
+  DX::Op::Observe->new(
+    vars => $vars,
+    builder => $builder,
+  );
+}
+
+sub _expand_op_act {
+  my ($self, $vars, $builder) = @_;
+  DX::Op::ProposeAction->new(
+    vars => $vars,
+    builder => $builder,
+  );
+}
+
+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);
+}
+
+sub _expand_op_prop {
+  my ($self, @args) = @_;
+  my %new; @new{qw(of name value)} = @args;
+  DX::Op::Prop->new(%new);
+}
+
+sub _expand_op_exists {
+  my ($self, $vars, @body) = @_;
+  DX::Op::Exists->new(
+    vars => $vars,
+    body => $self->expand_and_link(DX::Op::Return->new, @body)
+  );
+}
+
+sub _expand_op_has_action {
+  my ($self, @args) = @_;
+  DX::Op::HasAction->new(arg_spec => \@args);
+}
+
 1;