Class: CapistranoMulticonfigParallel::Application

Inherits:
Object
  • Object
show all
Includes:
ApplicationHelper, Celluloid, Celluloid::Logger
Defined in:
lib/capistrano_multiconfig_parallel/application.rb

Overview

finds app dependencies, shows menu and delegates jobs to celluloid manager rubocop:disable ClassLength

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from ApplicationHelper

parse_task_string, strip_characters_from_string

Methods included from StagesHelper

check_stage_path, checks_paths, fetch_stages, fetch_stages_paths, stages_paths

Methods included from CoreHelper

app_configuration, app_debug_enabled?, app_logger, ask_confirm, check_terminal_tty, debug_websocket?, execute_with_rescue, find_config_type, find_loaded_gem, find_worker_log, force_confirmation, format_error, log_error, log_to_file, rescue_error, rescue_interrupt, show_warning, websocket_config, websocket_server_config

Methods included from InternalHelper

config_file, custom_commands, default_internal_config, detect_root, enable_main_log_file, find_env_multi_cap_root, internal_config_directory, internal_config_file, log_directory, main_log_file, root, try_detect_capfile

Constructor Details

#initialize ⇒ Application

Returns a new instance of Application.



11
12
13
14
15
16
17
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 11

def initialize
  Celluloid.boot
  @stages = fetch_stages
  @stage_apps = multi_apps? ? @stages.map { |stage| stage.split(':').reverse[1] }.uniq : []
  collect_command_line_tasks(CapistranoMulticonfigParallel.original_args)
  @jobs = []
end

Instance Attribute Details

#application ⇒ Object (readonly)

Returns the value of attribute application.



9
10
11
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 9

def application
  @application
end

#args ⇒ Object (readonly)

Returns the value of attribute args.



9
10
11
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 9

def args
  @args
end

#argv ⇒ Object (readonly)

Returns the value of attribute argv.



9
10
11
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 9

def argv
  @argv
end

#branch_backup ⇒ Object (readonly)

Returns the value of attribute branch_backup.



9
10
11
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 9

def branch_backup
  @branch_backup
end

#condition ⇒ Object (readonly)

Returns the value of attribute condition.



9
10
11
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 9

def condition
  @condition
end

#default_stage ⇒ Object (readonly)

Returns the value of attribute default_stage.



9
10
11
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 9

def default_stage
  @default_stage
end

#dependency_tracker ⇒ Object (readonly)

Returns the value of attribute dependency_tracker.



9
10
11
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 9

def dependency_tracker
  @dependency_tracker
end

#jobs ⇒ Object (readonly)

Returns the value of attribute jobs.



9
10
11
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 9

def jobs
  @jobs
end

#manager ⇒ Object (readonly)

Returns the value of attribute manager.



9
10
11
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 9

def manager
  @manager
end

#name ⇒ Object (readonly)

Returns the value of attribute name.



9
10
11
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 9

def name
  @name
end

#stage ⇒ Object (readonly)

Returns the value of attribute stage.



9
10
11
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 9

def stage
  @stage
end

#stage_apps ⇒ Object (readonly)

Returns the value of attribute stage_apps.



9
10
11
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 9

def stage_apps
  @stage_apps
end

#stages ⇒ Object (readonly)

Returns the value of attribute stages.



9
10
11
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 9

def stages
  @stages
end

#top_level_tasks ⇒ Object (readonly)

Returns the value of attribute top_level_tasks.



9
10
11
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 9

def top_level_tasks
  @top_level_tasks
end

Instance Method Details

#backup_the_branch ⇒ Object 



50
51
52
53
54
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 50

def backup_the_branch
  return if custom_command? || @argv['BRANCH'].blank?
  @branch_backup = @argv['BRANCH'].to_s
  @argv['BRANCH'] = nil
end

#call_task_deploy_app(options = {}) ⇒ Object 



177
178
179
180
181
182
183
184
185
186
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 177

def call_task_deploy_app(options = {})
  options = options.stringify_keys
  main_box_name = @argv['BOX'].blank? ? '' : @argv['BOX']
  stage = options.fetch('stage', @default_stage)
  if app_configuration.development_stages.include?(stage) && main_box_name.present? && /^[a-z0-9,]+/.match(main_box_name)
    execute_on_multiple_boxes(main_box_name, options)
  else
    prepare_job(options)
  end
end

#can_tag_staging? ⇒ Boolean

Returns:

  • (Boolean) 


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

def can_tag_staging?
  wants_deploy_production? && tag_staging_exists? && fetch_multi_stages.include?('staging')
end

#check_before_starting ⇒ Object 



93
94
95
96
97
98
99
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 93

def check_before_starting
  CapistranoMulticonfigParallel.enable_logging
  @dependency_tracker = CapistranoMulticonfigParallel::DependencyTracker.new(Actor.current)
  @default_stage = app_configuration.development_stages.present? ? app_configuration.development_stages.first : 'development'
  @condition = Celluloid::Condition.new
  @manager = CapistranoMulticonfigParallel::CelluloidManager.new(Actor.current)
end

#check_multi_stages(custom_stages) ⇒ Object 



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

def check_multi_stages(custom_stages)
  can_tag_staging? ? custom_stages.reject { |u| u == 'production' } : custom_stages
end

#collect_command_line_tasks(args) ⇒ Object

:nodoc:



75
76
77
78
79
80
81
82
83
84
85
86
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 75

def collect_command_line_tasks(args) # :nodoc:
  @argv = {}
  @top_level_tasks = []
  args.each do |arg|
    if arg =~ /^(\w+)=(.*)$/m
      @argv[Regexp.last_match(1)] = Regexp.last_match(2)
    else
      @top_level_tasks << arg unless arg =~ /^-/
    end
  end
  @top_level_tasks.push(Rake.application.default_task_name) if @top_level_tasks.blank?
end

#collect_jobs(options = {}, &_block) ⇒ Object 



101
102
103
104
105
106
107
108
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 101

def collect_jobs(options = {}, &_block)
  options = prepare_options(options)
  options = options.stringify_keys
  apps = @dependency_tracker.fetch_apps_needed_for_deployment(options['app'], options['action'])
  backup_the_branch if multi_apps?
  deploy_multiple_apps(apps, options)
  deploy_app(options) if !custom_command? || !multi_apps?
end

#custom_command? ⇒ Boolean

Returns:

  • (Boolean) 


56
57
58
59
60
61
62
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 56

def custom_command?
  if multi_apps?
    !@stages.include?(@top_level_tasks.first) && custom_commands.include?(@top_level_tasks.first)
  else
    !@stages.include?(@top_level_tasks.second) && @stages.include?(@top_level_tasks.first) && custom_commands.include?(@top_level_tasks.second)
  end
end

#deploy_app(options = {}) ⇒ Object 



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

def deploy_app(options = {})
  options = options.stringify_keys
  app = options['app'].is_a?(Hash) ? options['app'] : { 'app' => options['app'] }
  branch = @branch_backup.present? ? @branch_backup : @argv['BRANCH'].to_s
  call_task_deploy_app({
    branch: branch,
    app: app,
    action: options['action']
  }.reverse_merge(options))
end

#deploy_multiple_apps(applications, options) ⇒ Object 



42
43
44
45
46
47
48
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 42

def deploy_multiple_apps(applications, options)
  options = options.stringify_keys
  return unless applications.present?
  applications.each do |app|
    deploy_app(options.merge('app' => app))
  end
end

#execute_on_multiple_boxes(main_box_name, options) ⇒ Object 



265
266
267
268
269
270
271
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 265

def execute_on_multiple_boxes(main_box_name, options)
  boxes = strip_characters_from_string(main_box_name).split(',').compact
  boxes.each do |box_name|
    options['env_options']['BOX'] = box_name
    prepare_job(options)
  end
end

#fetch_app_additional_env_options(variable) ⇒ Object 



245
246
247
248
249
250
251
252
253
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 245

def fetch_app_additional_env_options(variable)
  options = {}
  return options if variable.blank?
  env_options = strip_characters_from_string(variable)
  env_options = env_options.split(' ')
  options = multi_fetch_argv(env_options)
  options.stringify_keys!
  options
end

#fetch_multi_stages ⇒ Object 



124
125
126
127
128
129
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 124

def fetch_multi_stages
  custom_stages = @argv['STAGES'].blank? ? '' : @argv['STAGES']
  custom_stages = strip_characters_from_string(custom_stages).split(',').compact if custom_stages.present?
  custom_stages = custom_stages.present? ? custom_stages : [@default_stage]
  custom_stages
end

#get_app_additional_env_options(app, app_message) ⇒ Object 



154
155
156
157
158
159
160
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 154

def get_app_additional_env_options(app, app_message)
  app_name = (app.is_a?(Hash) && app[:app].present?) ? app[:app].camelcase : app
  app_name = app_name.present? ? app_name : 'current application'
  message = "Please write additional ENV options for #{app_name} for #{app_message}"
  app_additional_env_options = ask_confirm(message, nil)
  fetch_app_additional_env_options(app_additional_env_options)
end

#initialize_data ⇒ Object 



68
69
70
71
72
73
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 68

def initialize_data
  @application = custom_command? ? nil : @top_level_tasks.first.split(':').reverse[1]
  @stage = custom_command? ? nil : @top_level_tasks.first.split(':').reverse[0]
  @stage = @stage.present? ? @stage : @default_stage
  @name, @args = parse_task_string(@top_level_tasks.second)
end

#multi_apps? ⇒ Boolean

Returns:

  • (Boolean) 


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

def multi_apps?
  @stages.find { |stage| stage.include?(':') }.present?
end

#multi_fetch_argv(args) ⇒ Object 



255
256
257
258
259
260
261
262
263
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 255

def multi_fetch_argv(args)
  options = {}
  args.each do |arg|
    if arg =~ /^(\w+)=(.*)$/m
      options[Regexp.last_match(1)] = Regexp.last_match(2)
    end
  end
  options
end

#prepare_job(options) ⇒ Object

rubocop:disable CyclomaticComplexity



210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 210

def prepare_job(options)
  options = options.stringify_keys
  branch_name = options.fetch('branch', {})
  app = options.fetch('app', {})
  app = app.fetch('app', '')
  box = options['env_options']['BOX']
  message = box.present? ? "BOX #{box}:" : "stage #{options['stage']}:"
  env_opts = get_app_additional_env_options(app, message)

  options['env_options'] = options['env_options'].reverse_merge(env_opts)

  env_options = branch_name.present? ? { 'BRANCH' => branch_name }.merge(options['env_options']) : options['env_options']
  job_env_options = custom_command? && env_options['ACTION'].present? ? env_options.except('ACTION') : env_options

  job = {
    id: SecureRandom.random_number(500),
    app: app,
    env: options['stage'],
    action: custom_command? && env_options['ACTION'].present? ? env_options['ACTION'] : options['action'],
    task_arguments: options['task_arguments'],
    env_options: job_env_options
  }
  @jobs << job.stringify_keys
end

#prepare_options(options) ⇒ Object 



235
236
237
238
239
240
241
242
243
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 235

def prepare_options(options)
  options = options.stringify_keys
  options['app'] = options.fetch('app', @application.to_s.clone)
  options['action'] = options.fetch('action', @name.to_s.clone)
  options['stage'] = options.fetch('stage', @stage.to_s.clone)
  options['env_options'] = options.fetch('env_options', @argv.clone)
  options['task_arguments'] = options.fetch('task_arguments', @args.clone)
  options
end

#process_jobs ⇒ Object 



110
111
112
113
114
115
116
117
118
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 110

def process_jobs
  return unless @jobs.present?
  FileUtils.rm Dir["#{log_directory}/worker_*.log"]
  if app_configuration.multi_secvential.to_s.downcase == 'true'
    @jobs.each { |job| CapistranoMulticonfigParallel::StandardDeploy.new(job) }
  else
    run_async_jobs
  end
end

#run ⇒ Object 



166
167
168
169
170
171
172
173
174
175
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 166

def run
  options = {}
  if custom_command?
    options = verify_options_custom_command(options)
    run_custom_command(options)
  else
    collect_jobs(options)
  end
  process_jobs
end

#run_async_jobs ⇒ Object 



188
189
190
191
192
193
194
195
196
197
198
199
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 188

def run_async_jobs
  return unless @jobs.present?
  @jobs.pmap do |job|
    @manager.async.delegate(job)
  end
  until @manager.registration_complete
    sleep(0.1) # keep current thread alive
  end
  return unless @manager.registration_complete
  @manager.async.process_jobs
  wait_jobs_termination
end

#run_custom_command(options) ⇒ Object 



33
34
35
36
37
38
39
40
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 33

def run_custom_command(options)
  custom_stages = fetch_multi_stages
  return if custom_stages.blank?
  custom_stages = check_multi_stages(custom_stages)
  custom_stages.each do |stage|
    collect_jobs(options.merge('stage' => stage))
  end
end

#start ⇒ Object 



19
20
21
22
23
24
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 19

def start
  verify_app_dependencies if multi_apps? && app_configuration.application_dependencies.present?
  check_before_starting
  initialize_data
  run
end

#tag_staging_exists? ⇒ Boolean

check exists task from capistrano-gitflow

Returns:

  • (Boolean) 


120
121
122
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 120

def tag_staging_exists? # check exists task from capistrano-gitflow
  find_loaded_gem('capistrano-gitflow').present?
end

#verify_app_dependencies ⇒ Object

Raises:

  • (ArgumentError) 


26
27
28
29
30
31
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 26

def verify_app_dependencies
  wrong = app_configuration.application_dependencies.find do |hash|
    !@stage_apps.include?(hash[:app]) || (hash[:dependencies].present? && hash[:dependencies].find { |val| !@stage_apps.include?(val) })
  end
  raise ArgumentError, "Invalid configuration for #{wrong.inspect}".red if wrong.present?
end

#verify_options_custom_command(options) ⇒ Object 



88
89
90
91
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 88

def verify_options_custom_command(options)
  options[:action] = @argv['ACTION'].present? ? @argv['ACTION'] : 'deploy'
  options
end

#wait_jobs_termination ⇒ Object 



201
202
203
204
205
206
207
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 201

def wait_jobs_termination
  return if app_configuration.multi_secvential.to_s.downcase == 'true'
  result = @condition.wait
  return unless result.present?
  @manager.terminate
  terminate
end

#wants_deploy_production? ⇒ Boolean

Returns:

  • (Boolean) 


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

def wants_deploy_production?
  (!custom_command? && @stage == 'production') || (custom_command? && fetch_multi_stages.include?('production'))
end

#worker_environments ⇒ Object 



162
163
164
 
# File 'lib/capistrano_multiconfig_parallel/application.rb', line 162

def worker_environments
  @jobs.map { |job| job['env'] }
end