self_cleaning_temp_dir.h revision 2a99a7e74a7f215066514fe81d2bfa6639d9eddd 
1// Copyright (c) 2011 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#ifndef CHROME_INSTALLER_UTIL_SELF_CLEANING_TEMP_DIR_H_
6#define CHROME_INSTALLER_UTIL_SELF_CLEANING_TEMP_DIR_H_
7
8#include "base/basictypes.h"
9#include "base/files/file_path.h"
10#include "base/gtest_prod_util.h"
11
12namespace installer {
13
14// A helper class for managing a temporary directory.  In relation to
15// base::ScopedTempDir, this class additionally cleans up all non-empty parent
16// directories of the temporary directory that are created by an instance.
17class SelfCleaningTempDir {
18 public:
19  typedef base::FilePath::StringType StringType;
20
21  SelfCleaningTempDir();
22
23  // Performs a Delete().
24  ~SelfCleaningTempDir();
25
26  // Creates a temporary directory named |temp_name| under |parent_dir|,
27  // creating intermediate directories as needed.
28  bool Initialize(const base::FilePath& parent_dir,
29                  const StringType& temp_name);
30
31  // Returns the temporary directory created in Initialize().
32  const base::FilePath& path() const { return temp_dir_; }
33
34  // Deletes the temporary directory created in Initialize() and all of its
35  // contents, as well as all empty intermediate directories.  Any of these that
36  // cannot be deleted immediately are scheduled for deletion upon reboot.
37  bool Delete();
38
39 private:
40  static void GetTopDirToCreate(const base::FilePath& temp_parent_dir,
41                                base::FilePath* base_dir);
42
43  // The topmost directory created.
44  base::FilePath base_dir_;
45
46  // The temporary directory.
47  base::FilePath temp_dir_;
48
49  FRIEND_TEST_ALL_PREFIXES(SelfCleaningTempDirTest, TopLevel);
50  FRIEND_TEST_ALL_PREFIXES(SelfCleaningTempDirTest, TopLevelPlusOne);
51  DISALLOW_COPY_AND_ASSIGN(SelfCleaningTempDir);
52};
53
54}  // namespace installer
55
56#endif  // CHROME_INSTALLER_UTIL_SELF_CLEANING_TEMP_DIR_H_
57