82 lines
2.1 KiB
Ruby
82 lines
2.1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Ci
|
|
module Pipeline
|
|
module Expression
|
|
class Lexer
|
|
include ::Gitlab::Utils::StrongMemoize
|
|
|
|
SyntaxError = Class.new(Expression::ExpressionError)
|
|
|
|
LEXEMES = [
|
|
Expression::Lexeme::Variable,
|
|
Expression::Lexeme::String,
|
|
Expression::Lexeme::Pattern,
|
|
Expression::Lexeme::Null,
|
|
Expression::Lexeme::Equals,
|
|
Expression::Lexeme::Matches,
|
|
Expression::Lexeme::NotEquals,
|
|
Expression::Lexeme::NotMatches
|
|
].freeze
|
|
|
|
NEW_LEXEMES = [
|
|
Expression::Lexeme::Variable,
|
|
Expression::Lexeme::String,
|
|
Expression::Lexeme::Pattern,
|
|
Expression::Lexeme::Null,
|
|
Expression::Lexeme::Equals,
|
|
Expression::Lexeme::Matches,
|
|
Expression::Lexeme::NotEquals,
|
|
Expression::Lexeme::NotMatches,
|
|
Expression::Lexeme::And,
|
|
Expression::Lexeme::Or
|
|
].freeze
|
|
|
|
MAX_TOKENS = 100
|
|
|
|
def initialize(statement, max_tokens: MAX_TOKENS)
|
|
@scanner = StringScanner.new(statement)
|
|
@max_tokens = max_tokens
|
|
end
|
|
|
|
def tokens
|
|
strong_memoize(:tokens) { tokenize }
|
|
end
|
|
|
|
def lexemes
|
|
tokens.map(&:to_lexeme)
|
|
end
|
|
|
|
private
|
|
|
|
def tokenize
|
|
tokens = []
|
|
|
|
@max_tokens.times do
|
|
@scanner.skip(/\s+/) # ignore whitespace
|
|
|
|
return tokens if @scanner.eos?
|
|
|
|
lexeme = available_lexemes.find do |type|
|
|
type.scan(@scanner).tap do |token|
|
|
tokens.push(token) if token.present?
|
|
end
|
|
end
|
|
|
|
unless lexeme.present?
|
|
raise Lexer::SyntaxError, 'Unknown lexeme found!'
|
|
end
|
|
end
|
|
|
|
raise Lexer::SyntaxError, 'Too many tokens!'
|
|
end
|
|
|
|
def available_lexemes
|
|
Feature.enabled?(:ci_variables_complex_expressions) ? NEW_LEXEMES : LEXEMES
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|