Added MediaGalleryDatabase

content/browser/media_gallery/media_gallery_database.h

Index: content/browser/media_gallery/media_gallery_database.h
diff --git a/content/browser/media_gallery/media_gallery_database.h b/content/browser/media_gallery/media_gallery_database.h
new file mode 100644
index 0000000000000000000000000000000000000000..e5646be5e0742f447d1c119bc71849f344e1c887
--- /dev/null
+++ b/content/browser/media_gallery/media_gallery_database.h
@@ -0,0 +1,63 @@
+// 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.
+
+#ifndef CONTENT_BROWSER_MEDIA_GALLERY_MEDIA_GALLERY_DATABASE_H_
+#define CONTENT_BROWSER_MEDIA_GALLERY_MEDIA_GALLERY_DATABASE_H_
+#pragma once
+
+#include "base/basictypes.h"
+#include "base/file_path.h"
+#include "content/browser/media_gallery/media_gallery_types.h"
+#include "content/common/content_export.h"
+#include "sql/connection.h"
+#include "sql/init_status.h"
+#include "sql/meta_table.h"
+
+namespace media_gallery {
+
+class CONTENT_EXPORT MediaGalleryDatabase {
+ public:
+  MediaGalleryDatabase();
+  virtual ~MediaGalleryDatabase();
+
+  // Must call this function to complete initialization. Will return true on
+  // success. On false, no other function should be called.
+  sql::InitStatus Init(const FilePath& database_path);
+
+  // Returns the current version that we will generate media gallery databases
+  // with.
+  static int GetCurrentVersion();
+
+  // Sets the id field of the input collection_row to the generated unique
+  // key value and returns the same. On failure, returns zero.
+  int CreateCollectionRow(CollectionRow* collection_row);
+
+  // Finds the row with the specified id and fills its data into the collection
+  // pointer passed by the caller. Returns true on success.
+  bool GetCollectionRow(CollectionId id, CollectionRow* collection);
+

[vandebo (ex-Chrome), 2012/03/07 23:28:31]

We'll also need something to update a row, but that can come later.

[tpayne, 2012/03/08 00:04:50]

Yes, that will come later.

+ protected:

[vandebo (ex-Chrome), 2012/03/07 23:28:31]

I think these can be private - There are no plans to inherit from
MediaGalleryDatabase, right?

[tpayne, 2012/03/08 00:04:50]

It is inherited by the test code.

+  virtual sql::Connection& GetDB();
+
+  // Initializes an already open database.
+  sql::InitStatus InitInternal(sql::Connection* db);
+  static bool DoesCollectionsTableExist(sql::Connection* db);
+
+ private:
+  static bool CreateCollectionsTable(sql::Connection* db);
+  sql::InitStatus EnsureCurrentVersion();
+  void FillCollectionRow(const sql::Statement& statement, CollectionRow* row);
+
+  sql::Connection db_;
+  sql::MetaTable meta_table_;
+
+  DISALLOW_COPY_AND_ASSIGN(MediaGalleryDatabase);
+};
+
+}  // namespace media_gallery
+
+std::ostream& operator<<(std::ostream& out,
+    const media_gallery::CollectionRow& row);
+
+#endif  // CONTENT_BROWSER_MEDIA_GALLERY_MEDIA_GALLERY_DATABASE_H_