webserver/webservd/temp_file_manager.cc

// Copyright 2015 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "webservd/temp_file_manager.h"

#include <base/check.h>
#include <base/files/file_util.h>
#include <base/format_macros.h>
#include <base/strings/stringprintf.h>

namespace webservd {

TempFileManager::TempFileManager(const base::FilePath& temp_dir_path,
                                 FileDeleterInterface* file_deleter)
    : temp_dir_path_{temp_dir_path}, file_deleter_{file_deleter} {}

TempFileManager::~TempFileManager() {
  for (const auto& pair : request_files_)
    DeleteFiles(pair.second);
}

base::FilePath TempFileManager::CreateTempFileName(
    const std::string& request_id) {
  std::vector<base::FilePath>& file_list_ref = request_files_[request_id];
  std::string name = base::StringPrintf("%s-%" PRIuS, request_id.c_str(),
                                        file_list_ref.size() + 1);
  base::FilePath file_name = temp_dir_path_.AppendASCII(name);
  file_list_ref.push_back(file_name);
  return file_name;
}

void TempFileManager::DeleteRequestTempFiles(const std::string& request_id) {
  auto iter = request_files_.find(request_id);
  if (iter == request_files_.end())
    return;
  DeleteFiles(iter->second);
  request_files_.erase(iter);
}

void TempFileManager::DeleteFiles(const std::vector<base::FilePath>& files) {
  for (const base::FilePath& file : files)
    CHECK(file_deleter_->DeleteFile(file));
}

bool FileDeleter::DeleteFile(const base::FilePath& path) {
  return base::DeleteFile(path);
}

}  // namespace webservd