aa7da16986
This adds an assertion that the body begins with "! " for 400 responses, and then updates the few places that didn't already do that.
30 lines
982 B
Ruby
30 lines
982 B
Ruby
# frozen_string_literal: true
|
|
|
|
UbiCli.on("vm").run_on("scp") do
|
|
skip_option_parsing("ubi vm location/(vm-name|_vm-ubid) [options] scp [scp-options] (local-path :remote-path | :remote-path local-path)")
|
|
|
|
args(2..., invalid_args_message: "must provide 2 paths: either 'local-path :remote-path' or ':remote-path local-path'")
|
|
|
|
run do |(*argv, path1, path2), opts|
|
|
remote_path1 = path1[0] == ":"
|
|
remote_path2 = path2[0] == ":"
|
|
|
|
if remote_path1 ^ remote_path2
|
|
handle_ssh(opts) do |user:, address:|
|
|
address = "[#{address}]" if address.include?(":")
|
|
remote = "#{user}@#{address}"
|
|
|
|
if remote_path1
|
|
path1 = "#{remote}#{path1}"
|
|
else
|
|
path2 = "#{remote}#{path2}"
|
|
end
|
|
|
|
["scp", *argv, "--", path1, path2]
|
|
end
|
|
else
|
|
error = "! Only one path should be remote (start with ':')"
|
|
[400, {"content-type" => "text/plain", "content-length" => error.bytesize.to_s}, [error]]
|
|
end
|
|
end
|
|
end
|