335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
|
# File 'lib/cnvrg/files.rb', line 335
def download_cnvrg_image(image_name, secret)
res =Cnvrg::API.request("images/#{image_name}/" + "download", 'POST', {secret:secret},true)
Cnvrg::CLI.is_response_success(res, true)
if res["result"]
download_resp = res
sts_path = download_resp["result"]["path_sts"]
uri = URI.parse(sts_path)
http_object = Net::HTTP.new(uri.host, uri.port)
http_object.use_ssl = true if uri.scheme == 'https'
request = Net::HTTP::Get.new(sts_path)
body = ""
http_object.start do |http|
response = http.request request
body = response.read_body
end
split = body.split("\n")
key = split[0]
iv = split[1]
access = Cnvrg::Helpers.decrypt(key, iv, download_resp["result"]["sts_a"])
secret = Cnvrg::Helpers.decrypt(key,iv, download_resp["result"]["sts_s"])
session = Cnvrg::Helpers.decrypt(key,iv, download_resp["result"]["sts_st"])
region = Cnvrg::Helpers.decrypt(key,iv, download_resp["result"]["region"])
bucket = Cnvrg::Helpers.decrypt(key,iv, download_resp["result"]["bucket"])
key = Cnvrg::Helpers.decrypt(key,iv, download_resp["result"]["key"])
client = Aws::S3::Client.new(
:access_key_id =>access,
:secret_access_key => secret,
:session_token => session,
:region => region,
:http_open_timeout => 60, :retry_limit => 20
)
File.open("/tmp/#{image_name}.tar", 'w+') do |file|
resp = client.get_object({bucket:bucket,
key:key}, target: file)
end
return true
end
rescue => e
Cnvrg::Logger.log_error(e)
return false
end
|