Class: Mamiya::Script

Inherits:
DSL
  • Object
show all
Defined in:
lib/mamiya/script.rb

Defined Under Namespace

Classes: CommandFailed

Instance Attribute Summary

Attributes inherited from DSL

#hooks, #tasks

Instance Method Summary collapse

Methods inherited from DSL

#[], add_hook, defaults, define_variable_accessor, #evaluate!, #initialize, #invoke, #load!, #load_path, #set, set_default, #set_default, #task, #use

Constructor Details

This class inherits a constructor from Mamiya::DSL

Instance Method Details

#build_from ⇒ Object 



142
143
144
 
# File 'lib/mamiya/script.rb', line 142

def build_from
  self[:build_from] && Pathname.new(self[:build_from])
end

#cd(*args) ⇒ Object 



136
137
138
139
140
 
# File 'lib/mamiya/script.rb', line 136

def cd(*args)
  logger.info "$ cd #{args[0]}"
  # FIXME: This could make conflict with other threads
  Dir.chdir *args
end

#current_path ⇒ Object 



158
159
160
 
# File 'lib/mamiya/script.rb', line 158

def current_path
  deploy_to && deploy_to.join('current')
end

#deploy_to ⇒ Object 



146
147
148
 
# File 'lib/mamiya/script.rb', line 146

def deploy_to
  self[:deploy_to] && Pathname.new(self[:deploy_to])
end

#release_path ⇒ Object 



150
151
152
 
# File 'lib/mamiya/script.rb', line 150

def release_path
  self[:release_path] && Pathname.new(self[:release_path])
end

#run(*args, allow_failure: false) ⇒ Object 



56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
 
# File 'lib/mamiya/script.rb', line 56

def run(*args, allow_failure: false)
  # TODO: Stop when fail
  actual = -> do
    started_at = Time.now
    run_id = generate_run_id()
    logger = self.logger["run:#{run_id}"]

    env = args.last.is_a?(Hash) ? args.pop : {}
    shellenv = env.empty? ? nil : "#{escape_env(env)} "

    logger.info("$ #{shellenv}#{args.shelljoin}")

    err_r, err_w = IO.pipe
    out_r, out_w = IO.pipe

    pid = spawn(env, *args.map(&:to_s), out: out_w, err: err_w)

    [out_w, err_w].each(&:close)

    buf = ""
    last_out = Time.now

    ths = {:info => out_r, :warn => err_r}.map do |severity, io|
      Thread.new {
        until io.eof?
          str = io.gets
          logger.__send__(severity, "  #{str.chomp}")
          buf << str
          last_out = Time.now
        end
      }.tap { |_| _.abort_on_exception = true }
    end

    timekeeper_th = Thread.new do
      l = logger['timekeeper']
      loop do
        if 90 < (Time.now - last_out)
          l.warn "! pid #{pid} still running; since #{started_at}"
        end
        sleep 60
      end
    end
    timekeeper_th.abort_on_exception = true

    pid, status = Process.waitpid2(pid)
    timekeeper_th.kill if timekeeper_th.alive?

    begin
      Timeout.timeout(3) { ths.each(&:join) }
    rescue Timeout::Error
    end
    ths.each { |_| _.alive? && _.kill }

    [out_r, err_r].each(&:close)

    unless allow_failure || status.success?
      failure_msg = "Execution failed (" \
        "status=#{status.exitstatus}" \
        " pid=#{status.pid}" \
        "#{status.signaled? ? "termsig=#{status.termsig.inspect} stopsig=#{status.stopsig.inspect}" : nil}" \
        "#{status.stopped? ? " stopped" : nil}" \
        "): #{args.inspect}"

      logger.error failure_msg
      raise CommandFailed, failure_msg

    end

    logger.info "* pid #{pid} completed: #{args.inspect}"

    buf
  end

  if defined? Bundler
    Bundler.with_clean_env(&actual)
  else
    actual.call
  end
end

#shared_path ⇒ Object 



154
155
156
 
# File 'lib/mamiya/script.rb', line 154

def shared_path
  deploy_to && deploy_to.join('shared')
end