class Kagaribi::Collection

def self.sanitize_key(key)
  key.tr("/", "-")
end

def delete(doc_key)
  with_retry("Kagaribi::Collection#delete") do
    ref = firestore.doc(full_doc_key(doc_key))
    ref.delete
  end
end

def exists?(doc_key)
  with_retry("Kagaribi::Collection#exists?") do
    ref = firestore.doc(full_doc_key(doc_key))
    snap = ref.get
    snap&.exists?
  end
end

def firestore
  @firestore ||= Google::Cloud::Firestore.new(database_id: database_id)
end

def full_doc_key(doc_key)
  "#{sanitized_collection_name}/#{sanitize_key(doc_key)}"
end

def get(doc_key)
  with_retry("Kagaribi::Collection#get") do
    ref = firestore.doc(full_doc_key(doc_key))
    snap = ref.get
    snap&.data || {}
  end
end

def initialize(collection_name, database_id: nil, logger: nil)
  @collection_name = collection_name
  @database_id = database_id
  @logger =
    if logger
      logger
    else
      Logger.new($stdout)
    end
end

def retryable_error?(error)
  case error
  when RuntimeError
    # e.g.
    # Could not load the default credentials. Browse to
    # https://developers.google.com/accounts/docs/application-default-credentials
    # for more information
    return true if error.message.include?("Could not load the default credentials.")
    # e.g.
    # Your credentials were not found. To set up Application Default
    # Credentials for your environment, see
    # https://cloud.google.com/docs/authentication/external/set-up-adc
    return true if error.message.include?("Your credentials were not found.")
    return false
  end
  true
end

def sanitize_key(key)
  Collection.sanitize_key(key)
end

def sanitized_collection_name
  sanitize_key(@collection_name)
end

def set(doc_key, data)
  with_retry("Kagaribi::Collection#set") do
    ref = firestore.doc(full_doc_key(doc_key))
    ref.set(data)
  end
end

def update(doc_key, data)
  with_retry("Kagaribi::Collection#update") do
    ref = firestore.doc(full_doc_key(doc_key))
    ref.update(data)
  end
end

def with_retry(label)
  yield
rescue TypeError, GRPC::Unavailable, RuntimeError, Signet::AuthorizationError, Google::Cloud::UnauthenticatedError => error
  raise error unless retryable_error?(error)
  retry_count ||= 0
  retry_count += 1
  raise error if retry_count > MAX_RETRY_COUNT
  logger.warn "[#{label}] collection_name=#{@collection_name}, retry_count=#{retry_count}, error=#{error}"
  sleep 1
  retry
end