inspec/lib/utils/nginx_parser.rb
Miah Johnson 659b4b373a Remove # encoding: utf8 magic comments
Signed-off-by: Miah Johnson <miah@chia-pet.org>
2019-05-07 16:06:23 -07:00

104 lines
2.8 KiB
Ruby

# author: Dominik Richter
# author: Christoph Hartmann
require 'parslet'
class NginxParser < Parslet::Parser
root :outermost
# only designed for rabbitmq config files for now:
rule(:outermost) { filler? >> exp.repeat }
rule(:filler?) { one_filler.repeat }
rule(:one_filler) { match('\s+') | match["\n"] | comment }
rule(:space) { match('\s+') }
rule(:comment) { str('#') >> (match["\n\r"].absent? >> any).repeat }
rule(:exp) {
section | assignment
}
rule(:assignment) {
(identifier >> values.maybe.as(:args)).as(:assignment) >> str(';') >> filler?
}
rule(:standard_identifier) {
(match('[a-zA-Z]') >> match('\S').repeat).as(:identifier) >> space >> space.repeat
}
rule(:quoted_identifier) {
str('"') >> (str('"').absent? >> any).repeat.as(:identifier) >> str('"') >> space.repeat
}
rule(:identifier) {
standard_identifier | quoted_identifier
}
rule(:standard_value) {
((match(/[#;{'"]/).absent? >> any) >> (
str('\\') >> any | match('[#;{]|\s').absent? >> any
).repeat).as(:value) >> space.repeat
}
rule(:single_quoted_value) {
str("'") >> (
str('\\') >> any | str("'").absent? >> any
).repeat.as(:value) >> str("'") >> space.repeat
}
rule(:double_quoted_value) {
str('"') >> (
str('\\') >> any | str('"').absent? >> any
).repeat.as(:value) >> str('"') >> space.repeat
}
rule(:quoted_value) {
single_quoted_value | double_quoted_value
}
rule(:value) {
standard_value | quoted_value
}
rule(:values) {
value.repeat >> space.maybe
}
rule(:section) {
identifier.as(:section) >> values.maybe.as(:args) >> str('{') >> filler? >> exp.repeat.as(:expressions) >> str('}') >> filler?
}
end
class NginxTransform < Parslet::Transform
Group = Struct.new(:id, :args, :body)
Exp = Struct.new(:key, :vals)
def self.assemble_binary(seq)
b = ErlangBitstream.new
seq.each { |i| b.add(i) }
b.value
end
rule(section: { identifier: simple(:x) }, args: subtree(:y), expressions: subtree(:z)) { Group.new(x.to_s, y, z) }
rule(assignment: { identifier: simple(:x), args: subtree(:y) }) { Exp.new(x.to_s, y) }
rule(value: simple(:x)) { x.to_s }
end
class NginxConfig
def self.parse(content)
lex = NginxParser.new.parse(content)
tree = NginxTransform.new.apply(lex)
gtree = NginxTransform::Group.new(nil, '', tree)
read_nginx_group(gtree)
rescue Parslet::ParseFailed => err
raise "Failed to parse NginX config: #{err}"
end
def self.read_nginx_group(t)
agg_conf = Hash.new([])
agg_conf['_'] = t.args unless t.args == ''
groups, conf = t.body.partition { |i| i.is_a? NginxTransform::Group }
conf.each { |x| agg_conf[x.key] += [x.vals] }
groups.each { |x| agg_conf[x.id] += [read_nginx_group(x)] }
agg_conf
end
end