Adds support for saving favicon size into history database.

chrome/browser/history/thumbnail_database.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/history/thumbnail_database.h"
 
 #include <algorithm>
 #include <string>
 
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/memory/ref_counted_memory.h"
 #include "base/string_util.h"
 #include "base/time.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/diagnostics/sqlite_diagnostics.h"
 #include "chrome/browser/history/history_publisher.h"
 #include "chrome/browser/history/top_sites.h"
 #include "chrome/browser/history/url_database.h"
 #include "chrome/common/thumbnail_score.h"
 #include "sql/statement.h"
 #include "sql/transaction.h"
 #include "ui/gfx/favicon_size.h"
 #include "ui/gfx/image/image_util.h"
+#include "ui/gfx/size.h"
 
 #if defined(OS_MACOSX)
 #include "base/mac/mac_util.h"
 #endif
 
 static void FillIconMapping(const sql::Statement& statement,
                             const GURL& page_url,
                             history::IconMapping* icon_mapping) {
   icon_mapping->mapping_id = statement.ColumnInt64(0);
   icon_mapping->icon_id = statement.ColumnInt64(1);
   icon_mapping->icon_type =
       static_cast<history::IconType>(statement.ColumnInt(2));
+  icon_mapping->icon_pixel_size =
+      gfx::Size(statement.ColumnInt(3), statement.ColumnInt(4));
   icon_mapping->page_url = page_url;
 }
 
 namespace history {
 
 // Version number of the database.
 static const int kCurrentVersionNumber = 6;
 static const int kCompatibleVersionNumber = 6;
 
 // Use 90 quality (out of 100) which is pretty high, because we're very
@@ skipping 349 matching lines @@
 bool ThumbnailDatabase::SetFaviconLastUpdateTime(FaviconID icon_id,
                                                  base::Time time) {
   sql::Statement statement(db_.GetCachedStatement(SQL_FROM_HERE,
       "UPDATE favicons SET last_updated=? WHERE id=?"));
   statement.BindInt64(0, time.ToTimeT());
   statement.BindInt64(1, icon_id);
 
   return statement.Run();
 }
 
-FaviconID ThumbnailDatabase::GetFaviconIDForFaviconURL(const GURL& icon_url,
-                                                       int required_icon_type,
-                                                       IconType* icon_type) {
+void ThumbnailDatabase::GetFaviconIDsForFaviconURL(
+    const GURL& icon_url,
+    int required_icon_type,
+    std::vector<FaviconIDAndSize>* favicon_id_size_listing) {
   sql::Statement statement(db_.GetCachedStatement(SQL_FROM_HERE,
-      "SELECT id, icon_type FROM favicons WHERE url=? AND (icon_type & ? > 0) "
-      "ORDER BY icon_type DESC"));
+      "SELECT id, icon_type, width, height FROM favicons "
+      "WHERE url=? AND (icon_type & ? > 0) ORDER BY icon_type DESC"));
   statement.BindString(0, URLDatabase::GURLToDatabaseURL(icon_url));
   statement.BindInt(1, required_icon_type);
 
-  if (!statement.Step())
-    return 0;  // not cached
+  IconType chosen_icon_type = INVALID_ICON;
+  while (statement.Step()) {
+    IconType icon_type = static_cast<IconType>(statement.ColumnInt(1));
+    if (chosen_icon_type == INVALID_ICON)
+      chosen_icon_type = icon_type;
+    else if (icon_type != chosen_icon_type)
+      break;
 
-  if (icon_type)
-    *icon_type = static_cast<IconType>(statement.ColumnInt(1));
-  return statement.ColumnInt64(0);
+    FaviconIDAndSize favicon_id_size;
+    favicon_id_size.icon_id = statement.ColumnInt64(0);
+    favicon_id_size.icon_size = gfx::Size(statement.ColumnInt(2),
+                                          statement.ColumnInt(3));
+    favicon_id_size_listing->push_back(favicon_id_size);
+  }
 }
 
 bool ThumbnailDatabase::GetFavicon(
     FaviconID icon_id,
     base::Time* last_updated,
     std::vector<unsigned char>* png_icon_data,
     GURL* icon_url,
+    gfx::Size* pixel_size,
     IconType* icon_type) {
   DCHECK(icon_id);
 
   sql::Statement statement(db_.GetCachedStatement(SQL_FROM_HERE,
-      "SELECT last_updated, image_data, url, icon_type "
+      "SELECT last_updated, image_data, url, icon_type, width, height "
       "FROM favicons WHERE id=?"));
   statement.BindInt64(0, icon_id);
 
   if (!statement.Step())
     return false;  // No entry for the id.
 
   if (last_updated)
     *last_updated = base::Time::FromTimeT(statement.ColumnInt64(0));
   if (statement.ColumnByteLength(1) > 0)
     statement.ColumnBlobAsVector(1, png_icon_data);
   if (icon_url)
     *icon_url = GURL(statement.ColumnString(2));
   if (icon_type)
     *icon_type = static_cast<history::IconType>(statement.ColumnInt(3));
+  if (pixel_size)
+    *pixel_size = gfx::Size(statement.ColumnInt(4), statement.ColumnInt(5));
 
   return true;
 }
 
 FaviconID ThumbnailDatabase::AddFavicon(const GURL& icon_url,
+                                        const gfx::Size& pixel_size,
                                         IconType icon_type) {
-  int default_size = icon_type == FAVICON ? gfx::kFaviconSize : 0;
   sql::Statement statement(db_.GetCachedStatement(SQL_FROM_HERE,
       "INSERT INTO favicons (url, icon_type, width, height) VALUES "
       "(?, ?, ?, ?)"));
   statement.BindString(0, URLDatabase::GURLToDatabaseURL(icon_url));
   statement.BindInt(1, icon_type);
-  statement.BindInt(2, default_size);
-  statement.BindInt(3, default_size);
+  statement.BindInt(2, pixel_size.width());
+  statement.BindInt(3, pixel_size.height());
 
   if (!statement.Run())
     return 0;
   return db_.GetLastInsertRowId();
 }
 
 bool ThumbnailDatabase::DeleteFavicon(FaviconID id) {
   sql::Statement statement(db_.GetCachedStatement(SQL_FROM_HERE,
       "DELETE FROM favicons WHERE id = ?"));
   statement.BindInt64(0, id);
 
   return statement.Run();
 }
 
-bool ThumbnailDatabase::GetIconMappingForPageURL(const GURL& page_url,
-                                                 IconType required_icon_type,
-                                                 IconMapping* icon_mapping) {
-  std::vector<IconMapping> icon_mappings;
-  if (!GetIconMappingsForPageURL(page_url, &icon_mappings))
+bool ThumbnailDatabase::GetIconMappingsForPageURL(
+    const GURL& page_url,
+    int required_icon_types,
+    std::vector<IconMapping>* icon_mappings) {
+  std::vector<IconMapping> unfiltered_icon_mappings;
+  if (!GetIconMappingsForPageURL(page_url, &unfiltered_icon_mappings))
     return false;
 
-  for (std::vector<IconMapping>::iterator m = icon_mappings.begin();
-      m != icon_mappings.end(); ++m) {
-    if (m->icon_type == required_icon_type) {
-      if (icon_mapping != NULL)
-        *icon_mapping = *m;
-      return true;
+  for (std::vector<IconMapping>::iterator m = unfiltered_icon_mappings.begin();
+       m != unfiltered_icon_mappings.end(); ++m) {
+    if (m->icon_type & required_icon_types) {
+      if (icon_mappings == NULL)
+        return true;
+
+      icon_mappings->push_back(*m);
     }
   }
 
-  return false;
+  return (icon_mappings == NULL) ? false : !icon_mappings->empty();
 }
 
 bool ThumbnailDatabase::GetIconMappingsForPageURL(
     const GURL& page_url,
     std::vector<IconMapping>* mapping_data) {
   sql::Statement statement(db_.GetCachedStatement(SQL_FROM_HERE,
-      "SELECT icon_mapping.id, icon_mapping.icon_id, favicons.icon_type "
+      "SELECT icon_mapping.id, icon_mapping.icon_id, favicons.icon_type, "
+      "favicons.width, favicons.height "
       "FROM icon_mapping "
       "INNER JOIN favicons "
       "ON icon_mapping.icon_id = favicons.id "
       "WHERE icon_mapping.page_url=? "
       "ORDER BY favicons.icon_type DESC"));
   statement.BindString(0, URLDatabase::GURLToDatabaseURL(page_url));
 
   bool result = false;
   while (statement.Step()) {
     result = true;
@@ skipping 311 matching lines @@
   // have width & height of 16px.
   success &= db_.Execute(
       "UPDATE favicons SET width='16', height='16' WHERE icon_type='1'");
 
   meta_table_.SetVersionNumber(6);
   meta_table_.SetCompatibleVersionNumber(std::min(6, kCompatibleVersionNumber));
   return success;
 }
 
 }  // namespace history