Class: WhiskeyDisk

Inherits:
Object
  • Object
show all
Defined in:
lib/whiskey_disk.rb,
lib/whiskey_disk/config.rb,
lib/whiskey_disk/config/filter.rb,
lib/whiskey_disk/config/abstract_filter.rb,
lib/whiskey_disk/config/filters/modules/scope_helper.rb,
lib/whiskey_disk/config/filters/project_scope_filter.rb,
lib/whiskey_disk/config/filters/default_domain_filter.rb,
lib/whiskey_disk/config/filters/add_project_name_filter.rb,
lib/whiskey_disk/config/filters/localize_domains_filter.rb,
lib/whiskey_disk/config/filters/environment_scope_filter.rb,
lib/whiskey_disk/config/filters/stringify_hash_keys_filter.rb,
lib/whiskey_disk/config/filters/add_environment_name_filter.rb,
lib/whiskey_disk/config/filters/default_config_target_filter.rb,
lib/whiskey_disk/config/filters/normalize_ssh_options_filter.rb,
lib/whiskey_disk/config/filters/hashify_domain_entries_filter.rb,
lib/whiskey_disk/config/filters/drop_empty_domain_roles_filter.rb,
lib/whiskey_disk/config/filters/check_for_duplicate_domains_filter.rb,
lib/whiskey_disk/config/filters/convert_role_strings_to_list_filter.rb,
lib/whiskey_disk/config/filters/select_project_and_environment_filter.rb

Defined Under Namespace

Classes: Config

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(options = {}) ⇒ WhiskeyDisk

Returns a new instance of WhiskeyDisk.



7
8
9
 
# File 'lib/whiskey_disk.rb', line 7

def initialize(options = {})
  @staleness_checks = true if options[:staleness_checks]
end

Instance Attribute Details

#config ⇒ Object 



15
16
17
 
# File 'lib/whiskey_disk.rb', line 15

def config
  @config ||= WhiskeyDisk::Config.new
end

#configuration ⇒ Object 



19
20
21
 
# File 'lib/whiskey_disk.rb', line 19

def configuration
  @configuration ||= config.fetch
end

#results ⇒ Object (readonly)

Returns the value of attribute results.



5
6
7
 
# File 'lib/whiskey_disk.rb', line 5

def results
  @results
end

Instance Method Details

#apply_staleness_check(commands) ⇒ Object 



90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
 
# File 'lib/whiskey_disk.rb', line 90

def apply_staleness_check(commands)
  needs(:deploy_to, :repository)
  
  check = "cd #{setting(:deploy_to)}; " +
          "ml=\`git log -1 --pretty=format:%H\`; " +
          "mr=\`git ls-remote #{setting(:repository)} refs/heads/#{branch}\`; "
  
  if setting(:deploy_config_to)
    check += "cd #{setting(:deploy_config_to)}; " +
             "cl=\`git log -1 --pretty=format:%H\`; " +
             "cr=\`git ls-remote #{setting(:config_repository)} refs/heads/#{config_branch}\`; "
  end
  
  check += "if [[ $ml != ${mr%%\t*} ]] " +
           (setting(:deploy_config_to) ? "|| [[ $cl != ${cr%%\t*} ]]" : '') +
           "; then #{commands}; else echo \"No changes to deploy.\"; fi"
end

#branch ⇒ Object 



60
61
62
 
# File 'lib/whiskey_disk.rb', line 60

def branch
  (setting(:branch) and setting(:branch) != '') ? setting(:branch) : 'master'
end

#buffer ⇒ Object 



11
12
13
 
# File 'lib/whiskey_disk.rb', line 11

def buffer
  @buffer ||= []
end

#build_command(domain, cmd) ⇒ Object 



131
132
133
 
# File 'lib/whiskey_disk.rb', line 131

def build_command(domain, cmd)
  "#{'set -x; ' if debugging?}" + encode_roles(domain['roles']) + cmd
end

#build_path(path) ⇒ Object 



236
237
238
239
 
# File 'lib/whiskey_disk.rb', line 236

def build_path(path)
  return path if path =~ %r{^/}
  File.join(setting(:deploy_to), path)
end

#bundle ⇒ Object 



112
113
114
115
 
# File 'lib/whiskey_disk.rb', line 112

def bundle
  return '' if buffer.empty?
  (staleness_checks_enabled? and check_staleness?) ? apply_staleness_check(join_commands) : join_commands
end

#capture_git_changes ⇒ Object 



289
290
291
292
 
# File 'lib/whiskey_disk.rb', line 289

def capture_git_changes
  needs(:deploy_to)
  enqueue "git diff --name-only ${ml}..HEAD > #{setting(:deploy_to)}/.whiskey_disk_git_changes"
end

#check_staleness? ⇒ Boolean

Returns:

  • (Boolean) 


31
32
33
 
# File 'lib/whiskey_disk.rb', line 31

def check_staleness?
  config.check_staleness?
end

#checkout_configuration_repository ⇒ Object 



261
262
263
264
 
# File 'lib/whiskey_disk.rb', line 261

def checkout_configuration_repository
  needs(:deploy_config_to, :config_repository)
  clone_repository(setting(:config_repository), setting(:deploy_config_to), config_branch)
end

#checkout_main_repository ⇒ Object 



256
257
258
259
 
# File 'lib/whiskey_disk.rb', line 256

def checkout_main_repository
  needs(:deploy_to, :repository)
  clone_repository(setting(:repository), setting(:deploy_to), branch)
end

#clone_repository(repo, path, my_branch) ⇒ Object 



216
217
218
219
220
 
# File 'lib/whiskey_disk.rb', line 216

def clone_repository(repo, path, my_branch)
  enqueue "cd #{parent_path(path)}"
  enqueue("if [ -e #{path} ]; then echo 'Repository already cloned to [#{path}].  Skipping.'; " +
          "else git clone #{repo} #{tail_path(path)} && #{safe_branch_checkout(path, my_branch)}; fi")
end

#config_branch ⇒ Object 



64
65
66
 
# File 'lib/whiskey_disk.rb', line 64

def config_branch
  (setting(:config_branch) and setting(:config_branch) != '') ? setting(:config_branch) : 'master'
end

#debugging? ⇒ Boolean

Returns:

  • (Boolean) 


23
24
25
 
# File 'lib/whiskey_disk.rb', line 23

def debugging?
  config.debug?
end

#domain_limit_match?(domain, limit) ⇒ Boolean

Returns:

  • (Boolean) 


117
118
119
 
# File 'lib/whiskey_disk.rb', line 117

def domain_limit_match?(domain, limit)
  domain.sub(%r{^.*@}, '') == limit
end

#domain_of_interest?(domain) ⇒ Boolean

Returns:

  • (Boolean) 


121
122
123
124
 
# File 'lib/whiskey_disk.rb', line 121

def domain_of_interest?(domain)
  return true unless limit = config.domain_limit
  domain_limit_match?(domain, limit)
end

#encode_roles(roles) ⇒ Object 



126
127
128
129
 
# File 'lib/whiskey_disk.rb', line 126

def encode_roles(roles)
  return '' unless roles and !roles.empty?
  "export WD_ROLES='#{roles.join(':')}'; "
end

#enqueue(command) ⇒ Object 



39
40
41
 
# File 'lib/whiskey_disk.rb', line 39

def enqueue(command)
  buffer << command
end

#ensure_config_parent_path_is_present ⇒ Object 



251
252
253
254
 
# File 'lib/whiskey_disk.rb', line 251

def ensure_config_parent_path_is_present
  needs(:deploy_config_to)
  enqueue "mkdir -p #{parent_path(setting(:deploy_config_to))}"
end

#ensure_main_parent_path_is_present ⇒ Object 



246
247
248
249
 
# File 'lib/whiskey_disk.rb', line 246

def ensure_main_parent_path_is_present
  needs(:deploy_to)
  enqueue "mkdir -p #{parent_path(setting(:deploy_to))}"
end

#env_vars ⇒ Object 



68
69
70
71
72
73
74
 
# File 'lib/whiskey_disk.rb', line 68

def env_vars
  return '' unless setting(:rake_env)
  setting(:rake_env).keys.inject('') do |buffer,k| 
    buffer += "#{k}='#{setting(:rake_env)[k]}' "
    buffer
  end
end

#flush ⇒ Object 



159
160
161
162
163
164
165
166
167
 
# File 'lib/whiskey_disk.rb', line 159

def flush
  needs(:domain)
  setting(:domain).each do |domain|
    next unless domain_of_interest?(domain['name'])
    puts "Deploying #{domain['name']}..."
    status = remote?(domain['name']) ? run(domain, bundle) : shell(domain, bundle)
    record_result(domain['name'], status)
  end
end

#has_config_repo? ⇒ Boolean

Returns:

  • (Boolean) 


52
53
54
 
# File 'lib/whiskey_disk.rb', line 52

def has_config_repo?
  ! (setting(:config_repository).nil? or setting(:config_repository) == '')
end

#if_file_present(path, cmd) ⇒ Object 



204
205
206
 
# File 'lib/whiskey_disk.rb', line 204

def if_file_present(path, cmd)
  "if [ -e #{path} ]; then #{cmd}; fi"
end

#if_task_defined(task, cmd) ⇒ Object 



208
209
210
 
# File 'lib/whiskey_disk.rb', line 208

def if_task_defined(task, cmd)
  %Q(rakep=`#{env_vars} #{rake_command} -P` && if [[ `echo "${rakep}" | grep #{task}` != "" ]]; then #{cmd}; fi )
end

#initialize_all_changes ⇒ Object 



283
284
285
286
287
 
# File 'lib/whiskey_disk.rb', line 283

def initialize_all_changes
  needs(:deploy_to)
  initialize_git_changes
  initialize_rsync_changes
end

#initialize_git_changes ⇒ Object 



272
273
274
275
276
 
# File 'lib/whiskey_disk.rb', line 272

def initialize_git_changes
  needs(:deploy_to)
  enqueue "rm -f #{setting(:deploy_to)}/.whiskey_disk_git_changes"
  snapshot_git_revision
end

#initialize_rsync_changes ⇒ Object 



278
279
280
281
 
# File 'lib/whiskey_disk.rb', line 278

def initialize_rsync_changes
  needs(:deploy_to)
  enqueue "rm -f #{setting(:deploy_to)}/.whiskey_disk_rsync_changes"
end

#join_commands ⇒ Object 



108
109
110
 
# File 'lib/whiskey_disk.rb', line 108

def join_commands
  buffer.collect {|c| "{ #{c} ; }"}.join(' && ')
end

#needs(*keys) ⇒ Object 



84
85
86
87
88
 
# File 'lib/whiskey_disk.rb', line 84

def needs(*keys)
  keys.each do |key|
    raise "No value for '#{key}' declared in configuration files [#{config.configuration_file}]" unless setting(key)
  end
end

#parent_path(path) ⇒ Object 



76
77
78
 
# File 'lib/whiskey_disk.rb', line 76

def parent_path(path)
  File.split(path).first
end

#project_name_specified? ⇒ Boolean

Returns:

  • (Boolean) 


56
57
58
 
# File 'lib/whiskey_disk.rb', line 56

def project_name_specified?
  setting(:project) != 'unnamed_project'
end

#rake_command ⇒ Object 



135
136
137
 
# File 'lib/whiskey_disk.rb', line 135

def rake_command
  (setting(:rake_command) and setting(:rake_command) != '') ? setting(:rake_command) : 'rake'
end

#record_result(domain, status) ⇒ Object 



169
170
171
172
 
# File 'lib/whiskey_disk.rb', line 169

def record_result(domain, status)
  @results ||= []
  @results << { 'domain' => domain, 'status' => status }
end

#refresh_checkout(path, repo_branch) ⇒ Object 



222
223
224
225
226
227
 
# File 'lib/whiskey_disk.rb', line 222

def refresh_checkout(path, repo_branch)
  enqueue "cd #{path}"
  enqueue "git fetch origin +refs/heads/#{repo_branch}:refs/remotes/origin/#{repo_branch} #{'&>/dev/null' unless debugging?}"
  enqueue "git checkout #{repo_branch} #{'&>/dev/null' unless debugging?}"
  enqueue "git reset --hard origin/#{repo_branch} #{'&>/dev/null' unless debugging?}"
end

#refresh_configuration ⇒ Object 



307
308
309
310
311
312
313
314
 
# File 'lib/whiskey_disk.rb', line 307

def refresh_configuration
  needs(:deploy_to, :deploy_config_to)
  raise "Must specify project name when using a configuration repository." unless project_name_specified?
  enqueue "echo Rsyncing configuration..."
  enqueue("rsync -a#{'v --progress' if debugging?} " + '--log-format="%t [%p] %i %n" ' +
          "#{setting(:deploy_config_to)}/#{setting(:project)}/#{setting(:config_target)}/ #{setting(:deploy_to)}/ " + 
          "> #{setting(:deploy_to)}/.whiskey_disk_rsync_changes")
end

#remote?(domain) ⇒ Boolean

Returns:

  • (Boolean) 


43
44
45
46
47
48
49
50
 
# File 'lib/whiskey_disk.rb', line 43

def remote?(domain)
  return false unless domain
  return false if domain == 'local'
  limit = config.domain_limit 
  return false if limit and domain_limit_match?(domain, limit)

  true
end

#run(domain, cmd) ⇒ Object 



139
140
141
 
# File 'lib/whiskey_disk.rb', line 139

def run(domain, cmd)
  ssh(domain, cmd)
end

#run_post_deploy_hooks ⇒ Object 



322
323
324
325
326
 
# File 'lib/whiskey_disk.rb', line 322

def run_post_deploy_hooks
  needs(:deploy_to)
  run_script(setting(:post_deploy_script))
  run_rake_task(setting(:deploy_to), "deploy:post_deploy")
end

#run_post_setup_hooks ⇒ Object 



316
317
318
319
320
 
# File 'lib/whiskey_disk.rb', line 316

def run_post_setup_hooks
  needs(:deploy_to)
  run_script(setting(:post_setup_script))
  run_rake_task(setting(:deploy_to), "deploy:post_setup")
end

#run_rake_task(path, task_name) ⇒ Object 



229
230
231
232
233
234
 
# File 'lib/whiskey_disk.rb', line 229

def run_rake_task(path, task_name)
  enqueue "echo Running rake #{task_name}..."
  enqueue "cd #{path}"
  enqueue(if_file_present("#{setting(:deploy_to)}/Rakefile", 
    if_task_defined(task_name, "#{env_vars} #{rake_command} #{'--trace' if debugging?} #{task_name} to=#{setting(:environment)}")))
end

#run_script(script) ⇒ Object 



241
242
243
244
 
# File 'lib/whiskey_disk.rb', line 241

def run_script(script)
  return unless script
  enqueue(%Q<cd #{setting(:deploy_to)}; echo "Running post script..."; #{env_vars} bash #{'-x' if debugging?} #{build_path(script)}>)
end

#safe_branch_checkout(path, my_branch) ⇒ Object 



212
213
214
 
# File 'lib/whiskey_disk.rb', line 212

def safe_branch_checkout(path, my_branch)
  %Q(cd #{path} && git checkout -b #{my_branch} origin/#{my_branch} || git checkout #{my_branch} origin/#{my_branch} || git checkout #{my_branch})
end

#setting(key) ⇒ Object 



27
28
29
 
# File 'lib/whiskey_disk.rb', line 27

def setting(key)
  configuration[key.to_s]
end

#shell(domain, cmd) ⇒ Object 



154
155
156
157
 
# File 'lib/whiskey_disk.rb', line 154

def shell(domain, cmd)
  puts "Running command locally: [#{cmd}]" if debugging?
  system('bash', '-c', build_command(domain, cmd))
end

#snapshot_git_revision ⇒ Object 



266
267
268
269
270
 
# File 'lib/whiskey_disk.rb', line 266

def snapshot_git_revision
  needs(:deploy_to)
  enqueue "cd #{setting(:deploy_to)}"
  enqueue %Q{ml=\`git log -1 --pretty=format:%H\`}
end

#ssh(domain, cmd) ⇒ Object 



143
144
145
146
147
148
149
150
151
152
 
# File 'lib/whiskey_disk.rb', line 143

def ssh(domain, cmd)
  args = []
  args << domain['name']
  args << '-v' if debugging?
  args += domain['ssh_options'] if domain['ssh_options']
  args << build_command(domain, cmd)

  puts "Running: ssh #{args.join(' ')}" if debugging?
  system('ssh', *args)
end

#staleness_checks_enabled? ⇒ Boolean

Returns:

  • (Boolean) 


35
36
37
 
# File 'lib/whiskey_disk.rb', line 35

def staleness_checks_enabled?
  !!@staleness_checks
end

#success? ⇒ Boolean

Returns:

  • (Boolean) 


199
200
201
202
 
# File 'lib/whiskey_disk.rb', line 199

def success?
  return true if !results or results.empty?
  results.all? {|result| result['status'] }
end

#summarize ⇒ Object 



187
188
189
190
191
192
193
194
195
196
197
 
# File 'lib/whiskey_disk.rb', line 187

def summarize
  puts "\nResults:"
  if results and not results.empty?
    total, successes, failures = summarize_results(results)
    puts "Total: #{total} deployment#{total == 1 ? '' : 's'}, " +
      "#{successes} success#{successes == 1 ? '' : 'es'}, " +
      "#{failures} failure#{failures == 1 ? '' : 's'}."
  else
    puts "No deployments to report."
  end       
end

#summarize_results(results) ⇒ Object 



174
175
176
177
178
179
180
181
182
183
184
185
 
# File 'lib/whiskey_disk.rb', line 174

def summarize_results(results)
  successes = failures = 0
  results.each do |result|
    puts "#{result['domain']} => #{result['status'] ? 'succeeded' : 'failed'}."
    if result['status']
      successes += 1 
    else
      failures += 1
    end
  end
  [successes + failures, successes, failures]
end

#tail_path(path) ⇒ Object 



80
81
82
 
# File 'lib/whiskey_disk.rb', line 80

def tail_path(path)
  File.split(path).last
end

#update_configuration_repository_checkout ⇒ Object 



301
302
303
304
305
 
# File 'lib/whiskey_disk.rb', line 301

def update_configuration_repository_checkout
  needs(:deploy_config_to)
  initialize_rsync_changes
  refresh_checkout(setting(:deploy_config_to), config_branch)
end

#update_main_repository_checkout ⇒ Object 



294
295
296
297
298
299
 
# File 'lib/whiskey_disk.rb', line 294

def update_main_repository_checkout
  needs(:deploy_to)
  initialize_git_changes
  refresh_checkout(setting(:deploy_to), branch)
  capture_git_changes
end