class Hubspot::Company < Hubspot::Resource
  self.id_field = "companyId"
  self.property_name_field = "name"

  ALL_PATH                = '/companies/v2/companies/paged'
  BATCH_UPDATE_PATH       = '/companies/v1/batch-async/update'
  CONTACTS_PATH           = '/companies/v2/companies/:id/contacts'
  CONTACT_IDS_PATH        = '/companies/v2/companies/:id/vids'
  CREATE_PATH             = '/companies/v2/companies/'
  DELETE_PATH             = '/companies/v2/companies/:id'
  FIND_PATH               = '/companies/v2/companies/:id'
  RECENTLY_CREATED_PATH   = '/companies/v2/companies/recent/created'
  RECENTLY_MODIFIED_PATH  = '/companies/v2/companies/recent/modified'
  SEARCH_DOMAIN_PATH      = '/companies/v2/domains/:domain/companies'
  UPDATE_PATH             = '/companies/v2/companies/:id'

  class << self
    def all(opts = {})
      Hubspot::PagedCollection.new(opts) do |options, offset, limit|
        response = Hubspot::Connection.get_json(
          ALL_PATH,
          options.merge(offset: offset, limit: limit)
        )

        companies = response["companies"].map { |result| from_result(result) }

        [companies, response["offset"], response["has-more"]]
      end
    end

    def search_domain(domain, opts = {})
      Hubspot::PagedCollection.new(opts) do |options, offset, limit|
        request = {
          "limit" => limit,
          "requestOptions" => options,
          "offset" => {
            "isPrimary" => true,
            "companyId" => offset
          }
        }

        response = Hubspot::Connection.post_json(
          SEARCH_DOMAIN_PATH,
          params: { domain: domain },
          body: request
        )

        companies = response["results"].map { |result| from_result(result) }

        [companies, response["offset"]["companyId"], response["hasMore"]]
      end
    end

    def recently_created(opts = {})
      Hubspot::PagedCollection.new(opts) do |options, offset, limit|
        response = Hubspot::Connection.get_json(
          RECENTLY_CREATED_PATH,
          {offset: offset, count: limit}
        )

        companies = response["results"].map { |result| from_result(result) }

        [companies, response["offset"], response["hasMore"]]
      end
    end

    def recently_modified(opts = {})
      Hubspot::PagedCollection.new(opts) do |options, offset, limit|
        response = Hubspot::Connection.get_json(
          RECENTLY_MODIFIED_PATH,
          {offset: offset, count: limit}
        )

        companies = response["results"].map { |result| from_result(result) }

        [companies, response["offset"], response["hasMore"]]
      end
    end

    def add_contact(id, contact_id)
      Hubspot::Association.create("Company", id, "Contact", contact_id)
    end

    def remove_contact(id, contact_id)
      Hubspot::Association.delete("Company", id, "Contact", contact_id)
    end

    def batch_update(companies, opts = {})
      request = companies.map do |company|
        # Use the specified options or update with the changes
        changes = opts.empty? ? company.changes : opts

        unless changes.empty?
          {
            "objectId" => company.id,
            "properties" => changes.map { |k, v| { "name" => k, "value" => v } }
          }
        end
      end

      # Remove any objects without changes and return if there is nothing to update
      request.compact!
      return true if request.empty?

      Hubspot::Connection.post_json(
        BATCH_UPDATE_PATH,
        params: {},
        body: request
      )

      true
    end
  end

  def contacts(opts = {})
    Hubspot::PagedCollection.new(opts) do |options, offset, limit|
      response = Hubspot::Connection.get_json(
        CONTACTS_PATH,
        {"id" => @id, "vidOffset" => offset, "count" => limit}
      )

      contacts = response["contacts"].map do |result|
        result["properties"] = Hubspot::Utils.properties_array_to_hash(result["properties"])
        Hubspot::Contact.from_result(result)
      end

      [contacts, response["vidOffset"], response["hasMore"]]
    end
  end

  def contact_ids(opts = {})
    Hubspot::PagedCollection.new(opts) do |options, offset, limit|
      response = Hubspot::Connection.get_json(
        CONTACT_IDS_PATH,
        {"id" => @id, "vidOffset" => offset, "count" => limit}
      )

      [response["vids"], response["vidOffset"], response["hasMore"]]
    end
  end

  def add_contact(contact)
    self.class.add_contact(@id, contact.to_i)
  end

  def remove_contact(contact)
    self.class.remove_contact(@id, contact.to_i)
  end
end