third_party/WebKit/Source/modules/push_messaging/ServiceWorkerRegistrationPush.cpp

// 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.

#include "modules/push_messaging/ServiceWorkerRegistrationPush.h"

#include "modules/push_messaging/PushManager.h"
#include "modules/serviceworkers/ServiceWorkerRegistration.h"

namespace blink {

ServiceWorkerRegistrationPush::ServiceWorkerRegistrationPush(
    ServiceWorkerRegistration* registration)
    : m_registration(registration) {}

ServiceWorkerRegistrationPush::~ServiceWorkerRegistrationPush() {}

const char* ServiceWorkerRegistrationPush::supplementName() {
  return "ServiceWorkerRegistrationPush";
}

ServiceWorkerRegistrationPush& ServiceWorkerRegistrationPush::from(
    ServiceWorkerRegistration& registration) {
  ServiceWorkerRegistrationPush* supplement =
      static_cast<ServiceWorkerRegistrationPush*>(
          Supplement<ServiceWorkerRegistration>::from(registration,
                                                      supplementName()));
  if (!supplement) {
    supplement = new ServiceWorkerRegistrationPush(&registration);
    provideTo(registration, supplementName(), supplement);
  }
  return *supplement;
}

PushManager* ServiceWorkerRegistrationPush::pushManager(
    ServiceWorkerRegistration& registration) {
  return ServiceWorkerRegistrationPush::from(registration).pushManager();
}

PushManager* ServiceWorkerRegistrationPush::pushManager() {
  if (!m_pushManager)
    m_pushManager = PushManager::create(m_registration);
  return m_pushManager.get();
}

DEFINE_TRACE(ServiceWorkerRegistrationPush) {
  visitor->trace(m_registration);
  visitor->trace(m_pushManager);
  Supplement<ServiceWorkerRegistration>::trace(visitor);
}

}  // namespace blink