From 00afd5c6fce4e8c9ddfd350ee434dcbd1c3725bd Mon Sep 17 00:00:00 2001 From: David Heinemeier Hansson Date: Wed, 1 Feb 2023 13:30:04 +0100 Subject: [PATCH] Yield accessory --- lib/mrsk/cli/accessory.rb | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/lib/mrsk/cli/accessory.rb b/lib/mrsk/cli/accessory.rb index aa993c3e..84106177 100644 --- a/lib/mrsk/cli/accessory.rb +++ b/lib/mrsk/cli/accessory.rb @@ -93,13 +93,15 @@ class Mrsk::Cli::Accessory < Mrsk::Cli::Base else raise "Unknown method: #{options[:method]}" end.inquiry - if runner.exec_over_ssh? - run_locally do - info "Launching command on #{accessory.host}" - exec accessory.exec_over_ssh(cmd, host: accessory.host) + with_accessory(name) do |accessory| + if runner.exec_over_ssh? + run_locally do + info "Launching command on #{accessory.host}" + exec accessory.exec_over_ssh(cmd, host: accessory.host) + end + else + on(accessory.host) { puts capture_with_info(*accessory.send(runner, cmd)) } end - else - on(accessory.host) { puts capture_with_info(*accessory.send(runner, cmd)) } end end