editor/libeditor/html/nsHTMLURIRefObject.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* ***** BEGIN LICENSE BLOCK *****
   3  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   4  *
   5  * The contents of this file are subject to the Mozilla Public License Version
   6  * 1.1 (the "License"); you may not use this file except in compliance with
   7  * the License. You may obtain a copy of the License at
   8  * http://www.mozilla.org/MPL/
   9  *
  10  * Software distributed under the License is distributed on an "AS IS" basis,
  11  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  12  * for the specific language governing rights and limitations under the
  13  * License.
  14  *
  15  * The Original Code is mozilla.org code.
  16  *
  17  * The Initial Developer of the Original Code is
  18  * Netscape Communications Corporation.
  19  * Portions created by the Initial Developer are Copyright (C) 1998
  20  * the Initial Developer. All Rights Reserved.
  21  *
  22  * Contributor(s):
  23  *
  24  * Alternatively, the contents of this file may be used under the terms of
  25  * either of the GNU General Public License Version 2 or later (the "GPL"),
  26  * or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  27  * in which case the provisions of the GPL or the LGPL are applicable instead
  28  * of those above. If you wish to allow use of your version of this file only
  29  * under the terms of either the GPL or the LGPL, and not to allow others to
  30  * use your version of this file under the terms of the MPL, indicate your
  31  * decision by deleting the provisions above and replace them with the notice
  32  * and other provisions required by the GPL or the LGPL. If you do not delete
  33  * the provisions above, a recipient may use your version of this file under
  34  * the terms of any one of the MPL, the GPL or the LGPL.
  35  *
  36  * ***** END LICENSE BLOCK ***** */
  37
  38 /* Here is the list, from beppe and glazman:
  39     href >> A, AREA, BASE, LINK
  40     src >> FRAME, IFRAME, IMG, INPUT, SCRIPT
  41     <META http-equiv="refresh" content="3,http://www.acme.com/intro.html">
  42     longdesc >> FRAME, IFRAME, IMG
  43     usemap >> IMG, INPUT, OBJECT
  44     action >> FORM
  45     background >> BODY
  46     codebase >> OBJECT, APPLET
  47     classid >> OBJECT
  48     data >> OBJECT
  49     cite >> BLOCKQUOTE, DEL, INS, Q
  50     profile >> HEAD
  51     ARCHIVE attribute on APPLET ; warning, it contains a list of URIs.
  52
  53     Easier way of organizing the list:
  54     a:      href
  55     area:   href
  56     base:   href
  57     body:   background
  58     blockquote: cite (not normally rewritable)
  59     link:   href
  60     frame:  src, longdesc
  61     iframe: src, longdesc
  62     input:  src, usemap
  63     form:   action
  64     img:    src, longdesc, usemap
  65     script: src
  66     applet: codebase, archive <list>
  67     object: codebase, data, classid, usemap
  68     head:   profile
  69     del:    cite
  70     ins:    cite
  71     q:      cite
  72  */
  73
  74 /* Here is how to open a channel for testing
  75    (from embed/qa/testembed/Tests.cpp):
  76
  77   nsCOMPtr<nsIChannel> theChannel;
  78   nsCString uri;
  79   nsCOMPtr<nsIURI> theURI;
  80   rv = NS_NewURI(getter_AddRefs(theURI), theSpec);
  81   if (!theURI)
  82     error;
  83   rv = NS_OpenURI(getter_AddRefs(theChannel), theURI, nsnull, theLoadGroup);
  84   if (!theChannel)
  85     error;
  86   nsCOMPtr<nsILoadGroup> theLoadGroup(do_CreateInstance(NS_LOADGROUP_CONTRACTID));
  87   if (!theLoadGroup)
  88     error;
  89                 nsCOMPtr<nsIStreamListener> listener(static_cast<nsIStreamListener*>(qaBrowserImpl));
  90                 //nsCOMPtr<nsIWeakReference> thisListener(do_GetWeakReference(listener));
  91                 //qaWebBrowser->AddWebBrowserListener(thisListener, NS_GET_IID(nsIStreamListener));
  92
  93                 // this calls nsIStreamListener::OnDataAvailable()
  94                 rv = theChannel->AsyncOpen(listener, nsnull);
  95
  96                 nsCOMPtr<nsIRequest> theRequest = do_QueryInterface(theChannel);
  97     // Now we can do things on nsIRequest (like what?)
  98  */
  99
 100 #include "nsHTMLURIRefObject.h"
 101
 102 #include "nsAString.h"
 103 #include "nsString.h"
 104 #include "nsIDOMAttr.h"
 105 #include "nsIDOMElement.h"
 106
 107 // String classes change too often and I can't keep up.
 108 // Set this macro to this week's approved case-insensitive compare routine.
 109 #define MATCHES(tagName, str) tagName.EqualsIgnoreCase(str)
 110
 111 nsHTMLURIRefObject::nsHTMLURIRefObject()
 112 {
 113   mCurAttrIndex = mAttributeCnt = 0;
 114 }
 115
 116 nsHTMLURIRefObject::~nsHTMLURIRefObject()
 117 {
 118 }
 119
 120 //Interfaces for addref and release and queryinterface
 121 NS_IMPL_ISUPPORTS1(nsHTMLURIRefObject, nsIURIRefObject)
 122
 123 NS_IMETHODIMP
 124 nsHTMLURIRefObject::Reset()
 125 {
 126   mCurAttrIndex = 0;
 127   return NS_OK;
 128 }
 129
 130 NS_IMETHODIMP
 131 nsHTMLURIRefObject::GetNextURI(nsAString & aURI)
 132 {
 133   if (!mNode)
 134     return NS_ERROR_NOT_INITIALIZED;
 135
 136   nsAutoString tagName;
 137   nsresult rv = mNode->GetNodeName(tagName);
 138   if (NS_FAILED(rv))
 139   return rv;
 140
 141   // Loop over attribute list:
 142   if (!mAttributes)
 143   {
 144     nsCOMPtr<nsIDOMElement> element (do_QueryInterface(mNode));
 145     if (!element)
 146       return NS_ERROR_INVALID_ARG;
 147
 148     mCurAttrIndex = 0;
 149     mNode->GetAttributes(getter_AddRefs(mAttributes));
 150     if (!mAttributes)
 151       return NS_ERROR_NOT_INITIALIZED;
 152
 153     rv = mAttributes->GetLength(&mAttributeCnt);
 154     NS_ENSURE_SUCCESS(rv, rv);
 155     if (!mAttributeCnt) return NS_ERROR_FAILURE;
 156     mCurAttrIndex = 0;
 157   }
 158 #ifdef DEBUG_akkana
 159   printf("Looking at tag '%s'\n",
 160          NS_LossyConvertUTF16toASCII(tagName).get());
 161 #endif
 162   while (mCurAttrIndex < mAttributeCnt)
 163   {
 164     nsCOMPtr<nsIDOMNode> attrNode;
 165     rv = mAttributes->Item(mCurAttrIndex++, getter_AddRefs(attrNode));
 166       // XXX Does Item() addref, or not?
 167       // The comparable code in nsEditor assumes it doesn't.
 168     NS_ENSURE_SUCCESS(rv, rv);
 169     NS_ENSURE_ARG_POINTER(attrNode);
 170     nsCOMPtr<nsIDOMAttr> curAttrNode (do_QueryInterface(attrNode));
 171     NS_ENSURE_ARG_POINTER(curAttrNode);
 172     nsString curAttr;
 173     rv = curAttrNode->GetName(curAttr);
 174     NS_ENSURE_SUCCESS(rv, rv);
 175
 176     // href >> A, AREA, BASE, LINK
 177 #ifdef DEBUG_akkana
 178     printf("Trying to match attribute '%s'\n",
 179            NS_LossyConvertUTF16toASCII(curAttr).get());
 180 #endif
 181     if (MATCHES(curAttr, "href"))
 182     {
 183       if (!MATCHES(tagName, "a") && !MATCHES(tagName, "area")
 184           && !MATCHES(tagName, "base") && !MATCHES(tagName, "link"))
 185         continue;
 186       rv = curAttrNode->GetValue(aURI);
 187       NS_ENSURE_SUCCESS(rv, rv);
 188       nsString uri (aURI);
 189       // href pointing to a named anchor doesn't count
 190       if (aURI.First() != PRUnichar('#'))
 191         return NS_OK;
 192       aURI.Truncate();
 193       return NS_ERROR_INVALID_ARG;
 194     }
 195     // src >> FRAME, IFRAME, IMG, INPUT, SCRIPT
 196     else if (MATCHES(curAttr, "src"))
 197     {
 198       if (!MATCHES(tagName, "img")
 199           && !MATCHES(tagName, "frame") && !MATCHES(tagName, "iframe")
 200           && !MATCHES(tagName, "input") && !MATCHES(tagName, "script"))
 201         continue;
 202       return curAttrNode->GetValue(aURI);
 203     }
 204     //<META http-equiv="refresh" content="3,http://www.acme.com/intro.html">
 205     else if (MATCHES(curAttr, "content"))
 206     {
 207       if (!MATCHES(tagName, "meta"))
 208         continue;
 209     }
 210     // longdesc >> FRAME, IFRAME, IMG
 211     else if (MATCHES(curAttr, "longdesc"))
 212     {
 213       if (!MATCHES(tagName, "img")
 214           && !MATCHES(tagName, "frame") && !MATCHES(tagName, "iframe"))
 215         continue;
 216     }
 217     // usemap >> IMG, INPUT, OBJECT
 218     else if (MATCHES(curAttr, "usemap"))
 219     {
 220       if (!MATCHES(tagName, "img")
 221           && !MATCHES(tagName, "input") && !MATCHES(tagName, "object"))
 222         continue;
 223     }
 224     // action >> FORM
 225     else if (MATCHES(curAttr, "action"))
 226     {
 227       if (!MATCHES(tagName, "form"))
 228         continue;
 229     }
 230     // background >> BODY
 231     else if (MATCHES(curAttr, "background"))
 232     {
 233       if (!MATCHES(tagName, "body"))
 234         continue;
 235     }
 236     // codebase >> OBJECT, APPLET
 237     else if (MATCHES(curAttr, "codebase"))
 238     {
 239       if (!MATCHES(tagName, "meta"))
 240         continue;
 241     }
 242     // classid >> OBJECT
 243     else if (MATCHES(curAttr, "classid"))
 244     {
 245       if (!MATCHES(tagName, "object"))
 246         continue;
 247     }
 248     // data >> OBJECT
 249     else if (MATCHES(curAttr, "data"))
 250     {
 251       if (!MATCHES(tagName, "object"))
 252         continue;
 253     }
 254     // cite >> BLOCKQUOTE, DEL, INS, Q
 255     else if (MATCHES(curAttr, "cite"))
 256     {
 257       if (!MATCHES(tagName, "blockquote") && !MATCHES(tagName, "q")
 258           && !MATCHES(tagName, "del") && !MATCHES(tagName, "ins"))
 259         continue;
 260     }
 261     // profile >> HEAD
 262     else if (MATCHES(curAttr, "profile"))
 263     {
 264       if (!MATCHES(tagName, "head"))
 265         continue;
 266     }
 267     // archive attribute on APPLET; warning, it contains a list of URIs.
 268     else if (MATCHES(curAttr, "archive"))
 269     {
 270       if (!MATCHES(tagName, "applet"))
 271         continue;
 272     }
 273   }
 274   // Return a code to indicate that there are no more,
 275   // to distinguish that case from real errors.
 276   return NS_ERROR_NOT_AVAILABLE;
 277 }
 278
 279 NS_IMETHODIMP
 280 nsHTMLURIRefObject::RewriteAllURIs(const nsAString & aOldPat,
 281                             const nsAString & aNewPat,
 282                             PRBool aMakeRel)
 283 {
 284 #ifdef DEBUG_akkana
 285   printf("Can't rewrite URIs yet\n");
 286 #endif
 287   return NS_ERROR_NOT_IMPLEMENTED;
 288 }
 289
 290 NS_IMETHODIMP
 291 nsHTMLURIRefObject::GetNode(nsIDOMNode** aNode)
 292 {
 293   if (!mNode)
 294     return NS_ERROR_NOT_INITIALIZED;
 295   if (!aNode)
 296     return NS_ERROR_NULL_POINTER;
 297   *aNode = mNode.get();
 298   NS_ADDREF(*aNode);
 299   return NS_OK;
 300 }
 301
 302 NS_IMETHODIMP
 303 nsHTMLURIRefObject::SetNode(nsIDOMNode *aNode)
 304 {
 305   mNode = aNode;
 306   nsAutoString dummyURI;
 307   if (NS_SUCCEEDED(GetNextURI(dummyURI)))
 308   {
 309     mCurAttrIndex = 0;    // Reset so we'll get the first node next time
 310     return NS_OK;
 311   }
 312
 313   // If there weren't any URIs in the attributes,
 314   // then don't accept this node.
 315   mNode = 0;
 316   return NS_ERROR_INVALID_ARG;
 317 }
 318
 319 nsresult NS_NewHTMLURIRefObject(nsIURIRefObject** aResult, nsIDOMNode* aNode)
 320 {
 321   nsHTMLURIRefObject* refObject = new nsHTMLURIRefObject();
 322   if (!refObject) return NS_ERROR_OUT_OF_MEMORY;
 323   nsresult rv = refObject->SetNode(aNode);
 324   if (NS_FAILED(rv)) {
 325     *aResult = 0;
 326     delete refObject;
 327     return rv;
 328   }
 329   return refObject->QueryInterface(NS_GET_IID(nsIURIRefObject),
 330                                    (void**)aResult);
 331 }
 332