/* -*- Mode: C++; tab-width: 2; 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/. */

/* Here is the list, from beppe and glazman:

    href >> A, AREA, BASE, LINK

    src >> FRAME, IFRAME, IMG, INPUT, SCRIPT

    <META http-equiv="refresh" content="3,http://www.acme.com/intro.html">

    longdesc >> FRAME, IFRAME, IMG

    usemap >> IMG, INPUT, OBJECT

    action >> FORM

    background >> BODY

    codebase >> OBJECT, APPLET

    classid >> OBJECT

    data >> OBJECT

    cite >> BLOCKQUOTE, DEL, INS, Q

    profile >> HEAD

    ARCHIVE attribute on APPLET ; warning, it contains a list of URIs.

    Easier way of organizing the list:

    a:      href

    area:   href

    base:   href

    body:   background

    blockquote: cite (not normally rewritable)

    link:   href

    frame:  src, longdesc

    iframe: src, longdesc

    input:  src, usemap

    form:   action

    img:    src, longdesc, usemap

    script: src

    applet: codebase, archive <list>

    object: codebase, data, classid, usemap

    head:   profile

    del:    cite

    ins:    cite

    q:      cite

 */

#include "HTMLURIRefObject.h"

#include "mozilla/mozalloc.h"

#include "mozilla/dom/Attr.h"

#include "mozilla/dom/Element.h"

#include "nsAString.h"

#include "nsDebug.h"

#include "nsDOMAttributeMap.h"

#include "nsError.h"

#include "nsID.h"

#include "nsIDOMElement.h"

#include "nsIDOMNode.h"

#include "nsISupportsUtils.h"

#include "nsString.h"

#include "nsGkAtoms.h"

namespace mozilla {

// String classes change too often and I can't keep up.

// Set this macro to this week's approved case-insensitive compare routine.

#define MATCHES(tagName, str) tagName.EqualsIgnoreCase(str)

HTMLURIRefObject::HTMLURIRefObject()

    : mCurAttrIndex(0), mAttributeCnt(0), mAttrsInited(false) {}

HTMLURIRefObject::~HTMLURIRefObject() {}

// Interfaces for addref and release and queryinterface

NS_IMPL_ISUPPORTS(HTMLURIRefObject, nsIURIRefObject)

NS_IMETHODIMP

HTMLURIRefObject::Reset() {

  mCurAttrIndex = 0;

  return NS_OK;

}

NS_IMETHODIMP

HTMLURIRefObject::GetNextURI(nsAString& aURI) {

  NS_ENSURE_TRUE(mNode, NS_ERROR_NOT_INITIALIZED);

  nsCOMPtr<dom::Element> element = do_QueryInterface(mNode);

  NS_ENSURE_TRUE(element, NS_ERROR_INVALID_ARG);

  // Loop over attribute list:

  if (!mAttrsInited) {

    mAttrsInited = true;

    mAttributeCnt = element->GetAttrCount();

    NS_ENSURE_TRUE(mAttributeCnt, NS_ERROR_FAILURE);

    mCurAttrIndex = 0;

  }

  while (mCurAttrIndex < mAttributeCnt) {

    BorrowedAttrInfo attrInfo = element->GetAttrInfoAt(mCurAttrIndex++);

    NS_ENSURE_ARG_POINTER(attrInfo.mName);

    // href >> A, AREA, BASE, LINK

    if (attrInfo.mName->Equals(nsGkAtoms::href)) {

      if (!element->IsAnyOfHTMLElements(nsGkAtoms::a, nsGkAtoms::area,

                                        nsGkAtoms::base, nsGkAtoms::link)) {

        continue;

      }

      attrInfo.mValue->ToString(aURI);

      // href pointing to a named anchor doesn't count

      if (StringBeginsWith(aURI, NS_LITERAL_STRING("#"))) {

        aURI.Truncate();

        return NS_ERROR_INVALID_ARG;

      }

      return NS_OK;

    }

    // src >> FRAME, IFRAME, IMG, INPUT, SCRIPT

    else if (attrInfo.mName->Equals(nsGkAtoms::src)) {

      if (!element->IsAnyOfHTMLElements(nsGkAtoms::img, nsGkAtoms::frame,

                                        nsGkAtoms::iframe, nsGkAtoms::input,

                                        nsGkAtoms::script)) {

        continue;

      }

      attrInfo.mValue->ToString(aURI);

      return NS_OK;

    }

    //<META http-equiv="refresh" content="3,http://www.acme.com/intro.html">

    else if (attrInfo.mName->Equals(nsGkAtoms::content)) {

      if (!element->IsHTMLElement(nsGkAtoms::meta)) {

        continue;

      }

      // XXXbz And if it is?

    }

    // longdesc >> FRAME, IFRAME, IMG

    else if (attrInfo.mName->Equals(nsGkAtoms::longdesc)) {

      if (!element->IsAnyOfHTMLElements(nsGkAtoms::img, nsGkAtoms::frame,

                                        nsGkAtoms::iframe)) {

        continue;

      }

      // XXXbz And if it is?

    }

    // usemap >> IMG, INPUT, OBJECT

    else if (attrInfo.mName->Equals(nsGkAtoms::usemap)) {

      if (!element->IsAnyOfHTMLElements(nsGkAtoms::img, nsGkAtoms::input,

                                        nsGkAtoms::object)) {

        continue;

      }

    }

    // action >> FORM

    else if (attrInfo.mName->Equals(nsGkAtoms::action)) {

      if (!element->IsHTMLElement(nsGkAtoms::form)) {

        continue;

      }

      // XXXbz And if it is?

    }

    // background >> BODY

    else if (attrInfo.mName->Equals(nsGkAtoms::background)) {

      if (!element->IsHTMLElement(nsGkAtoms::body)) {

        continue;

      }

      // XXXbz And if it is?

    }

    // codebase >> OBJECT

    else if (attrInfo.mName->Equals(nsGkAtoms::codebase)) {

      if (!element->IsHTMLElement(nsGkAtoms::object)) {

        continue;

      }

      // XXXbz And if it is?

    }

    // classid >> OBJECT

    else if (attrInfo.mName->Equals(nsGkAtoms::classid)) {

      if (!element->IsHTMLElement(nsGkAtoms::object)) {

        continue;

      }

      // XXXbz And if it is?

    }

    // data >> OBJECT

    else if (attrInfo.mName->Equals(nsGkAtoms::data)) {

      if (!element->IsHTMLElement(nsGkAtoms::object)) {

        continue;

      }

      // XXXbz And if it is?

    }

    // cite >> BLOCKQUOTE, DEL, INS, Q

    else if (attrInfo.mName->Equals(nsGkAtoms::cite)) {

      if (!element->IsAnyOfHTMLElements(nsGkAtoms::blockquote, nsGkAtoms::q,

                                        nsGkAtoms::del, nsGkAtoms::ins)) {

        continue;

      }

      // XXXbz And if it is?

    }

    // profile >> HEAD

    else if (attrInfo.mName->Equals(nsGkAtoms::profile)) {

      if (!element->IsHTMLElement(nsGkAtoms::head)) {

        continue;

      }

      // XXXbz And if it is?

    }

  }

  // Return a code to indicate that there are no more,

  // to distinguish that case from real errors.

  return NS_ERROR_NOT_AVAILABLE;

}

NS_IMETHODIMP

HTMLURIRefObject::RewriteAllURIs(const nsAString& aOldPat,

                                 const nsAString& aNewPat, bool aMakeRel) {

  return NS_ERROR_NOT_IMPLEMENTED;

}

NS_IMETHODIMP

HTMLURIRefObject::GetNode(nsIDOMNode** aNode) {

  NS_ENSURE_TRUE(mNode, NS_ERROR_NOT_INITIALIZED);

  NS_ENSURE_TRUE(aNode, NS_ERROR_NULL_POINTER);

  *aNode = mNode.get();

  NS_ADDREF(*aNode);

  return NS_OK;

}

NS_IMETHODIMP

HTMLURIRefObject::SetNode(nsIDOMNode* aNode) {

  mNode = aNode;

  nsAutoString dummyURI;

  if (NS_SUCCEEDED(GetNextURI(dummyURI))) {

    mCurAttrIndex = 0;  // Reset so we'll get the first node next time

    return NS_OK;

  }

  // If there weren't any URIs in the attributes,

  // then don't accept this node.

  mNode = nullptr;

  return NS_ERROR_INVALID_ARG;

}

}  // namespace mozilla

nsresult NS_NewHTMLURIRefObject(nsIURIRefObject** aResult, nsIDOMNode* aNode) {

  RefPtr<mozilla::HTMLURIRefObject> refObject = new mozilla::HTMLURIRefObject();

  nsresult rv = refObject->SetNode(aNode);

  if (NS_FAILED(rv)) {

    *aResult = 0;

    return rv;

  }

  refObject.forget(aResult);

  return NS_OK;

}