set_args.hpp

Go to the documentation of this file.

// Copyright (c) 2006, 2007 Julio M. Merino Vidal
// Copyright (c) 2008 Ilya Sokolov, Boris Schaeling
// Copyright (c) 2009 Boris Schaeling
// Copyright (c) 2010 Felipe Tanus, Boris Schaeling
// Copyright (c) 2011, 2012 Jeff Flinn, Boris Schaeling
//
// Distributed under 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)
 
#pragma once
 
#include <sstream>
 
#include <boost/algorithm/string/predicate.hpp>
#include <boost/process/windows/initializers/initializer_base.hpp>
#include <boost/range/algorithm/copy.hpp>
#include <boost/range/begin.hpp>
#include <boost/range/end.hpp>
#include <boost/shared_array.hpp>
 
namespace boost::process::windows::initializers
{
 
    template <class Range>
    class set_args_ : public initializer_base
    {
    private:
        using ConstIterator = typename Range::const_iterator;
        using String = typename Range::value_type;
        using Char = typename String::value_type;
        using OStringStream = std::basic_ostringstream<Char>;
 
    public:
        explicit set_args_(const Range& args)
        {
            ConstIterator it = boost::const_begin(args);
            ConstIterator end = boost::const_end(args);
            if (it != end)
            {
                exe_ = *it;
                OStringStream os;
                for (; it != end; ++it)
                {
                    if (boost::algorithm::contains(*it, String(1, static_cast<Char>(' '))))
                    {
                        os << static_cast<Char>('"') << *it << static_cast<Char>('"');
                    }
                    else
                    {
                        os << *it;
                    }
                    os << static_cast<Char>(' ');
                }
                String s = os.str();
                cmd_line_.reset(new Char[s.size() + 1]);
                boost::copy(s, cmd_line_.get());
                cmd_line_[s.size()] = 0;
            }
            else
            {
                cmd_line_.reset(new Char[1]());
            }
        }
 
        template <class WindowsExecutor>
        void
        on_CreateProcess_setup(WindowsExecutor& e) const
        {
            e.cmd_line = cmd_line_.get();
            if (!e.exe && !exe_.empty())
            {
                e.exe = exe_.c_str();
            }
        }
 
    private:
        boost::shared_array<Char> cmd_line_;
        String exe_;
    };
 
    template <class Range>
    set_args_<Range>
    set_args(const Range& range)
    {
        return set_args_<Range>(range);
    }
 
} // namespace boost::process::windows::initializers