Class: Jets::Cfn::Upload

Inherits:
Object
  • Object
show all
Includes:
ActionView::Helpers::NumberHelper, AwsServices
Defined in:
lib/jets/cfn/upload.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from AwsServices

#apigateway, #cfn, #lambda, #logs, #s3, #s3_resource, #sns, #sqs, #sts

Methods included from AwsServices::StackStatus

#lookup, #stack_exists?, #stack_in_progress?

Constructor Details

#initialize(bucket_name) ⇒ Upload

Returns a new instance of Upload.



10
11
12
# File 'lib/jets/cfn/upload.rb', line 10

def initialize(bucket_name)
  @bucket_name = bucket_name
end

Instance Attribute Details

#bucket_nameObject (readonly)

Returns the value of attribute bucket_name.



9
10
11
# File 'lib/jets/cfn/upload.rb', line 9

def bucket_name
  @bucket_name
end

Instance Method Details

#add_rack_assets(public_folders) ⇒ Object



67
68
69
70
# File 'lib/jets/cfn/upload.rb', line 67

def add_rack_assets(public_folders)
  return public_folders unless Jets.rack?
  public_folders + %w[rack/public]
end

#cache_controlObject

If cache_control is provided, then it will set the entire cache-control header. If only max_age is provided, then we’ll generate a cache_control header. Using max_age is the shorter and simply way of setting the cache_control header.



118
119
120
121
122
123
124
125
# File 'lib/jets/cfn/upload.rb', line 118

def cache_control
  cache_control = Jets.config.assets.cache_control
  unless cache_control
    max_age = Jets.config.assets.max_age # defaults to 3600 in jets/application.rb
    cache_control = "public, max-age=#{max_age}"
  end
  cache_control
end

#identical_on_s3?(full_path) ⇒ Boolean

Returns:

  • (Boolean)


102
103
104
105
106
107
108
109
110
111
112
113
# File 'lib/jets/cfn/upload.rb', line 102

def identical_on_s3?(full_path)
  local_md5 = ::Digest::MD5.file(full_path)
  key = s3_key(full_path)
  begin
    resp = s3.head_object(bucket: bucket_name, key: key)
  rescue Aws::S3::Errors::NotFound
    return false
  end

  remote_md5 = resp.etag.delete_prefix('"').delete_suffix('"')
  local_md5 == remote_md5
end

#pretty_time(total_seconds) ⇒ Object



128
129
130
131
132
133
134
135
136
# File 'lib/jets/cfn/upload.rb', line 128

def pretty_time(total_seconds)
  minutes = (total_seconds / 60) % 60
  seconds = total_seconds % 60
  if total_seconds < 60
    "#{seconds.to_i}s"
  else
    "#{minutes.to_i}m #{seconds.to_i}s"
  end
end

#s3_key(full_path) ⇒ Object



97
98
99
100
# File 'lib/jets/cfn/upload.rb', line 97

def s3_key(full_path)
  relative_path = full_path.sub(Jets.root.to_s, '')
  "jets/#{relative_path}"
end

#uploadObject



14
15
16
17
18
# File 'lib/jets/cfn/upload.rb', line 14

def upload
  upload_cfn_templates
  upload_zip_files
  upload_assets
end

#upload_asset_folder(folder) ⇒ Object



72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
# File 'lib/jets/cfn/upload.rb', line 72

def upload_asset_folder(folder)
  expression = "#{Jets.root}#{folder}/**/*"
  group_size = 10
  Dir.glob(expression).each_slice(group_size) do |paths|
    threads = []
    paths.each do |full_path|
      next unless File.file?(full_path)

      threads << Thread.new do
        upload_to_s3(full_path)
      end
    end
    threads.each(&:join)
  end
end

#upload_assetsObject



52
53
54
55
56
57
# File 'lib/jets/cfn/upload.rb', line 52

def upload_assets
  puts "Checking for modified public assets and uploading to S3."
  start_time = Time.now
  upload_public_assets
  puts "Time for public assets to s3: #{pretty_time(Time.now-start_time).colorize(:green)}"
end

#upload_cfn_templatesObject



20
21
22
23
24
25
26
27
28
29
30
# File 'lib/jets/cfn/upload.rb', line 20

def upload_cfn_templates
  puts "Uploading CloudFormation templates to S3."
  expression = "#{Jets::Naming.template_path_prefix}-*"
  Dir.glob(expression).each do |path|
    next unless File.file?(path)

    key = "jets/cfn-templates/#{File.basename(path)}"
    obj = s3_resource.bucket(bucket_name).object(key)
    obj.upload_file(path)
  end
end

#upload_public_assetsObject



59
60
61
62
63
64
65
# File 'lib/jets/cfn/upload.rb', line 59

def upload_public_assets
  public_folders = %w[public]
  public_folders = add_rack_assets(public_folders)
  public_folders.each do |folder|
    upload_asset_folder(folder)
  end
end

#upload_to_s3(full_path) ⇒ Object



88
89
90
91
92
93
94
95
# File 'lib/jets/cfn/upload.rb', line 88

def upload_to_s3(full_path)
  return if identical_on_s3?(full_path)

  key = s3_key(full_path)
  obj = s3_resource.bucket(bucket_name).object(key)
  puts "Uploading s3://#{bucket_name}/#{key}" # uncomment to see and debug
  obj.upload_file(full_path, acl: "public-read", cache_control: cache_control)
end

#upload_zip_file(path) ⇒ Object



40
41
42
43
44
45
46
47
48
49
50
# File 'lib/jets/cfn/upload.rb', line 40

def upload_zip_file(path)
  file_size = number_to_human_size(File.size(path))

  puts "Uploading #{path} (#{file_size}) to S3"
  start_time = Time.now
  s3_key = "jets/code/#{File.basename(path)}"
  obj = s3_resource.bucket(bucket_name).object(s3_key)
  obj.upload_file(path)
  puts "Uploaded to s3://#{bucket_name}/#{s3_key}".colorize(:green)
  puts "Time to upload code to s3: #{pretty_time(Time.now-start_time).colorize(:green)}"
end

#upload_zip_filesObject



32
33
34
35
36
37
38
# File 'lib/jets/cfn/upload.rb', line 32

def upload_zip_files
  puts "Uploading code zip files to S3."
  zip_area = "#{Jets.build_root}/stage/zips"
  Dir.glob("#{zip_area}/*").each do |file|
    upload_zip_file(file)
  end
end