module Prism

def self.lex_compat(source, **options)
  LexCompat.new(source, **options).result
end

def self.lex_ripper(source)
  LexRipper.new(source).result
end

def self.load(source, serialized)
  Serialize.load(source, serialized)
end

def self.parse_failure?(source, **options)
  !parse_success?(source, **options)
end

def self.parse_file_failure?(filepath, **options)
  !parse_file_success?(filepath, **options)
end

def dump(code, **options)
  LibRubyParser::PrismBuffer.with do |buffer|
    LibRubyParser.pm_serialize_parse(buffer.pointer, code, code.bytesize, dump_options(options))
    buffer.read
  end
end

def dump_file(filepath, **options)
  LibRubyParser::PrismString.with(filepath) do |string|
    dump(string.read, **options, filepath: filepath)
  end
end

def dump_options(options)
  template = +""
  values = []
  template << "L"
  if (filepath = options[:filepath])
    values.push(filepath.bytesize, filepath.b)
    template << "A*"
  else
    values << 0
  end
  template << "l"
  values << options.fetch(:line, 1)
  template << "L"
  if (encoding = options[:encoding])
    name = encoding.name
    values.push(name.bytesize, name.b)
    template << "A*"
  else
    values << 0
  end
  template << "C"
  values << (options.fetch(:frozen_string_literal, false) ? 1 : 0)
  template << "C"
  values << { nil => 0, "3.3.0" => 1, "latest" => 0 }.fetch(options[:version])
  template << "L"
  if (scopes = options[:scopes])
    values << scopes.length
    scopes.each do |scope|
      template << "L"
      values << scope.length
      scope.each do |local|
        name = local.name
        template << "L"
        values << name.bytesize
        template << "A*"
        values << name.b
      end
    end
  else
    values << 0
  end
  values.pack(template)
end

def lex(code, **options)
  LibRubyParser::PrismBuffer.with do |buffer|
    LibRubyParser.pm_serialize_lex(buffer.pointer, code, code.bytesize, dump_options(options))
    Serialize.load_tokens(Source.new(code), buffer.read)
  end
end

def lex_file(filepath, **options)
  LibRubyParser::PrismString.with(filepath) do |string|
    lex(string.read, **options, filepath: filepath)
  end
end

def parse(code, **options)
  Prism.load(code, dump(code, **options))
end

def parse_comments(code, **options)
  LibRubyParser::PrismBuffer.with do |buffer|
    LibRubyParser.pm_serialize_parse_comments(buffer.pointer, code, code.bytesize, dump_options(options))
    source = Source.new(code)
    loader = Serialize::Loader.new(source, buffer.read)
    loader.load_header
    loader.load_encoding
    loader.load_start_line
    loader.load_comments
  end
end

def parse_file(filepath, **options)
  LibRubyParser::PrismString.with(filepath) do |string|
    parse(string.read, **options, filepath: filepath)
  end
end

def parse_file_comments(filepath, **options)
  LibRubyParser::PrismString.with(filepath) do |string|
    parse_comments(string.read, **options, filepath: filepath)
  end
end

def parse_file_success?(filepath, **options)
  LibRubyParser::PrismString.with(filepath) do |string|
    parse_success?(string.read, **options, filepath: filepath)
  end
end

def parse_lex(code, **options)
  LibRubyParser::PrismBuffer.with do |buffer|
    LibRubyParser.pm_serialize_parse_lex(buffer.pointer, code, code.bytesize, dump_options(options))
    source = Source.new(code)
    loader = Serialize::Loader.new(source, buffer.read)
    tokens = loader.load_tokens
    node, comments, magic_comments, data_loc, errors, warnings = loader.load_nodes
    tokens.each { |token,| token.value.force_encoding(loader.encoding) }
    ParseResult.new([node, tokens], comments, magic_comments, data_loc, errors, warnings, source)
  end
end

def parse_lex_file(filepath, **options)
  LibRubyParser::PrismString.with(filepath) do |string|
    parse_lex(string.read, **options, filepath: filepath)
  end
end

def parse_success?(code, **options)
  LibRubyParser.pm_parse_success_p(code, code.bytesize, dump_options(options))
end