dom/quota/StorageHelpers.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this file,
   5  * You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "mozilla/dom/quota/StorageHelpers.h"
   8
   9 #include "mozilla/dom/quota/QuotaCommon.h"
  10 #include "mozilla/dom/quota/ResultExtensions.h"
  11
  12 namespace mozilla::dom::quota {
  13
  14 AutoDatabaseAttacher::AutoDatabaseAttacher(
  15     nsCOMPtr<mozIStorageConnection> aConnection,
  16     nsCOMPtr<nsIFile> aDatabaseFile, const nsLiteralCString& aSchemaName)
  17     : mConnection(std::move(aConnection)),
  18       mDatabaseFile(std::move(aDatabaseFile)),
  19       mSchemaName(aSchemaName),
  20       mAttached(false) {}
  21
  22 AutoDatabaseAttacher::~AutoDatabaseAttacher() {
  23   if (mAttached) {
  24     QM_WARNONLY_TRY(MOZ_TO_RESULT(Detach()));
  25   }
  26 }
  27
  28 nsresult AutoDatabaseAttacher::Attach() {
  29   MOZ_ASSERT(mConnection);
  30   MOZ_ASSERT(mDatabaseFile);
  31   MOZ_ASSERT(!mAttached);
  32
  33 #ifdef DEBUG
  34   {
  35     QM_TRY_INSPECT(const bool& exists,
  36                    MOZ_TO_RESULT_INVOKE_MEMBER(mDatabaseFile, Exists));
  37
  38     MOZ_ASSERT(exists);
  39   }
  40 #endif
  41
  42   QM_TRY_INSPECT(const auto& path, MOZ_TO_RESULT_INVOKE_MEMBER_TYPED(
  43                                        nsString, mDatabaseFile, GetPath));
  44
  45   QM_TRY_INSPECT(
  46       const auto& stmt,
  47       MOZ_TO_RESULT_INVOKE_MEMBER_TYPED(
  48           nsCOMPtr<mozIStorageStatement>, mConnection, CreateStatement,
  49           "ATTACH DATABASE :path AS "_ns + mSchemaName + ";"_ns));
  50
  51   QM_TRY(MOZ_TO_RESULT(stmt->BindStringByName("path"_ns, path)));
  52   QM_TRY(MOZ_TO_RESULT(stmt->Execute()));
  53
  54   mAttached = true;
  55
  56   return NS_OK;
  57 }
  58
  59 nsresult AutoDatabaseAttacher::Detach() {
  60   MOZ_ASSERT(mConnection);
  61   MOZ_ASSERT(mAttached);
  62
  63   QM_TRY(MOZ_TO_RESULT(
  64       mConnection->ExecuteSimpleSQL("DETACH DATABASE "_ns + mSchemaName)));
  65
  66   mAttached = false;
  67   return NS_OK;
  68 }
  69
  70 }  // namespace mozilla::dom::quota