file_output_stream.cc

// Copyright (c) 2018 GitHub, Inc.
// Use of this source code is governed by the MIT license that can be
// found in the LICENSE file.

#include "atom/common/api/file_output_stream.h"

#include "base/path_service.h"
#include "base/strings/stringprintf.h"
#include "brightray/browser/brightray_paths.h"
#include "uv.h"  // NOLINT(build/include)

namespace atom {

FileOutputStream::FileOutputStream(base::File* file) : file_(file) {}

bool FileOutputStream::IsComplete() const {
  return is_complete_;
}

int FileOutputStream::GetChunkSize() {
  return 65536;
}

void FileOutputStream::EndOfStream() {
  is_complete_ = true;
}

v8::OutputStream::WriteResult FileOutputStream::WriteAsciiChunk(char* data,
                                                                int size) {
  auto bytes_written = file_->WriteAtCurrentPos(data, size);
  return bytes_written == size ? kContinue : kAbort;
}

// static
base::FilePath FileOutputStream::GetFilePath() {
  base::FilePath user_data;
  PathService::Get(brightray::DIR_USER_DATA, &user_data);

  auto now = uv_hrtime();
  auto sec = static_cast<unsigned int>(now / 1000000);
  auto usec = static_cast<unsigned int>(now % 1000000);
  auto filename = base::StringPrintf("heapdump-%u.%u.heapsnapshot", sec, usec);

  return user_data.AppendASCII(filename.c_str());
}

}  // namespace atom