summaryrefslogtreecommitdiffstats
path: root/chrome/installer/util/self_cleaning_temp_dir.h
blob: d9e3ad5ba5404a1f82f1dd2db292d27d499bdc5c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
// Copyright (c) 2011 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 CHROME_INSTALLER_UTIL_SELF_CLEANING_TEMP_DIR_H_
#define CHROME_INSTALLER_UTIL_SELF_CLEANING_TEMP_DIR_H_
#pragma once

#include "base/basictypes.h"
#include "base/file_path.h"
#include "base/gtest_prod_util.h"

namespace installer {

// A helper class for managing a temporary directory.  In relation to
// ScopedTempDir, this class additionally cleans up all non-empty parent
// directories of the temporary directory that are created by an instance.
class SelfCleaningTempDir {
 public:
  typedef FilePath::StringType StringType;

  SelfCleaningTempDir();

  // Performs a Delete().
  ~SelfCleaningTempDir();

  // Creates a temporary directory named |temp_name| under |parent_dir|,
  // creating intermediate directories as needed.
  bool Initialize(const FilePath& parent_dir, const StringType& temp_name);

  // Returns the temporary directory created in Initialize().
  const FilePath& path() const { return temp_dir_; }

  // Deletes the temporary directory created in Initialize() and all of its
  // contents, as well as all empty intermediate directories.  Any of these that
  // cannot be deleted immediately are scheduled for deletion upon reboot.
  bool Delete();

 private:
  static void GetTopDirToCreate(const FilePath& temp_parent_dir,
                                FilePath* base_dir);

  // The topmost directory created.
  FilePath base_dir_;

  // The temporary directory.
  FilePath temp_dir_;

  FRIEND_TEST_ALL_PREFIXES(SelfCleaningTempDirTest, TopLevel);
  FRIEND_TEST_ALL_PREFIXES(SelfCleaningTempDirTest, TopLevelPlusOne);
  DISALLOW_COPY_AND_ASSIGN(SelfCleaningTempDir);
};

}  // namespace installer

#endif  // CHROME_INSTALLER_UTIL_SELF_CLEANING_TEMP_DIR_H_