PrivateClickMeasurementManager.h [plain text]
#pragma once
#include "NetworkProcess.h"
#include "NetworkResourceLoadParameters.h"
#include <WebCore/PrivateClickMeasurement.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 {
enum class PrivateClickMeasurementAttributionType : bool { Unattributed, Attributed };
class PrivateClickMeasurementManager : public CanMakeWeakPtr<PrivateClickMeasurementManager> {
WTF_MAKE_FAST_ALLOCATED;
public:
using RegistrableDomain = WebCore::RegistrableDomain;
using PrivateClickMeasurement = WebCore::PrivateClickMeasurement;
using SourceSite = WebCore::PrivateClickMeasurement::SourceSite;
using AttributeOnSite = WebCore::PrivateClickMeasurement::AttributeOnSite;
using AttributionTriggerData = WebCore::PrivateClickMeasurement::AttributionTriggerData;
explicit PrivateClickMeasurementManager(NetworkSession&, NetworkProcess&, PAL::SessionID);
void storeUnattributed(PrivateClickMeasurement&&);
void handleAttribution(AttributionTriggerData&&, 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 markAllUnattributedAsExpiredForTesting();
void markAttributedPrivateClickMeasurementsAsExpiredForTesting(CompletionHandler<void()>&&);
void startTimer(Seconds);
private:
void clearSentAttributions(Vector<PrivateClickMeasurement>&&);
void attribute(const SourceSite&, const AttributeOnSite&, AttributionTriggerData&&);
void fireConversionRequest(const PrivateClickMeasurement&);
void firePendingAttributionRequests();
void clearExpired();
bool featureEnabled() const;
bool debugModeEnabled() const;
WebCore::Timer m_firePendingAttributionRequestsTimer;
bool m_isRunningTest { false };
Optional<URL> m_attributionBaseURLForTesting;
WeakPtr<NetworkSession> m_networkSession;
Ref<NetworkProcess> m_networkProcess;
PAL::SessionID m_sessionID;
Function<void(NetworkResourceLoadParameters&&, CompletionHandler<void(const WebCore::ResourceError&, const WebCore::ResourceResponse&)>&&)> m_pingLoadFunction;
};
}