mirror of
https://github.com/haml/haml.git
synced 2022-11-09 12:33:31 -05:00
29 lines
902 B
Ruby
29 lines
902 B
Ruby
# frozen_string_literal: true
|
|
module Haml
|
|
# 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])
|
|
else
|
|
raise UnexpectedExpression.new("[HAML BUG] Unexpected Temple expression '#{type}' is given!")
|
|
end
|
|
end
|
|
end
|
|
end
|