require 'ridley/resources/data_bag_item_resource'

module Ridley
  # @author Jamie Winsor <reset@riotgames.com>
  class DataBagResource < Ridley::Resource
    set_resource_path "data"
    represented_by Ridley::DataBagObject

    attr_reader :item_resource

    finalizer do
      item_resource.terminate if item_resource && item_resource.alive?
    end

    # @param [Celluloid::Registry] connection_registry
    # @param [String] data_bag_secret
    def initialize(connection_registry, data_bag_secret)
      super(connection_registry)
      @item_resource = DataBagItemResource.new_link(connection_registry, data_bag_secret)
    end

    # @param [String, #chef_id] object
    #
    # @return [nil, Ridley::DataBagResource]
    def find(object)
      chef_id = object.respond_to?(:chef_id) ? object.chef_id : object
      connection.get("#{self.class.resource_path}/#{chef_id}")
      new(name: chef_id)
    rescue Errors::HTTPNotFound
      nil
    end
  end
end