56 lines
1.3 KiB
C++
56 lines
1.3 KiB
C++
#include "mozilla/dom/DocGroup.h"
|
|
#include "mozilla/dom/TabGroup.h"
|
|
#include "mozilla/Telemetry.h"
|
|
#include "nsIURI.h"
|
|
#include "nsIEffectiveTLDService.h"
|
|
#include "mozilla/StaticPtr.h"
|
|
#include "mozilla/ClearOnShutdown.h"
|
|
#include "nsIDocShell.h"
|
|
|
|
namespace mozilla {
|
|
namespace dom {
|
|
|
|
/* static */ void
|
|
DocGroup::GetKey(nsIPrincipal* aPrincipal, nsACString& aKey)
|
|
{
|
|
aKey.Truncate();
|
|
nsCOMPtr<nsIURI> uri;
|
|
nsresult rv = aPrincipal->GetURI(getter_AddRefs(uri));
|
|
// GetBaseDomain works fine if |uri| is null, but it outputs a warning
|
|
// which ends up cluttering the logs.
|
|
if (NS_SUCCEEDED(rv) && uri) {
|
|
nsCOMPtr<nsIEffectiveTLDService> tldService =
|
|
do_GetService(NS_EFFECTIVETLDSERVICE_CONTRACTID);
|
|
if (tldService) {
|
|
rv = tldService->GetBaseDomain(uri, 0, aKey);
|
|
if (NS_FAILED(rv)) {
|
|
aKey.Truncate();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void
|
|
DocGroup::RemoveDocument(nsIDocument* aDocument)
|
|
{
|
|
MOZ_ASSERT(mDocuments.Contains(aDocument));
|
|
mDocuments.RemoveElement(aDocument);
|
|
}
|
|
|
|
DocGroup::DocGroup(TabGroup* aTabGroup, const nsACString& aKey)
|
|
: mKey(aKey), mTabGroup(aTabGroup)
|
|
{
|
|
// This method does not add itself to mTabGroup->mDocGroups as the caller does it for us.
|
|
}
|
|
|
|
DocGroup::~DocGroup()
|
|
{
|
|
MOZ_ASSERT(mDocuments.IsEmpty());
|
|
mTabGroup->mDocGroups.RemoveEntry(mKey);
|
|
}
|
|
|
|
NS_IMPL_ISUPPORTS(DocGroup, nsISupports)
|
|
|
|
}
|
|
}
|