lib/DX/RuleSet.pm

   1 package DX::RuleSet;
   2
   3 use Moo;
   4 use DX::Op::CallRule;
   5 use DX::Op::MemberOf;
   6 use DX::Op::ApplyConstraint;
   7 use DX::Op::Return;
   8 use DX::Op::Cut;
   9 use DX::Op::Backtrack;
  10 use DX::Op::Observe;
  11 use DX::Op::Not;
  12 use DX::Op::ProposeAction;
  13 use DX::Op::Materialize;
  14 use DX::Op::Prop;
  15 use List::Util qw(reduce);
  16
  17 has rules => (is => 'ro', default => sub { {} });
  18
  19 sub add_rule {
  20   my ($self, $name, $vars, @body) = @_;
  21   my $full_name = join('/', $name, scalar @$vars);
  22   push @{$self->rules->{$full_name}}, $self->_make_rule($vars, @body);
  23   return $self;
  24 }
  25
  26 sub _make_rule {
  27   my ($self, $vars, @body) = @_;
  28   my $head = $self->_expand_and_link(DX::Op::Return->new, @body);
  29   [ $vars, $head ];
  30 }
  31
  32 sub _expand_and_link {
  33   my ($self, $last, @body) = @_;
  34   return reduce { $b->but(next => $a) }
  35            $last,
  36            reverse map $self->expand(@$_), @body;
  37 }
  38
  39 sub expand {
  40   my ($self, $type, @rest) = @_;
  41   if ($self->can(my $expand_meth = "_expand_op_${type}")) {
  42     return $self->$expand_meth(@rest);
  43   }
  44   return $self->_expand_call($type, @rest);
  45 }
  46
  47 sub _expand_call {
  48   my ($self, $name, @args) = @_;
  49   DX::Op::CallRule->new(rule_name => $name, rule_args => \@args);
  50 }
  51
  52 sub _expand_op_cut { return DX::Op::Cut->new }
  53
  54 sub _expand_op_fail { return DX::Op::Backtrack->new }
  55
  56 sub _expand_op_not {
  57   my ($self, @contents) = @_;
  58   my $cut = DX::Op::Cut->new(next => DX::Op::Backtrack->new);
  59   DX::Op::Not->new(
  60     body => $self->_expand_and_link($cut, @contents)
  61   );
  62 }
  63
  64 sub _expand_op_member_of {
  65   my ($self, $member_var, $coll_var) = @_;
  66   DX::Op::MemberOf->new(
  67     member_var => $member_var,
  68     coll_var => $coll_var,
  69   );
  70 }
  71
  72 sub _expand_op_constrain {
  73   my ($self, $vars, $constraint) = @_;
  74   DX::Op::ApplyConstraint->new(
  75     vars => $vars,
  76     constraint => $constraint
  77   );
  78 }
  79
  80 sub _expand_op_observe {
  81   my ($self, $vars, $builder) = @_;
  82   DX::Op::Observe->new(
  83     vars => $vars,
  84     builder => $builder,
  85   );
  86 }
  87
  88 sub _expand_op_act {
  89   my ($self, $vars, $builder) = @_;
  90   DX::Op::ProposeAction->new(
  91     vars => $vars,
  92     builder => $builder,
  93   );
  94 }
  95
  96 sub _expand_op_materialize {
  97   my ($self, $var_name) = @_;
  98   DX::Op::Materialize->new(var_name => $var_name);
  99 }
 100
 101 sub _expand_op_prop {
 102   my ($self, @args) = @_;
 103   my %new; @new{qw(of name value)} = @args;
 104   DX::Op::Prop->new(%new);
 105 }
 106
 107 1;