public
Rubygem
Description: Merb Core: All you need. None you don't.
Homepage: http://www.merbivore.com
Clone URL: git://github.com/wycats/merb-core.git
Search Repo:
ezmobius (author)
Sun Mar 23 15:07:39 -0700 2008
commit  312181aade867323834c5031b7e9707f931e9ded
tree    6280af600d4c35194850e2d3476d431de40c1d54
parent  142d85ca7cf7c7a4f5a93b09be12978b39059125
merb-core / lib / merb-core / core_ext / kernel.rb
100644 304 lines (283 sloc) 10.117 kb
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
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
module Kernel
  # Loads the given string as a gem. Execution is deferred to
  # the Merb::BootLoader::Dependencies.run during bootup.
  #
  # ==== Parameters
  # name<String>:: The name of the gem to load.
  # *ver<Gem::Requirement, Gem::Version, Array, ~to_str>::
  # Version requirements to be passed to Gem.activate.
  def dependency(name, *ver)
    Merb::BootLoader::Dependencies.dependencies << [name, ver]
  end
  
  # Loads the given string as a gem.
  #
  # ==== Parameters
  # name<String>:: The name of the gem to load.
  # *ver<Gem::Requirement, Gem::Version, Array, ~to_str>::
  # Version requirements to be passed to Gem.activate.
  #
  # ==== Notes
  # If the gem cannot be found, the method will attempt to require the string
  # as a library.
  #
  # This new version tries to load the file via ROOT/gems first before moving
  # off to the system gems (so if you have a lower version of a gem in
  # ROOT/gems, it'll still get loaded).
  def load_dependency(name, *ver)
    try_framework = Merb.frozen?
    begin
      # If this is a piece of merb, and we're frozen, try to require
      # first, so we can pick it up from framework/,
      # otherwise try activating the gem
      if name =~ /^merb/ && try_framework
        require name
      else
        gem(name, *ver) if ver
        require name
        Merb.logger.info!("loading gem '#{name}' from #{__app_file_trace__.first} ...")
      end
    rescue LoadError
      if try_framework
        try_framework = false
        retry
      else
        Merb.logger.info!("loading gem '#{name}' from #{__app_file_trace__.first} ...")
        # Failed requiring as a gem, let's try loading with a normal require.
        require name
      end
    end
  end
 
  # Loads both gem and library dependencies that are passed in as arguments.
  # Execution is deferred to the Merb::BootLoader::Dependencies.run during bootup.
  #
  # ==== Parameters
  # *args<String, Hash, Array>:: The dependencies to load.
  def dependencies(*args)
    args.each do |arg|
      case arg
      when String then dependency(arg)
      when Hash then arg.each { |r,v| dependency(r, v) }
      when Array then arg.each { |r| dependency(r) }
      end
    end
  end
  
  # Loads both gem and library dependencies that are passed in as arguments.
  #
  # ==== Parameters
  # *args<String, Hash, Array>:: The dependencies to load.
  #
  # ==== Notes
  # Each argument can be:
  # String:: Single dependency.
  # Hash::
  # Multiple dependencies where the keys are names and the values versions.
  # Array:: Multiple string dependencies.
  #
  # ==== Examples
  # dependencies "RedCloth" # Loads the the RedCloth gem
  # dependencies "RedCloth", "merb_helpers" # Loads RedCloth and merb_helpers
  # dependencies "RedCloth" => "3.0" # Loads RedCloth 3.0
  def load_dependencies(*args)
    args.each do |arg|
      case arg
      when String then load_dependency(arg)
      when Hash then arg.each { |r,v| load_dependency(r, v) }
      when Array then arg.each { |r| load_dependency(r) }
      end
    end
  end
  
  # Does a basic require, and prints a message if an error occurs.
  #
  # ==== Parameters
  # library<~to_s>:: The library to attempt to include.
  # message<String>:: The error to add to the log upon failure. Defaults to nil.
  def rescue_require(library, message = nil)
    require library
  rescue LoadError, RuntimeError
    Merb.logger.error!(message) if message
  end
  
  # Used in Merb.root/config/init.rb to tell Merb which ORM (Object Relational
  # Mapper) you wish to use. Currently Merb has plugins to support
  # ActiveRecord, DataMapper, and Sequel.
  #
  # ==== Parameters
  # orm<~to_s>:: The ORM to use.
  #
  # ==== Examples
  # # This line goes in dependencies.yml
  # use_orm :datamapper
  #
  # # This will use the DataMapper generator for your ORM
  # $ ruby script/generate model MyModel
  def use_orm(orm)
    if !Merb.generator_scope.include?(:merb_default) && !Merb.generator_scope.include?(orm.to_sym)
      raise "Don't call use_orm more than once"
    end
    begin
      Merb.generator_scope.delete(:merb_default)
      orm_plugin = orm.to_s.match(/^merb_/) ? orm.to_s : "merb_#{orm}"
      Merb.generator_scope.unshift(orm.to_sym) unless Merb.generator_scope.include?(orm.to_sym)
      Kernel.dependency(orm_plugin)
    rescue LoadError => e
      Merb.logger.warn!("The #{orm_plugin} gem was not found. You may need to install it.")
      raise e
    end
  end
  
  # Used in Merb.root/config/init.rb to tell Merb which testing framework to
  # use. Currently Merb has plugins to support RSpec and Test::Unit.
  #
  # ==== Parameters
  # test_framework<Symbol>::
  # The test framework to use. Currently only supports :rspec and :test_unit.
  #
  # ==== Examples
  # # This line goes in dependencies.yml
  # use_test :rspec
  #
  # # This will now use the RSpec generator for tests
  # $ ruby script/generate controller MyController
  def use_test(test_framework, *test_dependencies)
    raise "use_test only supports :rspec and :test_unit currently" unless
      [:rspec, :test_unit].include?(test_framework.to_sym)
    Merb.generator_scope.delete(:rspec)
    Merb.generator_scope.delete(:test_unit)
    Merb.generator_scope.push(test_framework.to_sym)
    
    dependencies test_dependencies if Merb.env == "test" || Merb.env.nil?
  end
  
  # ==== Returns
  # Array[String]:: A stack trace of the applications files.
  def __app_file_trace__
    caller.select do |call|
      call.include?(Merb.root) && !call.include?(Merb.root + "/framework")
    end.map do |call|
      file, line = call.scan(Regexp.new("#{Merb.root}/(.*):(.*)")).first
      "#{file}:#{line}"
    end
  end
 
  # ==== Parameters
  # i<Fixnum>:: The caller number. Defaults to 1.
  #
  # ==== Returns
  # Array[Array]:: The file, line and method of the caller.
  #
  # ==== Examples
  # __caller_info__(1)
  # # => ['/usr/lib/ruby/1.8/irb/workspace.rb', '52', 'irb_binding']
  def __caller_info__(i = 1)
    file, line, meth = caller[i].scan(/(.*?):(\d+):in `(.*?)'/).first
  end
 
  # ==== Parameters
  # file<String>:: The file to read.
  # line<Fixnum>:: The line number to look for.
  # size<Fixnum>::
  # Number of lines to include above and below the the line to look for.
  # Defaults to 4.
  #
  # ==== Returns
  # Array[Array]::
  # Triplets containing the line number, the line and whether this was the
  # searched line.
  #
  # ==== Examples
  # __caller_lines__('/usr/lib/ruby/1.8/debug.rb', 122, 2) # =>
  # [
  # [ 120, " def check_suspend", false ],
  # [ 121, " return if Thread.critical", false ],
  # [ 122, " while (Thread.critical = true; @suspend_next)", true ],
  # [ 123, " DEBUGGER__.waiting.push Thread.current", false ],
  # [ 124, " @suspend_next = false", false ]
  # ]
  def __caller_lines__(file, line, size = 4)
    return [['Template Error!', "problem while rendering", false]] if file =~ /\(erubis\)/
    lines = File.readlines(file)
    current = line.to_i - 1
 
    first = current - size
    first = first < 0 ? 0 : first
 
    last = current + size
    last = last > lines.size ? lines.size : last
 
    log = lines[first..last]
 
    area = []
 
    log.each_with_index do |line, index|
      index = index + first + 1
      area << [index, line.chomp, index == current + 1]
    end
 
    area
  end
  
  # Takes a block, profiles the results of running the block 100 times and
  # writes out the results in a file.
  #
  # ==== Parameters
  # name<~to_s>::
  # The file name. The result will be written out to
  # Merb.root/"log/#{name}.html".
  # min<Fixnum>::
  # Minimum percentage of the total time a method must take for it to be
  # included in the result. Defaults to 1.
  #
  # ==== Returns
  # String:: The result of the profiling.
  #
  # ==== Notes
  # Requires ruby-prof (<tt>sudo gem install ruby-prof</tt>)
  #
  # ==== Examples
  # __profile__("MyProfile", 5, 30) do
  # rand(10)**rand(10)
  # puts "Profile run"
  # end
  #
  # Assuming that the total time taken for #puts calls was less than 5% of the
  # total time to run, #puts won't appear in the profile report.
  # The code block will be run 30 times.
  def __profile__(name, min=1, iter=100)
    require 'ruby-prof' unless defined?(RubyProf)
    return_result = ''
    result = RubyProf.profile do
      iter.times{return_result = yield}
    end
    printer = RubyProf::GraphHtmlPrinter.new(result)
    path = File.join(Merb.root, 'log', "#{name}.html")
    File.open(path, 'w') do |file|
     printer.print(file, {:min_percent => min,
                          :print_file => true})
    end
    return_result
  end
  
  # Extracts an options hash if it is the last item in the args array. Used
  # internally in methods that take *args.
  #
  # ==== Parameters
  # args<Array>:: The arguments to extract the hash from.
  #
  # ==== Examples
  # def render(*args,&blk)
  # opts = extract_options_from_args!(args) || {}
  # # [...]
  # end
  def extract_options_from_args!(args)
    args.pop if Hash === args.last
  end
  
  # Checks that the given objects quack like the given conditions.
  #
  # ==== Parameters
  # opts<Hash>::
  # Conditions to enforce. Each key will receive a quacks_like? call with the
  # value (see Object#quacks_like? for details).
  #
  # ==== Raises
  # ArgumentError:: An object failed to quack like a condition.
  def enforce!(opts = {})
    opts.each do |k,v|
      raise ArgumentError, "#{k.inspect} doesn't quack like #{v.inspect}" unless k.quacks_like?(v)
    end
  end
  
  unless Kernel.respond_to?(:debugger)
 
    # Define debugger method so that code even works if debugger was not
    # requested. Drops a note to the logs that Debugger was not available.
    def debugger
       Merb.logger.info! "\n***** Debugger requested, but was not " +
                        "available: Start server with --debugger " +
                        "to enable *****\n"
    end
  end
end