// Boost.Filesystem mbcopy.cpp ---------------------------------------------//

// Copyright Beman Dawes 2005

// Use, modification, and distribution is subject to the Boost Software
// License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt)

// Copy the files in a directory, using mbpath to represent the new file names
// See http://../doc/path.htm#mbpath for more information

// See deprecated_test for tests of deprecated features
#define BOOST_FILESYSTEM_NO_DEPRECATED

#include <boost/filesystem/config.hpp>
#ifdef BOOST_FILESYSTEM_NARROW_ONLY
#error This compiler or standard library does not support wide-character strings or paths
#endif

#include "mbpath.hpp"
#include <iostream>
#include <boost/filesystem/operations.hpp>
#include <boost/filesystem/fstream.hpp>
#include "../src/utf8_codecvt_facet.hpp"

namespace fs = boost::filesystem;

namespace {

// we can't use boost::filesystem::copy_file() because the argument types
// differ, so provide a not-very-smart replacement.
void copy_file(const fs::wpath& from, const user::mbpath& to)
{
 fs::ifstream from_file(from, std::ios_base::in | std::ios_base::binary);
 if (!from_file)
 {
 std::cout << "input open failed\n";
 return;
 }

 fs::ofstream to_file(to, std::ios_base::out | std::ios_base::binary);
 if (!to_file)
 {
 std::cout << "output open failed\n";
 return;
 }

 char c;
 while (from_file.get(c))
 {
 to_file.put(c);
 if (to_file.fail())
 {
 std::cout << "write error\n";
 return;
 }
 }

 if (!from_file.eof())
 {
 std::cout << "read error\n";
 }
}

} // namespace

int main(int argc, char* argv[])
{
 if (argc != 2)
 {
 std::cout << "Copy files in the current directory to a target directory\n"
 << "Usage: mbcopy <target-dir>\n";
 return 1;
 }

 // For encoding, use Boost UTF-8 codecvt
 std::locale global_loc = std::locale();
 std::locale loc(global_loc, new fs::detail::utf8_codecvt_facet);
 user::mbpath_traits::imbue(loc);

 std::string target_string(argv[1]);
 user::mbpath target_dir(user::mbpath_traits::to_internal(target_string));

 if (!fs::is_directory(target_dir))
 {
 std::cout << "Error: " << argv[1] << " is not a directory\n";
 return 1;
 }

 for (fs::wdirectory_iterator it(L".");
 it != fs::wdirectory_iterator(); ++it)
 {
 if (fs::is_regular_file(it->status()))
 {
 copy_file(*it, target_dir / it->path().filename());
 }
 }

 return 0;
}