michael@0: /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 4 -*-
michael@0:  *
michael@0:  * This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: #include "nsIconChannel.h"
michael@0: #include "nsIconURI.h"
michael@0: #include "nsIconProtocolHandler.h"
michael@0: #include "nsIURL.h"
michael@0: #include "nsCRT.h"
michael@0: #include "nsCOMPtr.h"
michael@0: #include "nsIComponentManager.h"
michael@0: #include "nsIServiceManager.h"
michael@0: #include "nsNetCID.h"
michael@0: 
michael@0: ////////////////////////////////////////////////////////////////////////////////
michael@0: 
michael@0: nsIconProtocolHandler::nsIconProtocolHandler() 
michael@0: {
michael@0: }
michael@0: 
michael@0: nsIconProtocolHandler::~nsIconProtocolHandler() 
michael@0: {}
michael@0: 
michael@0: NS_IMPL_ISUPPORTS(nsIconProtocolHandler, nsIProtocolHandler, nsISupportsWeakReference)
michael@0: 
michael@0:     
michael@0: ////////////////////////////////////////////////////////////////////////////////
michael@0: // nsIProtocolHandler methods:
michael@0: 
michael@0: NS_IMETHODIMP nsIconProtocolHandler::GetScheme(nsACString &result) 
michael@0: {
michael@0:   result = "moz-icon";
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsIconProtocolHandler::GetDefaultPort(int32_t *result) 
michael@0: {
michael@0:   *result = 0;
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsIconProtocolHandler::AllowPort(int32_t port, const char *scheme, bool *_retval)
michael@0: {
michael@0:     // don't override anything.  
michael@0:     *_retval = false;
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsIconProtocolHandler::GetProtocolFlags(uint32_t *result)
michael@0: {
michael@0:   *result = URI_NORELATIVE | URI_NOAUTH | URI_IS_UI_RESOURCE |
michael@0:             URI_IS_LOCAL_RESOURCE;
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsIconProtocolHandler::NewURI(const nsACString &aSpec,
michael@0:                                             const char *aOriginCharset, // ignored
michael@0:                                             nsIURI *aBaseURI,
michael@0:                                             nsIURI **result) 
michael@0: {
michael@0:   
michael@0:   nsCOMPtr<nsIURI> uri = new nsMozIconURI();
michael@0:   if (!uri) return NS_ERROR_OUT_OF_MEMORY;
michael@0: 
michael@0:   nsresult rv = uri->SetSpec(aSpec);
michael@0:   if (NS_FAILED(rv)) return rv;
michael@0: 
michael@0:   NS_ADDREF(*result = uri);
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP nsIconProtocolHandler::NewChannel(nsIURI* url, nsIChannel* *result)
michael@0: {
michael@0:   NS_ENSURE_ARG_POINTER(url);
michael@0:   nsIconChannel* channel = new nsIconChannel;
michael@0:   if (!channel)
michael@0:     return NS_ERROR_OUT_OF_MEMORY;
michael@0:   NS_ADDREF(channel);
michael@0: 
michael@0:   nsresult rv = channel->Init(url);
michael@0:   if (NS_FAILED(rv)) {
michael@0:     NS_RELEASE(channel);
michael@0:     return rv;
michael@0:   }
michael@0: 
michael@0:   *result = channel;
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: ////////////////////////////////////////////////////////////////////////////////