/
engine_tracking.rb
102 lines (82 loc) · 2.4 KB
/
engine_tracking.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
# frozen_string_literal: true
require 'sinatra/base'
module Sinatra
# Adds methods like `haml?` that allow helper methods to check whether they
# are called from within a template.
module EngineTracking
attr_reader :current_engine
# @return [Boolean] Returns true if current engine is `:erb`.
def erb?
@current_engine == :erb
end
# Returns true if the current engine is `:erubi`, or `Tilt[:erb]` is set
# to Tilt::ErubiTemplate.
#
# @return [Boolean] Returns true if current engine is `:erubi`.
def erubi?
@current_engine == :erubi or
(erb? && Tilt[:erb] == Tilt::ErubiTemplate)
end
# @return [Boolean] Returns true if current engine is `:haml`.
def haml?
@current_engine == :haml
end
# @return [Boolean] Returns true if current engine is `:sass`.
def sass?
@current_engine == :sass
end
# @return [Boolean] Returns true if current engine is `:scss`.
def scss?
@current_engine == :scss
end
# @return [Boolean] Returns true if current engine is `:builder`.
def builder?
@current_engine == :builder
end
# @return [Boolean] Returns true if current engine is `:liquid`.
def liquid?
@current_engine == :liquid
end
# @return [Boolean] Returns true if current engine is `:markdown`.
def markdown?
@current_engine == :markdown
end
# @return [Boolean] Returns true if current engine is `:rdoc`.
def rdoc?
@current_engine == :rdoc
end
# @return [Boolean] Returns true if current engine is `:markaby`.
def markaby?
@current_engine == :markaby
end
# @return [Boolean] Returns true if current engine is `:nokogiri`.
def nokogiri?
@current_engine == :nokogiri
end
# @return [Boolean] Returns true if current engine is `:slim`.
def slim?
@current_engine == :slim
end
# @return [Boolean] Returns true if current engine is `:ruby`.
def ruby?
@current_engine == :ruby
end
def initialize(*)
@current_engine = :ruby
super
end
# @param engine [Symbol, String] Name of Engine to shift to.
def with_engine(engine)
engine_was = @current_engine
@current_engine = engine.to_sym
yield
ensure
@current_engine = engine_was
end
private
def render(engine, *)
with_engine(engine) { super }
end
end
helpers EngineTracking
end