switch recheck to using an on_completion_step

[scpubgit/DX.git] / lib / DX / SearchState.pm

diff --git a/lib/DX/SearchState.pm b/lib/DX/SearchState.pm
index 1548f11..553b304 100644 (file)
--- a/lib/DX/SearchState.pm
+++ b/lib/DX/SearchState.pm
@@ -2,12 +2,13 @@ package DX::SearchState;
 
 use Types::Standard qw(Maybe);
 use DX::Step::Backtrack;
-use DX::Step::InvokeNextPredicate;
+use DX::Step::ConsiderProposition;
+use DX::Step::MarkAsSolution;
 use DX::Class;
 
 has current_hypothesis => (is => 'ro', isa => Hypothesis, required => 1);
 
-has next_step => (is => 'ro', isa => Maybe[Step]);
+has next_step => (is => 'ro', isa => Maybe[Step], required => 1);
 
 has propositions => (is => 'ro', isa => PropositionSequence, required => 1);
 
@@ -15,6 +16,10 @@ has alternatives => (is => 'ro', isa => AlternativeList, required => 1);
 
 has is_solution_state => (is => 'ro', required => 1);
 
+has on_exhaustion_step => (is => 'ro', isa => Maybe[Step], required => 1);
+
+has on_solution_step => (is => 'ro', isa => Maybe[Step], required => 1);
+
 sub next_proposition {
   my ($self, $hyp) = @_;
   $hyp ||= $self->current_hypothesis;
@@ -31,13 +36,15 @@ sub new_for {
     propositions => $props,
     (@{$props->members}
       ? (
-          next_step => DX::Step::InvokeNextPredicate->new(
+          next_step => DX::Step::ConsiderProposition->new(
             proposition => $props->members->[0],
           ),
           is_solution_state => 0,
         )
-      : ( is_solution_state => 1 )
+      : ( next_step => undef, is_solution_state => 1 )
     ),
+    on_exhaustion_step => undef,
+    on_solution_step => DX::Step::MarkAsSolution->new,
   );
 }