Sha256: 973553686c99bb38d007224d3672e541f97f64ce055c7d263c63658e900885f5

Contents?: true

Size: 1.96 KB

Versions: 12

Compression:

Stored size: 1.96 KB

Contents

require "erb"
require "foreman/export"

class Foreman::Export::Runit < Foreman::Export::Base
  ENV_VARIABLE_REGEX = /([a-zA-Z_]+[a-zA-Z0-9_]*)=(\S+)/

  def export(location, options={})
    error("Must specify a location") unless location

    app = options[:app] || File.basename(engine.directory)
    user = options[:user] || app
    log_root = options[:log] || "/var/log/#{app}"
    template_root = options[:template]

    concurrency = Foreman::Utils.parse_concurrency(options[:concurrency])

    run_template = export_template('runit', 'run.erb', template_root)
    log_run_template = export_template('runit', 'log_run.erb', template_root)

    engine.procfile.entries.each do |process|
      1.upto(concurrency[process.name]) do |num|
        process_directory     = "#{location}/#{app}-#{process.name}-#{num}"
        process_env_directory = "#{process_directory}/env"
        process_log_directory = "#{process_directory}/log"

        create_directory process_directory
        create_directory process_env_directory
        create_directory process_log_directory

        run = ERB.new(run_template).result(binding)
        write_file "#{process_directory}/run", run
        FileUtils.chmod 0755, "#{process_directory}/run"

        port = engine.port_for(process, num, options[:port])
        environment_variables = {'PORT' => port}.
            merge(engine.environment).
            merge(inline_variables(process.command))

        environment_variables.each_pair do |var, env|
          write_file "#{process_env_directory}/#{var.upcase}", env
        end

        log_run = ERB.new(log_run_template).result(binding)
        write_file "#{process_log_directory}/run", log_run
        FileUtils.chmod 0755, "#{process_log_directory}/run"
      end
    end

  end

  private
  def create_directory(location)
    say "creating: #{location}"
    FileUtils.mkdir(location)
  end

  def inline_variables(command)
    variable_name_regex =
    Hash[*command.scan(ENV_VARIABLE_REGEX).flatten]
  end
end

Version data entries

12 entries across 12 versions & 1 rubygems

Version Path
foreman-0.37.0.pre3-mingw32 lib/foreman/export/runit.rb
foreman-0.37.0.pre3-java lib/foreman/export/runit.rb
foreman-0.37.0.pre3 lib/foreman/export/runit.rb
foreman-0.37.0.pre2-java lib/foreman/export/runit.rb
foreman-0.37.0.pre2 lib/foreman/export/runit.rb
foreman-0.37.0.pre1-java lib/foreman/export/runit.rb
foreman-0.37.0.pre1 lib/foreman/export/runit.rb
foreman-0.36.1 lib/foreman/export/runit.rb
foreman-0.36.0 lib/foreman/export/runit.rb
foreman-0.35.0 lib/foreman/export/runit.rb
foreman-0.34.1 lib/foreman/export/runit.rb
foreman-0.34.0 lib/foreman/export/runit.rb