module GdsApi::TestHelpers::ContentStore

def content_item_for_base_path(base_path)
  super.merge("base_path" => base_path)
end

def content_store_endpoint(draft: false)
  draft ? Plek.find("draft-content-store") : Plek.find("content-store")
end

def stub_content_store_does_not_have_item(base_path, options = {})
  endpoint = content_store_endpoint(draft: options[:draft])
  stub_request(:get, "#{endpoint}/content#{base_path}").to_return(status: 404, headers: {})
  stub_request(:get, "#{endpoint}/incoming-links#{base_path}").to_return(status: 404, headers: {})
end

def stub_content_store_has_gone_item(base_path, body = gone_content_item_for_base_path(base_path), options = {})
  body = body.to_json unless body.is_a?(String)
  endpoint = content_store_endpoint(draft: options[:draft])
  stub_request(:get, "#{endpoint}/content#{base_path}").to_return(
    status: 410,
    body:,
    headers: {},
  )
end

def stub_content_store_has_incoming_links(base_path, links)
  url = "#{content_store_endpoint}/incoming-links#{base_path}"
  body = links.to_json
  stub_request(:get, url).to_return(body:)
end

def stub_content_store_has_item(base_path, body = content_item_for_base_path(base_path), options = {})
  max_age = options.fetch(:max_age, 900)
  visibility = options[:private] ? "private" : "public"
  body = body.to_json unless body.is_a?(String)
  endpoint = content_store_endpoint(draft: options[:draft])
  stub_request(:get, "#{endpoint}/content#{base_path}").to_return(
    status: 200,
    body:,
    headers: {
      cache_control: "#{visibility}, max-age=#{max_age}",
      date: Time.now.httpdate,
    },
  )
end

def stub_content_store_isnt_available
  stub_request(:any, /#{content_store_endpoint}\/.*/).to_return(status: 503)
end