106 lines
3 KiB
Ruby
106 lines
3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Ci
|
|
module Pipeline
|
|
module Expression
|
|
class Parser
|
|
ParseError = Class.new(Expression::ExpressionError)
|
|
|
|
def initialize(tokens)
|
|
@tokens = tokens.to_enum
|
|
@nodes = []
|
|
end
|
|
|
|
def tree
|
|
results = []
|
|
|
|
tokens =
|
|
if ::Gitlab::Ci::Features.ci_if_parenthesis_enabled?
|
|
tokens_rpn
|
|
else
|
|
legacy_tokens_rpn
|
|
end
|
|
|
|
tokens.each do |token|
|
|
case token.type
|
|
when :value
|
|
results.push(token.build)
|
|
when :logical_operator
|
|
right_operand = results.pop
|
|
left_operand = results.pop
|
|
|
|
token.build(left_operand, right_operand).tap do |res|
|
|
results.push(res)
|
|
end
|
|
else
|
|
raise ParseError, "Unprocessable token found in parse tree: #{token.type}"
|
|
end
|
|
end
|
|
|
|
raise ParseError, 'Unreachable nodes in parse tree' if results.count > 1
|
|
raise ParseError, 'Empty parse tree' if results.count < 1
|
|
|
|
results.pop
|
|
end
|
|
|
|
def self.seed(statement)
|
|
new(Expression::Lexer.new(statement).tokens)
|
|
end
|
|
|
|
private
|
|
|
|
# Parse the expression into Reverse Polish Notation
|
|
# (See: Shunting-yard algorithm)
|
|
# Taken from: https://en.wikipedia.org/wiki/Shunting-yard_algorithm#The_algorithm_in_detail
|
|
def tokens_rpn
|
|
output = []
|
|
operators = []
|
|
|
|
@tokens.each do |token|
|
|
case token.type
|
|
when :value
|
|
output.push(token)
|
|
when :logical_operator
|
|
output.push(operators.pop) while token.lexeme.consume?(operators.last&.lexeme)
|
|
|
|
operators.push(token)
|
|
when :parenthesis_open
|
|
operators.push(token)
|
|
when :parenthesis_close
|
|
output.push(operators.pop) while token.lexeme.consume?(operators.last&.lexeme)
|
|
|
|
raise ParseError, 'Unmatched parenthesis' unless operators.last
|
|
|
|
operators.pop if operators.last.lexeme.type == :parenthesis_open
|
|
end
|
|
end
|
|
|
|
output.concat(operators.reverse)
|
|
end
|
|
|
|
# To be removed with `ci_if_parenthesis_enabled`
|
|
def legacy_tokens_rpn
|
|
output = []
|
|
operators = []
|
|
|
|
@tokens.each do |token|
|
|
case token.type
|
|
when :value
|
|
output.push(token)
|
|
when :logical_operator
|
|
if operators.any? && token.lexeme.precedence >= operators.last.lexeme.precedence
|
|
output.push(operators.pop)
|
|
end
|
|
|
|
operators.push(token)
|
|
end
|
|
end
|
|
|
|
output.concat(operators.reverse)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|