Source/modules/push_messaging/PushController.h

// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef PushController_h
#define PushController_h

#include "core/page/Page.h"
#include "platform/Supplementable.h"
#include "wtf/Forward.h"
#include "wtf/Noncopyable.h"
#include "wtf/PassOwnPtr.h"

namespace blink {

class WebPushClient;

class PushController final : public NoBaseWillBeGarbageCollected<PushController>, public WillBeHeapSupplement<Page> {
    WILL_BE_USING_GARBAGE_COLLECTED_MIXIN(PushController);
    WTF_MAKE_NONCOPYABLE(PushController);

public:
    static PassOwnPtrWillBeRawPtr<PushController> create(WebPushClient*);
    static const char* supplementName();
    static PushController* from(Page* page) { return static_cast<PushController*>(WillBeHeapSupplement<Page>::from(page, supplementName())); }
    static WebPushClient* clientFrom(Page*);

    WebPushClient* client() const { return m_client; }

    virtual void trace(Visitor* visitor) override { WillBeHeapSupplement<Page>::trace(visitor); }

private:
    explicit PushController(WebPushClient*);

    WebPushClient* m_client;
};

void providePushControllerTo(Page&, WebPushClient*);

} // namespace blink

#endif // PushController_h