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 nsresult
  56 nsDownloadHistory::RegisterSelf(nsIComponentManager *aCompMgr,
  57                                 nsIFile *aPath,
  58                                 const char *aLoaderStr,
  59                                 const char *aType,
  60                                 const nsModuleComponentInfo *aInfo)
  61 {
  62   nsCOMPtr<nsIComponentRegistrar> compReg(do_QueryInterface(aCompMgr));
  63   if (!compReg)
  64     return NS_ERROR_UNEXPECTED;
  65
  66   PRBool registered;
  67   nsresult rv = compReg->IsContractIDRegistered(NS_DOWNLOADHISTORY_CONTRACTID,
  68                                                 &registered);
  69   NS_ENSURE_SUCCESS(rv, rv);
  70
  71   // If someone has already registered the contractID, we don't want to register
  72   // it ourselves.  We do want to register though in case someone wants to access
  73   // this implementation (by CID).
  74   if (registered) {
  75     return compReg->RegisterFactoryLocation(GetCID(), "nsDownloadHistory",
  76                                             nsnull, aPath, aLoaderStr, aType);
  77   }
  78
  79   return compReg->RegisterFactoryLocation(GetCID(), "nsDownloadHistory",
  80                                           NS_DOWNLOADHISTORY_CONTRACTID,
  81                                           aPath, aLoaderStr, aType);
  82 }
  83
  84 ////////////////////////////////////////////////////////////////////////////////
  85 //// nsIDownloadHistory
  86
  87 NS_IMETHODIMP
  88 nsDownloadHistory::AddDownload(nsIURI *aSource,
  89                                nsIURI *aReferrer,
  90                                PRTime aStartTime)
  91 {
  92   NS_ENSURE_ARG_POINTER(aSource);
  93
  94   nsCOMPtr<nsIGlobalHistory2> history =
  95     do_GetService("@mozilla.org/browser/global-history;2");
  96   if (!history)
  97     return NS_ERROR_NOT_AVAILABLE;
  98
  99   PRBool visited;
 100   nsresult rv = history->IsVisited(aSource, &visited);
 101   NS_ENSURE_SUCCESS(rv, rv);
 102
 103   rv = history->AddURI(aSource, PR_FALSE, PR_TRUE, aReferrer);
 104   NS_ENSURE_SUCCESS(rv, rv);
 105
 106   if (!visited) {
 107     nsCOMPtr<nsIObserverService> os =
 108       do_GetService("@mozilla.org/observer-service;1");
 109     if (os)
 110       os->NotifyObservers(aSource, NS_LINK_VISITED_EVENT_TOPIC, nsnull);
 111   }
 112
 113   return NS_OK;
 114 }