summaryrefslogtreecommitdiffstats
path: root/toolkit/xre/nsAppStartupNotifier.cpp
blob: 8a00ff88a86e4ae4ac521dd3a41fe5bf180d017f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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/. */

#include "nsAppStartupNotifier.h"
#include "nsServiceManagerUtils.h"
#include "nsComponentManagerUtils.h"
#include "nsCOMPtr.h"
#include "nsString.h"
#include "nsICategoryManager.h"
#include "nsIObserver.h"
#include "nsXPCOM.h"
#include "mozilla/SimpleEnumerator.h"

using namespace mozilla;

/* static */
nsresult nsAppStartupNotifier::NotifyObservers(const char* aCategory) {
  NS_ENSURE_ARG(aCategory);
  nsresult rv;

  // now initialize all startup listeners
  nsCOMPtr<nsICategoryManager> categoryManager =
      do_GetService(NS_CATEGORYMANAGER_CONTRACTID, &rv);
  NS_ENSURE_SUCCESS(rv, rv);

  nsDependentCString category(aCategory);

  nsCOMPtr<nsISimpleEnumerator> enumerator;
  rv = categoryManager->EnumerateCategory(category, getter_AddRefs(enumerator));
  if (NS_WARN_IF(NS_FAILED(rv))) {
    return rv;
  }

  for (auto& categoryEntry : SimpleEnumerator<nsICategoryEntry>(enumerator)) {
    nsAutoCString contractId;
    categoryEntry->GetValue(contractId);

    nsCOMPtr<nsISupports> startupInstance;

    // If we see the word "service," in the beginning
    // of the contractId then we create it as a service
    // if not we do a createInstance
    if (StringBeginsWith(contractId, "service,"_ns)) {
      startupInstance = do_GetService(contractId.get() + 8, &rv);
    } else {
      startupInstance = do_CreateInstance(contractId.get(), &rv);
    }

    if (NS_SUCCEEDED(rv)) {
      // Try to QI to nsIObserver
      nsCOMPtr<nsIObserver> startupObserver =
          do_QueryInterface(startupInstance, &rv);
      if (NS_SUCCEEDED(rv)) {
        rv = startupObserver->Observe(nullptr, aCategory, nullptr);

        // mainly for debugging if you want to know if your observer worked.
        NS_ASSERTION(NS_SUCCEEDED(rv), "Startup Observer failed!\n");
      }
    } else {
#ifdef DEBUG
      nsAutoCString warnStr("Cannot create startup observer : ");
      warnStr += contractId.get();
      NS_WARNING(warnStr.get());
#endif
    }
  }

  return NS_OK;
}