shaka-packager/packager/file/file_util.cc

// Copyright 2016 Google Inc. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd

#include "packager/file/file_util.h"

#include <inttypes.h>

#include "packager/base/files/file_path.h"
#include "packager/base/files/file_util.h"
#include "packager/base/process/process_handle.h"
#include "packager/base/strings/stringprintf.h"
#include "packager/base/threading/platform_thread.h"
#include "packager/base/time/time.h"

namespace shaka {
namespace {
// Create a temp file name using process id, thread id and current time.
std::string TempFileName() {
  const int32_t pid = static_cast<int32_t>(base::GetCurrentProcId());
  const int32_t tid = static_cast<int32_t>(base::PlatformThread::CurrentId());
  const int64_t current_time = base::Time::Now().ToInternalValue();
  return base::StringPrintf("packager-tempfile-%x-%x-%" PRIx64, pid, tid,
                            current_time);
}
}  // namespace

bool TempFilePath(const std::string& temp_dir, std::string* temp_file_path) {
  if (temp_dir.empty()) {
    base::FilePath file_path;
    if (!base::CreateTemporaryFile(&file_path)) {
      LOG(ERROR) << "Failed to create temporary file.";
      return false;
    }
    *temp_file_path = file_path.AsUTF8Unsafe();
  } else {
    *temp_file_path =
        base::FilePath::FromUTF8Unsafe(temp_dir)
            .Append(base::FilePath::FromUTF8Unsafe(TempFileName()))
            .AsUTF8Unsafe();
  }
  return true;
}

}  // namespace shaka