# frozen_string_literal: true

require "helper"
require "json"

RSpec.describe HTTP2::Header do
  folders = %w[
    go-hpack
    haskell-http2-diff
    haskell-http2-diff-huffman
    haskell-http2-linear
    haskell-http2-linear-huffman
    haskell-http2-naive
    haskell-http2-naive-huffman
    haskell-http2-static
    haskell-http2-static-huffman
    #hyper-hpack
    nghttp2
    nghttp2-16384-4096
    nghttp2-change-table-size
    node-http2-hpack
  ]

  context "Decompressor" do
    folders.each do |folder|
      next if folder.include?("#")

      path = File.expand_path("hpack-test-case/#{folder}", File.dirname(__FILE__))
      next unless Dir.exist?(path)

      context folder.to_s do
        Dir.foreach(path) do |file|
          next unless file.include?(".json")

          it "should decode #{file}" do
            story = JSON.parse(File.read("#{path}/#{file}"))
            cases = story["cases"]
            table_size = cases[0]["header_table_size"] || 4096
            @dc = Decompressor.new(table_size: table_size)
            cases.each do |c|
              wire = [c["wire"]].pack("H*").force_encoding(Encoding::BINARY)
              @emitted = @dc.decode(HTTP2::Buffer.new(wire))
              headers = c["headers"].flat_map(&:to_a)
              expect(@emitted).to eq headers
            end
          end
        end
      end
    end
  end

  context "Compressor" do
    %w[
      LINEAR
      NAIVE
      SHORTER
      STATIC
    ].each do |mode|
      next if mode.include?("#")

      ["", "H"].each do |huffman|
        encoding_mode = :"#{mode}#{huffman}"
        encoding_options = HTTP2::Header::EncodingContext.const_get(encoding_mode)
        [4096, 512].each do |table_size|
          options = { table_size: table_size }
          options.update(encoding_options)

          context "with #{mode}#{huffman} mode and table_size #{table_size}" do
            path = File.expand_path("hpack-test-case/raw-data", File.dirname(__FILE__))
            Dir.foreach(path) do |file|
              next unless file.include?(".json")

              it "should encode #{file}" do
                story = JSON.parse(File.read("#{path}/#{file}"))
                cases = story["cases"]
                @cc = Compressor.new(options)
                @dc = Decompressor.new(options)
                cases.each do |c|
                  headers = c["headers"].flat_map(&:to_a)
                  wire = @cc.encode(headers)
                  decoded = @dc.decode(HTTP2::Buffer.new(wire))
                  expect(decoded).to eq headers
                end
              end
            end
          end
        end
      end
    end
  end
end