Remove dependency on //content from history DownloadDatabase

components/history/core/browser/download_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/download_database.h"
+#include "components/history/core/browser/download_database.h"
 
 #include <limits>
 #include <string>
 #include <vector>
 
 #include "base/debug/alias.h"
 #include "base/files/file_path.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/metrics/histogram.h"
 #include "base/stl_util.h"
 #include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/time/time.h"
 #include "build/build_config.h"
-#include "chrome/browser/history/download_row.h"
+#include "components/history/core/browser/download_constants.h"
+#include "components/history/core/browser/download_row.h"
 #include "components/history/core/browser/history_types.h"
-#include "content/public/browser/download_interrupt_reasons.h"
-#include "content/public/browser/download_item.h"
 #include "sql/statement.h"
 
-using content::DownloadItem;
-
 namespace history {
 
 namespace {
 
 // Reason for dropping a particular record.
 enum DroppedReason {
   DROPPED_REASON_BAD_STATE = 0,
   DROPPED_REASON_BAD_DANGER_TYPE = 1,
   DROPPED_REASON_BAD_ID = 2,
   DROPPED_REASON_DUPLICATE_ID = 3,
@@ skipping 19 matching lines @@
   statement.BindString16(col, path.value());
 }
 base::FilePath ColumnFilePath(sql::Statement& statement, int col) {
   return base::FilePath(statement.ColumnString16(col));
 }
 
 #endif
 
 }  // namespace
 
-// These constants and the transformation functions below are used to allow
-// DownloadItem::DownloadState and DownloadDangerType to change without
-// breaking the database schema.
-// They guarantee that the values of the |state| field in the database are one
-// of the values returned by StateToInt, and that the values of the |state|
-// field of the DownloadRows returned by QueryDownloads() are one of the values
-// returned by IntToState().
-const int DownloadDatabase::kStateInvalid = -1;
-const int DownloadDatabase::kStateInProgress = 0;
-const int DownloadDatabase::kStateComplete = 1;
-const int DownloadDatabase::kStateCancelled = 2;
-const int DownloadDatabase::kStateBug140687 = 3;
-const int DownloadDatabase::kStateInterrupted = 4;
-
-const int DownloadDatabase::kDangerTypeInvalid = -1;
-const int DownloadDatabase::kDangerTypeNotDangerous = 0;
-const int DownloadDatabase::kDangerTypeDangerousFile = 1;
-const int DownloadDatabase::kDangerTypeDangerousUrl = 2;
-const int DownloadDatabase::kDangerTypeDangerousContent = 3;
-const int DownloadDatabase::kDangerTypeMaybeDangerousContent = 4;
-const int DownloadDatabase::kDangerTypeUncommonContent = 5;
-const int DownloadDatabase::kDangerTypeUserValidated = 6;
-const int DownloadDatabase::kDangerTypeDangerousHost = 7;
-const int DownloadDatabase::kDangerTypePotentiallyUnwanted = 8;
-
-int DownloadDatabase::StateToInt(DownloadItem::DownloadState state) {
-  switch (state) {
-    case DownloadItem::IN_PROGRESS: return DownloadDatabase::kStateInProgress;
-    case DownloadItem::COMPLETE: return DownloadDatabase::kStateComplete;
-    case DownloadItem::CANCELLED: return DownloadDatabase::kStateCancelled;
-    case DownloadItem::INTERRUPTED: return DownloadDatabase::kStateInterrupted;
-    case DownloadItem::MAX_DOWNLOAD_STATE:
-      NOTREACHED();
-      return DownloadDatabase::kStateInvalid;
-  }
-  NOTREACHED();
-  return DownloadDatabase::kStateInvalid;
-}
-
-DownloadItem::DownloadState DownloadDatabase::IntToState(int state) {
-  switch (state) {
-    case DownloadDatabase::kStateInProgress: return DownloadItem::IN_PROGRESS;
-    case DownloadDatabase::kStateComplete: return DownloadItem::COMPLETE;
-    case DownloadDatabase::kStateCancelled: return DownloadItem::CANCELLED;
-    // We should not need kStateBug140687 here because MigrateDownloadsState()
-    // is called in HistoryDatabase::Init().
-    case DownloadDatabase::kStateInterrupted: return DownloadItem::INTERRUPTED;
-    default: return DownloadItem::MAX_DOWNLOAD_STATE;
-  }
-}
-
-int DownloadDatabase::DangerTypeToInt(content::DownloadDangerType danger_type) {
-  switch (danger_type) {
-    case content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS:
-      return DownloadDatabase::kDangerTypeNotDangerous;
-    case content::DOWNLOAD_DANGER_TYPE_DANGEROUS_FILE:
-      return DownloadDatabase::kDangerTypeDangerousFile;
-    case content::DOWNLOAD_DANGER_TYPE_DANGEROUS_URL:
-      return DownloadDatabase::kDangerTypeDangerousUrl;
-    case content::DOWNLOAD_DANGER_TYPE_DANGEROUS_CONTENT:
-      return DownloadDatabase::kDangerTypeDangerousContent;
-    case content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT:
-      return DownloadDatabase::kDangerTypeMaybeDangerousContent;
-    case content::DOWNLOAD_DANGER_TYPE_UNCOMMON_CONTENT:
-      return DownloadDatabase::kDangerTypeUncommonContent;
-    case content::DOWNLOAD_DANGER_TYPE_USER_VALIDATED:
-      return DownloadDatabase::kDangerTypeUserValidated;
-    case content::DOWNLOAD_DANGER_TYPE_DANGEROUS_HOST:
-      return DownloadDatabase::kDangerTypeDangerousHost;
-    case content::DOWNLOAD_DANGER_TYPE_POTENTIALLY_UNWANTED:
-      return DownloadDatabase::kDangerTypePotentiallyUnwanted;
-    case content::DOWNLOAD_DANGER_TYPE_MAX:
-      NOTREACHED();
-      return DownloadDatabase::kDangerTypeInvalid;
-  }
-  NOTREACHED();
-  return DownloadDatabase::kDangerTypeInvalid;
-}
-
-content::DownloadDangerType DownloadDatabase::IntToDangerType(int danger_type) {
-  switch (danger_type) {
-    case DownloadDatabase::kDangerTypeNotDangerous:
-      return content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS;
-    case DownloadDatabase::kDangerTypeDangerousFile:
-      return content::DOWNLOAD_DANGER_TYPE_DANGEROUS_FILE;
-    case DownloadDatabase::kDangerTypeDangerousUrl:
-      return content::DOWNLOAD_DANGER_TYPE_DANGEROUS_URL;
-    case DownloadDatabase::kDangerTypeDangerousContent:
-      return content::DOWNLOAD_DANGER_TYPE_DANGEROUS_CONTENT;
-    case DownloadDatabase::kDangerTypeMaybeDangerousContent:
-      return content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT;
-    case DownloadDatabase::kDangerTypeUncommonContent:
-      return content::DOWNLOAD_DANGER_TYPE_UNCOMMON_CONTENT;
-    case DownloadDatabase::kDangerTypeUserValidated:
-      return content::DOWNLOAD_DANGER_TYPE_USER_VALIDATED;
-    case DownloadDatabase::kDangerTypeDangerousHost:
-      return content::DOWNLOAD_DANGER_TYPE_DANGEROUS_HOST;
-    case DownloadDatabase::kDangerTypePotentiallyUnwanted:
-      return content::DOWNLOAD_DANGER_TYPE_POTENTIALLY_UNWANTED;
-    default:
-      return content::DOWNLOAD_DANGER_TYPE_MAX;
-  }
-}
-
-DownloadDatabase::DownloadDatabase()
+DownloadDatabase::DownloadDatabase(
+    DownloadInterruptReason download_interrupt_no_reason,
+    DownloadInterruptReason download_interrupt_crash)
     : owning_thread_set_(false),
       owning_thread_(0),
-      in_progress_entry_cleanup_completed_(false) {
+      in_progress_entry_cleanup_completed_(false),
+      download_interrupt_no_reason_(download_interrupt_no_reason),
+      download_interrupt_crash_(download_interrupt_crash) {
 }
 
 DownloadDatabase::~DownloadDatabase() {
 }
 
 bool DownloadDatabase::EnsureColumnExists(
     const std::string& name, const std::string& type) {
   std::string add_col = "ALTER TABLE downloads ADD COLUMN " + name + " " + type;
   return GetDB().DoesColumnExist("downloads", name.c_str()) ||
          GetDB().Execute(add_col.c_str());
@@ skipping 58 matching lines @@
       "  total_bytes, state, danger_type, interrupt_reason, end_time, opened ) "
       "SELECT id, full_path, full_path, "
       "       CASE start_time WHEN 0 THEN 0 ELSE "
       "            (start_time + 11644473600) * 1000000 END, "
       "       received_bytes, total_bytes, "
       "       state, ?, ?, "
       "       CASE end_time WHEN 0 THEN 0 ELSE "
       "            (end_time + 11644473600) * 1000000 END, "
       "       opened "
       "FROM downloads_tmp"));
-  statement_populate.BindInt(0, content::DOWNLOAD_INTERRUPT_REASON_NONE);
+  statement_populate.BindInt(0, download_interrupt_no_reason_);
   statement_populate.BindInt(1, kDangerTypeNotDangerous);
   if (!statement_populate.Run())
     return false;
 
   // Create new chain table and populate it.
   if (!GetDB().Execute(kReasonPathDangerUrlChainSchema))
     return false;
 
   if (!GetDB().Execute("INSERT INTO downloads_url_chains "
                        "  ( id, chain_index, url) "
@@ skipping 25 matching lines @@
   const char kSchema[] =
       "CREATE TABLE downloads ("
       "id INTEGER PRIMARY KEY,"             // Primary key.
       "current_path LONGVARCHAR NOT NULL,"  // Current disk location
       "target_path LONGVARCHAR NOT NULL,"   // Final disk location
       "start_time INTEGER NOT NULL,"        // When the download was started.
       "received_bytes INTEGER NOT NULL,"    // Total size downloaded.
       "total_bytes INTEGER NOT NULL,"       // Total size of the download.
       "state INTEGER NOT NULL,"             // 1=complete, 4=interrupted
       "danger_type INTEGER NOT NULL,"       // Danger type, validated.
-      "interrupt_reason INTEGER NOT NULL,"  // content::DownloadInterruptReason
+      "interrupt_reason INTEGER NOT NULL,"  // DownloadInterruptReason
       "end_time INTEGER NOT NULL,"          // When the download completed.
       "opened INTEGER NOT NULL,"            // 1 if it has ever been opened
                                             // else 0
       "referrer VARCHAR NOT NULL,"          // HTTP Referrer
       "by_ext_id VARCHAR NOT NULL,"         // ID of extension that started the
                                             // download
       "by_ext_name VARCHAR NOT NULL,"       // name of extension
       "etag VARCHAR NOT NULL,"              // ETag
       "last_modified VARCHAR NOT NULL,"     // Last-Modified header
       "mime_type VARCHAR(255) NOT NULL,"    // MIME type.
@@ skipping 17 matching lines @@
     return (!GetDB().DoesTableExist("downloads_url_chain") &&
             GetDB().Execute(kSchema) && GetDB().Execute(kUrlChainSchema));
   }
 }
 
 uint32 DownloadDatabase::GetNextDownloadId() {
   sql::Statement select_max_id(GetDB().GetUniqueStatement(
       "SELECT max(id) FROM downloads"));
   bool result = select_max_id.Step();
   DCHECK(result);
-  // If there are zero records in the downloads table, then max(id) will return
-  // 0 = kInvalidId, so GetNextDownloadId() will set *id = kInvalidId + 1.
-  // If there is at least one record but all of the |id|s are <= kInvalidId,
-  // then max(id) will return <= kInvalidId, so GetNextDownloadId should return
-  // kInvalidId + 1. Note that any records with |id <= kInvalidId| will be
-  // dropped in QueryDownloads()
+  // If there are zero records in the downloads table, then max(id) will
+  // return 0 = kInvalidDownloadId, so GetNextDownloadId() will set
+  // *id = kInvalidDownloadId + 1.
+  //
+  // If there is at least one record but all of the |id|s are
+  // <= kInvalidDownloadId, then max(id) will return <= kInvalidDownloadId,
+  // so GetNextDownloadId() should return kInvalidDownloadId + 1.
+  //
+  // Note that any records with |id <= kInvalidDownloadId| will be dropped in
+  // QueryDownloads().
+  //
   // SQLITE doesn't have unsigned integers.
   return 1 + static_cast<uint32>(std::max(
-      static_cast<int64>(content::DownloadItem::kInvalidId),
+      static_cast<int64>(kInvalidDownloadId),
       select_max_id.ColumnInt64(0)));
 }
 
 bool DownloadDatabase::DropDownloadTable() {
   return GetDB().Execute("DROP TABLE downloads");
 }
 
 void DownloadDatabase::QueryDownloads(
     std::vector<DownloadRow>* results) {
   EnsureInProgressEntriesCleanedUp();
@@ skipping 22 matching lines @@
     info->id = static_cast<uint32>(signed_id);
     info->current_path = ColumnFilePath(statement_main, column++);
     info->target_path = ColumnFilePath(statement_main, column++);
     info->mime_type = statement_main.ColumnString(column++);
     info->original_mime_type = statement_main.ColumnString(column++);
     info->start_time = base::Time::FromInternalValue(
         statement_main.ColumnInt64(column++));
     info->received_bytes = statement_main.ColumnInt64(column++);
     info->total_bytes = statement_main.ColumnInt64(column++);
     int state = statement_main.ColumnInt(column++);
-    info->state = IntToState(state);
-    if (info->state == DownloadItem::MAX_DOWNLOAD_STATE)
+    info->state = IntToDownloadState(state);
+    if (info->state == kStateInvalid)
       UMA_HISTOGRAM_COUNTS("Download.DatabaseInvalidState", state);
-    info->danger_type = IntToDangerType(statement_main.ColumnInt(column++));
-    info->interrupt_reason = static_cast<content::DownloadInterruptReason>(
+    info->danger_type =
+        IntToDownloadDangerType(statement_main.ColumnInt(column++));
+    info->interrupt_reason = static_cast<DownloadInterruptReason>(
         statement_main.ColumnInt(column++));
     info->end_time = base::Time::FromInternalValue(
         statement_main.ColumnInt64(column++));
     info->opened = statement_main.ColumnInt(column++) != 0;
     info->referrer_url = GURL(statement_main.ColumnString(column++));
     info->by_ext_id = statement_main.ColumnString(column++);
     info->by_ext_name = statement_main.ColumnString(column++);
     info->etag = statement_main.ColumnString(column++);
     info->last_modified = statement_main.ColumnString(column++);
 
     // If the record is corrupted, note that and drop it.
     // http://crbug.com/251269
     DroppedReason dropped_reason = DROPPED_REASON_MAX;
-    if (signed_id <= static_cast<int64>(content::DownloadItem::kInvalidId)) {
+    if (signed_id <= static_cast<int64>(kInvalidDownloadId)) {
       // SQLITE doesn't have unsigned integers.
       dropped_reason = DROPPED_REASON_BAD_ID;
     } else if (!ids.insert(info->id).second) {
       dropped_reason = DROPPED_REASON_DUPLICATE_ID;
       NOTREACHED() << info->id;
-    } else if (info->state == DownloadItem::MAX_DOWNLOAD_STATE) {
+    } else if (info->state == kStateInvalid) {
       dropped_reason = DROPPED_REASON_BAD_STATE;
-    } else if (info->danger_type == content::DOWNLOAD_DANGER_TYPE_MAX) {
+    } else if (info->danger_type == kDangerTypeInvalid) {
       dropped_reason = DROPPED_REASON_BAD_DANGER_TYPE;
     }
     if (dropped_reason != DROPPED_REASON_MAX) {
       UMA_HISTOGRAM_ENUMERATION("Download.DatabaseRecordDropped",
                                 dropped_reason,
                                 DROPPED_REASON_MAX + 1);
     } else {
       DCHECK(!ContainsKey(info_map, info->id));
       uint32 id = info->id;
       info_map[id] = info.release();
     }
   }
 
   sql::Statement statement_chain(GetDB().GetCachedStatement(
       SQL_FROM_HERE,
       "SELECT id, chain_index, url FROM downloads_url_chains "
       "ORDER BY id, chain_index"));
 
   while (statement_chain.Step()) {
     int column = 0;
     // See the comment above about SQLITE lacking unsigned integers.
     int64 signed_id = statement_chain.ColumnInt64(column++);
     int chain_index = statement_chain.ColumnInt(column++);
 
-    if (signed_id <= static_cast<int64>(content::DownloadItem::kInvalidId))
+    if (signed_id <= static_cast<int64>(kInvalidDownloadId))
       continue;
     uint32 id = static_cast<uint32>(signed_id);
 
     // Note that these DCHECKs may trip as a result of corrupted databases.
     // We have them because in debug builds the chances are higher there's
     // an actual bug than that the database is corrupt, but we handle the
     // DB corruption case in production code.
 
     // Confirm the id has already been seen--if it hasn't, discard the
     // record.
@@ skipping 29 matching lines @@
       results->push_back(*row);
     }
     delete row;
     it->second = NULL;
   }
 }
 
 bool DownloadDatabase::UpdateDownload(const DownloadRow& data) {
   EnsureInProgressEntriesCleanedUp();
 
-  DCHECK_NE(content::DownloadItem::kInvalidId, data.id);
-  int state = StateToInt(data.state);
-  if (state == kStateInvalid) {
+  DCHECK_NE(kInvalidDownloadId, data.id);
+  if (data.state == kStateInvalid) {
     NOTREACHED();
     return false;
   }
-  int danger_type = DangerTypeToInt(data.danger_type);
-  if (danger_type == kDangerTypeInvalid) {
+  if (data.danger_type == kDangerTypeInvalid) {
     NOTREACHED();
     return false;
   }
 
   sql::Statement statement(GetDB().GetCachedStatement(SQL_FROM_HERE,
       "UPDATE downloads "
       "SET current_path=?, target_path=?, "
       "mime_type=?, original_mime_type=?, "
       "received_bytes=?, state=?, "
       "danger_type=?, interrupt_reason=?, end_time=?, total_bytes=?, "
       "opened=?, by_ext_id=?, by_ext_name=?, etag=?, last_modified=? "
       "WHERE id=?"));
   int column = 0;
   BindFilePath(statement, data.current_path, column++);
   BindFilePath(statement, data.target_path, column++);
   statement.BindString(column++, data.mime_type);
   statement.BindString(column++, data.original_mime_type);
   statement.BindInt64(column++, data.received_bytes);
-  statement.BindInt(column++, state);
-  statement.BindInt(column++, danger_type);
+  statement.BindInt(column++, static_cast<int>(data.state));
+  statement.BindInt(column++, static_cast<int>(data.danger_type));
   statement.BindInt(column++, static_cast<int>(data.interrupt_reason));
   statement.BindInt64(column++, data.end_time.ToInternalValue());
   statement.BindInt64(column++, data.total_bytes);
   statement.BindInt(column++, (data.opened ? 1 : 0));
   statement.BindString(column++, data.by_ext_id);
   statement.BindString(column++, data.by_ext_name);
   statement.BindString(column++, data.etag);
   statement.BindString(column++, data.last_modified);
   statement.BindInt(column++, data.id);
 
   return statement.Run();
 }
 
 void DownloadDatabase::EnsureInProgressEntriesCleanedUp() {
   if (in_progress_entry_cleanup_completed_)
     return;
 
   sql::Statement statement(GetDB().GetCachedStatement(SQL_FROM_HERE,
       "UPDATE downloads SET state=?, interrupt_reason=? WHERE state=?"));
   statement.BindInt(0, kStateInterrupted);
-  statement.BindInt(1, content::DOWNLOAD_INTERRUPT_REASON_CRASH);
+  statement.BindInt(1, download_interrupt_crash_);
   statement.BindInt(2, kStateInProgress);
 
   statement.Run();
   in_progress_entry_cleanup_completed_ = true;
 }
 
 bool DownloadDatabase::CreateDownload(const DownloadRow& info) {
-  DCHECK_NE(content::DownloadItem::kInvalidId, info.id);
+  DCHECK_NE(kInvalidDownloadId, info.id);
   EnsureInProgressEntriesCleanedUp();
 
   if (info.url_chain.empty())
     return false;
 
-  int state = StateToInt(info.state);
-  if (state == kStateInvalid)
+  if (info.state == kStateInvalid)
     return false;
 
-  int danger_type = DangerTypeToInt(info.danger_type);
-  if (danger_type == kDangerTypeInvalid)
+  if (info.danger_type == kDangerTypeInvalid)
     return false;
 
   {
     sql::Statement statement_insert(GetDB().GetCachedStatement(
         SQL_FROM_HERE,
         "INSERT INTO downloads "
         "(id, current_path, target_path, "
         " mime_type, original_mime_type, "
         " start_time, "
         " received_bytes, total_bytes, state, danger_type, interrupt_reason, "
         " end_time, opened, referrer, by_ext_id, by_ext_name, etag, "
         " last_modified) "
         "VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"));
 
     int column = 0;
     statement_insert.BindInt(column++, info.id);
     BindFilePath(statement_insert, info.current_path, column++);
     BindFilePath(statement_insert, info.target_path, column++);
     statement_insert.BindString(column++, info.mime_type);
     statement_insert.BindString(column++, info.original_mime_type);
     statement_insert.BindInt64(column++, info.start_time.ToInternalValue());
     statement_insert.BindInt64(column++, info.received_bytes);
     statement_insert.BindInt64(column++, info.total_bytes);
-    statement_insert.BindInt(column++, state);
-    statement_insert.BindInt(column++, danger_type);
+    statement_insert.BindInt(column++, static_cast<int>(info.state));
+    statement_insert.BindInt(column++, static_cast<int>(info.danger_type));
     statement_insert.BindInt(column++, info.interrupt_reason);
     statement_insert.BindInt64(column++, info.end_time.ToInternalValue());
     statement_insert.BindInt(column++, info.opened ? 1 : 0);
     statement_insert.BindString(column++, info.referrer_url.spec());
     statement_insert.BindString(column++, info.by_ext_id);
     statement_insert.BindString(column++, info.by_ext_name);
     statement_insert.BindString(column++, info.etag);
     statement_insert.BindString(column++, info.last_modified);
     if (!statement_insert.Run()) {
       // GetErrorCode() returns a bitmask where the lower byte is a more general
@@ skipping 70 matching lines @@
 size_t DownloadDatabase::CountDownloads() {
   EnsureInProgressEntriesCleanedUp();
 
   sql::Statement statement(GetDB().GetCachedStatement(SQL_FROM_HERE,
       "SELECT count(*) from downloads"));
   statement.Step();
   return statement.ColumnInt(0);
 }
 
 }  // namespace history