lib/kpm/sha1_checker.rb



# frozen_string_literal: true

require 'logger'
require 'yaml'
require 'pathname'

module KPM
  class Sha1Checker
    def self.from_file(sha1_file, logger = nil)
      Sha1Checker.new(sha1_file, logger)
    end

    def initialize(sha1_file, logger = nil)
      @sha1_file = sha1_file
      init!

      if logger.nil?
        @logger = Logger.new(STDOUT)
        @logger.level = Logger::INFO
      else
        @logger = logger
      end
    end

    def sha1(coordinates)
      sha1_cache[coordinates]
    end

    def all_sha1
      sha1_cache
    end

    def add_or_modify_entry!(coordinates, remote_sha1)
      sha1_cache[coordinates] = remote_sha1
      save!
    end

    def remove_entry!(coordinates)
      sha1_cache.delete(coordinates)
      nexus_cache.delete(coordinates)
      save!
    end

    def artifact_info(coordinates)
      nexus_cache[coordinates]
    end

    def cache_artifact_info(coordinates_with_maybe_latest, artifact_info)
      return if artifact_info.nil?

      if coordinates_with_maybe_latest.end_with?('LATEST')
        return nil if artifact_info[:version].nil?

        coordinates = coordinates_with_maybe_latest.gsub(/LATEST$/, artifact_info[:version])
      else
        coordinates = coordinates_with_maybe_latest
      end

      # See BaseArtifact#artifact_info
      nexus_keys = %i[sha1 version repository_path is_tgz]
      nexus_cache[coordinates] = artifact_info.select { |key, _| nexus_keys.include? key }
      save!
    end

    def killbill_info(version)
      killbill_cache[version]
    end

    def cache_killbill_info(version, dependencies)
      killbill_cache[version] = dependencies
      save!
    end

    private

    def sha1_cache
      @sha1_config['sha1'] ||= {}
    end

    def nexus_cache
      @sha1_config['nexus'] ||= {}
    end

    def killbill_cache
      @sha1_config['killbill'] ||= {}
    end

    def save!
      Dir.mktmpdir do |tmp_destination_dir|
        tmp_file = File.join(tmp_destination_dir, File.basename(@sha1_file))
        File.open(tmp_file, 'w') do |file|
          file.write(@sha1_config.to_yaml)
        end
        FileUtils.copy(tmp_file, @sha1_file)
      end
      reload!
    end

    def init!
      unless File.exist?(@sha1_file)
        create_sha1_directory_if_missing
        init_config = {}
        init_config['sha1'] = {}
        File.open(@sha1_file, 'w') do |file|
          file.write(init_config.to_yaml)
        end
      end
      reload!
    end

    def create_sha1_directory_if_missing
      sha1_dir = Pathname(@sha1_file).dirname
      FileUtils.mkdir_p(sha1_dir) unless File.directory?(sha1_dir)
    end

    def reload!
      @sha1_config = YAML.load_file(@sha1_file)
    end
  end
end