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