From 3454d95a84ba162d21cae76635d656e7e8c4c7a5 Mon Sep 17 00:00:00 2001
From: Oliver Gorwits <oliver@cpan.org>
Date: Wed, 20 Sep 2017 20:53:52 +0100
Subject: [PATCH] capture result on main phase as well

---
 lib/App/Netdisco/Worker/Runner.pm | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/lib/App/Netdisco/Worker/Runner.pm b/lib/App/Netdisco/Worker/Runner.pm
index 1b3ac3f4..8c7fe1ba 100644
--- a/lib/App/Netdisco/Worker/Runner.pm
+++ b/lib/App/Netdisco/Worker/Runner.pm
@@ -71,6 +71,7 @@ sub run_workers {
   my $hook = shift or return $self->jobstat->error('missing hook param');
   my $store = Dancer::Factory::Hook->instance();
   my $check = ($hook eq 'nd2_core_check');
+  my $main  = ($hook eq 'nd2_core_main');
 
   return unless scalar @{ $store->get_hooks_for($hook) };
   debug "running workers for hook: $hook";
@@ -80,7 +81,7 @@ sub run_workers {
       my $retval = $worker->($self->job);
       # could die or return undef or a scalar or Status or another class
       $self->jobstat($retval)
-        if $check and ref $retval eq 'App::Netdisco::Worker::Status';
+        if ($check or $main) and ref $retval eq 'App::Netdisco::Worker::Status';
     }
     catch { $self->jobstat->error($_) if $check };