AdClickAttributionManager.h [plain text]
#pragma once
#include "NetworkProcess.h"
#include "NetworkResourceLoadParameters.h"
#include <WebCore/AdClickAttribution.h>
#include <WebCore/RegistrableDomain.h>
#include <WebCore/ResourceError.h>
#include <WebCore/ResourceResponse.h>
#include <WebCore/Timer.h>
#include <pal/SessionID.h>
#include <wtf/CompletionHandler.h>
#include <wtf/HashMap.h>
#include <wtf/WeakPtr.h>
#include <wtf/text/WTFString.h>
namespace WebKit {
class AdClickAttributionManager : public CanMakeWeakPtr<AdClickAttributionManager> {
WTF_MAKE_FAST_ALLOCATED;
public:
using RegistrableDomain = WebCore::RegistrableDomain;
using AdClickAttribution = WebCore::AdClickAttribution;
using Source = WebCore::AdClickAttribution::Source;
using Destination = WebCore::AdClickAttribution::Destination;
using Conversion = WebCore::AdClickAttribution::Conversion;
explicit AdClickAttributionManager(NetworkProcess& networkProcess, PAL::SessionID sessionID)
: m_firePendingConversionRequestsTimer(*this, &AdClickAttributionManager::firePendingConversionRequests)
, m_pingLoadFunction([](NetworkResourceLoadParameters&& params, CompletionHandler<void(const WebCore::ResourceError&, const WebCore::ResourceResponse&)>&& completionHandler) {
UNUSED_PARAM(params);
completionHandler(WebCore::ResourceError(), WebCore::ResourceResponse());
})
, m_networkProcess(networkProcess)
, m_sessionID(sessionID)
{
}
void storeUnconverted(AdClickAttribution&&);
void handleConversion(Conversion&&, const URL& requestURL, const WebCore::ResourceRequest& redirectRequest);
void clear();
void clearForRegistrableDomain(const RegistrableDomain&);
void toString(CompletionHandler<void(String)>&&) const;
void setPingLoadFunction(Function<void(NetworkResourceLoadParameters&&, CompletionHandler<void(const WebCore::ResourceError&, const WebCore::ResourceResponse&)>&&)>&& pingLoadFunction) { m_pingLoadFunction = WTFMove(pingLoadFunction); }
void setOverrideTimerForTesting(bool value) { m_isRunningTest = value; }
void setConversionURLForTesting(URL&&);
void markAllUnconvertedAsExpiredForTesting();
private:
void startTimer(Seconds);
void convert(const Source&, const Destination&, Conversion&&);
void fireConversionRequest(const AdClickAttribution&);
void firePendingConversionRequests();
void clearExpired();
bool debugModeEnabled() const;
HashMap<std::pair<Source, Destination>, AdClickAttribution> m_unconvertedAdClickAttributionMap;
HashMap<std::pair<Source, Destination>, AdClickAttribution> m_convertedAdClickAttributionMap;
WebCore::Timer m_firePendingConversionRequestsTimer;
Function<void(NetworkResourceLoadParameters&&, CompletionHandler<void(const WebCore::ResourceError&, const WebCore::ResourceResponse&)>&&)> m_pingLoadFunction;
bool m_isRunningTest { false };
Optional<URL> m_conversionBaseURLForTesting;
Ref<NetworkProcess> m_networkProcess;
PAL::SessionID m_sessionID;
};
}