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:
Michael S. Klishin (author)
Fri May 16 07:15:50 -0700 2008
commit  9d44defce29f7aeb23d8e31745b5fa854a772e76
tree    b63ee4fae7dc5b99d2f89daac93c6eecf652e323
parent  c7c69e8f6a0bce3a1a4ccce13d334fafd6e24685
merb-core / lib / merb-core / core_ext / kernel.rb
100644 322 lines (298 sloc) 10.339 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
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
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}' ...")
      end
    rescue LoadError
      if try_framework
        try_framework = false
        retry
      else
        Merb.logger.info!("loading gem '#{name}' ...")
        # 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
  # use_orm :datamapper
  #
  # # This will use the DataMapper generator for your ORM
  # $ merb-gen model ActivityEvent
  #
  # ==== Notes
  # If for some reason this is called more than once, latter
  # call takes over other.
  def use_orm(orm)
    begin
      register_orm(orm)
      orm_plugin = "merb_#{orm}"
      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
 
 
  # Registers ORM at generator scope.
  #
  # ==== Parameters
  # orm<~to_sym>::
  # ORM alias, like :activerecord, :datamapper or :sequel.
  #--
  # @private
  def register_orm(orm)
    Merb.orm_generator_scope.replace([orm])
  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
  # use_test :rspec
  #
  # # This will now use the RSpec generator for tests
  # $ merb-gen model ActivityEvent
  def use_test(test_framework, *test_dependencies)
    raise "use_test only supports :rspec and :test_unit currently" unless supported_test_framework?(test_framework)
    register_test_framework(test_framework)
 
    dependencies test_dependencies if Merb.env == "test" || Merb.env.nil?
  end
 
  # Check whether Merb supports test framework. Currently Merb has plugins to support RSpec and Test::Unit.
  #
  # ==== Parameters
  # test_framework<Symbol>::
  # The test framework to check. Currently only supports :rspec and :test_unit.
  #--
  # @semi-public
  def supported_test_framework?(test_framework)
    [:rspec, :test_unit].include?(test_framework.to_sym)
  end
 
  # Register test framework at generator scope. Currently Merb has plugins to support RSpec and Test::Unit.
  #
  # ==== Parameters
  # test_framework<Symbol>::
  # The test framework to check. Currently only supports :rspec and :test_unit but the check is performed before registration if you use API.
  #--
  # @private
  def register_test_framework(test_framework)
    Merb.test_framework_generator_scope.replace([test_framework])
  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
  # specified number of times and generates HTML report.
  #
  # ==== 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 in the example above.
  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