dom/indexedDB/DatabaseFileManagerImpl.h

   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
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #ifndef DOM_INDEXEDDB_DATABASEFILEMANAGERIMPL_H_
   8 #define DOM_INDEXEDDB_DATABASEFILEMANAGERIMPL_H_
   9
  10 #include "DatabaseFileManager.h"
  11
  12 // local includes
  13 #include "ActorsParentCommon.h"
  14
  15 // global includes
  16 #include "mozilla/dom/quota/Assertions.h"
  17 #include "mozilla/dom/quota/QuotaCommon.h"
  18 #include "nsIFile.h"
  19 #include "nsString.h"
  20
  21 namespace mozilla::dom::indexedDB {
  22
  23 template <typename KnownDirEntryOp, typename UnknownDirEntryOp>
  24 Result<Ok, nsresult> DatabaseFileManager::TraverseFiles(
  25     nsIFile& aDirectory, KnownDirEntryOp&& aKnownDirEntryOp,
  26     UnknownDirEntryOp&& aUnknownDirEntryOp) {
  27   quota::AssertIsOnIOThread();
  28
  29   QM_TRY_INSPECT(const bool& exists,
  30                  MOZ_TO_RESULT_INVOKE_MEMBER(aDirectory, Exists));
  31
  32   if (!exists) {
  33     return Ok{};
  34   }
  35
  36   QM_TRY(quota::CollectEachFile(
  37       aDirectory,
  38       [&aKnownDirEntryOp, &aUnknownDirEntryOp](
  39           const nsCOMPtr<nsIFile>& file) -> Result<Ok, nsresult> {
  40         QM_TRY_INSPECT(const auto& dirEntryKind, quota::GetDirEntryKind(*file));
  41
  42         switch (dirEntryKind) {
  43           case quota::nsIFileKind::ExistsAsDirectory: {
  44             QM_TRY_INSPECT(
  45                 const auto& leafName,
  46                 MOZ_TO_RESULT_INVOKE_MEMBER_TYPED(nsString, file, GetLeafName));
  47
  48             if (leafName.Equals(kJournalDirectoryName)) {
  49               QM_TRY(std::forward<KnownDirEntryOp>(aKnownDirEntryOp)(
  50                   *file, /* isDirectory */ true));
  51
  52               break;
  53             }
  54
  55             Unused << WARN_IF_FILE_IS_UNKNOWN(*file);
  56
  57             QM_TRY(std::forward<UnknownDirEntryOp>(aUnknownDirEntryOp)(
  58                 *file, /* isDirectory */ true));
  59
  60             break;
  61           }
  62
  63           case quota::nsIFileKind::ExistsAsFile: {
  64             QM_TRY_INSPECT(
  65                 const auto& leafName,
  66                 MOZ_TO_RESULT_INVOKE_MEMBER_TYPED(nsString, file, GetLeafName));
  67
  68             nsresult rv;
  69             leafName.ToInteger64(&rv);
  70             if (NS_SUCCEEDED(rv)) {
  71               QM_TRY(std::forward<KnownDirEntryOp>(aKnownDirEntryOp)(
  72                   *file, /* isDirectory */ false));
  73
  74               break;
  75             }
  76
  77             Unused << WARN_IF_FILE_IS_UNKNOWN(*file);
  78
  79             QM_TRY(std::forward<UnknownDirEntryOp>(aUnknownDirEntryOp)(
  80                 *file, /* isDirectory */ false));
  81
  82             break;
  83           }
  84
  85           case quota::nsIFileKind::DoesNotExist:
  86             // Ignore files that got removed externally while iterating.
  87             break;
  88         }
  89
  90         return Ok{};
  91       }));
  92
  93   return Ok{};
  94 }
  95
  96 }  // namespace mozilla::dom::indexedDB
  97
  98 #endif  // DOM_INDEXEDDB_DATABASEFILEMANAGERIMPL_H_