[Activity Log] when extension is uninstalled, delete data about it

chrome/browser/extensions/activity_log/fullstream_ui_policy.cc

 // Copyright 2013 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/extensions/activity_log/fullstream_ui_policy.h"
 
 #include "base/callback.h"
 #include "base/command_line.h"
 #include "base/files/file_path.h"
 #include "base/json/json_reader.h"
@@ skipping 264 matching lines @@
     statement.BindString(0, restrict_urls[i].spec());
 
     if (!statement.Run()) {
       LOG(ERROR) << "Removing arg URL from database failed: "
                  << statement.GetSQLStatement();
       return;
     }
   }
 }
 
+void FullStreamUIPolicy::DoRemoveExtensionData(
+    const std::string& extension_id) {
+  if (extension_id.empty())
+    return;
+
+  sql::Connection* db = GetDatabaseConnection();
+  if (!db) {
+    LOG(ERROR) << "Unable to connect to database";
+    return;
+  }
+
+  // Make sure any queued in memory are sent to the database before cleaning.
+  activity_database()->AdviseFlush(ActivityDatabase::kFlushImmediately);
+
+  std::string sql_str = base::StringPrintf(
+      "DELETE FROM %s WHERE extension_id=?", kTableName);
+  sql::Statement statement;
+  statement.Assign(
+      db->GetCachedStatement(sql::StatementID(SQL_FROM_HERE), sql_str.c_str()));
+  statement.BindString(0, extension_id);
+  if (!statement.Run()) {
+    LOG(ERROR) << "Removing URLs for extension "
+               << extension_id << "from database failed: "
+               << statement.GetSQLStatement();
+  }
+}
+
 void FullStreamUIPolicy::DoDeleteDatabase() {
   sql::Connection* db = GetDatabaseConnection();
   if (!db) {
     LOG(ERROR) << "Unable to connect to database";
     return;
   }
 
   queued_actions_.clear();
 
   // Not wrapped in a transaction because the deletion should happen even if
@@ skipping 49 matching lines @@
                  page_url,
                  arg_url,
                  days_ago),
       callback);
 }
 
 void FullStreamUIPolicy::RemoveURLs(const std::vector<GURL>& restrict_urls) {
   ScheduleAndForget(this, &FullStreamUIPolicy::DoRemoveURLs, restrict_urls);
 }
 
+void FullStreamUIPolicy::RemoveExtensionData(const std::string& extension_id) {
+  ScheduleAndForget(
+      this, &FullStreamUIPolicy::DoRemoveExtensionData, extension_id);
+}
+
 void FullStreamUIPolicy::DeleteDatabase() {
   ScheduleAndForget(this, &FullStreamUIPolicy::DoDeleteDatabase);
 }
 
 scoped_refptr<Action> FullStreamUIPolicy::ProcessArguments(
     scoped_refptr<Action> action) const {
   return action;
 }
 
 void FullStreamUIPolicy::ProcessAction(scoped_refptr<Action> action) {
   // TODO(mvrable): Right now this argument stripping updates the Action object
   // in place, which isn't good if there are other users of the object.  When
   // database writing is moved to policy class, the modifications should be
   // made locally.
   action = ProcessArguments(action);
   ScheduleAndForget(this, &FullStreamUIPolicy::QueueAction, action);
 }
 
 void FullStreamUIPolicy::QueueAction(scoped_refptr<Action> action) {
   if (activity_database()->is_db_valid()) {
     queued_actions_.push_back(action);
     activity_database()->AdviseFlush(queued_actions_.size());
   }
 }
 
 }  // namespace extensions