#
# Author:: Hedgehog (<hedgehogshiatus@gmail.com>)
# Copyright:: Copyright (c) 2011 Hedgehog.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
module ::Cuken
  module Api
    module Chef
      module Cookbook

        def parse_to_cookbooks_path(hsh)
          case
            when !(hsh['cookbook'].nil? || hsh['cookbook'].empty?)
              ckbk = hsh['cookbook']
              ckbk_src = "cookbooks/#{ckbk}"
            when !(hsh['site-cookbook'].nil? || hsh['site-cookbook'].empty?)
              ckbk = hsh['site-cookbook']
              ckbk_src = "site-cookbooks/#{ckbk}"
            else
              src =""
          end
          return ckbk, ckbk_src
        end

        def check_cookbook_table_presence(table, expect_presence = true)
          table.hashes.each do |hsh|
            ckbk, ckbk_src = parse_to_cookbooks_path(hsh)
            full_cookbook_src = find_path_in_cookbook_folders(ckbk, ckbk_src, ckbk)
            announce_or_puts(%{Looking for cookbook: #{full_cookbook_src}}) if @announce_env
            check_cookbook_presence(full_cookbook_src, expect_presence)
          end
        end

        def check_remote_cookbook_table_presence(table, expect_presence = true)
          books = cookbooks_list
          table.hashes.each do |hsh|
            if expect_presence
              books.should include(hsh['cookbook']) if not (hsh['cookbook'].nil? || hsh['cookbook'].empty?)
              books.should include(hsh['site-cookbook']) if not (hsh['site-cookbook'].nil? || hsh['site-cookbook'].empty?)
            else
              books.should_not include(hsh['cookbook']) if not (hsh['cookbook'].nil? || hsh['cookbook'].empty?)
              books.should_not include(hsh['site-cookbook']) if not (hsh['site-cookbook'].nil? || hsh['site-cookbook'].empty?)
            end
          end
        end

        def find_path_in_cookbook_folders(ckbk, ckbk_src, path_fragment1, path_fagment2='')
          verify_cookbook_folders
          full_data_bag_src = nil
          in_chef_root do
            list1 = chef.cookbooks_paths.find_all { |dir| Pathname(dir + path_fragment1 + path_fagment2).exist? }
            list2 = chef.cookbook_paths.find_all { |dir| (dir.to_s[/#{ckbk_src}/] && Pathname(dir+path_fagment2).exist?) }
            loc = list2[0] || ((list1[0] + ckbk) if list1[0] && list1[0].exist?)
            if loc.nil? || !(loc.exist?)
              # TODO: error handling if data bags or cookbooks do not exist
            else
              full_data_bag_src = (loc + path_fagment2).expand_path.realdirpath.to_s
            end
          end
          full_data_bag_src
        end

        def verify_cookbook_folders
          if (chef.cookbooks_paths.empty? && chef.cookbook_paths.empty?)
            check_default_cookbooks_paths
          end
        end

        def check_default_cookbooks_paths
          in_chef_root do
            ['site-cookbooks', 'cookbooks'].each do |dir|
              path = (Pathname('.')+dir)
              chef.cookbooks_paths << path if path.exist?
            end
            if chef.cookbooks_paths.empty?
              raise(RuntimeError, "chef.cookbooks_paths AND chef.cookbook_paths cannot both be empty.", caller)
            end
            chef.cookbooks_paths.uniq
          end
        end

        def check_cookbook_presence(ckbk, expect_presence = true)
          chef.cookbook_paths.each do |pn|
            curr_ckbk = pn.basename.to_s
            condition = Pathname(pn).exist?
            given_ckbk = Pathname(ckbk).basename.to_s if ckbk
            result = given_ckbk == curr_ckbk
            if condition && result
              given_ckbk.should == curr_ckbk
              announce_or_puts(%{Found cookbook: #{pn}}) if @announce_env
            end
          end
        end

        def cookbooks_load(table)
          table.hashes.each do |hsh|
            ckbk, ckbk_src = parse_to_cookbooks_path(hsh)
            full_cookbook_src = find_path_in_cookbook_folders(ckbk, ckbk_src, ckbk)
            cookbook_load(ckbk)
          end
        end

        def cookbooks_delete(table)
          table.hashes.each do |hsh|
            ckbk, ckbk_src = parse_to_cookbooks_path(hsh)
            full_cookbook_src = find_path_in_cookbook_folders(ckbk, ckbk_src, ckbk)
            version = hsh['version'] ? hsh['version'] : 'all'
            cookbook_delete(ckbk, version)
          end
        end

        def cookbook_load(ckbk)

          knife_config_file_error_handling()

          ckbk_pth = nil
          in_chef_root do
            ckbk_pth = unless chef.cookbooks_paths.empty?
                  (chef.cookbooks_paths.collect { |pn| pn.expand_path.to_s }).join(':')
              end
          end

          argv = ['cookbook', 'upload', ckbk]
          ( argv << '--cookbook-path' << ckbk_pth ) if ckbk_pth
          run_knife(argv)
        end

        def cookbook_delete(ckbk, version='all')

          knife_config_file_error_handling()

          ckbk_pth = nil
          in_chef_root do
            ckbk_pth = unless chef.cookbooks_paths.empty?
                  (chef.cookbooks_paths.collect { |pn| pn.expand_path.to_s }).join(':')
              end
          end

          argv = ['cookbook', 'delete', ckbk]
          version == 'all' ? argv << '--all' : argv << version
          argv << '--yes' << '--no-editor'
          run_knife(argv)
        end

        def cookbooks_list

          knife_config_file_error_handling()

          argv = ['cookbook', 'list']
          run_knife(argv)
        end

      end
    end
  end
end