discourse/lib/compression/engine.rb

# frozen_string_literal: true

module Compression
  class Engine
    UnsupportedFileExtension = Class.new(StandardError)

    def self.default_strategies
      [
        Compression::Zip.new,
        Compression::Pipeline.new([Compression::Tar.new, Compression::Gzip.new]),
        Compression::Gzip.new,
        Compression::Tar.new
      ]
    end

    def self.engine_for(filename, strategies: default_strategies)
      strategy = strategies.detect(-> { raise UnsupportedFileExtension }) { |e| e.can_handle?(filename) }
      new(strategy)
    end

    def initialize(strategy)
      @strategy = strategy
    end

    def decompress(dest_path, compressed_file_path, allow_non_root_folder: false)
      @strategy.decompress(dest_path, compressed_file_path, allow_non_root_folder: false)
    end

    def compress(path, target_name)
      @strategy.compress(path, target_name)
    end
  end
end