Module: QueryReviewer::ControllerExtensions

Defined in:
lib/query_reviewer/controller_extensions.rb

Defined Under Namespace

Classes: QueryViewBase

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.included(base) ⇒ Object 



10
11
12
13
14
15
16
17
 
# File 'lib/query_reviewer/controller_extensions.rb', line 10

def self.included(base)
  if QueryReviewer::CONFIGURATION["inject_view"]
    alias_name = defined?(Rails::Railtie) ? :process_action : :perform_action
    base.alias_method_chain(alias_name, :query_review)
  end
  base.alias_method_chain :process, :query_review
  base.helper_method :query_review_output
end

Instance Method Details

#add_query_output_to_view(total_time) ⇒ Object 



32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
 
# File 'lib/query_reviewer/controller_extensions.rb', line 32

def add_query_output_to_view(total_time)
  if request.xhr?
    if cookies["query_review_enabled"]
      if !response.content_type || response.content_type.include?("text/html")
        response.body += "<script type=\"text/javascript\">"+query_review_output(true, total_time)+"</script>"
      elsif response.content_type && response.content_type.include?("text/javascript")
        response.body += ";\n"+query_review_output(true, total_time)
      end
    end
  else
    if !response.instance_eval{ @body.is_a?(File) || @body.is_a?(Enumerator) } &&
         response.body.is_a?(String) && response.body.match(/<\/body>/i)
      idx = (response.body =~ /<\/body>/i)
      html = query_review_output(false, total_time)
      response.body = response.body.insert(idx, html)
    end
  end
end

#perform_action_with_query_review(*args) ⇒ Object Also known as: process_action_with_query_review 



51
52
53
54
55
56
57
58
59
 
# File 'lib/query_reviewer/controller_extensions.rb', line 51

def perform_action_with_query_review(*args)
  Thread.current["query_reviewer_enabled"] = query_reviewer_output_enabled?

  t1 = Time.now
  r = defined?(Rails::Railtie) ? process_action_without_query_review(*args) : perform_action_without_query_review(*args)
  t2 = Time.now
  add_query_output_to_view(t2 - t1)
  r
end

#process_with_query_review(*args) ⇒ Object

:nodoc:



62
63
64
65
 
# File 'lib/query_reviewer/controller_extensions.rb', line 62

def process_with_query_review(*args) #:nodoc:
  Thread.current["queries"] = SqlQueryCollection.new
  process_without_query_review(*args)
end

#query_review_output(ajax = false, total_time = nil) ⇒ Object 



19
20
21
22
23
24
25
26
27
28
29
30
 
# File 'lib/query_reviewer/controller_extensions.rb', line 19

def query_review_output(ajax = false, total_time = nil)
  faux_view = QueryViewBase.new([File.join(File.dirname(__FILE__), "views")], {}, self)
  queries = Thread.current["queries"]
  queries.analyze!
  faux_view.instance_variable_set("@queries", queries)
  faux_view.instance_variable_set("@total_time", total_time)
  if ajax
    js = faux_view.render(:partial => "/box_ajax.js")
  else
    html = faux_view.render(:partial => "/box")
  end
end

#query_reviewer_output_enabled? ⇒ Boolean

Returns . Returns whether or not the user has indicated that query_reviewer output is enabled.

Returns:

  • (Boolean) —

    . Returns whether or not the user has indicated that query_reviewer output is enabled



68
69
70
71
72
 
# File 'lib/query_reviewer/controller_extensions.rb', line 68

def query_reviewer_output_enabled?
  cookie_enabled = (CONFIGURATION["enabled"] == true and cookies["query_review_enabled"])
  session_enabled = (CONFIGURATION["enabled"] == "based_on_session" and session["query_review_enabled"])
  return cookie_enabled || session_enabled
end