class ChefCLI::Policyfile::ChefServerCookbookSource

def ==(other)
  other.is_a?(self.class) && other.uri == uri && other.preferred_cookbooks == preferred_cookbooks
end

def default_source_args
  [:chef_server, uri]
end

def desc
  "chef_server(#{uri})"
end

def full_chef_server_graph
  @full_chef_server_graph ||=
    begin
      http_connection_for(uri.to_s).get("/universe")
    end
end

def http_connection_for(base_url)
  @http_connections[base_url] ||=
    ChefServerAPIMulti.new(base_url,
      signing_key_filename: chef_config.client_key,
      client_name: chef_config.node_name)
end

def initialize(uri, chef_config: nil)
  @uri = SourceURI.parse(uri)
  @http_connections = {}
  @chef_config = chef_config
  @preferred_cookbooks = []
  yield self if block_given?
end

def null?
  false
end

def preferred_for(*cookbook_names)
  preferred_cookbooks.concat(cookbook_names)
end

def preferred_source_for?(cookbook_name)
  preferred_cookbooks.include?(cookbook_name)
end

def source_options_for(cookbook_name, cookbook_version)
  {
    chef_server: uri.to_s,
    version: cookbook_version,
    http_client: http_connection_for(uri.to_s),
  }
end

def universe_graph
  @universe_graph ||= begin
    full_chef_server_graph.inject({}) do |normalized_graph, (cookbook_name, metadata_by_version)|
      normalized_graph[cookbook_name] = metadata_by_version.inject({}) do |deps_by_version, (version, metadata)|
        deps_by_version[version] = metadata["dependencies"]
        deps_by_version
      end
      normalized_graph
    end
  end
end