chrome/browser/webdata/web_data_service.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/webdata/web_data_service.h"
   6
   7 #include "base/bind.h"
   8 #include "base/stl_util.h"
   9 #include "chrome/browser/chrome_notification_types.h"
  10 #include "chrome/browser/search_engines/template_url.h"
  11 #include "chrome/browser/webdata/keyword_table.h"
  12 #include "chrome/browser/webdata/logins_table.h"
  13 #include "chrome/browser/webdata/web_apps_table.h"
  14 #include "chrome/browser/webdata/web_intents_table.h"
  15 #include "components/signin/core/webdata/token_service_table.h"
  16 #include "components/webdata/common/web_database_service.h"
  17 #include "content/public/browser/browser_thread.h"
  18 #include "content/public/browser/notification_details.h"
  19 #include "content/public/browser/notification_service.h"
  20 #include "content/public/browser/notification_source.h"
  21 #include "third_party/skia/include/core/SkBitmap.h"
  22
  23 ////////////////////////////////////////////////////////////////////////////////
  24 //
  25 // WebDataService implementation.
  26 //
  27 ////////////////////////////////////////////////////////////////////////////////
  28
  29 using base::Bind;
  30 using base::Time;
  31 using content::BrowserThread;
  32
  33 WDAppImagesResult::WDAppImagesResult() : has_all_images(false) {}
  34
  35 WDAppImagesResult::~WDAppImagesResult() {}
  36
  37 WDKeywordsResult::WDKeywordsResult()
  38   : default_search_provider_id(0),
  39     builtin_keyword_version(0) {
  40 }
  41
  42 WDKeywordsResult::~WDKeywordsResult() {}
  43
  44 WebDataService::WebDataService(scoped_refptr<WebDatabaseService> wdbs,
  45                                const ProfileErrorCallback& callback)
  46     : WebDataServiceBase(wdbs, callback,
  47           BrowserThread::GetMessageLoopProxyForThread(BrowserThread::UI)) {
  48 }
  49
  50 //////////////////////////////////////////////////////////////////////////////
  51 //
  52 // Keywords.
  53 //
  54 //////////////////////////////////////////////////////////////////////////////
  55
  56 void WebDataService::AddKeyword(const TemplateURLData& data) {
  57   wdbs_->ScheduleDBTask(
  58       FROM_HERE, Bind(&WebDataService::AddKeywordImpl, this, data));
  59 }
  60
  61 void WebDataService::RemoveKeyword(TemplateURLID id) {
  62   wdbs_->ScheduleDBTask(
  63       FROM_HERE, Bind(&WebDataService::RemoveKeywordImpl, this, id));
  64 }
  65
  66 void WebDataService::UpdateKeyword(const TemplateURLData& data) {
  67   wdbs_->ScheduleDBTask(
  68       FROM_HERE, Bind(&WebDataService::UpdateKeywordImpl, this, data));
  69 }
  70
  71 WebDataServiceBase::Handle WebDataService::GetKeywords(
  72     WebDataServiceConsumer* consumer) {
  73   return wdbs_->ScheduleDBTaskWithResult(FROM_HERE,
  74       Bind(&WebDataService::GetKeywordsImpl, this), consumer);
  75 }
  76
  77 void WebDataService::SetDefaultSearchProvider(const TemplateURL* url) {
  78   wdbs_->ScheduleDBTask(FROM_HERE,
  79       Bind(&WebDataService::SetDefaultSearchProviderImpl, this,
  80            url ? url->id() : 0));
  81 }
  82
  83 void WebDataService::SetBuiltinKeywordVersion(int version) {
  84   wdbs_->ScheduleDBTask(FROM_HERE,
  85       Bind(&WebDataService::SetBuiltinKeywordVersionImpl, this, version));
  86 }
  87
  88 //////////////////////////////////////////////////////////////////////////////
  89 //
  90 // Web Apps
  91 //
  92 //////////////////////////////////////////////////////////////////////////////
  93
  94 void WebDataService::SetWebAppImage(const GURL& app_url,
  95                                     const SkBitmap& image) {
  96   wdbs_->ScheduleDBTask(FROM_HERE,
  97       Bind(&WebDataService::SetWebAppImageImpl, this, app_url, image));
  98 }
  99
 100 void WebDataService::SetWebAppHasAllImages(const GURL& app_url,
 101                                            bool has_all_images) {
 102   wdbs_->ScheduleDBTask(FROM_HERE,
 103       Bind(&WebDataService::SetWebAppHasAllImagesImpl, this, app_url,
 104            has_all_images));
 105 }
 106
 107 void WebDataService::RemoveWebApp(const GURL& app_url) {
 108   wdbs_->ScheduleDBTask(FROM_HERE,
 109       Bind(&WebDataService::RemoveWebAppImpl, this, app_url));
 110 }
 111
 112 WebDataServiceBase::Handle WebDataService::GetWebAppImages(
 113     const GURL& app_url, WebDataServiceConsumer* consumer) {
 114   return wdbs_->ScheduleDBTaskWithResult(FROM_HERE,
 115       Bind(&WebDataService::GetWebAppImagesImpl, this, app_url), consumer);
 116 }
 117
 118 ////////////////////////////////////////////////////////////////////////////////
 119
 120 WebDataService::WebDataService()
 121     : WebDataServiceBase(NULL, ProfileErrorCallback(),
 122           BrowserThread::GetMessageLoopProxyForThread(BrowserThread::UI)) {
 123 }
 124
 125 WebDataService::~WebDataService() {
 126 }
 127
 128 ////////////////////////////////////////////////////////////////////////////////
 129 //
 130 // Keywords implementation.
 131 //
 132 ////////////////////////////////////////////////////////////////////////////////
 133
 134 WebDatabase::State WebDataService::AddKeywordImpl(
 135     const TemplateURLData& data, WebDatabase* db) {
 136   KeywordTable::FromWebDatabase(db)->AddKeyword(data);
 137   return WebDatabase::COMMIT_NEEDED;
 138 }
 139
 140 WebDatabase::State WebDataService::RemoveKeywordImpl(
 141     TemplateURLID id, WebDatabase* db) {
 142   DCHECK(id);
 143   KeywordTable::FromWebDatabase(db)->RemoveKeyword(id);
 144   return WebDatabase::COMMIT_NEEDED;
 145 }
 146
 147 WebDatabase::State WebDataService::UpdateKeywordImpl(
 148     const TemplateURLData& data, WebDatabase* db) {
 149   if (!KeywordTable::FromWebDatabase(db)->UpdateKeyword(data)) {
 150     NOTREACHED();
 151     return WebDatabase::COMMIT_NOT_NEEDED;
 152   }
 153  return WebDatabase::COMMIT_NEEDED;
 154 }
 155
 156 scoped_ptr<WDTypedResult> WebDataService::GetKeywordsImpl(WebDatabase* db) {
 157   WDKeywordsResult result;
 158   KeywordTable::FromWebDatabase(db)->GetKeywords(&result.keywords);
 159   result.default_search_provider_id =
 160       KeywordTable::FromWebDatabase(db)->GetDefaultSearchProviderID();
 161   result.builtin_keyword_version =
 162       KeywordTable::FromWebDatabase(db)->GetBuiltinKeywordVersion();
 163   return scoped_ptr<WDTypedResult>(
 164       new WDResult<WDKeywordsResult>(KEYWORDS_RESULT, result));
 165 }
 166
 167 WebDatabase::State WebDataService::SetDefaultSearchProviderImpl(
 168     TemplateURLID id, WebDatabase* db) {
 169   if (!KeywordTable::FromWebDatabase(db)->SetDefaultSearchProviderID(id)) {
 170     NOTREACHED();
 171     return WebDatabase::COMMIT_NOT_NEEDED;
 172   }
 173   return WebDatabase::COMMIT_NEEDED;
 174 }
 175
 176 WebDatabase::State WebDataService::SetBuiltinKeywordVersionImpl(
 177     int version, WebDatabase* db) {
 178   if (!KeywordTable::FromWebDatabase(db)->SetBuiltinKeywordVersion(version)) {
 179     NOTREACHED();
 180     return WebDatabase::COMMIT_NOT_NEEDED;
 181   }
 182   return WebDatabase::COMMIT_NEEDED;
 183 }
 184
 185 ////////////////////////////////////////////////////////////////////////////////
 186 //
 187 // Web Apps implementation.
 188 //
 189 ////////////////////////////////////////////////////////////////////////////////
 190
 191 WebDatabase::State WebDataService::SetWebAppImageImpl(
 192     const GURL& app_url, const SkBitmap& image, WebDatabase* db) {
 193   WebAppsTable::FromWebDatabase(db)->SetWebAppImage(app_url, image);
 194   return WebDatabase::COMMIT_NEEDED;
 195 }
 196
 197 WebDatabase::State WebDataService::SetWebAppHasAllImagesImpl(
 198     const GURL& app_url, bool has_all_images, WebDatabase* db) {
 199   WebAppsTable::FromWebDatabase(db)->SetWebAppHasAllImages(app_url,
 200                                                            has_all_images);
 201   return WebDatabase::COMMIT_NEEDED;
 202 }
 203
 204 WebDatabase::State WebDataService::RemoveWebAppImpl(
 205     const GURL& app_url, WebDatabase* db) {
 206   WebAppsTable::FromWebDatabase(db)->RemoveWebApp(app_url);
 207   return WebDatabase::COMMIT_NEEDED;
 208 }
 209
 210 scoped_ptr<WDTypedResult> WebDataService::GetWebAppImagesImpl(
 211     const GURL& app_url, WebDatabase* db) {
 212   WDAppImagesResult result;
 213   result.has_all_images =
 214       WebAppsTable::FromWebDatabase(db)->GetWebAppHasAllImages(app_url);
 215   WebAppsTable::FromWebDatabase(db)->GetWebAppImages(app_url, &result.images);
 216   return scoped_ptr<WDTypedResult>(
 217       new WDResult<WDAppImagesResult>(WEB_APP_IMAGES, result));
 218 }