sul-dlss/was_robot_suite

# frozen_string_literal: true

require 'zip'

module Dor
  module WasCrawl
    # Extracts WARC files from a WACZ and delete
    class WarcExtractorService
      def self.extract(base_path, wacz_filename)
        new(base_path, wacz_filename).extract
      end

      def initialize(base_path, wacz_filename)
        @base_path = base_path
        @wacz_filename = wacz_filename
      end

      def extract
        Zip::File.open(wacz_filepath) do |wacz_file|
          wacz_file.glob('archive/*.warc.gz').each do |warc_entry|
            filename = warc_entry.name.delete_prefix('archive/')
            # Prefixing with WACZ filename to make unique.
            warc_entry.extract(File.join(base_path, "#{wacz_basename}-#{filename}"))
          end
        end
        File.delete(wacz_filepath)
      end

      private

      attr_reader :base_path, :wacz_filename

      def wacz_filepath
        File.join(base_path, wacz_filename)
      end

      def wacz_basename
        @wacz_basename ||= File.basename(wacz_filename, '.wacz')
      end
    end
  end
end