Module: Babushka::RunHelpers

87
88
89
90
91
92
93

# File 'lib/babushka/helpers/run_helpers.rb', line 87

def added_by_babushka nlines
  if nlines == 1
    "This line added #{_by_babushka}"
  else
    "These #{nlines} lines added #{_by_babushka}"
  end
end

76
77
78
79

# File 'lib/babushka/helpers/run_helpers.rb', line 76

def append_to_file text, file, opts = {}
  text = text.to_s
  shell %Q{echo "\n# #{added_by_babushka(text.split("\n").length)}\n#{text.gsub('"', '\"')}" >> #{file}}, opts
end

95
96
97
98
99
100
101
102
103

# File 'lib/babushka/helpers/run_helpers.rb', line 95

def babushka_config? path
  if !path.p.exists?
    unmet "the config hasn't been generated yet"
  elsif !grep(/Generated by babushka/, path)
    unmet "the config needs to be regenerated"
  else
    true
  end
end

15
16
17
18
19

# File 'lib/babushka/helpers/run_helpers.rb', line 15

def bundle_rake cmd, &block
  cd var(:rails_root) do
    shell "bundle exec rake #{cmd} --trace RAILS_ENV=#{var :app_env}", :as => var(:username), :log => true, &block
  end
end

38
39
40
41
42
43
44
45

# File 'lib/babushka/helpers/run_helpers.rb', line 38

def change_line line, replacement, filename
  path = filename.p

  log "Patching #{path}"
  shell "cat > #{path}", :as => path.owner, :input => path.readlines.map {|l|
    l.gsub(/^(\s*)(#{Regexp.escape(line)})/, "\\1# #{edited_by_babushka}\n\\1# was: \\2\n\\1#{replacement}")
  }.join("")
end

65
66
67
68
69
70

# File 'lib/babushka/helpers/run_helpers.rb', line 65

def change_with_sed keyword, from, to, file
  # Remove the incorrect setting if it's there
  shell("#{sed} -ri 's/^#{keyword}\s+#{from}//' #{file}", :sudo => !File.writable?(file))
  # Add the correct setting unless it's already there
  grep(/^#{keyword}\s+#{to}/, file) or shell("echo '#{keyword} #{to}' >> #{file}", :sudo => !File.writable?(file))
end

21
22
23
24
25

# File 'lib/babushka/helpers/run_helpers.rb', line 21

def check_file file_name, method_name
  File.send(method_name, file_name).tap {|result|
    log_error "#{file_name} failed #{method_name.to_s.sub(/[?!]$/, '')} check." unless result
  }
end

84
85
86

# File 'lib/babushka/helpers/run_helpers.rb', line 84

def edited_by_babushka
  "This line edited #{_by_babushka}"
end

126
127
128
129
130
131
132

# File 'lib/babushka/helpers/run_helpers.rb', line 126

def erb_path_for erb
  if erb.to_s.starts_with? '/'
    erb # absolute path
  elsif load_path
    File.dirname(load_path) / erb # directory this dep is in, plus relative path
  end
end

27
28
29
30
31
32
33
34
35
36

# File 'lib/babushka/helpers/run_helpers.rb', line 27

def grep pattern, file
  if (path = file.p).exists?
    output = if pattern.is_a? String
      path.readlines.select {|l| l[pattern] }
    elsif pattern.is_a? Regexp
      path.readlines.grep pattern
    end
    output unless output.blank?
  end
end

7
8
9

# File 'lib/babushka/helpers/run_helpers.rb', line 7

def hostname
  shell 'hostname -f'
end

47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

# File 'lib/babushka/helpers/run_helpers.rb', line 47

def insert_into_file insert_before, path, lines, opts = {}
  opts.defaults! :comment_char => '#', :insert_after => nil
  nlines = lines.split("\n").length
  before, after = path.p.readlines.cut {|l| l.strip == insert_before.strip }

  log "Patching #{path}"
  if after.empty? || (opts[:insert_after] && before.last.strip != opts[:insert_after].strip)
    log_error "Couldn't find the spot to write to in #{path}."
  else
    shell "cat > #{path}", :as => path.owner, :sudo => !File.writable?(path), :input => [
      before,
      added_by_babushka(nlines).start_with(opts[:comment_char] + ' ').end_with("\n"),
      lines.end_with("\n"),
      after
    ].join
  end
end

134
135
136
137
138

# File 'lib/babushka/helpers/run_helpers.rb', line 134

def log_and_open message, url
  log "#{message} Hit Enter to open the download page.", :newline => false
  read_from_prompt ' '
  shell "open #{url}"
end

140
141
142
143

# File 'lib/babushka/helpers/run_helpers.rb', line 140

def mysql cmd, username = 'root', include_password = true
  password_segment = "--password='#{var :db_password}'" if include_password
  shell "echo \"#{cmd.gsub('"', '\"').end_with(';')}\" | mysql -u #{username} #{password_segment}"
end

11
12
13

# File 'lib/babushka/helpers/run_helpers.rb', line 11

def rake cmd, &block
  sudo "rake #{cmd} RAILS_ENV=#{var :app_env}", :as => var(:username), &block
end

110
111
112
113
114
115
116
117
118
119
120
121
122
123
124

# File 'lib/babushka/helpers/run_helpers.rb', line 110

def render_erb erb, opts = {}
  if (path = erb_path_for(erb)).nil?
    log_error "If you use #render_erb within a dynamically defined dep, you have to give the full path to the erb template."
  elsif !File.exists?(path) && !opts[:optional]
    log_error "Couldn't find erb to render at #{path}."
  elsif File.exists?(path)
    Renderable.new(opts[:to]).render(path, opts.merge(:context => self)).tap {|result|
      if result
        log "Rendered #{opts[:to]}."
      else
        log_error "Couldn't render #{opts[:to]}."
      end
    }
  end
end

72
73
74

# File 'lib/babushka/helpers/run_helpers.rb', line 72

def sed
  Base.host.linux? ? 'sed' : 'gsed'
end

105
106
107
108

# File 'lib/babushka/helpers/run_helpers.rb', line 105

def yaml path
  require 'yaml'
  YAML.load_file path.p
end