56 lines
1.3 KiB
Ruby
56 lines
1.3 KiB
Ruby
module Pigeon
|
|
class Parser
|
|
class DuplicateKeyError < StandardError; end
|
|
|
|
def self.parse(db, tokens)
|
|
raise "NO!" unless db.is_a?(Pigeon::Database)
|
|
self.new(db, tokens).parse
|
|
end
|
|
|
|
def initialize(db, tokens)
|
|
reset_scratchpad
|
|
@db = db
|
|
@tokens = tokens
|
|
@results = []
|
|
end
|
|
|
|
def parse()
|
|
@tokens.each_with_index do |token, i|
|
|
case token.first
|
|
when :AUTHOR then set(:author, token.last)
|
|
when :KIND then set(:kind, token.last)
|
|
when :DEPTH then set(:depth, token.last)
|
|
when :PREV then set(:prev, token.last)
|
|
when :LIPMAA then set(:lipmaa, token.last)
|
|
when :HEADER_END then set(:body, {})
|
|
when :BODY_ENTRY then set(token[1], token[2], @scratchpad[:body])
|
|
when :BODY_END then nil
|
|
when :SIGNATURE then set(:signature, token.last)
|
|
when :MESSAGE_END then finish_this_message!
|
|
end
|
|
end
|
|
@results
|
|
end
|
|
|
|
private
|
|
|
|
def reset_scratchpad
|
|
@scratchpad = {}
|
|
end
|
|
|
|
def finish_this_message!
|
|
@scratchpad.freeze
|
|
@results.push(@db.ingest(**@scratchpad))
|
|
reset_scratchpad
|
|
end
|
|
|
|
def set(key, value, hash = @scratchpad)
|
|
if hash[key]
|
|
raise DuplicateKeyError, "Found duplicate keys: #{key}"
|
|
else
|
|
hash[key] = value
|
|
end
|
|
end
|
|
end
|
|
end
|