module KineticSdk
  class Task

    # Add a category
    #
    # @param category [Hash] Category properties
    #   - +name+ - name of the category
    #   - +handlers+ - array of handler definitionIds associated to the category
    #   - +policyRules+ - array of policy rule names associated to the category
    #   - +trees+ - array of tree (routine) definitionIds associated to the category
    # @param headers [Hash] hash of headers to send, default is basic authentication and accept JSON content type
    # @return [KineticSdk::Utils::KineticHttpResponse] object, with +code+, +message+, +content_string+, and +content+ properties
    def add_category(category, headers=default_headers)
      @logger.info("Add category \"#{category['name']}\"")
      post("#{@api_url}/categories", category, headers)
    end

    # Delete a Category
    #
    # @param name [String] Category name
    # @param headers [Hash] hash of headers to send, default is basic authentication
    # @return [KineticSdk::Utils::KineticHttpResponse] object, with +code+, +message+, +content_string+, and +content+ properties
    def delete_category(name, headers=header_basic_auth)
      @logger.info("Deleting Category \"#{name}\"")
      delete("#{@api_url}/categories/#{encode(name)}", headers)
    end

    # Delete all Categories
    #
    # @param headers [Hash] hash of headers to send, default is basic authentication
    # @return [KineticSdk::Utils::KineticHttpResponse] object, with +code+, +message+, +content_string+, and +content+ properties
    def delete_categories(headers=header_basic_auth)
      @logger.info("Deleting all categories")
      (find_categories(headers).content["categories"] || []).each do |category|
        delete_category(category['name'], headers)
      end
    end

    # Export a Category
    #
    # @param category [String|Hash] Category name, or hash of category properties
    #   - +name+ - name of the category
    #   - +handlers+ - array of handler definitionIds associated to the category
    #   - +policyRules+ - array of policy rule names associated to the category
    #   - +trees+ - array of tree (routine) definitionIds associated to the category
    # @param headers [Hash] hash of headers to send, default is basic authentication
    # @return nil
    def export_category(category, headers=header_basic_auth)
      raise StandardError.new "An export directory must be defined to export a category." if @options[:export_directory].nil?
      if category.is_a? String
        response = find_category(category, { "include" => "handlers,trees,policyRules" }, headers)
        category = response.content
      end
      @logger.info("Exporting category \"#{category['name']}\" to #{@options[:export_directory]}.")
      # Create the category directory if it doesn't yet exist
      category_dir = FileUtils::mkdir_p(File.join(@options[:export_directory], "categories"))
      category_file = File.join(category_dir, "#{category['name'].slugify}.json")

      # Just export the associated key values
      category['handlers'].collect! { |h| h['definitionId'] }.sort!
      category['trees'].collect! { |t| t['definitionId'] }.sort!
      category['policyRules'].collect! { |pr| pr['name'] }.sort!

      # write the file
      File.write(category_file, JSON.pretty_generate(category))
      @logger.info("Exported category: #{category['name']} to #{category_file}")
    end

    # Export Categories
    #
    # @param headers [Hash] hash of headers to send, default is basic authentication
    # @return nil
    def export_categories(headers=header_basic_auth)
      raise StandardError.new "An export directory must be defined to export categories." if @options[:export_directory].nil?
      (find_categories({ "include" => "handlers,trees,policyRules" }).content["categories"] || []).each do |category|
        export_category(category)
      end
    end

    # Import Categories
    #
    # @param headers [Hash] hash of headers to send, default is basic authentication
    # @return nil
    def import_categories(headers=default_headers)
      raise StandardError.new "An export directory must be defined to import categories from." if @options[:export_directory].nil?
      @logger.info("Importing all Categories in Export Directory")
      Dir["#{@options[:export_directory]}/categories/*.json"].sort.each do |file|
        category = JSON.parse(File.read(file))
        add_category(category, headers)
      end
    end

    # Find all categories
    #
    # @param params [Hash] Query parameters that are added to the URL, such as +include+
    # @param headers [Hash] hash of headers to send, default is basic authentication
    # @return [KineticSdk::Utils::KineticHttpResponse] object, with +code+, +message+, +content_string+, and +content+ properties
    def find_categories(params={}, headers=header_basic_auth)
      @logger.info("Finding all categories")
      get("#{@api_url}/categories", params, headers)
    end

    # Find a category
    #
    # @param name [String] name of the category
    # @param params [Hash] Query parameters that are added to the URL, such as +include+
    # @param headers [Hash] hash of headers to send, default is basic authentication
    # @return [KineticSdk::Utils::KineticHttpResponse] object, with +code+, +message+, +content_string+, and +content+ properties
    def find_category(name, params={}, headers=header_basic_auth)
      @logger.info("Finding Category \"#{name}\"")
      get("#{@api_url}/categories/#{encode(name)}", params, headers)
    end

    # Update a category
    #
    # @param original_name [String] name of the category before updating
    # @param body [Hash] the updated property values
    # @param headers [Hash] hash of headers to send, default is basic authentication and accept JSON content type
    # @return [KineticSdk::Utils::KineticHttpResponse] object, with +code+, +message+, +content_string+, and +content+ properties
    #
    # Example
    #
    #     update_category("Foo", { "name" => "Bar" })
    #
    def update_category(original_name, body={}, headers=default_headers)
      @logger.info("Updating Category \"#{original_name}\"")
      put("#{@api_url}/categories/#{encode(original_name)}", body, headers)
    end


    # Add a handler to a category
    #
    # @param handler_id [String] the handler id to add
    # @param category_name [String] name of the category to associate the handler
    # @param headers [Hash] hash of headers to send, default is basic authentication and accept JSON content type
    # @return [KineticSdk::Utils::KineticHttpResponse] object, with +code+, +message+, +content_string+, and +content+ properties
    def add_handler_to_category(handler_id, category_name, headers=default_headers)
      body = { "definitionId" => handler_id }
      @logger.info("Adding handler \"#{handler_id}\" to category \"#{category_name}\"")
      post("#{@api_url}/categories/#{encode(category_name)}/handlers", body, headers)
    end

    # Remove a handler from a category
    #
    # @param handler_id [String] the handler id to remove
    # @param category_name [String] name of the category to remove the handler from
    # @param headers [Hash] hash of headers to send, default is basic authentication and accept JSON content type
    # @return [KineticSdk::Utils::KineticHttpResponse] object, with +code+, +message+, +content_string+, and +content+ properties
    def remove_handler_from_category(handler_id, category_name, headers=default_headers)
      @logger.info("Removing handler \"#{handler_id}\" from category \"#{category_name}\"")
      delete("#{@api_url}/categories/#{encode(category_name)}/handlers/#{encode(handler_id)}", headers)
    end

    # Add a global routine to a category
    #
    # @param routine_id [String] the global routine id to add
    # @param category_name [String] name of the category to associate the global routine
    # @param headers [Hash] hash of headers to send, default is basic authentication and accept JSON content type
    # @return [KineticSdk::Utils::KineticHttpResponse] object, with +code+, +message+, +content_string+, and +content+ properties
    def add_routine_to_category(routine_id, category_name, headers=default_headers)
      body = { "definitionId" => routine_id }
      @logger.info("Adding routine \"#{routine_id}\" to category \"#{category_name}\"")
      post("#{@api_url}/categories/#{encode(category_name)}/routines", body, headers)
    end

    # Remove a global routine from a category
    #
    # @param routine_id [String] the global routine id to remove
    # @param category_name [String] name of the category to remove the global routine from
    # @param headers [Hash] hash of headers to send, default is basic authentication and accept JSON content type
    # @return [KineticSdk::Utils::KineticHttpResponse] object, with +code+, +message+, +content_string+, and +content+ properties
    def remove_routine_from_category(routine_id, category_name, headers=default_headers)
      @logger.info("Removing routine \"#{routine_id}\" from category \"#{category_name}\"")
      delete("#{@api_url}/categories/#{encode(category_name)}/routines/#{encode(routine_id)}", headers)
    end

    # Add a policy rule to a category
    #
    # @param policy_rule_type [String] the type of policy rule
    # @param policy_rule_name [String] the name of policy rule
    # @param category_name [String] name of the category to associate the policy rule
    # @param headers [Hash] hash of headers to send, default is basic authentication and accept JSON content type
    # @return [KineticSdk::Utils::KineticHttpResponse] object, with +code+, +message+, +content_string+, and +content+ properties
    def add_policy_rule_to_category(policy_rule_type, policy_rule_name, category_name, headers=default_headers)
      body = { "type" => policy_rule_type, "name" => policy_rule_name }
      @logger.info("Adding policy rule \"#{policy_rule_type} - #{policy_rule_name}\" to category \"#{category_name}\"")
      post("#{@api_url}/categories/#{encode(category_name)}/policyRules", body, headers)
    end

    # Remove a policy rule from a category
    #
    # @param policy_rule_type [String] the type of policy rule
    # @param policy_rule_name [String] the name of policy rule
    # @param category_name [String] name of the category to remove the policy rule from
    # @param headers [Hash] hash of headers to send, default is basic authentication and accept JSON content type
    # @return [KineticSdk::Utils::KineticHttpResponse] object, with +code+, +message+, +content_string+, and +content+ properties
    def remove_policy_rule_from_category(policy_rule_type, policy_rule_name, category_name, headers=default_headers)
      @logger.info("Removing policy rule \"#{policy_rule_type} - #{policy_rule_name}\" from category \"#{category_name}\"")
      delete("#{@api_url}/categories/#{encode(category_name)}/policyRules/#{encode(policy_rule_type)}/#{encode(policy_rule_name)}", headers)
    end

  end
end