class Inspec::Lockfile

def self.from_dependency_set(dep_set)
  lockfile_content = {
    'lockfile_version' => CURRENT_LOCKFILE_VERSION,
    'depends' => dep_set.to_array,
  }
  new(lockfile_content)
end

def self.from_file(path)
  parsed_content = YAML.load(File.read(path))
  version = parsed_content['lockfile_version']
  fail "No lockfile_version set in #{path}!" if version.nil?
  validate_lockfile_version!(version.to_i)
  new(parsed_content)
end

def self.validate_lockfile_version!(version)
  if version < MINIMUM_SUPPORTED_VERSION
    fail <<EOF
 lockfile specifies a lockfile_version of #{version} which is
r than the minimum supported version #{MINIMUM_SUPPORTED_VERSION}.
se create a new lockfile for this project by running:
inspec vendor
  elsif version > CURRENT_LOCKFILE_VERSION
    fail <<EOF
 lockfile claims to be version #{version} which is greater than
most recent lockfile version(#{CURRENT_LOCKFILE_VERSION}).
 may happen if you are using an older version of inspec than was
 to create the lockfile.
  end
end

def initialize(lockfile_content_hash)
  version = lockfile_content_hash['lockfile_version']
  @version = version.to_i
  parse_content_hash(lockfile_content_hash)
end

def mutate_hash_keys_with(hash, fun)
  hash.each_with_object({}) do |v, memo|
    key = fun.call(v[0])
    value = if v[1].is_a?(Hash)
              mutate_hash_keys_with(v[1], fun)
            elsif v[1].is_a?(Array)
              v[1].map do |i|
                i.is_a?(Hash) ? mutate_hash_keys_with(i, fun) : i
              end
            else
              v[1]
            end
    memo[key] = value
  end
end

def parse_content_hash(lockfile_content_hash)
  case version
  when 1
    parse_content_hash_1(lockfile_content_hash)
  else
    # If we've gotten here, there is likely a mistake in the
    # lockfile version validation in the constructor.
    fail "No lockfile parser for version #{version}"
  end
end

def parse_content_hash_1(lockfile_content_hash)
  @deps = if lockfile_content_hash['depends']
            lockfile_content_hash['depends'].map { |i| symbolize_keys(i) }
          end
end

def stringify_keys(hash)
  mutate_hash_keys_with(hash, proc { |i| i.to_s })
end

def symbolize_keys(hash)
  mutate_hash_keys_with(hash, proc { |i| i.to_sym })
end

def to_yaml
  {
    'lockfile_version' => CURRENT_LOCKFILE_VERSION,
    'depends' => @deps.map { |i| stringify_keys(i) },
  }.to_yaml
end