layout/xul/nsRepeatService.cpp
author Dragana Damjanovic <dd.mozilla@gmail.com>
Tue, 08 Jan 2019 20:40:35 +0000
changeset 506654 da0b07a8386c07b7b4c54a5c001fbedc64fa79eb
parent 505383 6f3709b3878117466168c40affa7bca0b60cf75b
child 509549 f0a91d36587266d7454a450c6044d573664fbed5
permissions -rw-r--r--
Bug 1515459 - Check if we reenter CreateShimError. r=mayhemer, a=RyanVM Differential Revision: https://phabricator.services.mozilla.com/D15297

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

//
// Eric Vaughan
// Netscape Communications
//
// See documentation in associated header file
//

#include "nsRepeatService.h"
#include "mozilla/StaticPtr.h"
#include "nsIDocument.h"
#include "nsIServiceManager.h"

using namespace mozilla;

static StaticAutoPtr<nsRepeatService> gRepeatService;

nsRepeatService::nsRepeatService()
    : mCallback(nullptr), mCallbackData(nullptr) {}

nsRepeatService::~nsRepeatService() {
  NS_ASSERTION(!mCallback && !mCallbackData,
               "Callback was not removed before shutdown");
}

/* static */ nsRepeatService* nsRepeatService::GetInstance() {
  if (!gRepeatService) {
    gRepeatService = new nsRepeatService();
  }
  return gRepeatService;
}

/*static*/ void nsRepeatService::Shutdown() { gRepeatService = nullptr; }

void nsRepeatService::Start(Callback aCallback, void* aCallbackData,
                            nsIDocument* aDocument,
                            const nsACString& aCallbackName,
                            uint32_t aInitialDelay) {
  MOZ_ASSERT(aCallback != nullptr, "null ptr");

  mCallback = aCallback;
  mCallbackData = aCallbackData;
  mCallbackName = aCallbackName;

  mRepeatTimer = NS_NewTimer(aDocument->EventTargetFor(TaskCategory::Other));

  if (mRepeatTimer) {
    InitTimerCallback(aInitialDelay);
  }
}

void nsRepeatService::Stop(Callback aCallback, void* aCallbackData) {
  if (mCallback != aCallback || mCallbackData != aCallbackData) return;

  // printf("Stopping repeat timer\n");
  if (mRepeatTimer) {
    mRepeatTimer->Cancel();
    mRepeatTimer = nullptr;
  }
  mCallback = nullptr;
  mCallbackData = nullptr;
}

void nsRepeatService::InitTimerCallback(uint32_t aInitialDelay) {
  if (!mRepeatTimer) {
    return;
  }

  mRepeatTimer->InitWithNamedFuncCallback(
      [](nsITimer* aTimer, void* aClosure) {
        // Use gRepeatService instead of nsRepeatService::GetInstance() (because
        // we don't want nsRepeatService::GetInstance() to re-create a new
        // instance for us, if we happen to get invoked after
        // nsRepeatService::Shutdown() has nulled out gRepeatService).
        nsRepeatService* rs = gRepeatService;
        if (!rs) {
          return;
        }

        if (rs->mCallback) {
          rs->mCallback(rs->mCallbackData);
        }

        rs->InitTimerCallback(REPEAT_DELAY);
      },
      nullptr, aInitialDelay, nsITimer::TYPE_ONE_SHOT, mCallbackName.Data());
}