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
|
# frozen_string_literal: true
module Hamlit
# A module to count lines of expected code. This would be faster than actual code generation
# and counting newlines in it.
module TempleLineCounter
class UnexpectedExpression < StandardError; end
def self.count_lines(exp)
type, *args = exp
case type
when :multi
args.map { |a| count_lines(a) }.reduce(:+) || 0
when :dynamic, :code
args.first.count("\n")
when :static
0 # It has not real newline "\n" but escaped "\\n".
when :case
arg, *cases = args
arg.count("\n") + cases.map do |cond, e|
(cond == :else ? 0 : cond.count("\n")) + count_lines(e)
end.reduce(:+)
when :escape
count_lines(args[1])
when :newline
1
else
raise UnexpectedExpression.new("[HAML BUG] Unexpected Temple expression '#{type}' is given!")
end
end
end
end
|
