-
Notifications
You must be signed in to change notification settings - Fork 1
/
utils.rb
126 lines (111 loc) · 3.69 KB
/
utils.rb
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
# utils.rb
# Created by John Clayton on 1/25/2010.
# Copyright 2010 Fivesquare Software, LLC. All rights reserved.
# Copyright 2011 Barnes & Noble. All rights reserved.
# 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.
#
# 3. Neither the name of <#organization#> nor the names of its contributors may
# be used to endorse or promote products derived from this software without
# specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "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 ICONFACTORY 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.
require 'yaml'
require 'ostruct'
class << YAML::DefaultResolver
alias_method :_node_import, :node_import
def node_import(node)
o = _node_import(node)
o.is_a?(Hash) ? ::OpenStruct.new(o) : o
end
end
class Array
def random
self[Kernel.rand(self.size)]
end
end
module Utils
module Shell
def self.do(command,&block)
output = `#{command}`
retval = $?
if block_given?
yield(retval == 0, output, command)
end
end
def shell(command,&block)
Shell.do(command,&block)
end
end
module Interface
def ask(question,options=[])
prompt = "% #{question} (#{options.join('')}): "
sync = STDOUT.sync
begin
STDOUT.sync = true
print(prompt)
STDIN.gets.chomp
ensure
STDOUT.sync = sync
end
end
end
module Configuration
attr_accessor :config
# Loads a yaml file by the given path
# If can't find the file at the given path:
# - tries ./file
# - tries ~/file
# - tries /etc/file
# - tries ./config.yml
# - gives up
def load_config(config_file, required=false)
begin
file_path = File.expand_path(config_file)
if File.exist?(file_path)
File.open(file_path) { |f| self.config = YAML.load(f) }
else
basename = File.basename(config_file)
if File.exists?("./#{basename}")
config_file = "./#{basename}"
elsif File.exist?("~/#{basename}")
config_file = "~/#{basename}"
elsif File.exist?("/etc/#{basename}")
config_file = "/etc/#{basename}"
elsif File.exist?("./config.yml")
config_file = "./config.yml"
elsif required
raise "Could not locate config file #{config_file} anywhere"
end
# No file loaded, and require is false, load new file or create empty config
file_path = File.expand_path(config_file)
if File.exist?(file_path)
File.open(file_path) { |f| self.config = YAML.load(f) }
else
puts "Not loading #{config_file}"
self.config = OpenStruct.new
end
end
file_path
rescue
raise "#{file_path} did not load properly (#{$!})"
end
end
end
end