897c31731fdb6195e9efbbe3beaa5a1876ca31ac
[scpubgit/DKit.git] / 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 List::Util qw(reduce);
9
10 has rules => (is => 'ro', default => sub { {} });
11
12 sub add_rule {
13   my ($self, $name, $vars, @body) = @_;
14   my $full_name = join('/', $name, scalar @$vars);
15   push @{$self->rules->{$full_name}}, $self->_make_rule($vars, @body);
16   return $self;
17 }
18
19 sub _make_rule {
20   my ($self, $vars, @body) = @_;
21   my $head = reduce { $b->but(next => $a) }
22                DX::Op::Return->new,
23                reverse map $self->expand(@$_), @body;
24   [ $vars, $head ];
25 }
26
27 sub expand {
28   my ($self, $type, @rest) = @_;
29   if ($self->can(my $expand_meth = "_expand_${type}")) {
30     return $self->$expand_meth(@rest);
31   }
32   return $self->_expand_call($type, @rest);
33 }
34
35 sub _expand_call {
36   my ($self, $name, @args) = @_;
37   DX::Op::CallRule->new(rule_name => $name, rule_args => \@args);
38 }
39
40 sub _expand_member_of {
41   my ($self, $member_var, $coll_var) = @_;
42   DX::Op::MemberOf->new(
43     member_var => $member_var,
44     coll_var => $coll_var,
45   );
46 }
47
48 sub _expand_constrain {
49   my ($self, $vars, $constraint) = @_;
50   DX::Op::ApplyConstraint->new(
51     vars => $vars,
52     constraint => $constraint
53   );
54 }
55
56 1;