docshell/base/nsDownloadHistory.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
   2  * vim: sw=2 ts=2 sts=2
   3  * ***** BEGIN LICENSE BLOCK *****
   4  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   5  *
   6  * The contents of this file are subject to the Mozilla Public License Version
   7  * 1.1 (the "License"); you may not use this file except in compliance with
   8  * the License. You may obtain a copy of the License at
   9  * http://www.mozilla.org/MPL/
  10  *
  11  * Software distributed under the License is distributed on an "AS IS" basis,
  12  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  13  * for the specific language governing rights and limitations under the
  14  * License.
  15  *
  16  * The Original Code is mozilla.org code.
  17  *
  18  * The Initial Developer of the Original Code is
  19  * Mozilla Corporation.
  20  * Portions created by the Initial Developer are Copyright (C) 2007
  21  * the Initial Developer. All Rights Reserved.
  22  *
  23  * Contributor(s):
  24  *   Shawn Wilsher <me@shawnwilsher.com> (Original Author)
  25  *
  26  * Alternatively, the contents of this file may be used under the terms of
  27  * either the GNU General Public License Version 2 or later (the "GPL"), or
  28  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  29  * in which case the provisions of the GPL or the LGPL are applicable instead
  30  * of those above. If you wish to allow use of your version of this file only
  31  * under the terms of either the GPL or the LGPL, and not to allow others to
  32  * use your version of this file under the terms of the MPL, indicate your
  33  * decision by deleting the provisions above and replace them with the notice
  34  * and other provisions required by the GPL or the LGPL. If you do not delete
  35  * the provisions above, a recipient may use your version of this file under
  36  * the terms of any one of the MPL, the GPL or the LGPL.
  37  *
  38  * ***** END LICENSE BLOCK ***** */
  39
  40 #include "nsDownloadHistory.h"
  41 #include "nsCOMPtr.h"
  42 #include "nsServiceManagerUtils.h"
  43 #include "nsIGlobalHistory.h"
  44 #include "nsIGlobalHistory2.h"
  45 #include "nsIObserverService.h"
  46 #include "nsIURI.h"
  47 #include "nsIComponentRegistrar.h"
  48 #include "nsDocShellCID.h"
  49
  50 ////////////////////////////////////////////////////////////////////////////////
  51 //// nsDownloadHistory
  52
  53 NS_IMPL_ISUPPORTS1(nsDownloadHistory, nsIDownloadHistory)
  54
  55 ////////////////////////////////////////////////////////////////////////////////
  56 //// nsIDownloadHistory
  57
  58 NS_IMETHODIMP
  59 nsDownloadHistory::AddDownload(nsIURI *aSource,
  60                                nsIURI *aReferrer,
  61                                PRTime aStartTime)
  62 {
  63   NS_ENSURE_ARG_POINTER(aSource);
  64
  65   nsCOMPtr<nsIGlobalHistory2> history =
  66     do_GetService("@mozilla.org/browser/global-history;2");
  67   if (!history)
  68     return NS_ERROR_NOT_AVAILABLE;
  69
  70   PRBool visited;
  71   nsresult rv = history->IsVisited(aSource, &visited);
  72   NS_ENSURE_SUCCESS(rv, rv);
  73
  74   rv = history->AddURI(aSource, PR_FALSE, PR_TRUE, aReferrer);
  75   NS_ENSURE_SUCCESS(rv, rv);
  76
  77   if (!visited) {
  78     nsCOMPtr<nsIObserverService> os =
  79       do_GetService("@mozilla.org/observer-service;1");
  80     if (os)
  81       os->NotifyObservers(aSource, NS_LINK_VISITED_EVENT_TOPIC, nsnull);
  82   }
  83
  84   return NS_OK;
  85 }