// Copyright 2008, Google Inc.
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
//    * Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
//    * Redistributions in binary form must reproduce the above
// copyright notice, this list of conditions and the following disclaimer
// in the documentation and/or other materials provided with the
// distribution.
//    * Neither the name of Google Inc. nor the names of its
// contributors may be used to endorse or promote products derived from
// this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

#include "chrome/browser/fav_icon_helper.h"

#include "base/gfx/image_operations.h"
#include "base/gfx/png_decoder.h"
#include "base/gfx/png_encoder.h"
#include "chrome/browser/navigation_entry.h"
#include "chrome/browser/navigation_controller.h"
#include "chrome/browser/tab_contents_delegate.h"
#include "chrome/browser/render_view_host.h"
#include "chrome/browser/web_contents.h"
#include "chrome/common/gfx/favicon_size.h"

FavIconHelper::FavIconHelper(WebContents* web_contents)
    : web_contents_(web_contents),
      got_fav_icon_url_(false),
      got_fav_icon_from_history_(false),
      fav_icon_expired_(false) {
}

void FavIconHelper::FetchFavIcon(const GURL& url) {
  cancelable_consumer_.CancelAllRequests();

  url_ = url;

  fav_icon_expired_ = got_fav_icon_from_history_ = got_fav_icon_url_ = false;

  // Request the favicon from the history service. In parallel to this the
  // renderer is going to notify us (well webcontents) when the favicon url is
  // available.
  if (GetHistoryService()) {
    GetHistoryService()->GetFavIconForURL(url_, &cancelable_consumer_,
        NewCallback(this, &FavIconHelper::OnFavIconDataForInitialURL));
  }
}

void FavIconHelper::SetFavIconURL(const GURL& icon_url) {
  NavigationEntry* entry = GetEntry();
  if (!entry)
    return;

  got_fav_icon_url_ = true;

  if (!GetHistoryService())
    return;

  if (!fav_icon_expired_ && entry->IsValidFavIcon() &&
      entry->GetFavIconURL() == icon_url) {
    // We already have the icon, no need to proceed.
    return;
  }

  entry->SetFavIconURL(icon_url);

  if (got_fav_icon_from_history_)
    DownloadFavIconOrAskHistory(entry);
}

Profile* FavIconHelper::profile() {
  return web_contents_->profile();
}

HistoryService* FavIconHelper::GetHistoryService() {
  return profile()->GetHistoryService(Profile::EXPLICIT_ACCESS);
}

void FavIconHelper::SetFavIcon(
    int download_id,
    const GURL& image_url,
    const SkBitmap& image) {
  DownloadRequests::iterator i = download_requests_.find(download_id);
  if (i == download_requests_.end()) {
    // Currently WebContents notifies us of ANY downloads so that it is
    // possible to get here.
    return;
  }

  const SkBitmap& sized_image =
      (image.width() == kFavIconSize && image.height() == kFavIconSize)
      ? image : ConvertToFavIconSize(image);

  if (GetHistoryService() && !profile()->IsOffTheRecord()) {
    std::vector<unsigned char> image_data;
    SkAutoLockPixels icon_lock(sized_image);
    PNGEncoder::Encode(
        reinterpret_cast<unsigned char*>(sized_image.getPixels()),
        PNGEncoder::FORMAT_BGRA, sized_image.width(),
        sized_image.height(), sized_image.width()* 4, false,
        &image_data);
    GetHistoryService()->SetFavIcon(i->second.url, i->second.fav_icon_url,
                                    image_data);
  }

  if (i->second.url == url_) {
    NavigationEntry* entry = GetEntry();
    if (entry)
      UpdateFavIcon(entry, sized_image);
  }

  download_requests_.erase(i);
}

void FavIconHelper::FavIconDownloadFailed(int download_id) {
  DownloadRequests::iterator i = download_requests_.find(download_id);
  if (i != download_requests_.end())
    download_requests_.erase(i);
}

void FavIconHelper::UpdateFavIcon(NavigationEntry* entry,
                                  const std::vector<unsigned char>& data) {
  SkBitmap image;
  PNGDecoder::Decode(&data, &image);
  UpdateFavIcon(entry, image);
}

void FavIconHelper::UpdateFavIcon(NavigationEntry* entry,
                                  const SkBitmap& image) {
  // No matter what happens, we need to mark the favicon as being set.
  entry->SetValidFavIcon(true);

  if (image.empty())
    return;

  entry->SetFavIcon(image);
  if (web_contents_->delegate()) {
    web_contents_->delegate()->NavigationStateChanged(
        web_contents_, TabContents::INVALIDATE_FAVICON);
  }
}

NavigationEntry* FavIconHelper::GetEntry() {
  NavigationEntry* entry = web_contents_->controller()->GetActiveEntry();
  if (entry && entry->GetURL() == url_ &&
      web_contents_->IsActiveEntry(entry->GetPageID())) {
    return entry;
  }
  // If the URL has changed out from under us (as will happen with redirects)
  // return NULL.
  return NULL;
}

void FavIconHelper::OnFavIconDataForInitialURL(
    HistoryService::Handle handle,
    bool know_favicon,
    scoped_refptr<RefCountedBytes> data,
    bool expired,
    GURL icon_url) {
  NavigationEntry* entry = GetEntry();
  if (!entry)
    return;

  got_fav_icon_from_history_ = true;

  fav_icon_expired_ = (know_favicon && expired);

  if (know_favicon && !entry->IsValidFavIcon() &&
      (!got_fav_icon_url_ || entry->GetFavIconURL() == icon_url)) {
    // The db knows the favicon (although it may be out of date) and the entry
    // doesn't have an icon. Set the favicon now, and if the favicon turns out
    // to be expired (or the wrong url) we'll fetch later on. This way the
    // user doesn't see a flash of the default favicon.
    entry->SetFavIconURL(icon_url);
    if (data && !data->data.empty())
      UpdateFavIcon(entry, data->data);
    entry->SetValidFavIcon(true);
  }

  if (know_favicon && !expired) {
    if (got_fav_icon_url_ && entry->GetFavIconURL() != icon_url) {
      // Mapping in the database is wrong. DownloadFavIconOrAskHistory will
      // update the mapping for this url and download the favicon if we don't
      // already have it.
      DownloadFavIconOrAskHistory(entry);
    }
  } else if (got_fav_icon_url_) {
    // We know the official url for the favicon, by either don't have the
    // favicon or its expired. Continue on to DownloadFavIconOrAskHistory to
    // either download or check history again.
    DownloadFavIconOrAskHistory(entry);
  } // else we haven't got the icon url. When we get it we'll ask the
    // renderer to download the icon.
}

void FavIconHelper::DownloadFavIconOrAskHistory(NavigationEntry* entry) {
  DCHECK(entry); // We should only get here if entry is valid.
  if (fav_icon_expired_) {
    // We have the mapping, but the favicon is out of date. Download it now.
    ScheduleDownload(entry);
  } else if (GetHistoryService()) {
    // We don't know the favicon, but we may have previously downloaded the
    // favicon for another page that shares the same favicon. Ask for the
    // favicon given the favicon URL.
    if (profile()->IsOffTheRecord()) {
      GetHistoryService()->GetFavIcon(
          entry->GetFavIconURL(),
          &cancelable_consumer_,
          NewCallback(this, &FavIconHelper::OnFavIconData));
    } else {
      // Ask the history service for the icon. This does two things:
      // 1. Attempts to fetch the favicon data from the database.
      // 2. If the favicon exists in the database, this updates the database to
      //    include the mapping between the page url and the favicon url.
      // This is asynchronous. The history service will call back when done.
      // Issue the request and associate the current page ID with it.
      GetHistoryService()->UpdateFavIconMappingAndFetch(
          entry->GetURL(),
          entry->GetFavIconURL(), &cancelable_consumer_,
          NewCallback(this, &FavIconHelper::OnFavIconData));
    }
  }
}

void FavIconHelper::OnFavIconData(
    HistoryService::Handle handle,
    bool know_favicon,
    scoped_refptr<RefCountedBytes> data,
    bool expired,
    GURL icon_url) {
  NavigationEntry* entry = GetEntry();
  if (!entry)
    return;

  // No need to update the favicon url. By the time we get here
  // UpdateFavIconURL will have set the favicon url.

  if (know_favicon && data && !data->data.empty()) {
    // There is a favicon, set it now. If expired we'll download the current
    // one again, but at least the user will get some icon instead of the
    // default and most likely the current one is fine anyway.
    UpdateFavIcon(entry, data->data);
  }

  if (!know_favicon || expired) {
    // We don't know the favicon, or it is out of date. Request the current one.
    ScheduleDownload(entry);
  }
}

void FavIconHelper::ScheduleDownload(NavigationEntry* entry) {
  const int download_id = web_contents_->render_view_host()->DownloadImage(
      entry->GetFavIconURL(), kFavIconSize);
  if (!download_id) {
    // Download request failed.
    return;
  }
  // Download ids should be unique.
  DCHECK(download_requests_.find(download_id) == download_requests_.end());
  download_requests_[download_id] =
      DownloadRequest(entry->GetURL(), entry->GetFavIconURL());
}

SkBitmap FavIconHelper::ConvertToFavIconSize(const SkBitmap& image) {
  int width = image.width();
  int height = image.height();
  if (width > 0 && height > 0) {
    calc_favicon_target_size(&width, &height);
    return gfx::ImageOperations::Resize(
          image, gfx::ImageOperations::RESIZE_LANCZOS3,
          gfx::Size(width, height));
  }
  return image;
}