| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/webdata/web_apps_table.h" | 5 #include "chrome/browser/webdata/web_apps_table.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "chrome/browser/history/history_database.h" | 8 #include "chrome/browser/history/history_database.h" |
| 9 #include "googleurl/src/gurl.h" | 9 #include "googleurl/src/gurl.h" |
| 10 #include "sql/statement.h" | 10 #include "sql/statement.h" |
| (...skipping 37 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 48 } | 48 } |
| 49 return true; | 49 return true; |
| 50 } | 50 } |
| 51 | 51 |
| 52 bool WebAppsTable::SetWebAppImage(const GURL& url, const SkBitmap& image) { | 52 bool WebAppsTable::SetWebAppImage(const GURL& url, const SkBitmap& image) { |
| 53 // Don't bother with a cached statement since this will be a relatively | 53 // Don't bother with a cached statement since this will be a relatively |
| 54 // infrequent operation. | 54 // infrequent operation. |
| 55 sql::Statement s(db_->GetUniqueStatement( | 55 sql::Statement s(db_->GetUniqueStatement( |
| 56 "INSERT OR REPLACE INTO web_app_icons " | 56 "INSERT OR REPLACE INTO web_app_icons " |
| 57 "(url, width, height, image) VALUES (?, ?, ?, ?)")); | 57 "(url, width, height, image) VALUES (?, ?, ?, ?)")); |
| 58 if (!s) | |
| 59 return false; | |
| 60 | |
| 61 std::vector<unsigned char> image_data; | 58 std::vector<unsigned char> image_data; |
| 62 gfx::PNGCodec::EncodeBGRASkBitmap(image, false, &image_data); | 59 gfx::PNGCodec::EncodeBGRASkBitmap(image, false, &image_data); |
| 63 | |
| 64 s.BindString(0, history::HistoryDatabase::GURLToDatabaseURL(url)); | 60 s.BindString(0, history::HistoryDatabase::GURLToDatabaseURL(url)); |
| 65 s.BindInt(1, image.width()); | 61 s.BindInt(1, image.width()); |
| 66 s.BindInt(2, image.height()); | 62 s.BindInt(2, image.height()); |
| 67 s.BindBlob(3, &image_data.front(), static_cast<int>(image_data.size())); | 63 s.BindBlob(3, &image_data.front(), static_cast<int>(image_data.size())); |
| 64 |
| 68 return s.Run(); | 65 return s.Run(); |
| 69 } | 66 } |
| 70 | 67 |
| 71 bool WebAppsTable::GetWebAppImages(const GURL& url, | 68 bool WebAppsTable::GetWebAppImages(const GURL& url, |
| 72 std::vector<SkBitmap>* images) { | 69 std::vector<SkBitmap>* images) { |
| 73 sql::Statement s(db_->GetUniqueStatement( | 70 sql::Statement s(db_->GetUniqueStatement( |
| 74 "SELECT image FROM web_app_icons WHERE url=?")); | 71 "SELECT image FROM web_app_icons WHERE url=?")); |
| 75 if (!s) { | 72 s.BindString(0, history::HistoryDatabase::GURLToDatabaseURL(url)); |
| 76 NOTREACHED() << "Statement prepare failed"; | 73 |
| 74 if (!s.is_valid()) { |
| 77 return false; | 75 return false; |
| 78 } | 76 } |
| 79 s.BindString(0, history::HistoryDatabase::GURLToDatabaseURL(url)); | 77 |
| 80 while (s.Step()) { | 78 while (s.Step()) { |
| 81 SkBitmap image; | 79 SkBitmap image; |
| 82 int col_bytes = s.ColumnByteLength(0); | 80 int col_bytes = s.ColumnByteLength(0); |
| 83 if (col_bytes > 0) { | 81 if (col_bytes > 0) { |
| 84 if (gfx::PNGCodec::Decode( | 82 if (gfx::PNGCodec::Decode( |
| 85 reinterpret_cast<const unsigned char*>(s.ColumnBlob(0)), | 83 reinterpret_cast<const unsigned char*>(s.ColumnBlob(0)), |
| 86 col_bytes, &image)) { | 84 col_bytes, &image)) { |
| 87 images->push_back(image); | 85 images->push_back(image); |
| 88 } else { | 86 } else { |
| 89 // Should only have valid image data in the db. | 87 // Should only have valid image data in the db. |
| 90 NOTREACHED(); | 88 NOTREACHED(); |
| 91 } | 89 } |
| 92 } | 90 } |
| 93 } | 91 } |
| 94 return true; | 92 return true; |
| 95 } | 93 } |
| 96 | 94 |
| 97 bool WebAppsTable::SetWebAppHasAllImages(const GURL& url, | 95 bool WebAppsTable::SetWebAppHasAllImages(const GURL& url, |
| 98 bool has_all_images) { | 96 bool has_all_images) { |
| 99 sql::Statement s(db_->GetUniqueStatement( | 97 sql::Statement s(db_->GetUniqueStatement( |
| 100 "INSERT OR REPLACE INTO web_apps (url, has_all_images) VALUES (?, ?)")); | 98 "INSERT OR REPLACE INTO web_apps (url, has_all_images) VALUES (?, ?)")); |
| 101 if (!s) { | |
| 102 NOTREACHED() << "Statement prepare failed"; | |
| 103 return false; | |
| 104 } | |
| 105 s.BindString(0, history::HistoryDatabase::GURLToDatabaseURL(url)); | 99 s.BindString(0, history::HistoryDatabase::GURLToDatabaseURL(url)); |
| 106 s.BindInt(1, has_all_images ? 1 : 0); | 100 s.BindInt(1, has_all_images ? 1 : 0); |
| 101 |
| 107 return s.Run(); | 102 return s.Run(); |
| 108 } | 103 } |
| 109 | 104 |
| 110 bool WebAppsTable::GetWebAppHasAllImages(const GURL& url) { | 105 bool WebAppsTable::GetWebAppHasAllImages(const GURL& url) { |
| 111 sql::Statement s(db_->GetUniqueStatement( | 106 sql::Statement s(db_->GetUniqueStatement( |
| 112 "SELECT has_all_images FROM web_apps WHERE url=?")); | 107 "SELECT has_all_images FROM web_apps WHERE url=?")); |
| 113 if (!s) { | |
| 114 NOTREACHED() << "Statement prepare failed"; | |
| 115 return false; | |
| 116 } | |
| 117 s.BindString(0, history::HistoryDatabase::GURLToDatabaseURL(url)); | 108 s.BindString(0, history::HistoryDatabase::GURLToDatabaseURL(url)); |
| 109 |
| 118 return (s.Step() && s.ColumnInt(0) == 1); | 110 return (s.Step() && s.ColumnInt(0) == 1); |
| 119 } | 111 } |
| 120 | 112 |
| 121 bool WebAppsTable::RemoveWebApp(const GURL& url) { | 113 bool WebAppsTable::RemoveWebApp(const GURL& url) { |
| 122 sql::Statement delete_s(db_->GetUniqueStatement( | 114 sql::Statement delete_s(db_->GetUniqueStatement( |
| 123 "DELETE FROM web_app_icons WHERE url = ?")); | 115 "DELETE FROM web_app_icons WHERE url = ?")); |
| 124 if (!delete_s) { | |
| 125 NOTREACHED() << "Statement prepare failed"; | |
| 126 return false; | |
| 127 } | |
| 128 delete_s.BindString(0, history::HistoryDatabase::GURLToDatabaseURL(url)); | 116 delete_s.BindString(0, history::HistoryDatabase::GURLToDatabaseURL(url)); |
| 117 |
| 129 if (!delete_s.Run()) | 118 if (!delete_s.Run()) |
| 130 return false; | 119 return false; |
| 131 | 120 |
| 132 sql::Statement delete_s2(db_->GetUniqueStatement( | 121 sql::Statement delete_s2(db_->GetUniqueStatement( |
| 133 "DELETE FROM web_apps WHERE url = ?")); | 122 "DELETE FROM web_apps WHERE url = ?")); |
| 134 if (!delete_s2) { | |
| 135 NOTREACHED() << "Statement prepare failed"; | |
| 136 return false; | |
| 137 } | |
| 138 delete_s2.BindString(0, history::HistoryDatabase::GURLToDatabaseURL(url)); | 123 delete_s2.BindString(0, history::HistoryDatabase::GURLToDatabaseURL(url)); |
| 124 |
| 139 return delete_s2.Run(); | 125 return delete_s2.Run(); |
| 140 } | 126 } |
| 141 | |
| OLD | NEW |