#
# Author:: John Keiser (<jkeiser@opscode.com>)
# Copyright:: Copyright (c) 2012 Opscode, Inc.
# License:: Apache License, Version 2.0
#
# 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.
#

require 'chef_fs/file_system/acls_dir'
require 'chef_fs/file_system/base_fs_dir'
require 'chef_fs/file_system/rest_list_dir'
require 'chef_fs/file_system/cookbooks_dir'
require 'chef_fs/file_system/data_bags_dir'
require 'chef_fs/file_system/nodes_dir'
require 'chef_fs/file_system/environments_dir'
require 'chef/rest'
require 'chef_fs/data_handler/client_data_handler'
require 'chef_fs/data_handler/role_data_handler'
require 'chef_fs/data_handler/user_data_handler'
require 'chef_fs/data_handler/group_data_handler'
require 'chef_fs/data_handler/container_data_handler'

module ChefFS
  module FileSystem
    class ChefServerRootDir < BaseFSDir
      def initialize(root_name, chef_config, options = {})
        super("", nil)
        @chef_server_url = chef_config[:chef_server_url]
        @chef_username = chef_config[:node_name]
        @chef_private_key = chef_config[:client_key]
        @environment = chef_config[:environment]
        @repo_mode = chef_config[:repo_mode]
        @root_name = root_name
        @cookbook_version = options[:cookbook_version] # Used in knife diff and download for server cookbook version
      end

      attr_reader :chef_server_url
      attr_reader :chef_username
      attr_reader :chef_private_key
      attr_reader :environment
      attr_reader :repo_mode
      attr_reader :cookbook_version

      def fs_description
        "Chef server at #{chef_server_url} (user #{chef_username}), repo_mode = #{repo_mode}"
      end

      def rest
        Chef::REST.new(chef_server_url, chef_username, chef_private_key)
      end

      def api_path
        ""
      end

      def path_for_printing
        "#{@root_name}/"
      end

      def can_have_child?(name, is_dir)
        is_dir && children.any? { |child| child.name == name }
      end

      def org
        @org ||= if URI.parse(chef_server_url).path =~ /^\/+organizations\/+([^\/]+)$/
          $1
        else
          nil
        end
      end

      def children
        @children ||= begin
          result = [
            CookbooksDir.new(self),
            DataBagsDir.new(self),
            EnvironmentsDir.new(self),
            RestListDir.new("roles", self, nil, ChefFS::DataHandler::RoleDataHandler.new)
          ]
          if repo_mode == 'hosted_everything'
            result += [
              AclsDir.new(self),
              RestListDir.new("clients", self, nil, ChefFS::DataHandler::ClientDataHandler.new),
              RestListDir.new("containers", self, nil, ChefFS::DataHandler::ContainerDataHandler.new),
              RestListDir.new("groups", self, nil, ChefFS::DataHandler::GroupDataHandler.new),
              NodesDir.new(self)
            ]
          elsif repo_mode != 'static'
            result += [
              RestListDir.new("clients", self, nil, ChefFS::DataHandler::ClientDataHandler.new),
              NodesDir.new(self),
              RestListDir.new("users", self, nil, ChefFS::DataHandler::UserDataHandler.new)
            ]
          end
          result.sort_by { |child| child.name }
        end
      end
    end
  end
end
