// Filename: p3dFileDownload.cxx
// Created by:  drose (11Jun09)
//
////////////////////////////////////////////////////////////////////
//
// PANDA 3D SOFTWARE
// Copyright (c) Carnegie Mellon University.  All rights reserved.
//
// All use of this software is subject to the terms of the revised BSD
// license.  You should have received a copy of this license along
// with this source code in a file named "LICENSE."
//
////////////////////////////////////////////////////////////////////

#include "p3dFileDownload.h"
#include "p3dInstanceManager.h"
#include "mkdir_complete.h"
#include "wstring_encode.h"

////////////////////////////////////////////////////////////////////
//     Function: P3DFileDownload::Constructor
//       Access: Public
//  Description: 
////////////////////////////////////////////////////////////////////
P3DFileDownload::
P3DFileDownload() {
}

////////////////////////////////////////////////////////////////////
//     Function: P3DFileDownload::Copy Constructor
//       Access: Public
//  Description: 
////////////////////////////////////////////////////////////////////
P3DFileDownload::
P3DFileDownload(const P3DFileDownload &copy) : 
  P3DDownload(copy)
{
  // We don't copy the filename.  You have to copy it yourself.
}

////////////////////////////////////////////////////////////////////
//     Function: P3DFileDownload::set_filename
//       Access: Public
//  Description: Supplies the target local filename for the download.
//               Returns true on success, false on failure.
////////////////////////////////////////////////////////////////////
bool P3DFileDownload::
set_filename(const string &filename) {
  _filename = filename;

  return open_file();
}

////////////////////////////////////////////////////////////////////
//     Function: P3DFileDownload::open_file
//       Access: Protected, Virtual
//  Description: Opens the local file for receiving the download.
//               Returns true on success, false on failure.
////////////////////////////////////////////////////////////////////
bool P3DFileDownload::
open_file() {
  if (!mkfile_complete(_filename, nout)) {
    nout << "Failed to create " << _filename << "\n";
    return false;
  }

  _file.clear();
#ifdef _WIN32
  wstring filename_w;
  if (string_to_wstring(filename_w, _filename)) {
    _file.open(filename_w.c_str(), ios::out | ios::trunc | ios::binary);
  }
#else // _WIN32
  _file.open(_filename.c_str(), ios::out | ios::trunc | ios::binary);
#endif  // _WIN32
  if (!_file) {
    nout << "Failed to open " << _filename << " in write mode\n";
    return false;
  }
  
  return true;
}

////////////////////////////////////////////////////////////////////
//     Function: P3DFileDownload::close_file
//       Access: Protected
//  Description: Closes the local file.
////////////////////////////////////////////////////////////////////
void P3DFileDownload::
close_file() {
  _file.close();
}

////////////////////////////////////////////////////////////////////
//     Function: P3DFileDownload::receive_data
//       Access: Protected, Virtual
//  Description: Called as new data is downloaded.  Returns true on
//               success, false on failure.
////////////////////////////////////////////////////////////////////
bool P3DFileDownload::
receive_data(const unsigned char *this_data, size_t this_data_size) {
  _file.write((const char *)this_data, this_data_size);

  if (!_file) {
    return false;
  }
  
  return true;
}

////////////////////////////////////////////////////////////////////
//     Function: P3DFileDownload::download_finished
//       Access: Protected, Virtual
//  Description: Intended to be overloaded to generate a callback
//               when the download finishes, either successfully or
//               otherwise.  The bool parameter is true if the
//               download was successful.
////////////////////////////////////////////////////////////////////
void P3DFileDownload::
download_finished(bool success) {
  P3DDownload::download_finished(success);
  _file.close();
}