module KineticSdk
  class Task

    # Add a source
    #
    # @param source [Hash] Source properties
    #   - +name+ - name of the source
    #   - +status+ - initial status of the source (Active | Inactive)
    #   - +type+ - name of one of the source consumers registered in the task engine
    #   - +properties+ - hash of properties specific to the source consumer
    #   - +policyRules+ - array of policy rules to associate with the source
    # @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
    #
    #     add_source({
    #       "name" => "Source Name",
    #       "status" => "Active",
    #       "type" => "Kinetic Request CE",
    #       "properties" => {
    #         "Space Slug" => "foo",
    #         "Web Server" => "http://localhost:8080/kinetic",
    #         "Proxy Username" => "integration-user",
    #         "Proxy Password" => "integration-password"
    #       },
    #       "policyRules" => []
    #     })
    #
    def add_source(source, headers=default_headers)
      name = source['name']
      @logger.info("Adding the #{name} source")
      post("#{@api_url}/sources", source, headers)
    end

    # Delete a Source
    #
    # @param name [String] name of the source
    # @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_source(name, headers=header_basic_auth)
      @logger.info("Deleting Source \"#{name}\"")
      delete("#{@api_url}/sources/#{encode(name)}", headers)
    end

    # Delete all Sources
    #
    # @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_sources(headers=header_basic_auth)
      @logger.info("Deleting all sources")
      (find_sources(headers).content['sources'] || []).each do |source|
        delete("#{@api_url}/sources/#{encode(source['name'])}", headers)
      end
    end

    # Export all sources to :source-slug.json file in export_directory/sources
    #
    # @param headers [Hash] hash of headers to send, default is basic authentication
    # @return nil
    def export_sources(headers=header_basic_auth)
      raise StandardError.new "An export directory must be defined to export sources." if @options[:export_directory].nil?
      response = find_sources({"include" => "policyRules"})
      (response.content["sourceRoots"] || []).each do |source|
        # determine which directory to write the file to
        if source['name'] != "-"
          # create the directory if it doesn't yet exist
          sources_dir = FileUtils::mkdir_p(File.join(@options[:export_directory], "sources"))
          source_file = File.join(sources_dir, "#{source['name'].slugify}.json")
          write_object_to_file(source_file, source)
        end
      end
    end

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

    # Find all sources
    #
    # @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_sources(params={}, headers=header_basic_auth)
      @logger.info("Finding all sources")
      get("#{@api_url}/sources", params, headers)
    end

    # Find a source
    #
    # @param name [String] name of the source
    # @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_source(name, params={}, headers=header_basic_auth)
      @logger.info("Finding source named \"#{name}\"")
      get("#{@api_url}/sources/#{encode(name)}", params, headers)
    end

    # Update a source
    #
    # @param source [Hash] hash of existing source properties that must contain 'name'
    #   - +name+ - name of the source
    # @param body [Hash] - source properties to update
    # @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
    #
    # Exammple
    #
    #     update_source(
    #       { "name" => "Kinetic Request CE" },
    #       {
    #         "name": "Possible New Source Name",
    #         "status": "Active|Unconfigured",
    #         "type": "Adhoc|...",
    #         "properties": {},
    #         "policyRules": [
    #           { "type": "API Access", "name": "Super User" },
    #           { "type": "API Access", "name": "Info User" },
    #           { "type": "Console Access", "name": "Super User" }
    #         ]
    #       }
    #     )
    #
    def update_source(source, body={}, headers=default_headers)
      @logger.info("Updating the \"#{source['name']}\" Source")
      put("#{@api_url}/sources/#{encode(source['name'])}", body, headers)
    end

    # Add policy rule to source
    #
    # @param policy_rule_type [String] the type of policy rule
    # @param policy_rule_name [String] the name of the policy rule
    # @param source_name [String] name of the source to add the policy rule to
    # @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_source(policy_rule_type, policy_rule_name, source_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 source \"#{source_name}\"")
      post("#{@api_url}/sources/#{encode(source_name)}/policyRules", body, headers)
    end

    # Remove policy rule from source
    #
    # @param policy_rule_type [String] the type of policy rule
    # @param policy_rule_name [String] the name of the policy rule
    # @param source_name [String] name of the source to add the policy rule to
    # @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_source(policy_rule_type, policy_rule_name, source_name, headers=default_headers)
      @logger.info("Removing policy rule \"#{policy_rule_type} - #{policy_rule_name}\" from source \"#{source_name}\"")
      delete("#{@api_url}/sources/#{encode(source_name)}/policyRules/#{encode(policy_rule_type)}/#{encode(policy_rule_name)}", headers)
    end

  end
end