aboutsummaryrefslogtreecommitdiffstats
path: root/Library/Homebrew/utils.rb
blob: 6ce87f1909f2260052ffffd698a6deff02b1e7d0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
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
135
136
137
138
#  Copyright 2009 Max Howell and other contributors.
#
#  Redistribution and use in source and binary forms, with or without
#  modification, are permitted provided that the following conditions
#  are met:
#
#  1. Redistributions of source code must retain the above copyright
#     notice, this list of conditions and the following disclaimer.
#  2. Redistributions in binary form must reproduce the above copyright
#     notice, this list of conditions and the following disclaimer in the
#     documentation and/or other materials provided with the distribution.
#
#  THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
#  IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
#  OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
#  IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
#  INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
#  NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
#  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
#  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
#  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
#  THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

# args are additional inputs to puts until a nil arg is encountered
def ohai title, *sput
  title = title[0, `/usr/bin/tput cols`.strip.to_i-4] unless ARGV.verbose?
  puts "\033[0;34m==>\033[0;0;1m #{title}\033[0;0m"
  puts *sput unless sput.empty?
end

# shows a warning in delicious pink
def opoo warning
  puts "\033[1;35m==>\033[0;0;1m Warning!\033[0;0m #{warning}"
end

def onoe error
  lines = error.to_s.split'\n'
  puts "\033[1;31m==>\033[0;0;1m Error\033[0;0m: #{lines.shift}"
  puts *lines unless lines.empty?
end

def pretty_duration s
  return "2 seconds" if s < 3 # avoids the plural problem ;)
  return "#{s.to_i} seconds" if s < 120
  return "%.1f minutes" % (s/60)
end

def interactive_shell
  pid=fork
  if pid.nil?
    # TODO make the PS1 var change pls
    #brown="\[\033[0;33m\]"
    #reset="\[\033[0m\]"
    #ENV['PS1']="Homebrew-#{HOMEBREW_VERSION} #{brown}\W#{reset}\$ "
    exec ENV['SHELL']
  end
  Process.wait pid
  raise SystemExit, "Aborting due to non-zero exit status" if $? != 0
end

# Kernel.system but with exceptions
def safe_system cmd, *args
  puts "#{cmd} #{args*' '}" if ARGV.verbose?
  exec_success = Kernel.system cmd, *args
  # some tools, eg. tar seem to confuse ruby and it doesn't propogate the
  # CTRL-C interrupt to us too, so execution continues, but the exit code os
  # still 2 so we raise our own interrupt
  raise Interrupt, cmd if $?.termsig == 2
  unless exec_success
    puts "Exit code: #{$?}"
    raise ExecutionError.new(cmd, args)
  end 
end

def curl *args
  safe_system 'curl', '-f#LA', HOMEBREW_USER_AGENT, *args unless args.empty?
end

def puts_columns items, cols = 4
  if $stdout.tty?
    items = items.join("\n") if items.is_a?(Array)
    items.concat("\n") unless items.empty?
    width=`/bin/stty size`.chomp.split(" ").last
    IO.popen("/usr/bin/pr -#{cols} -t", "w"){|io| io.write(items) }
  else
    puts *items
  end
end

def exec_editor *args
  editor=ENV['EDITOR']
  if editor.nil?
    if system "/usr/bin/which -s mate"
      editor='mate'
    else
      editor='/usr/bin/vim'
    end
  end
  # we split the editor because especially on mac "mate -w" is common
  # but we still want to use the comma-delimited version of exec because then
  # we don't have to escape args, and escaping 100% is tricky
  exec *(editor.split+args)
end

# provide an absolute path to a command or this function will search the PATH
def arch_for_command cmd
    archs = []
    cmd = `/usr/bin/which #{cmd}` if not Pathname.new(cmd).absolute?

    IO.popen("/usr/bin/file #{cmd}").readlines.each do |line|
      case line
      when /Mach-O executable ppc/
        archs << :ppc7400
      when /Mach-O 64-bit executable ppc64/
        archs << :ppc64
      when /Mach-O executable i386/
        archs << :i386
      when /Mach-O 64-bit executable x86_64/
        archs << :x86_64
      end
    end

    return archs
end


# replaces before with after for the file path
def inreplace path, before, after
  before=Regexp.escape before.to_s
  before.gsub! "/", "\\/" # I guess not escaped as delimiter varies
  after=after.to_s
  after.gsub! "\\", "\\\\"
  after.gsub! "/", "\\/"
  after.gsub! "$", "\\$"

  # FIXME use proper Ruby for teh exceptions!
  safe_system "/usr/bin/perl", "-pi", "-e", "s/#{before}/#{after}/g", path
end