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

#include "nsCOMPtr.h"

#include "nsIContent.h"

#include "nsFrameLoader.h"

#include "mozilla/AsyncEventDispatcher.h"

#include "mozilla/dom/HTMLIFrameElement.h"

#include "mozilla/dom/XULFrameElement.h"

#include "mozilla/dom/XULFrameElementBinding.h"

namespace mozilla {

namespace dom {

NS_IMPL_CYCLE_COLLECTION_CLASS(XULFrameElement)

NS_IMPL_CYCLE_COLLECTION_TRAVERSE_BEGIN_INHERITED(XULFrameElement,

                                                  nsXULElement)

  NS_IMPL_CYCLE_COLLECTION_TRAVERSE(mFrameLoader);

  NS_IMPL_CYCLE_COLLECTION_TRAVERSE(mOpener);

NS_IMPL_CYCLE_COLLECTION_TRAVERSE_END

NS_IMPL_CYCLE_COLLECTION_UNLINK_BEGIN_INHERITED(XULFrameElement,

                                                nsXULElement)

  if (tmp->mFrameLoader) {

    tmp->mFrameLoader->Destroy();

  }

  NS_IMPL_CYCLE_COLLECTION_UNLINK(mFrameLoader, mOpener)

NS_IMPL_CYCLE_COLLECTION_UNLINK_END

NS_IMPL_ISUPPORTS_CYCLE_COLLECTION_INHERITED(XULFrameElement,

                                             nsXULElement,

                                             nsIFrameLoaderOwner)

JSObject*

XULFrameElement::WrapNode(JSContext *aCx, JS::Handle<JSObject*> aGivenProto)

{

  return XULFrameElement_Binding::Wrap(aCx, this, aGivenProto);

}

nsIDocShell*

XULFrameElement::GetDocShell()

{

  RefPtr<nsFrameLoader> frameLoader = GetFrameLoader();

  return frameLoader ? frameLoader->GetDocShell(IgnoreErrors()) : nullptr;

}

already_AddRefed<nsIWebNavigation>

XULFrameElement::GetWebNavigation()

{

  nsCOMPtr<nsIDocShell> docShell = GetDocShell();

  nsCOMPtr<nsIWebNavigation> webnav = do_QueryInterface(docShell);

  return webnav.forget();

}

already_AddRefed<nsPIDOMWindowOuter>

XULFrameElement::GetContentWindow()

{

  nsCOMPtr<nsIDocShell> docShell = GetDocShell();

  if (docShell) {

    nsCOMPtr<nsPIDOMWindowOuter> win = docShell->GetWindow();

    return win.forget();

  }

  return nullptr;

}

nsIDocument*

XULFrameElement::GetContentDocument()

{

  nsCOMPtr<nsPIDOMWindowOuter> win = GetContentWindow();

  return win ? win->GetDoc() : nullptr;

}

void

XULFrameElement::LoadSrc()

{

  if (!IsInUncomposedDoc() ||

      !OwnerDoc()->GetRootElement()) {

      return;

  }

  RefPtr<nsFrameLoader> frameLoader = GetFrameLoader();

  if (!frameLoader) {

    // Check if we have an opener we need to be setting

    nsCOMPtr<nsPIDOMWindowOuter> opener = mOpener;

    if (!opener) {

      // If we are a primary xul-browser, we want to take the opener property!

      nsCOMPtr<nsPIDOMWindowOuter> window = OwnerDoc()->GetWindow();

      if (AttrValueIs(kNameSpaceID_None, nsGkAtoms::primary,

                      nsGkAtoms::_true, eIgnoreCase) && window) {

        opener = window->TakeOpenerForInitialContentBrowser();

      }

    }

    mOpener = nullptr;

    // false as the last parameter so that xul:iframe/browser/editor

    // session history handling works like dynamic html:iframes.

    // Usually xul elements are used in chrome, which doesn't have

    // session history at all.

    mFrameLoader = nsFrameLoader::Create(this, opener, false);

    if (NS_WARN_IF(!mFrameLoader)) {

      return;

    }

    (new AsyncEventDispatcher(this,

                              NS_LITERAL_STRING("XULFrameLoaderCreated"),

                              CanBubble::eYes))->RunDOMEventWhenSafe();

  }

  mFrameLoader->LoadFrame(false);

}

void

XULFrameElement::SwapFrameLoaders(HTMLIFrameElement& aOtherLoaderOwner,

                                  ErrorResult& rv)

{

  aOtherLoaderOwner.SwapFrameLoaders(this, rv);

}

void

XULFrameElement::SwapFrameLoaders(XULFrameElement& aOtherLoaderOwner,

                                  ErrorResult& rv)

{

  if (&aOtherLoaderOwner == this) {

    // nothing to do

    return;

  }

  aOtherLoaderOwner.SwapFrameLoaders(this, rv);

}

void

XULFrameElement::SwapFrameLoaders(nsIFrameLoaderOwner* aOtherLoaderOwner,

                                  mozilla::ErrorResult& rv)

{

  RefPtr<nsFrameLoader> loader = GetFrameLoader();

  RefPtr<nsFrameLoader> otherLoader = aOtherLoaderOwner->GetFrameLoader();

  if (!loader || !otherLoader) {

    rv.Throw(NS_ERROR_NOT_IMPLEMENTED);

    return;

  }

  rv = loader->SwapWithOtherLoader(otherLoader, this, aOtherLoaderOwner);

}

nsresult

XULFrameElement::BindToTree(nsIDocument* aDocument,

                            nsIContent* aParent,

                            nsIContent* aBindingParent)

{

  nsresult rv = nsXULElement::BindToTree(aDocument, aParent, aBindingParent);

  NS_ENSURE_SUCCESS(rv, rv);

  if (aDocument) {

    NS_ASSERTION(!nsContentUtils::IsSafeToRunScript(),

                 "Missing a script blocker!");

    // We're in a document now.  Kick off the frame load.

    LoadSrc();

  }

  return NS_OK;

}

void

XULFrameElement::UnbindFromTree(bool aDeep, bool aNullParent)

{

  RefPtr<nsFrameLoader> frameLoader = GetFrameLoader();

  if (frameLoader) {

    frameLoader->Destroy();

  }

  mFrameLoader = nullptr;

  nsXULElement::UnbindFromTree(aDeep, aNullParent);

}

void

XULFrameElement::DestroyContent()

{

  RefPtr<nsFrameLoader> frameLoader = GetFrameLoader();

  if (frameLoader) {

    frameLoader->Destroy();

  }

  mFrameLoader = nullptr;

  nsXULElement::DestroyContent();

}

nsresult

XULFrameElement::AfterSetAttr(int32_t aNamespaceID, nsAtom* aName,

                              const nsAttrValue* aValue,

                              const nsAttrValue* aOldValue,

                              nsIPrincipal* aSubjectPrincipal,

                              bool aNotify)

{

  if (aNamespaceID == kNameSpaceID_None && aName == nsGkAtoms::src && aValue) {

    LoadSrc();

  }

  return nsXULElement::AfterSetAttr(aNamespaceID, aName,

                                    aValue, aOldValue, aSubjectPrincipal, aNotify);

}

} // namespace dom

} // namespace mozilla