module Berkshelf
  module RSpec
    module ChefAPI
      # Return an array of Hashes containing cookbooks and their information
      #
      # @return [Array]
      def get_cookbooks
        ridley.cookbook.all
      end

      def upload_cookbook(path)
        cached = CachedCookbook.from_store_path(path)
        ridley.cookbook.upload(cached.path, name: cached.cookbook_name)
      end

      # Remove the version of the given cookbook from the Chef Server defined
      # in your Knife config.
      #
      # @param [#to_s] name
      # @param [#to_s] version
      def purge_cookbook(name, version = nil)
        if version.nil?
          ridley.cookbook.delete_all(name)
        else
          ridley.cookbook.delete(name, version)
        end
      rescue Ridley::Errors::HTTPNotFound
        true
      end

      def server_has_cookbook?(name, version = nil)
        versions = ridley.cookbook.versions(name)

        if version.nil?
          !versions.empty?
        else
          !versions.find { |ver| ver == version }.nil?
        end
      rescue Ridley::Errors::HTTPNotFound
        false
      end

      def generate_cookbook(path, name, version, options = {})
        path = Pathname.new(path)
        cookbook_path = path.join("#{name}-#{version}")
        directories = [
          "recipes",
          "templates/default",
          "files/default",
          "attributes",
          "definitions",
          "providers",
          "resources"
        ]
        files = [
          "recipes/default.rb",
          "templates/default/template.erb",
          "files/default/file.h",
          "attributes/default.rb"
        ]

        directories.each do |directory|
          FileUtils.mkdir_p(cookbook_path.join(directory))
        end

        files.each do |file|
          FileUtils.touch(cookbook_path.join(file))
        end

        metadata = <<-EOF
name "#{name}"
version "#{version}"
EOF

        if options[:dependencies]
          options[:dependencies].each do |name, constraint|
            metadata << "depends '#{name}', '#{constraint}'\n"
          end
        end

        if options[:recommendations]
          options[:recommendations].each do |name, constraint|
            metadata << "recommends '#{name}', '#{constraint}'\n"
          end
        end

        File.open(cookbook_path.join("metadata.rb"), 'w+') do |f|
          f.write metadata
        end
        
        cookbook_path
      end

      private

        def ridley
          @ridley ||= Ridley.new(
            server_url: Berkshelf::Chef::Config[:chef_server_url],
            client_name: Berkshelf::Chef::Config[:node_name],
            client_key: Berkshelf::Chef::Config[:client_key],
            ssl: { verify: false }
          )
        end
    end
  end
end